# configure packaging

function(rocprofiler_systems_parse_release)
    if(EXISTS /etc/lsb-release AND NOT IS_DIRECTORY /etc/lsb-release)
        file(READ /etc/lsb-release _LSB_RELEASE)
        if(_LSB_RELEASE)
            string(
                REGEX REPLACE
                "DISTRIB_ID=(.*)\nDISTRIB_RELEASE=(.*)\nDISTRIB_CODENAME=.*"
                "\\1-\\2"
                _SYSTEM_NAME
                "${_LSB_RELEASE}"
            )
        endif()
    elseif(EXISTS /etc/os-release AND NOT IS_DIRECTORY /etc/os-release)
        file(READ /etc/os-release _OS_RELEASE)
        if(_OS_RELEASE)
            string(REPLACE "\"" "" _OS_RELEASE "${_OS_RELEASE}")
            string(REPLACE "-" " " _OS_RELEASE "${_OS_RELEASE}")
            string(
                REGEX REPLACE
                "NAME=.*\nVERSION=([0-9\.]+).*\nID=([a-z]+).*"
                "\\2-\\1"
                _SYSTEM_NAME
                "${_OS_RELEASE}"
            )
        endif()
    endif()
    string(TOLOWER "${_SYSTEM_NAME}" _SYSTEM_NAME)
    if(NOT _SYSTEM_NAME)
        set(_SYSTEM_NAME "${CMAKE_SYSTEM_NAME}")
    endif()
    set(_SYSTEM_NAME "${_SYSTEM_NAME}" PARENT_SCOPE)
endfunction()

# parse either /etc/lsb-release or /etc/os-release
rocprofiler_systems_parse_release()

if(NOT _SYSTEM_NAME)
    set(_SYSTEM_NAME "${CMAKE_SYSTEM_NAME}")
endif()

# Add packaging directives
set(CPACK_PACKAGE_NAME ${PROJECT_NAME})
set(CPACK_PACKAGE_VENDOR "Advanced Micro Devices, Inc.")
set(CPACK_PACKAGE_DESCRIPTION_SUMMARY
    "Runtime instrumentation and binary rewriting for Perfetto via Dyninst"
)
set(CPACK_PACKAGE_VERSION_MAJOR "${PROJECT_VERSION_MAJOR}")
set(CPACK_PACKAGE_VERSION_MINOR "${PROJECT_VERSION_MINOR}")
set(CPACK_PACKAGE_VERSION_PATCH "${PROJECT_VERSION_PATCH}")

set(CPACK_PACKAGE_CONTACT "https://github.com/ROCm/rocprofiler-systems")
set(CPACK_RESOURCE_FILE_LICENSE "${PROJECT_SOURCE_DIR}/LICENSE.md")
set(CPACK_INCLUDE_TOPLEVEL_DIRECTORY OFF)

# For handling the project rebranding from "omnitrace" to "rocprofiler-systems"
set(OMNITRACE_PACKAGE_NAME "omnitrace")

set(ROCPROFSYS_CPACK_SYSTEM_NAME
    "${_SYSTEM_NAME}"
    CACHE STRING
    "System name, e.g. Linux or Ubuntu-20.04"
)
set(ROCPROFSYS_CPACK_PACKAGE_SUFFIX "")

if(ROCPROFSYS_USE_ROCM)
    set(ROCPROFSYS_CPACK_PACKAGE_SUFFIX
        "${ROCPROFSYS_CPACK_PACKAGE_SUFFIX}-ROCm-${ROCmVersion_NUMERIC_VERSION}"
    )
endif()

if(ROCPROFSYS_USE_PAPI)
    set(ROCPROFSYS_CPACK_PACKAGE_SUFFIX "${ROCPROFSYS_CPACK_PACKAGE_SUFFIX}-PAPI")
endif()

if(ROCPROFSYS_USE_OMPT)
    set(ROCPROFSYS_CPACK_PACKAGE_SUFFIX "${ROCPROFSYS_CPACK_PACKAGE_SUFFIX}-OMPT")
endif()

if(ROCPROFSYS_USE_MPI)
    set(VALID_MPI_IMPLS "mpich" "openmpi")
    if("${MPI_C_COMPILER_INCLUDE_DIRS};${MPI_C_HEADER_DIR}" MATCHES "openmpi")
        set(ROCPROFSYS_MPI_IMPL "openmpi")
    elseif("${MPI_C_COMPILER_INCLUDE_DIRS};${MPI_C_HEADER_DIR}" MATCHES "mpich")
        set(ROCPROFSYS_MPI_IMPL "mpich")
    else()
        message(
            WARNING
            "MPI implementation could not be determined. Please set ROCPROFSYS_MPI_IMPL to one of the following for CPack: ${VALID_MPI_IMPLS}"
        )
    endif()
    if(ROCPROFSYS_MPI_IMPL AND NOT "${ROCPROFSYS_MPI_IMPL}" IN_LIST VALID_MPI_IMPLS)
        message(
            SEND_ERROR
            "Invalid ROCPROFSYS_MPI_IMPL (${ROCPROFSYS_MPI_IMPL}). Should be one of: ${VALID_MPI_IMPLS}"
        )
    else()
        rocprofiler_systems_add_feature(ROCPROFSYS_MPI_IMPL
                                        "MPI implementation for CPack DEBIAN depends"
        )
    endif()

    if("${ROCPROFSYS_MPI_IMPL}" STREQUAL "openmpi")
        set(ROCPROFSYS_MPI_IMPL_UPPER "OpenMPI")
    elseif("${ROCPROFSYS_MPI_IMPL}" STREQUAL "mpich")
        set(ROCPROFSYS_MPI_IMPL_UPPER "MPICH")
    else()
        set(ROCPROFSYS_MPI_IMPL_UPPER "MPI")
    endif()
    set(ROCPROFSYS_CPACK_PACKAGE_SUFFIX
        "${ROCPROFSYS_CPACK_PACKAGE_SUFFIX}-${ROCPROFSYS_MPI_IMPL_UPPER}"
    )
endif()

if(ROCPROFSYS_USE_PYTHON)
    set(_ROCPROFSYS_PYTHON_NAME "Python3")
    foreach(_VER ${ROCPROFSYS_PYTHON_VERSIONS})
        if("${_VER}" VERSION_LESS 3.0.0)
            set(_ROCPROFSYS_PYTHON_NAME "Python")
        endif()
    endforeach()
    set(ROCPROFSYS_CPACK_PACKAGE_SUFFIX "${ROCPROFSYS_CPACK_PACKAGE_SUFFIX}-Python3")
endif()

set(CPACK_PACKAGE_FILE_NAME
    "${CPACK_PACKAGE_NAME}-${ROCPROFSYS_VERSION}-${ROCPROFSYS_CPACK_SYSTEM_NAME}${ROCPROFSYS_CPACK_PACKAGE_SUFFIX}"
)
if(DEFINED ENV{CPACK_PACKAGE_FILE_NAME})
    set(CPACK_PACKAGE_FILE_NAME $ENV{CPACK_PACKAGE_FILE_NAME})
endif()

set(ROCPROFSYS_PACKAGE_FILE_NAME
    ${CPACK_PACKAGE_NAME}-${ROCPROFSYS_VERSION}-${ROCPROFSYS_CPACK_SYSTEM_NAME}${ROCPROFSYS_CPACK_PACKAGE_SUFFIX}
)
rocprofiler_systems_add_feature(ROCPROFSYS_PACKAGE_FILE_NAME "CPack filename")

if(ROCM_DEP_ROCMCORE OR ROCPROFILER_DEP_ROCMCORE)
    set(_DEBIAN_PACKAGE_DEPENDS "rocm-core")
    set(_RPM_PACKAGE_REQUIRES "rocm-core")
else()
    set(_DEBIAN_PACKAGE_DEPENDS "")
    set(_RPM_PACKAGE_REQUIRES "")
endif()

# -------------------------------------------------------------------------------------- #
#
# Debian package specific variables
#
# -------------------------------------------------------------------------------------- #

set(CPACK_DEBIAN_PACKAGE_HOMEPAGE "https://github.com/ROCm/rocprofiler-systems")
set(CPACK_DEBIAN_PACKAGE_RELEASE
    "${ROCPROFSYS_CPACK_SYSTEM_NAME}${ROCPROFSYS_CPACK_PACKAGE_SUFFIX}"
)
string(
    REGEX REPLACE
    "([a-zA-Z])-([0-9])"
    "\\1\\2"
    CPACK_DEBIAN_PACKAGE_RELEASE
    "${CPACK_DEBIAN_PACKAGE_RELEASE}"
)
string(REPLACE "-" "~" CPACK_DEBIAN_PACKAGE_RELEASE "${CPACK_DEBIAN_PACKAGE_RELEASE}")

if(ROCPROFSYS_USE_PAPI AND NOT ROCPROFSYS_BUILD_PAPI)
    list(APPEND _DEBIAN_PACKAGE_DEPENDS libpapi-dev libpfm4)
endif()
if(NOT ROCPROFSYS_BUILD_DYNINST)
    if(NOT ROCPROFSYS_BUILD_BOOST)
        foreach(
            _BOOST_COMPONENT
            atomic
            system
            thread
            date-time
            filesystem
            timer
        )
            list(
                APPEND _DEBIAN_PACKAGE_DEPENDS
                "libboost-${_BOOST_COMPONENT}-dev (>= 1.67.0)"
            )
        endforeach()
    endif()
    if(NOT ROCPROFSYS_BUILD_TBB)
        list(APPEND _DEBIAN_PACKAGE_DEPENDS "libtbb-dev (>= 2018.6)")
    endif()
    if(NOT ROCPROFSYS_BUILD_LIBIBERTY)
        list(APPEND _DEBIAN_PACKAGE_DEPENDS "libiberty-dev (>= 20170913)")
    endif()
endif()
if(ROCmVersion_FOUND)
    set(_AMD_SMI_SUFFIX
        " (>= ${ROCmVersion_MAJOR_VERSION}.0.0.${ROCmVersion_NUMERIC_VERSION})"
    )
endif()
if(ROCPROFSYS_USE_ROCM)
    list(APPEND _DEBIAN_PACKAGE_DEPENDS "amd-smi-lib${_AMD_SMI_SUFFIX}")
    list(APPEND _DEBIAN_PACKAGE_DEPENDS "rocprofiler-sdk (>= ${rocprofiler-sdk_VERSION})")
endif()
if(ROCPROFSYS_USE_MPI)
    if("${ROCPROFSYS_MPI_IMPL}" STREQUAL "openmpi")
        list(APPEND _DEBIAN_PACKAGE_DEPENDS "libopenmpi-dev")
    elseif("${ROCPROFSYS_MPI_IMPL}" STREQUAL "mpich")
        list(APPEND _DEBIAN_PACKAGE_DEPENDS "libmpich-dev")
    endif()
endif()
if(ROCPROFSYS_BUILD_TESTING)
    list(APPEND _DEBIAN_PACKAGE_DEPENDS "rocdecode-test")
    list(APPEND _DEBIAN_PACKAGE_DEPENDS "rocjpeg-test")
endif()
string(REPLACE ";" ", " _DEBIAN_PACKAGE_DEPENDS "${_DEBIAN_PACKAGE_DEPENDS}")
set(CPACK_DEBIAN_PACKAGE_DEPENDS
    "${_DEBIAN_PACKAGE_DEPENDS}"
    CACHE STRING
    "Debian package dependencies"
    FORCE
)

set(CPACK_DEBIAN_FILE_NAME "DEB-DEFAULT")
set(CPACK_DEBIAN_PACKAGE_SHLIBDEPS ON)

# Handle the project rebranding from "omnitrace" to "rocprofiler-systems"
set(CPACK_DEBIAN_PACKAGE_PROVIDES ${OMNITRACE_PACKAGE_NAME})
set(CPACK_DEBIAN_PACKAGE_REPLACES ${OMNITRACE_PACKAGE_NAME})
set(CPACK_DEBIAN_PACKAGE_BREAKS ${OMNITRACE_PACKAGE_NAME})

# -------------------------------------------------------------------------------------- #
#
# RPM package specific variables
#
# -------------------------------------------------------------------------------------- #

if(DEFINED CPACK_PACKAGING_INSTALL_PREFIX)
    set(CPACK_RPM_EXCLUDE_FROM_AUTO_FILELIST_ADDITION "${CPACK_PACKAGING_INSTALL_PREFIX}")
endif()

set(CPACK_RPM_PACKAGE_RELEASE
    "${ROCPROFSYS_CPACK_SYSTEM_NAME}${ROCPROFSYS_CPACK_PACKAGE_SUFFIX}"
)
string(
    REGEX REPLACE
    "([a-zA-Z])-([0-9])"
    "\\1\\2"
    CPACK_RPM_PACKAGE_RELEASE
    "${CPACK_RPM_PACKAGE_RELEASE}"
)
string(REPLACE "-" "~" CPACK_RPM_PACKAGE_RELEASE "${CPACK_RPM_PACKAGE_RELEASE}")

# Handle the project rebranding from "omnitrace" to "rocprofiler-systems"
set(CPACK_RPM_PACKAGE_OBSOLETES "${OMNITRACE_PACKAGE_NAME} <= 1.13.0")
set(CPACK_RPM_PACKAGE_CONFLICTS ${OMNITRACE_PACKAGE_NAME})
set(_RPM_PACKAGE_PROVIDES ${OMNITRACE_PACKAGE_NAME})

if(ROCPROFSYS_BUILD_LIBUNWIND)
    list(APPEND _RPM_PACKAGE_PROVIDES "libunwind.so.99()(64bit)")
    list(APPEND _RPM_PACKAGE_PROVIDES "libunwind-x86_64.so.99()(64bit)")
    list(APPEND _RPM_PACKAGE_PROVIDES "libunwind-setjmp.so.0()(64bit)")
    list(APPEND _RPM_PACKAGE_PROVIDES "libunwind-ptrace.so.0()(64bit)")
endif()

string(REPLACE ";" ", " CPACK_RPM_PACKAGE_PROVIDES "${_RPM_PACKAGE_PROVIDES}")
set(CPACK_RPM_PACKAGE_PROVIDES
    "${CPACK_RPM_PACKAGE_PROVIDES}"
    CACHE STRING
    "RPM package provides"
    FORCE
)

if(ROCPROFSYS_USE_MPI)
    if("${ROCPROFSYS_MPI_IMPL}" STREQUAL "openmpi")
        list(APPEND _RPM_PACKAGE_REQUIRES "libopenmpi-devel")
    elseif("${ROCPROFSYS_MPI_IMPL}" STREQUAL "mpich")
        list(APPEND _RPM_PACKAGE_REQUIRES "libmpich-devel")
    endif()
endif()

if(ROCPROFSYS_USE_ROCM)
    if(ROCPROFSYS_BUILD_TESTING)
        list(APPEND _RPM_PACKAGE_REQUIRES "rocdecode-test")
        list(APPEND _RPM_PACKAGE_REQUIRES "rocjpeg-test")
    endif()
endif()

string(REPLACE ";" ", " _RPM_PACKAGE_REQUIRES "${_RPM_PACKAGE_REQUIRES}")
set(CPACK_RPM_PACKAGE_REQUIRES
    ${_RPM_PACKAGE_REQUIRES}
    CACHE STRING
    "RPM package requires"
    FORCE
)
set(CPACK_RPM_SPEC_MORE_DEFINE "%undefine __brp_mangle_shebangs")
set(CPACK_RPM_PACKAGE_LICENSE "MIT")
set(CPACK_RPM_FILE_NAME "RPM-DEFAULT")
set(CPACK_RPM_PACKAGE_RELEASE_DIST ON)
set(CPACK_RPM_PACKAGE_AUTOPROV ON)
set(CPACK_RPM_PACKAGE_AUTOREQ ON)

# -------------------------------------------------------------------------------------- #
#
# Prepare final CPACK parameters
#
# -------------------------------------------------------------------------------------- #

set(CPACK_PACKAGE_VERSION
    "${CPACK_PACKAGE_VERSION_MAJOR}.${CPACK_PACKAGE_VERSION_MINOR}.${CPACK_PACKAGE_VERSION_PATCH}"
)

if(DEFINED ENV{ROCM_LIBPATCH_VERSION})
    set(CPACK_PACKAGE_VERSION "${CPACK_PACKAGE_VERSION}.$ENV{ROCM_LIBPATCH_VERSION}")
endif()

if(DEFINED ENV{CPACK_DEBIAN_PACKAGE_RELEASE})
    set(CPACK_DEBIAN_PACKAGE_RELEASE $ENV{CPACK_DEBIAN_PACKAGE_RELEASE})
endif()

if(DEFINED ENV{CPACK_RPM_PACKAGE_RELEASE})
    set(CPACK_RPM_PACKAGE_RELEASE $ENV{CPACK_RPM_PACKAGE_RELEASE})
endif()

rocprofiler_systems_add_feature(CPACK_PACKAGE_NAME "Package name")
rocprofiler_systems_add_feature(CPACK_PACKAGE_VERSION "Package version")
rocprofiler_systems_add_feature(CPACK_PACKAGING_INSTALL_PREFIX
                                "Package installation prefix"
)

rocprofiler_systems_add_feature(CPACK_DEBIAN_FILE_NAME "Debian file name")
rocprofiler_systems_add_feature(CPACK_DEBIAN_PACKAGE_RELEASE
                                "Debian package release version"
)
rocprofiler_systems_add_feature(CPACK_DEBIAN_PACKAGE_DEPENDS
                                "Debian package dependencies"
)
rocprofiler_systems_add_feature(CPACK_DEBIAN_PACKAGE_SHLIBDEPS
                                "Debian package shared library dependencies"
)

rocprofiler_systems_add_feature(CPACK_RPM_FILE_NAME "RPM file name")
rocprofiler_systems_add_feature(CPACK_RPM_PACKAGE_RELEASE "RPM package release version")
rocprofiler_systems_add_feature(CPACK_RPM_PACKAGE_AUTOREQPROV
                                "RPM package auto generate requires and provides"
)
rocprofiler_systems_add_feature(CPACK_RPM_PACKAGE_REQUIRES "RPM package requires")
rocprofiler_systems_add_feature(CPACK_RPM_PACKAGE_PROVIDES "RPM package provides")

include(CPack)