#!/bin/bash

# === 设置变量 ===
API_KEY="sk-154"  # ← 请替换为你的真实 DeepSeek 密钥
CONFIG_PATH="/root/.config/shell_gpt"
SGPT_BIN="/usr/local/bin/sgpt"

echo "📦 正在检查 Python 与 pip3..."

# 安装 pip3（如果未安装）
if ! command -v pip3 &>/dev/null; then
  echo "⚙️ pip3 不存在，正在安装 python3 和 pip3..."
  dnf install -y python3 python3-pip || {
    echo "❌ pip3 安装失败，请检查网络或源配置"; exit 1;
  }
fi

# 创建 pip 软链接（如果未定义）
if ! command -v pip &>/dev/null; then
  ln -s /usr/bin/pip3 /usr/bin/pip
fi

echo "✅ pip 已准备好"

# 安装 shell-gpt（用清华源）
echo "📥 安装 shell-gpt 中..."
pip install -U shell-gpt -i https://pypi.tuna.tsinghua.edu.cn/simple || {
  echo "❌ shell-gpt 安装失败"; exit 1;
}

# 获取 sgpt 命令路径
SGPT_BIN=$(find /usr/local -type f -name sgpt | head -n 1)
echo "✅ sgpt 路径为：$SGPT_BIN"

# 写入配置文件
echo "🛠️ 写入配置到 $CONFIG_PATH/.sgptrc..."
mkdir -p "$CONFIG_PATH"
cat > "$CONFIG_PATH/.sgptrc" <<EOF
API_BASE_URL=https://api.deepseek.com/v1
OPENAI_API_KEY=$API_KEY
DEFAULT_MODEL=deepseek-reasoner
PRETTIFY_MARKDOWN=true
SHOW_FUNCTIONS_OUTPUT=false
EOF
chmod 600 "$CONFIG_PATH/.sgptrc"

# 测试 sgpt 问答
echo "🧪 正在测试 sgpt 接口是否可用："
$SGPT_BIN "用一句话解释 TCP 三次握手"

echo -e "\n✅ 部署成功！现在你可以直接运行 sgpt 进行提问了。"

    1  mkdir -p /root/scripts/gpt_make
    3  cd /root/scripts/gpt_make
    5  vim setup_shell_gpt.sh
    7  chmod +x setup_shell_gpt.sh
   11  vim setup_shell_gpt.sh
   12  ./setup_shell_gpt.sh
   13  sgpt "用一句话解释 TCP 三次握手"