#!/usr/bin/env bash
# ============================================================
# 大时代 生产环境检查脚本
#
# 用法:
#   ./scripts/check-prod-env.sh
#   ./scripts/check-prod-env.sh --strict
#
# 检查内容:
#   - Python / Node / npm 是否可用
#   - 后端关键 Python 模块是否已安装
#   - frontend/package-lock.json 与 npm ci 是否可消费
#   - .env 是否存在以及关键变量是否配置
#   - 前端是否可构建
# ============================================================
set -euo pipefail

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

STRICT=false
for arg in "$@"; do
  case "$arg" in
    --strict) STRICT=true ;;
    *) echo -e "${YELLOW}忽略未知参数: ${arg}${NC}" ;;
  esac
done

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

WARNINGS=0

ok() {
  echo -e "${GREEN}✔${NC} $1"
}

warn() {
  WARNINGS=$((WARNINGS + 1))
  echo -e "${YELLOW}⚠${NC} $1"
}

fail() {
  echo -e "${RED}✘${NC} $1"
  exit 1
}

require_cmd() {
  local cmd="$1"
  command -v "${cmd}" >/dev/null 2>&1 || fail "未找到命令: ${cmd}"
}

check_python_modules() {
  python - <<'PY'
mods = [
    'fastapi', 'uvicorn', 'yaml', 'httpx', 'cryptography', 'websockets',
    'rich', 'dotenv', 'pandas_market_calendars', 'finnhub', 'openai',
    'anthropic', 'agentscope', 'pydantic'
]
missing = []
for m in mods:
    try:
        __import__(m)
    except Exception as exc:
        missing.append((m, f"{type(exc).__name__}: {exc}"))

if missing:
    for name, err in missing:
        print(f"MISSING {name} {err}")
    raise SystemExit(1)
print("OK")
PY
}

check_env_file() {
  if [ ! -f .env ]; then
    if ${STRICT}; then
      fail "未找到 .env，生产环境请先基于 env.template 配置"
    fi
    warn "未找到 .env，生产部署前需要补齐"
    return
  fi

  set -a
  # shellcheck disable=SC1091
  source .env
  set +a

  [ -n "${MODEL_NAME:-}" ] || warn "MODEL_NAME 未配置"
  [ -n "${OPENAI_API_KEY:-}" ] || warn "OPENAI_API_KEY 未配置"
  [ -n "${FINNHUB_API_KEY:-}" ] || warn "FINNHUB_API_KEY 未配置（live 模式必需）"
  ok ".env 已加载"
}

check_frontend_install() {
  [ -f frontend/package-lock.json ] || fail "frontend/package-lock.json 缺失，生产部署建议保留锁文件"
  (
    cd frontend
    npm ci --dry-run >/tmp/bigtime-npm-ci.log 2>&1 || {
      cat /tmp/bigtime-npm-ci.log
      exit 1
    }
  )
  if rg -n "@emoji-mart/react|@lobehub/ui|ERESOLVE overriding peer dependency" /tmp/bigtime-npm-ci.log >/dev/null 2>&1; then
    warn "frontend npm ci 存在已知非阻塞 peer warning（@lobehub/icons 依赖链），可忽略"
  elif rg -n "npm warn" /tmp/bigtime-npm-ci.log >/dev/null 2>&1; then
    warn "frontend npm ci 存在 warning，建议查看 /tmp/bigtime-npm-ci.log"
  else
    ok "frontend npm ci --dry-run 通过"
  fi
}

check_frontend_build() {
  (
    cd frontend
    npm run build >/tmp/bigtime-frontend-build.log 2>&1 || {
      cat /tmp/bigtime-frontend-build.log
      exit 1
    }
  )
  ok "frontend 构建通过"
}

echo -e "${CYAN}大时代 · 生产环境检查${NC}"

require_cmd python
require_cmd node
require_cmd npm

ok "python: $(python -V 2>&1)"
ok "node: $(node -v)"
ok "npm: $(npm -v)"

check_python_modules && ok "后端关键 Python 模块已安装"
check_env_file
check_frontend_install
check_frontend_build

if [ "${WARNINGS}" -gt 0 ]; then
  echo -e "${YELLOW}检查完成：有 ${WARNINGS} 项 warning${NC}"
  ${STRICT} && exit 1 || exit 0
fi

echo -e "${GREEN}检查完成：环境可用于生产部署${NC}"