enable

#!/usr/bin/env bash

##      N.B.!!! This script load/enable the environment variable only for O2Physics and it's dependencies !!!
##      then it adds to PATH the location of the example analysis task found within this repository!

[[ "${BASH_SOURCE[0]}" -ef "${0}" ]] && { echo -e "This script should be sourced! use:\nsource $(realpath ${0})"; exit 1; }
SCRIPT_PATH=$(realpath -e -P "${BASH_SOURCE[0]}")
SCRIPT_DIR=$(cd -P $(dirname -- "${SCRIPT_PATH}") >/dev/null 2>&1 && pwd)

TAG_ARG="${1}"
shift

###   DEFAULTS   ###
export CVMFS_TAG="daily-20260126-0000-1"

# Steering of the environment:
# O2OPENACCESS_SW_TAG_CVMFS : set the string name for CVMFS installation
export O2OPENACCESS_SW_TAG_CVMFS=${O2OPENACCESS_SW_TAG_CVMFS:-${CVMFS_TAG}}

# If argument is passed, overwrite any defaults or env variables
[[ -n "${TAG_ARG}" ]] && O2OPENACCESS_SW_TAG_CVMFS="${TAG_ARG}"

#####################################################
## HELPERS
SourceIfValid () { [[ -f "${1}" ]] && source "${1}"; }

# Check if 2nd arg is found in the list(terms delimited by ":") in the 1st arg
ITEM_IN_PATH () {
local LIST_STR ITEM path_arr n
[[ -z "${1}" ]] && return 1;
[[ -z "${2}" ]] && return 1;
LIST_STR="${1}"
ITEM="${2}"
IFS=':' read -ra path_arr <<< "${LIST_STR}"
for (( n=0; n < ${#path_arr[*]}; n++)); do
    [[ "$(realpath -q ${ITEM})" == "$(realpath -q ${path_arr[n]})" ]] && return 0;
done
return 1;
}

# Insert/Add to $PATH if and only if not already present
__PATH_INS () { [[ -z "${1}" ]] && return 1; ITEM_IN_PATH "${PATH}" "${1}" && return 0; export PATH="${1}${PATH:+:}${PATH}"; }
__PATH_ADD () { [[ -z "${1}" ]] && return 1; ITEM_IN_PATH "${PATH}" "${1}" && return 0; export PATH="${PATH}${PATH:+:}${1}"; }

######################################################################

## if O2PHYSICS present (environemnt already enabled then just add path and return(we are sourced))
[[ -n "${O2PHYSICS_ROOT}" ]] && { __PATH_INS "${SCRIPT_DIR}/bin" ; return 0 ; }

# apply local customizations
SourceIfValid ./.aliopendata_cfg  # apply local directory customization

# construct location for etc/profile.d/init.sh
stat "/cvmfs/alice.cern.ch/bin" &> /dev/null || { echo "CVMFS is not available!! check for presence of /cvmfs/alice.cern.ch/bin!"; return 1; }
O2PHYS_TAG="${O2OPENACCESS_SW_TAG_CVMFS}"
WORK_DIR="/cvmfs/alice.cern.ch"
DISABLE_GPU=1

# check the arch of package
ALIBUILD_ARCH_PREFIX="el9-x86_64/Packages"
O2PHYS_DIR="${WORK_DIR}/${ALIBUILD_ARCH_PREFIX}/O2Physics/${O2PHYS_TAG}"

# it packages are not el9-x86_64/Packages let's try with el7-x86_64
if [[ ! -d "${O2PHYS_DIR}" ]]; then
    ALIBUILD_ARCH_PREFIX="el7-x86_64/Packages"
    O2PHYS_DIR="${WORK_DIR}/${ALIBUILD_ARCH_PREFIX}/O2Physics/${O2PHYS_TAG}"
    [[ ! -d "${O2PHYS_DIR}" ]] && { echo "O2Physics was not found in ${O2PHYS_DIR}"; return 1; }
fi

export WORK_DIR DISABLE_GPU O2PHYS_DIR

O2PHYSICS_INIT="${O2PHYS_DIR}/etc/profile.d/init.sh"
[[ ! -f "${O2PHYSICS_INIT}" ]] && { echo "${O2PHYSICS_INIT} not available"; return 1; }

source "${O2PHYSICS_INIT}" &> /dev/null  # if we load from cvmfs ignore any errors

# add to PATH built executables
__PATH_INS "${SCRIPT_DIR}/bin"