#!/usr/bin/env bash
# Sundy Handbook · RTK installer for macOS / Linux.
#
# Recommended:
#   bash <(curl -fsSL https://sundy.tumpai.site/install-rtk.sh)
#
# Options:
#   --no-init      Install RTK only; skip Claude Code / Codex integration
#   --claude-only  Configure Claude Code only
#   --codex-only   Configure Codex only
#   -y, --yes      Skip confirmation

set -e

ASSUME_YES="${RTK_SUNDY_YES:-0}"
RUN_INIT="${RTK_INIT:-1}"
INIT_CLAUDE=1
INIT_CODEX=1

for arg in "$@"; do
  case "$arg" in
    --no-init) RUN_INIT=0 ;;
    --claude-only) INIT_CLAUDE=1; INIT_CODEX=0 ;;
    --codex-only) INIT_CLAUDE=0; INIT_CODEX=1 ;;
    -y|--yes) ASSUME_YES=1 ;;
    -h|--help)
      cat <<HELP
用法: bash <(curl -fsSL .../install-rtk.sh) [选项]
  --no-init      只安装 RTK，不初始化 Claude Code / Codex
  --claude-only  只初始化 Claude Code
  --codex-only   只初始化 Codex
  -y, --yes      跳过确认
环境变量:
  RTK_SUNDY_YES=1
  RTK_INIT=0
HELP
      exit 0 ;;
    *) echo "Unknown option: $arg" >&2; exit 1 ;;
  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}OK${C_RST} $*"; }
info()  { echo "${C_BLU}INFO${C_RST} $*"; }
warn()  { echo "${C_YLW}WARN${C_RST} $*"; }
fail()  { echo "${C_RED}ERROR${C_RST} $*" >&2; exit 1; }
title() { echo; echo "${C_BLD}${C_BLU}==> $*${C_RST}"; }

NON_INTERACTIVE=0
if [ ! -t 0 ]; then
  if ( : </dev/tty ) 2>/dev/null; then
    exec </dev/tty
    :
  else
    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（Windows 请使用 install-rtk.ps1）" ;;
esac

echo
cat <<BANNER
${C_BLD}${C_BLU}╔══════════════════════════════════════════════╗
║   Sundy · RTK 省 Token 一键安装             ║
║   平台: ${PLATFORM} (${ARCH})
╚══════════════════════════════════════════════╝${C_RST}

本脚本会做这些事：
  1. 安装 Rust Token Killer（rtk-ai/rtk）
  2. 验证没有装成同名的 Rust Type Kit
  3. 全局初始化 Claude Code hook 和 Codex 指令
  4. 运行 rtk gain 查看节省统计

注意：不要使用 cargo install rtk；RTK 官方文档说明这个名字存在同名包风险。
BANNER

if [ "$ASSUME_YES" != "1" ]; then
  GO=""
  if read -t 60 -r -p "${C_BLD}是否继续？[Y/n]（60s 内不输入默认 Y）${C_RST} " GO; then
    :
  else
    echo
    GO="Y"
  fi
  GO="${GO:-Y}"
  case "$GO" in
    [Yy]*) ;;
    *) info "已取消。"; exit 0 ;;
  esac
fi

export PATH="$HOME/.local/bin:/opt/homebrew/bin:/usr/local/bin:$PATH"

rtk_is_valid() {
  command -v rtk >/dev/null 2>&1 && rtk gain >/dev/null 2>&1
}

title "1/4 安装 RTK"
if rtk_is_valid; then
  ok "RTK 已安装：$(rtk --version 2>/dev/null || echo unknown)"
else
  if command -v rtk >/dev/null 2>&1; then
    warn "检测到 rtk 命令，但 rtk gain 失败；这可能是同名的 Rust Type Kit。将安装 Rust Token Killer 并让它优先进入 PATH。"
  fi

  if command -v brew >/dev/null 2>&1; then
    info "使用 Homebrew 安装：brew install rtk-ai/tap/rtk"
    brew install rtk-ai/tap/rtk || fail "Homebrew 安装 RTK 失败，请确认网络可以访问 GitHub"
  else
    info "使用官方 quick install 安装"
    curl -fsSL https://raw.githubusercontent.com/rtk-ai/rtk/master/install.sh | sh || fail "官方安装脚本执行失败，请确认网络可以访问 GitHub"
  fi
fi

if ! rtk_is_valid; then
  fail "安装后仍无法通过 rtk gain 验证。请检查 PATH 或卸载错误的同名 rtk。"
fi
ok "RTK 可用：$(rtk --version)"

title "2/4 初始化 AI 工具"
if [ "$RUN_INIT" = "1" ]; then
  if [ "$INIT_CLAUDE" = "1" ]; then
    if rtk init -g --auto-patch; then
      ok "已初始化 Claude Code 全局 hook"
    else
      warn "Claude Code 初始化失败，可稍后手动运行：rtk init -g --auto-patch"
    fi
  fi

  if [ "$INIT_CODEX" = "1" ]; then
    if rtk init -g --codex; then
      ok "已初始化 Codex 全局指令"
    else
      warn "Codex 初始化失败，可稍后手动运行：rtk init -g --codex"
    fi
  fi
else
  info "已跳过初始化"
fi

title "3/4 验证"
rtk verify || warn "rtk verify 未完全通过；RTK 命令本身仍可使用"

title "4/4 节省统计"
rtk gain || true

echo
cat <<DONE
${C_BLD}${C_GRN}RTK 安装完成。${C_RST}

重启 Claude Code / Codex 后生效。常用命令：
  rtk git status
  rtk ls -la
  rtk test npm test
  rtk summary npm run build
  rtk gain
DONE