#!/bin/bash
cd "$(dirname "$0")"

echo ""
echo "============================================"
echo "  yangjinming.org - Claude Code (Mac)"
echo "============================================"
echo ""

if ! command -v node &> /dev/null; then
    echo "[ERROR] Install Node.js first: brew install node"
    read -p "Press Enter to exit..."
    exit 1
fi
echo "[OK] Node.js: $(node --version)"

read -p "Enter your API Key (sk-...): " USER_KEY
[ -z "$USER_KEY" ] && { echo "[ERROR] Key empty"; read; exit 1; }

SHELL_CONFIG="$HOME/.zshrc"
[ "$SHELL" = "/bin/bash" ] && SHELL_CONFIG="$HOME/.bash_profile"
touch "$SHELL_CONFIG"

echo ""
echo "[1/2] Installing claude-code..."
npm install -g @anthropic-ai/claude-code 2>&1 | tail -3

echo ""
echo "[2/2] Setting env vars in $SHELL_CONFIG..."
grep -v "ANTHROPIC_BASE_URL\|ANTHROPIC_API_KEY\|ANTHROPIC_AUTH_TOKEN\|yangjinming.org Claude" "$SHELL_CONFIG" > "$SHELL_CONFIG.tmp" 2>/dev/null || true
mv "$SHELL_CONFIG.tmp" "$SHELL_CONFIG" 2>/dev/null || true
cat >> "$SHELL_CONFIG" << ENV_EOF

# yangjinming.org Claude Code (added by setup script)
export ANTHROPIC_BASE_URL="https://yangjinming.org/anthropic"
export ANTHROPIC_API_KEY="$USER_KEY"
export ANTHROPIC_AUTH_TOKEN="$USER_KEY"
ENV_EOF

echo ""
echo "[SUCCESS] Open NEW Terminal, run: claude"
echo "Console: https://yangjinming.org/console.html"
read -p "Press Enter to exit..."