#!/bin/bash -e

# MIT License
#
# Copyright (c) 2025 Advanced Micro Devices, Inc. All rights reserved.
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.

SCRIPT_DIR=$(realpath $(dirname ${BASH_SOURCE[0]}))
cd $(dirname ${SCRIPT_DIR})
echo -e "Working directory: $(pwd)"

umask 0000

verbose-run()
{
    echo -e "\n##### Executing \"${@}\"... #####\n"
    eval $@
}

: ${RPM_FILE:=""}
: ${DEB_FILE:=""}
: ${STGZ_FILE:=""}
: ${SUDO_CMD:=""}

usage()
{
    print_option() { printf "    --%-10s %-24s     %s\n" "${1}" "${2}" "${3}"; }
    echo "Options:"
    print_option stgz "<FILE>" "Run installation test on STGZ package"
    print_option deb "<FILE>" "Run installation test on DEB package"
    print_option rpm "<FILE>" "Run installation test on RPM package"
    print_option sudo "" "Execute commands with sudo"
}

while [[ $# -gt 0 ]]
do
    ARG=${1}
    shift

    case "${ARG}" in
        --clean)
            CLEAN=1
            continue
            ;;
        --fresh)
            FRESH=1
            continue
            ;;
    esac

    VAL=""
    while [[ $# -gt 0 ]]
    do
        VAL=${1}
        shift
        break
    done

    if [ -z "${VAL}" ]; then
        echo "Error! Missing value for argument \"${ARG}\""
        usage
        exit -1
    fi

    case "${ARG}" in
        ? | -h | --help)
            usage
            exit 0
            ;;
        --stgz)
            STGZ_FILE=${VAL}
            ;;
        --deb)
            DEB_FILE=${VAL}
            ;;
        --rpm)
            RPM_FILE=${VAL}
            ;;
        --sudo)
            SUDO_CMD=sudo
            ;;
        *)
            echo -e "Error! Unknown option : ${ARG}"
            usage
            exit -1
            ;;
    esac
done

remove-pycache()
{
    rm -rf ${1}/lib/python/site-packages/rocprofsys/__pycache__
}

setup-env()
{
    rm -rf ${1}/*
    export PATH=${1}/bin:${PATH}
    export LD_LIBRARY_PATH=${1}/lib:/opt/rocm/lib:${LD_LIBRARY_PATH}
}

test-install()
{
    ROCPROFSYS_MODULE_PATH="${1}/lib/python/site-packages/rocprofsys"
    verbose-run rocprof-sys-instrument --help
    verbose-run rocprof-sys-avail --help
    verbose-run rocprof-sys-avail --all

    if [ -d $ROCPROFSYS_MODULE_PATH ]; then
        verbose-run which python3
        verbose-run python3 --version
        PYTHON_VERSION=$(python3 -c "import sys; print(f'python-{sys.version_info.major}{sys.version_info.minor}')")

        if [ -z "$(find $ROCPROFSYS_MODULE_PATH -name "*${PYTHON_VERSION}*" -print -quit)" ]; then
            echo "Error: No library found for Python version ${PYTHON_VERSION} in $ROCPROFSYS_MODULE_PATH"
            exit 1
        fi

        verbose-run rocprof-sys-python --help
    fi
}

change-directory()
{
    if [ ! -f "${1}" ]; then
        if [ -f "/home/rocprofiler-systems/${1}" ]; then
            cd /home/rocprofiler-systems
        elif [ -f "/home/rocprofiler-systems/docker/${1}" ]; then
            cd /home/rocprofiler-systems/docker
        fi
    fi
    realpath ${1}
}

test-stgz()
{
    if [ -z "${1}" ]; then return; fi

    local INSTALLER=$(change-directory ${1})
    mkdir /opt/rocprofiler-systems-stgz
    setup-env /opt/rocprofiler-systems-stgz

    verbose-run ${INSTALLER} --prefix=/opt/rocprofiler-systems-stgz --skip-license --exclude-dir

    test-install /opt/rocprofiler-systems-stgz
}

test-deb()
{
    if [ -z "${1}" ]; then return; fi

    local INSTALLER=$(change-directory ${1})
    setup-env /opt/rocprofiler-systems

    verbose-run ${SUDO_CMD} dpkg --contents ${INSTALLER}
    verbose-run ${SUDO_CMD} dpkg -i ${INSTALLER}

    test-install /opt/rocprofiler-systems
    remove-pycache /opt/rocprofiler-systems
    verbose-run apt-get remove -y rocprofiler-systems
    if [ -d /opt/rocprofiler-systems ]; then
        find /opt/rocprofiler-systems -type f
    fi
}

test-rpm()
{
    if [ -z "${1}" ]; then return; fi

    local INSTALLER=$(change-directory ${1})
    setup-env /opt/rocprofiler-systems

    verbose-run ${SUDO_CMD} rpm -ql -p ${INSTALLER}
    verbose-run ${SUDO_CMD} rpm -v -i -p ${INSTALLER} --nodeps

    test-install /opt/rocprofiler-systems
    remove-pycache /opt/rocprofiler-systems
    verbose-run rpm -e rocprofiler-systems
    if [ -d /opt/rocprofiler-systems ]; then
        find /opt/rocprofiler-systems -type f
    fi
}

test-stgz ${STGZ_FILE}
test-deb  ${DEB_FILE}
test-rpm  ${RPM_FILE}