#!/bin/bash
set -ex

OLD_PATH="/nfs/ofs-llm-ssd/user/gogongxt/miniconda3"
NEW_PATH="/nfs/gogongxt/miniconda3"

# 1. 修复 base 环境的 bin 目录
if [ -d "$NEW_PATH/bin" ]; then
    cd "$NEW_PATH/bin"
    # -r 参数表示如果 grep 没有输出，xargs 不会执行后面的 sed
    grep -rIl "$OLD_PATH" . | xargs -r sed -i "s|$OLD_PATH|$NEW_PATH|g"
fi

# 2. 修复 Conda 自带的配置和环境激活脚本
if [ -d "$NEW_PATH/etc/profile.d" ]; then
    cd "$NEW_PATH/etc/profile.d"
    grep -rIl "$OLD_PATH" . | xargs -r sed -i "s|$OLD_PATH|$NEW_PATH|g"
fi

# 3. 修复 base 环境的激活钩子脚本
if [ -d "$NEW_PATH/etc/conda/activate.d" ]; then
    cd "$NEW_PATH/etc/conda/activate.d"
    grep -rIl "$OLD_PATH" . | xargs -r sed -i "s|$OLD_PATH|$NEW_PATH|g"
fi

# 4. 修复所有虚拟环境的 bin 目录（pip 和 python 路径的关键）
if [ -d "$NEW_PATH/envs" ]; then
    cd "$NEW_PATH/envs"
    # 递归匹配所有虚拟环境 bin 目录下的脚本
    grep -rIl "$OLD_PATH" ./*/bin/ 2>/dev/null | xargs -r sed -i "s|$OLD_PATH|$NEW_PATH|g"
    # 递归匹配所有虚拟环境的激活钩子脚本
    grep -rIl "$OLD_PATH" ./*/etc/conda/activate.d/ 2>/dev/null | xargs -r sed -i "s|$OLD_PATH|$NEW_PATH|g"
fi

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh .
bash Miniconda3-latest-Linux-x86_64.sh -b -u -p $NEW_PATH

echo "🎉 修复完成！请重新开启终端检查效果。"