#!/bin/bash

# SPDX-FileCopyrightText: Copyright (c) 2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved. 
# SPDX-License-Identifier: LicenseRef-NvidiaProprietary
#
# NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
# property and proprietary rights in and to this material, related
# documentation and any modifications thereto. Any use, reproduction, 
# disclosure or distribution of this material and related documentation 
# without an express license agreement from NVIDIA CORPORATION or 
# its affiliates is strictly prohibited.

if [ "${BASH_SOURCE[0]}" == "${0}" ]; then
    echo "Error: This script needs to be sourced. Please run as 'source ${BASH_SOURCE[0]} install-directory-param'" >&2
    exit 1
fi

if [ -z "$1" ]; then
    echo "Error: This script needs to be called with install directory parameter.\
        Please run as 'source ${BASH_SOURCE[0]} install-directory-param'" >&2
    exit 1
fi

install_dir=$1
if grep -qs "debian" /etc/os-release 2>/dev/null; then
    NSYS_ARCH=$(uname -m)
elif [[ -f /etc/redhat-release ]]; then
    NSYS_ARCH=$(rpm --eval '%{_arch}')
fi
export NSYS_ARCH

# if fontconfig not exists, then it should be at least installed via dependencies installer
if [ ! -x "$(command -v fc-list)" ]; then
    export FONTCONFIG_FILE=$install_dir/etc/fonts/fonts.conf
    export FONTCONFIG_PATH=$install_dir/etc/fonts/
fi

export PATH=$install_dir/usr/sbin:$install_dir/usr/bin:$install_dir/bin${PATH:+:$PATH}
export LD_LIBRARY_PATH=$install_dir/lib:$install_dir/lib64:$install_dir/usr/lib:$install_dir/usr/lib64${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}
export LD_LIBRARY_PATH=$install_dir/usr/lib/$NSYS_ARCH-linux-gnu:$install_dir/usr/lib64/$NSYS_ARCH-linux-gnu${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}
export LD_LIBRARY_PATH=$install_dir/lib/$NSYS_ARCH-linux-gnu:$install_dir/lib64/$NSYS_ARCH-linux-gnu${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}
