#!/usr/bin/env bash
# ============================================
# Claude Code + DeepSeek API 一键安装脚本
# 适用: Ubuntu 20.04 / 22.04 / 24.04
# 用法: 把本文件拷到 Ubuntu 家目录, 执行:
#   chmod +x install-claude-deepseek.sh && ./install-claude-deepseek.sh
# ============================================

set -e

echo "==> [1/4] 安装基础依赖 (会要求输入 sudo 密码)"
sudo apt update
sudo apt install -y curl git ca-certificates

echo "==> [2/4] 安装 Node.js 20 (通过 nvm)"
export NVM_NODEJS_ORG_MIRROR="https://npmmirror.com/mirrors/node"  # 国内加速, 海外网络可删除此行
if [ ! -s "$HOME/.nvm/nvm.sh" ]; then
  curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/master/install.sh | bash
fi
export NVM_DIR="$HOME/.nvm"
# shellcheck disable=SC1091
. "$NVM_DIR/nvm.sh"
nvm install 20
nvm alias default 20
echo "Node 版本: $(node -v)"

echo "==> [3/4] 安装 Claude Code"
npm install -g @anthropic-ai/claude-code
claude --version

echo "==> [4/4] 配置 DeepSeek API"
read -rp "粘贴你的 DeepSeek API Key (sk- 开头, 回车确认): " DSEEK_KEY
if [ -z "$DSEEK_KEY" ]; then
  echo "未输入 Key, 请稍后手动编辑 ~/.bashrc 添加配置"; exit 1
fi

# 先清除旧配置行, 避免重复写入 (可重复运行本脚本)
sed -i '/ANTHROPIC_BASE_URL/d;/ANTHROPIC_AUTH_TOKEN/d;/ANTHROPIC_MODEL/d;/ANTHROPIC_SMALL_FAST_MODEL/d' ~/.bashrc
cat >> ~/.bashrc << EOF
export ANTHROPIC_BASE_URL=https://api.deepseek.com/anthropic
export ANTHROPIC_AUTH_TOKEN=$DSEEK_KEY
export ANTHROPIC_MODEL=deepseek-chat
export ANTHROPIC_SMALL_FAST_MODEL=deepseek-chat
EOF

# 当前终端立即生效
export ANTHROPIC_BASE_URL=https://api.deepseek.com/anthropic
export ANTHROPIC_AUTH_TOKEN="$DSEEK_KEY"
export ANTHROPIC_MODEL=deepseek-chat
export ANTHROPIC_SMALL_FAST_MODEL=deepseek-chat

echo ""
echo "========================================="
echo "✅ 安装完成! 接下来:"
echo "   1. 重开一个终端 (或执行: source ~/.bashrc)"
echo "   2. 进入项目目录, 运行: claude"
echo "   3. 在 claude 里输入 /model 确认显示 deepseek-chat"
echo "========================================="