#!/usr/bin/env bash

set -eo pipefail

REPO_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
SHARED_ROOT="${SHARED_ROOT:-/shared_work/xuan}"
CONDA_BASE_PREFIX="${CONDA_BASE_PREFIX:-$SHARED_ROOT/miniforge3}"
# Ignore inherited CONDA_EXE values from unrelated hosts and use the configured prefix.
CONDA_EXE_PATH="$CONDA_BASE_PREFIX/bin/conda"
ENV_NAME="${ENV_NAME:-xvla-stable}"
ENV_PREFIX="${ENV_PREFIX:-$SHARED_ROOT/conda-envs/$ENV_NAME}"
CONDARC_PATH="${CONDARC_PATH:-$REPO_ROOT/.condarc.shared}"

export CONDARC="$CONDARC_PATH"
export CONDA_PKGS_DIRS="$SHARED_ROOT/conda-pkgs"
export CONDA_ENVS_PATH="$SHARED_ROOT/conda-envs"
export PIP_CACHE_DIR="$SHARED_ROOT/.cache/pip"
export HF_HOME="$SHARED_ROOT/.cache/huggingface"
export TRANSFORMERS_CACHE="$HF_HOME/transformers"
export HUGGINGFACE_HUB_CACHE="$HF_HOME/hub"
export TORCH_HOME="$SHARED_ROOT/.cache/torch"
export MPLCONFIGDIR="$SHARED_ROOT/.cache/matplotlib"
export XDG_CACHE_HOME="$SHARED_ROOT/.cache/xdg"
export PYTHONNOUSERSITE=1
export PYTHONPATH="$REPO_ROOT${PYTHONPATH:+:$PYTHONPATH}"

unset CONDA_DEFAULT_ENV
unset CONDA_EXE
unset CONDA_PREFIX
unset CONDA_PREFIX_1
unset CONDA_PROMPT_MODIFIER
unset CONDA_PYTHON_EXE
unset CONDA_SHLVL
unset CONDA_CONFIG_DIR
unset _CE_CONDA
unset _CE_M

if [[ ! -x "$CONDA_EXE_PATH" ]]; then
  echo "Conda executable not found at $CONDA_EXE_PATH" >&2
  return 1 2>/dev/null || exit 1
fi

had_nounset=0
if [[ $- == *u* ]]; then
  had_nounset=1
  set +u
fi

__conda_setup="$("$CONDA_EXE_PATH" shell.bash hook)"
eval "$__conda_setup"
unset __conda_setup

conda activate "$ENV_PREFIX"

if [[ "$had_nounset" -eq 1 ]]; then
  set -u
fi

echo "Activated shared X-VLA environment: $ENV_PREFIX"