@echo off

:start
cls
echo ========================================
echo   Welcome to use Globalai Claude Code
echo ========================================
echo.

:input_token
set "TOKEN="
set /p TOKEN=请输入您的sk密钥:

:: 检查是否为空
if "%TOKEN%"=="" (
    echo.
    echo [错误] 输入不能为空，请重新输入！
    echo.
    goto input_token
)

:: 检查是否以 sk- 开头
echo %TOKEN% | findstr /b "sk-" >nul
if errorlevel 1 (
    echo.
    echo [错误] 密钥格式不正确！请输入以 "sk-" 开头的密钥。
    echo.
    goto input_token
)

set BASE_URL=https://globalai.vip

:: 设置环境变量
setx ANTHROPIC_AUTH_TOKEN "%TOKEN%" >nul
setx ANTHROPIC_BASE_URL "%BASE_URL%" >nul

:: 设置 settings.json 路径
set "SETTINGS_DIR=%USERPROFILE%\.claude"
set "SETTINGS_FILE=%SETTINGS_DIR%\settings.json"

:: 检查 .claude 目录是否存在，不存在则创建
if not exist "%SETTINGS_DIR%" (
    mkdir "%SETTINGS_DIR%"
)

:: 创建或覆盖 settings.json 文件
(
echo {
echo   "env": {
echo     "ANTHROPIC_AUTH_TOKEN": "%TOKEN%",
echo     "ANTHROPIC_BASE_URL": "https://globalai.vip",
echo     "CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC": "1",
echo     "API_TIMEOUT_MS": "600000"
echo   },
echo   "permissions": {
echo     "allow": [],
echo     "deny": []
echo   },
echo   "alwaysThinkingEnabled": false
echo }
) > "%SETTINGS_FILE%"

echo.
echo ========================================
echo   [OK] 环境变量设置完成！
echo   [OK] settings.json 配置完成！
echo ========================================
echo.
echo   ANTHROPIC_AUTH_TOKEN = %TOKEN:~0,10%***（已隐藏）
echo   配置文件路径: %SETTINGS_FILE%
echo.
echo   注意: 请重新打开Claude Code终端或重启插件以使变量生效。
echo.

:menu
echo ========================================
echo   请选择操作:
echo   1. 重新设置密钥
echo   2. 退出
echo ========================================
set /p CHOICE=请输入选项 (1/2):

if "%CHOICE%"=="1" goto start
if "%CHOICE%"=="2" goto end

echo.
echo [错误] 无效选项，请重新选择！
echo.
goto menu

:end
echo.
echo 感谢使用，再见！
pause