#!/usr/bin/env bash
# =============================================================================
# Fischer AgentKit — 一键测试环境脚本
# =============================================================================
#
# 启动完整测试环境：后端 API + E2E 测试（使用独立端口，不影响开发环境）
#
# 用法：
#   bash scripts/test-start.sh              # 启动测试环境并运行全部测试
#   bash scripts/test-start.sh --unit       # 仅单元测试
#   bash scripts/test-start.sh --e2e        # 仅 E2E 测试
#   bash scripts/test-start.sh --bitable    # 仅 bitable 模块测试
#   bash scripts/test-start.sh --skip-server # 跳过服务启动（使用已有服务）
#   bash scripts/test-start.sh --help        # 帮助
# =============================================================================

set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
cd "$PROJECT_ROOT"

: "${E2E_PORT:=18765}"
: "${E2E_API_KEY:=ak_live_e2e_test_key_000000000000000000000000000000000000000000000000}"

# ── 颜色 ────────────────────────────────────────────────────────────────────

RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
CYAN='\033[0;36m'
NC='\033[0m'

# ── 帮助 ────────────────────────────────────────────────────────────────────

show_help() {
  cat <<-'EOF'
Fischer AgentKit — 测试环境

用法: bash scripts/test-start.sh [选项]

选项:
  --unit         仅运行单元测试（pytest -m "not integration"）
  --e2e          仅运行 E2E 测试
  --bitable      仅运行 bitable 模块测试
  --skip-server  跳过服务启动（使用已有服务）
  --help         显示帮助

环境变量:
  E2E_PORT       测试服务器端口（默认: 18765）
  E2E_API_KEY    测试 API Key（默认: 内置测试 Key）
EOF
}

# ── 参数解析 ────────────────────────────────────────────────────────────────

RUN_MODE="all"
SKIP_SERVER=0

while [[ $# -gt 0 ]]; do
  case $1 in
    --unit)        RUN_MODE="unit";     shift ;;
    --e2e)         RUN_MODE="e2e";      shift ;;
    --bitable)     RUN_MODE="bitable";  shift ;;
    --skip-server) SKIP_SERVER=1;       shift ;;
    --help|-h)     show_help; exit 0 ;;
    *) shift ;;
  esac
done

# ── 日志函数 ────────────────────────────────────────────────────────────────

info()   { echo -e "${BLUE}[INFO]${NC}  $*"; }
ok()     { echo -e "${GREEN}[OK]${NC}    $*"; }
warn()   { echo -e "${YELLOW}[WARN]${NC}  $*"; }
fail()   { echo -e "${RED}[FAIL]${NC}  $*"; }

# ── 服务管理 ────────────────────────────────────────────────────────────────

start_test_server() {
  if [[ $SKIP_SERVER -eq 1 ]]; then
    info "SKIP_SERVER=1，跳过服务启动"
    return 0
  fi

  info "启动测试服务器（端口 $E2E_PORT）..."
  if lsof -i :$E2E_PORT 2>/dev/null | grep -q LISTEN; then
    ok "测试服务器已在端口 $E2E_PORT 运行"
    return 0
  fi

  export AGENTKIT_E2E_MODE=1
  export AGENTKIT_WS_TIMEOUT=0
  export AGENTKIT_API_KEY="$E2E_API_KEY"

  source .venv/bin/activate 2>/dev/null || true
  python3 -m agentkit.cli.main serve --host 127.0.0.1 --port "$E2E_PORT" &
  SERVER_PID=$!

  # 等待就绪
  local attempt=0
  while [[ $attempt -lt 60 ]]; do
    if curl -sf "http://127.0.0.1:$E2E_PORT/api/v1/health" &>/dev/null; then
      ok "测试服务器就绪 (PID $SERVER_PID，端口 $E2E_PORT)"
      return 0
    fi
    sleep 0.5
    ((attempt++))
  done

  fail "测试服务器启动超时"
  kill $SERVER_PID 2>/dev/null || true
  exit 1
}

stop_test_server() {
  if [[ $SKIP_SERVER -eq 1 ]]; then return 0; fi
  PID=$(lsof -ti :$E2E_PORT 2>/dev/null || true)
  if [[ -n "$PID" ]]; then
    kill $PID 2>/dev/null && ok "测试服务器已停止" || true
  fi
}

# ── 测试执行 ────────────────────────────────────────────────────────────────

run_unit_tests() {
  info "运行单元测试..."
  source .venv/bin/activate 2>/dev/null || true
  pytest -m "not integration" -q --tb=short
  ok "单元测试完成"
}

run_bitable_tests() {
  info "运行 bitable 模块测试..."
  source .venv/bin/activate 2>/dev/null || true
  pytest tests/unit/bitable/ -v --tb=short
  ok "bitable 测试完成"
}

run_e2e_tests() {
  info "运行 E2E 测试..."
  FE_DIR="$PROJECT_ROOT/src/agentkit/server/frontend"
  cd "$FE_DIR"

  if [[ ! -d node_modules ]]; then
    info "安装前端依赖..."
    npm install
  fi

  export AGENTKIT_SERVER_URL="http://127.0.0.1:$E2E_PORT"
  export AGENTKIT_API_KEY="$E2E_API_KEY"

  npm run test:e2e
  ok "E2E 测试完成"
  cd "$PROJECT_ROOT"
}

# ── 主流程 ─────────────────────────────────────────────────────────────────

echo ""
echo -e "${CYAN}═══════════════════════════════════════════════════${NC}"
echo -e "${CYAN}  Fischer AgentKit — 测试环境${NC}"
echo -e "${CYAN}═══════════════════════════════════════════════════${NC}"
echo ""

trap stop_test_server EXIT INT TERM

case $RUN_MODE in
  unit)
    run_unit_tests
    ;;
  e2e)
    start_test_server
    run_e2e_tests
    ;;
  bitable)
    run_bitable_tests
    ;;
  all)
    info "运行全部测试..."
    run_unit_tests
    echo ""
    run_bitable_tests
    echo ""
    start_test_server
    run_e2e_tests
    ;;
esac

echo ""
ok "测试完成！"