#!/bin/bash
##############################################################################
# Setup script for CEPCSW:
# - build the cepcsw
#
# Usage:
# $ bash build.sh
# or:
# $ 
#
# Author: Tao Lin <lintao@ihep.ac.cn>
##############################################################################

function info:() {
    echo "INFO: $*" 1>&2
}

function error:() {
    echo "ERROR: $*" 1>&2
}

function check-cepcsw-envvar() {
    if [ "${k4_version}" = "nightlies" ]; then
        source /cvmfs/sw-nightlies.hsf.org/key4hep/setup.sh
    else
        source /cvmfs/sw.hsf.org/key4hep/setup.sh -r ${k4_version}
    fi

    # fix the order of compiler
    local ccdir=$(dirname $CC)
    export PATH=$ccdir:$PATH
}

function build-dir() {
    local blddir=build

    if [ -n "${bldtool}" ]; then
        blddir=${blddir}.${bldtool}
    fi

    # If detect the extra env var, append it to the build dir
    if [ -n "${k4_version}" ]; then
        blddir=${blddir}.${k4_version}
    fi
    if [ -n "${k4_platform}" ]; then
        blddir=${blddir}.${k4_platform}
    fi

    echo $blddir
}

function check-working-builddir() {
    local blddir=$(build-dir)
    if [ ! -d "$blddir" ]; then
        mkdir $blddir || {
            error: "Failed to create $blddir"
            return 1
        }
    fi
}

function run-cmake() {
    local blddir=$(build-dir)

    if [ ! -d "$blddir" ]; then
        error: "Failed to create $blddir"
        return 1
    fi
    local bldgen
    if [ -n "$bldtool" ] ; then
        case $bldtool in
            ninja)
                bldgen="-G Ninja"
                ;;
            *)
                ;;
        esac
    fi

    cd $blddir

    # locate the pandorapfa
    local pandorapfa=$(echo $CMAKE_PREFIX_PATH | tr ':' '\n' | grep pandorapfa | head -n1)
    info: "Find PandoraPFA: $pandorapfa"
    if [ -z "$pandorapfa" ]; then
        error: "Failed to find the PandoraPFA"
        return 1
    fi
    
    local pandorapfa_cmake_modules=${pandorapfa}/cmakemodules
    info: "Find PandoraPFA cmake: ${pandorapfa_cmake_modules}"
    if [ ! -d "${pandorapfa_cmake_modules}" ]; then
        error: "Failed to find the cmake modules for PandoraPFA: ${pandorapfa_cmake_modules}"
        return 1
    fi

    local cmake_modules=${pandorapfa_cmake_modules}

    cmake .. -DHOST_BINARY_TAG=${k4_platform} ${bldgen} \
          -DCMAKE_MODULE_PATH=${cmake_modules} || {
        error: "Failed to cmake"
        return 1
    }

}

function run-make() {
    cmake --build .
}

function run-install() {
    cmake --install .
}

##############################################################################
# Parse the command line options
##############################################################################

# The current default platform
k4_platform=${K4_PLATFORM:-x86_64-linux-gcc11-opt}
k4_version=${K4_VERSION:-2024-03-10} # or nightlies
bldtool=${CEPCSW_BLDTOOL} # make, ninja # set in env var

check-cepcsw-envvar || exit -1

check-working-builddir || exit -1

run-cmake || exit -1

run-make || exit -1

run-install || exit -1