#!/usr/bin/env bash
# Tumpai Handbook · Codex CLI macOS / Linux 一键安装脚本
#
# 推荐用法（最稳）:
#   bash <(curl -fsSL https://tumpai-handbook.pages.dev/install.sh)
#
# 也支持（管道方式，已加固兜底）:
#   curl -fsSL https://tumpai-handbook.pages.dev/install.sh | bash
#
# 全自动（CI / 无人值守）:
#   CODEX_API_KEY=sk-xxx CODEX_YES=1 bash <(curl -fsSL https://tumpai-handbook.pages.dev/install.sh)

set -e

BASE_URL="${CODEX_BASE_URL:-https://api.tumpai.site:2053/v1}"
MODEL_DEFAULT="${CODEX_MODEL:-gpt-5.5}"
ENV_KEY_NAME="TUMPAI_API_KEY"
PROVIDER_NAME="tumpai"

# --- 参数解析 ---
ASSUME_YES="${CODEX_YES:-0}"
for arg in "$@"; do
  case "$arg" in
    -y|--yes) ASSUME_YES=1 ;;
    -h|--help)
      cat <<HELP
用法: bash <(curl -fsSL .../install.sh) [选项]
  -y, --yes      跳过所有确认（非交互模式）
环境变量:
  CODEX_API_KEY  预填密钥，跳过手动输入
  CODEX_BASE_URL 预填接口地址，默认 https://api.tumpai.site:2053/v1
  CODEX_MODEL    默认模型 (gpt-5.5 / gpt-5.4)
  CODEX_YES=1    等同 --yes
HELP
      exit 0 ;;
  esac
done

# --- 颜色 ---
if [ -t 1 ]; then
  C_RED=$'\033[31m'; C_GRN=$'\033[32m'; C_YLW=$'\033[33m'
  C_BLU=$'\033[34m'; C_BLD=$'\033[1m';  C_RST=$'\033[0m'
else
  C_RED=""; C_GRN=""; C_YLW=""; C_BLU=""; C_BLD=""; C_RST=""
fi

ok()    { echo "${C_GRN}✔${C_RST} $*"; }
info()  { echo "${C_BLU}ℹ${C_RST} $*"; }
warn()  { echo "${C_YLW}⚠${C_RST} $*"; }
fail()  { echo "${C_RED}✘${C_RST} $*" >&2; exit 1; }
title() { echo; echo "${C_BLD}${C_BLU}==> $*${C_RST}"; }

# --- stdin 兜底：管道执行时把 stdin 切回真终端 ---
NON_INTERACTIVE=0
if [ ! -t 0 ]; then
  if [ -r /dev/tty ] && exec </dev/tty 2>/dev/null; then
    :
  else
    warn "未检测到可交互终端，进入非交互模式（需提前设置 CODEX_API_KEY 环境变量）"
    NON_INTERACTIVE=1
    ASSUME_YES=1
  fi
fi

# --- 系统识别 ---
OS="$(uname -s)"
ARCH="$(uname -m)"
case "$OS" in
  Darwin) PLATFORM="mac" ;;
  Linux)  PLATFORM="linux" ;;
  *) fail "暂不支持的系统：$OS（仅支持 macOS / Linux，Windows 请使用 install.ps1）" ;;
esac

# --- BANNER（不再 clear，避免提示被滚出屏幕）---
echo
cat <<BANNER
${C_BLD}${C_BLU}╔══════════════════════════════════════════════╗
║   Tumpai · Codex CLI 一键安装                ║
║   平台: ${PLATFORM} (${ARCH})
╚══════════════════════════════════════════════╝${C_RST}

本脚本会做这些事：
  1. 检查并安装 Node.js（已有则跳过）
  2. 安装 Codex CLI（npm 全局）
  3. 让你输入服务方发放的 API Key（不会上传任何地方）
  4. 写入 ~/.codex/config.toml 和 shell 环境变量
  5. 验证连通性

整个过程约 1-3 分钟。按 Ctrl+C 可随时中止。
BANNER

# --- 是否继续 ---
if [ "$ASSUME_YES" = "1" ]; then
  info "已设置自动确认（--yes / CODEX_YES=1），继续安装..."
else
  GO=""
  if read -t 60 -r -p "${C_BLD}是否继续？[Y/n]（60s 内不输入默认 Y）${C_RST} " GO; then
    :
  else
    echo
    info "60 秒未输入，默认继续"
    GO="Y"
  fi
  GO="${GO:-Y}"
  case "$GO" in
    [Yy]*) ;;
    *) info "已取消。"; exit 0 ;;
  esac
fi

# --- npm 镜像加速（中国大陆 + 没用代理时） ---
maybe_setup_npm_mirror() {
  if [ -z "${HTTP_PROXY:-}${HTTPS_PROXY:-}${ALL_PROXY:-}" ]; then
    if command -v npm >/dev/null 2>&1; then
      local cur
      cur="$(npm config get registry 2>/dev/null || echo '')"
      if [ "$cur" = "https://registry.npmjs.org/" ]; then
        info "未检测到代理；提示：如果 npm 安装很慢，可执行：npm config set registry https://registry.npmmirror.com"
      fi
    fi
  fi
}

# --- 1. 准备 Node ---
title "1/5 检查 Node.js"
if command -v node >/dev/null 2>&1; then
  NODE_V="$(node -v)"
  ok "已安装 Node $NODE_V"
else
  warn "未检测到 Node.js，开始安装..."
  if [ "$PLATFORM" = "mac" ]; then
    if ! command -v brew >/dev/null 2>&1; then
      info "未发现 Homebrew，开始安装 Homebrew（约 2-5 分钟，可能需要输入 sudo 密码）"
      info "提示：如果在中国大陆且没开代理，Homebrew 安装可能极慢；建议先开 Clash/V2Ray TUN 模式后重试"
      /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
      if [ -d /opt/homebrew/bin ]; then
        eval "$(/opt/homebrew/bin/brew shellenv)"
      fi
    fi
    brew install node
  else
    if command -v apt-get >/dev/null 2>&1; then
      info "使用 apt 安装 Node 20"
      curl -fsSL https://deb.nodesource.com/setup_20.x | sudo -E bash -
      sudo apt-get install -y nodejs
    elif command -v dnf >/dev/null 2>&1; then
      curl -fsSL https://rpm.nodesource.com/setup_20.x | sudo -E bash -
      sudo dnf install -y nodejs
    elif command -v yum >/dev/null 2>&1; then
      curl -fsSL https://rpm.nodesource.com/setup_20.x | sudo -E bash -
      sudo yum install -y nodejs
    else
      fail "无法识别的 Linux 发行版，请手动安装 Node.js 18+ 后重试：https://nodejs.org/"
    fi
  fi
  ok "Node 安装完成：$(node -v)"
fi

maybe_setup_npm_mirror

# --- 2. 装 codex ---
title "2/5 安装 Codex CLI"
if command -v codex >/dev/null 2>&1; then
  ok "已安装 codex（$(codex --version 2>/dev/null || echo 'unknown')），将更新到最新版"
fi
if [ "$PLATFORM" = "linux" ] && [ "$(id -u)" != "0" ]; then
  if [ ! -w "$(npm prefix -g 2>/dev/null || echo /usr/lib)" ]; then
    info "全局 npm 目录不可写，将使用 sudo"
    sudo npm install -g @openai/codex || fail "安装 codex 失败"
  else
    npm install -g @openai/codex || fail "安装 codex 失败"
  fi
else
  npm install -g @openai/codex || fail "安装 codex 失败"
fi
ok "Codex 已安装：$(codex --version 2>/dev/null || echo '已就绪')"

# --- 3. 让用户输入 key ---
title "3/5 录入 API 密钥"

API_KEY="${CODEX_API_KEY:-}"
if [ -n "$API_KEY" ]; then
  ok "已从环境变量 CODEX_API_KEY 读取密钥（${#API_KEY} 字符）"
elif [ "$NON_INTERACTIVE" = "1" ]; then
  fail "非交互模式但未提供 CODEX_API_KEY，请重试：CODEX_API_KEY=sk-xxx bash <(curl -fsSL .../install.sh)"
else
  echo "请把服务方发放的 API Key 粘贴进来（输入会显示在屏幕上）："
  echo "  · 这是访问 https://api.tumpai.site:2053/v1 的客户端密钥"
  echo "  · 不是管理台登录密钥"
  echo "  · 不会上传，只写到本机 ~/.codex/config.toml"
  echo
  while [ -z "$API_KEY" ]; do
    read -rp "${C_BLD}🔑 API Key: ${C_RST}" API_KEY
    if [ -z "$API_KEY" ]; then
      warn "密钥为空，请重新输入"
      continue
    fi
    if [ "${#API_KEY}" -lt 10 ]; then
      warn "密钥太短（${#API_KEY} 字符），看起来不太像，确定吗？"
      read -rp "继续使用此密钥？[y/N] " YN
      case "$YN" in [Yy]*) break ;; *) API_KEY="" ;; esac
    fi
  done
  ok "已收到密钥（${#API_KEY} 字符）"
fi

# 选模型
MODEL="$MODEL_DEFAULT"
if [ "$NON_INTERACTIVE" = "1" ] || [ "$ASSUME_YES" = "1" ]; then
  ok "使用默认模型：$MODEL"
else
  echo
  echo "选择默认模型："
  echo "  ${C_BLD}1${C_RST}) gpt-5.5  ← 推荐（综合最强）"
  echo "  ${C_BLD}2${C_RST}) gpt-5.4  （高性价比）"
  if read -t 30 -rp "请选择 [1]（30s 内不输入默认 1）: " MSEL; then :; else echo; MSEL=""; fi
  MSEL="${MSEL:-1}"
  case "$MSEL" in
    2) MODEL="gpt-5.4" ;;
    *) MODEL="gpt-5.5" ;;
  esac
  ok "默认模型：$MODEL"
fi

# --- 4. 写配置 ---
title "4/5 写入配置"
CFG_DIR="$HOME/.codex"
CFG_FILE="$CFG_DIR/config.toml"
mkdir -p "$CFG_DIR"

if [ -f "$CFG_FILE" ]; then
  BAK="$CFG_FILE.bak-$(date +%Y%m%d-%H%M%S)"
  cp "$CFG_FILE" "$BAK"
  ok "已备份旧配置到 $BAK"
fi

cat > "$CFG_FILE" <<EOF
# Tumpai Codex CLI · 自动生成于 $(date '+%Y-%m-%d %H:%M:%S')
model = "$MODEL"
model_provider = "$PROVIDER_NAME"
approval_policy = "never"
sandbox_mode = "danger-full-access"
model_reasoning_effort = "low"

[model_providers.$PROVIDER_NAME]
name = "Tumpai"
base_url = "$BASE_URL"
wire_api = "responses"
env_key = "$ENV_KEY_NAME"

[features]
apps = false
EOF
ok "已写入 $CFG_FILE"

SHELL_RC=""
case "$SHELL" in
  *zsh)  SHELL_RC="$HOME/.zshrc" ;;
  *bash)
    if [ "$PLATFORM" = "mac" ]; then
      SHELL_RC="$HOME/.bash_profile"   # macOS 的 bash 读 .bash_profile，不读 .bashrc
    else
      SHELL_RC="$HOME/.bashrc"
    fi
    ;;
  *fish) SHELL_RC="$HOME/.config/fish/config.fish" ;;
  *)     SHELL_RC="$HOME/.profile" ;;
esac

# 收集所有要写的 rc 文件
# - zsh 额外写 .zshenv：所有启动模式（含 non-interactive）都读
# - macOS bash 同时写 .bash_profile + .bashrc：兼容用户自己 source 的习惯
# - Linux bash 同时写 .bashrc + .bash_profile：兼容 login shell
RC_FILES=("$SHELL_RC")
case "$SHELL" in
  *zsh)
    RC_FILES+=("$HOME/.zshenv")
    ;;
  *bash)
    if [ "$PLATFORM" = "mac" ]; then
      RC_FILES+=("$HOME/.bashrc")
    else
      RC_FILES+=("$HOME/.bash_profile")
    fi
    ;;
esac

# 清理旧标记块
for rc in "${RC_FILES[@]}"; do
  [ -f "$rc" ] || continue
  if [ "$PLATFORM" = "mac" ]; then
    sed -i '' "/# >>> codex-tumpai >>>/,/# <<< codex-tumpai <<</d" "$rc" 2>/dev/null || true
  else
    sed -i "/# >>> codex-tumpai >>>/,/# <<< codex-tumpai <<</d" "$rc" 2>/dev/null || true
  fi
done

# 写入 export
case "$SHELL" in
  *fish)
    mkdir -p "$(dirname "$SHELL_RC")"
    cat >> "$SHELL_RC" <<EOF

# >>> codex-tumpai >>>
set -gx $ENV_KEY_NAME "$API_KEY"
# <<< codex-tumpai <<<
EOF
    ;;
  *)
    for rc in "${RC_FILES[@]}"; do
      mkdir -p "$(dirname "$rc")"
      cat >> "$rc" <<EOF

# >>> codex-tumpai >>>
export $ENV_KEY_NAME="$API_KEY"
# <<< codex-tumpai <<<
EOF
    done
    ;;
esac

if [ "${#RC_FILES[@]}" -gt 1 ]; then
  ok "已写入 ${RC_FILES[*]}（开新终端即可生效）"
else
  ok "已写入 $SHELL_RC（开新终端或运行 'source $SHELL_RC' 生效）"
fi

export $ENV_KEY_NAME="$API_KEY"

# --- 5. 验证 ---
title "5/5 验证连通性"
HTTP_CODE="$(curl -s -o /tmp/codex_check.txt -w "%{http_code}" \
  --max-time 15 \
  -H "Authorization: Bearer $API_KEY" \
  "$BASE_URL/models" || echo "000")"

case "$HTTP_CODE" in
  200) ok "服务可达，密钥有效 ✅" ;;
  401|403) warn "服务可达，但密钥被拒绝（HTTP $HTTP_CODE）。请检查密钥是否正确或联系服务方。" ;;
  000) warn "无法连接 $BASE_URL，请检查网络（也可能是临时波动，可稍后再试）" ;;
  *) warn "返回 HTTP $HTTP_CODE，未必有问题，可继续测试" ;;
esac
rm -f /tmp/codex_check.txt

echo
cat <<DONE
${C_BLD}${C_GRN}╔══════════════════════════════════════════════╗
║   🎉 安装完成！                              ║
╚══════════════════════════════════════════════╝${C_RST}

${C_BLD}${C_YLW}⚠ 重要：当前终端还没加载新环境变量${C_RST}
${C_BLD}${C_YLW}    如果直接跑 codex 会报 "Missing environment variable: $ENV_KEY_NAME"${C_RST}

${C_BLD}请按下面 ${C_GRN}任选一种${C_BLD} 让 codex 立即可用：${C_RST}

${C_BLD}方法 1（推荐 · 当前终端立即生效）${C_RST}
  ${C_BLD}${C_BLU}source $SHELL_RC && codex${C_RST}

${C_BLD}方法 2（最干净 · 关掉本终端窗口，开一个新终端）${C_RST}
  然后直接运行：${C_BLD}${C_BLU}codex${C_RST}

${C_BLD}方法 3（一次性测试，不写 rc）${C_RST}
  ${C_BLD}${C_BLU}export $ENV_KEY_NAME='<你刚输入的那个 key>'${C_RST}
  ${C_BLD}${C_BLU}codex${C_RST}

────────────────────────────────────────────────
首次试用建议命令：
  ${C_BLD}${C_BLU}codex "用 python 写一个猜数字游戏"${C_RST}

如果遇到问题：
  · ${C_BLD}Missing environment variable${C_RST}：执行 ${C_BLD}source $SHELL_RC${C_RST}
  · ${C_BLD}command not found: codex${C_RST}：开新终端，让 PATH 重新加载
  · ${C_BLD}401/403${C_RST}：检查密钥（${C_BLD}cat ~/.codex/config.toml${C_RST}）
  · 其他：访问手册 ${C_BLU}https://tumpai-handbook.pages.dev/${C_RST}

DONE

# --- 自动尝试在当前 shell 让 env 立即可用（如果脚本是 sourced 而不是 piped）---
# 注意：bash <(...) 和 | bash 都是子进程，无法影响父 shell
# 但留这一行，让 sourcing 用法（极少数高级用户）也能直接生效
export $ENV_KEY_NAME="$API_KEY"