#!/usr/bin/env bash
# launch.sh — dev 环境一键检测与启动
#
# 模式：
#   ./scripts/launch.sh              容器组 (system-db + pi-agent) + 宿主 bin/server + Vite
#   ./scripts/launch.sh --no-docker  非容器：localpg + PI sidecar + bin/agent-server + Vite
#                                    （共享 KB 容器 shuimu-kb-shared 仍可保留）
#
# FORCE_BUILD=1 强制重建；日志 .tmp/launch/{server,web,pi-sidecar}.log

set -uo pipefail

ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
COMPOSE_FILE="docker-compose.dev.yml"
COMPOSE_PROJECT="agent-dev"
EXCLUDED_SERVICES="app"
LOG_DIR="$ROOT/.tmp/launch"
BACKEND_BIN_LEGACY="$ROOT/bin/server"
BACKEND_BIN_SINGLE="$ROOT/bin/agent-server"
BACKEND_URL="http://127.0.0.1:8080/api/v1/health"
FRONTEND_PORT=3100
PI_AGENT_CONTAINER="shuimu-pi-agent-dev"
PI_SIDECAR_PORT="${PI_AGENT_RUNNER_PORT:-18865}"
NO_DOCKER=0

for arg in "$@"; do
  case "$arg" in
    --no-docker) NO_DOCKER=1 ;;
    -h|--help)
      echo "usage: $0 [--no-docker]"
      exit 0
      ;;
  esac
done

mkdir -p "$LOG_DIR"

info() { printf '\033[1;34m[launch]\033[0m %s\n' "$*"; }
ok()   { printf '\033[1;32m  ✅ %s\033[0m\n' "$*"; }
skip() { printf '\033[1;32m  ⏭️  %s\033[0m\n' "$*"; }
act()  { printf '\033[1;33m  🔄 %s\033[0m\n' "$*"; }
fail() { printf '\033[1;31m  ❌ %s\033[0m\n' "$*"; }

wait_http() {
  local url="$1" retries="${2:-30}" i
  for ((i = 0; i < retries; i++)); do
    curl -sf -m 2 "$url" >/dev/null 2>&1 && return 0
    sleep 1
  done
  return 1
}

wait_port() {
  local port="$1" retries="${2:-30}" i
  for ((i = 0; i < retries; i++)); do
    lsof -nP -iTCP:"$port" -sTCP:LISTEN >/dev/null 2>&1 && return 0
    sleep 1
  done
  return 1
}

stop_pid_on_port() {
  local port="$1" pid
  pid="$(lsof -nP -tiTCP:"$port" -sTCP:LISTEN 2>/dev/null | head -n1 || true)"
  if [[ -n "$pid" ]]; then
    act "停止占用 :$port 的进程 PID $pid"
    kill "$pid" 2>/dev/null || true
    for ((i = 0; i < 10; i++)); do
      lsof -nP -iTCP:"$port" -sTCP:LISTEN >/dev/null 2>&1 || return 0
      sleep 1
    done
    lsof -nP -tiTCP:"$port" -sTCP:LISTEN 2>/dev/null | xargs -r kill -9 2>/dev/null || true
  fi
}

# ---------- --no-docker path ----------
launch_no_docker() {
  info "模式: --no-docker（localpg + PI sidecar + agent-server + Vite）"
  cd "$ROOT"

  # Stop conflicting docker app/events/pi-agent (keep shared KB; system-db stopped after migrate)
  if docker info >/dev/null 2>&1; then
    act "停止 dev 容器 app / events / pi-agent（保留 shared KB）..."
    docker compose -f "$COMPOSE_FILE" -p "$COMPOSE_PROJECT" stop app events pi-agent 2>/dev/null || true
  fi

  # Ensure env flags for managed PG
  export LOCALPG_MODE="${LOCALPG_MODE:-managed}"
  export PI_AGENT_RUNNER_URL="${PI_AGENT_RUNNER_URL:-http://127.0.0.1:${PI_SIDECAR_PORT}}"
  export MINIMAX_MODEL_ID="${MINIMAX_MODEL_ID:-MiniMax-M2.7-highspeed}"
  export PI_AGENT_MODEL="${PI_AGENT_MODEL:-MiniMax-M2.7-highspeed}"
  export DATA_DIR="${DATA_DIR:-$ROOT/data/dev}"
  export ENV="${ENV:-dev}"

  # PI sidecar
  info "检测 PI sidecar :$PI_SIDECAR_PORT ..."
  if wait_http "http://127.0.0.1:${PI_SIDECAR_PORT}/health" 2; then
    skip "PI sidecar 已健康"
  else
    stop_pid_on_port "$PI_SIDECAR_PORT"
    act "启动 PI sidecar（日志 $LOG_DIR/pi-sidecar.log）..."
    (
      set -a
      # shellcheck disable=SC1091
      [[ -f "$ROOT/.env" ]] && . "$ROOT/.env"
      set +a
      export PORT="$PI_SIDECAR_PORT"
      export DATA_DIR="$DATA_DIR/workspaces"
      export PI_CODING_AGENT_DIR="$DATA_DIR/../pi-agent"
      export MINIMAX_MODEL_ID PI_AGENT_MODEL
      nohup bash "$ROOT/deploy/pi-agent-sidecar/start.sh" >"$LOG_DIR/pi-sidecar.log" 2>&1 &
    )
    if wait_http "http://127.0.0.1:${PI_SIDECAR_PORT}/health" 60; then
      ok "PI sidecar 就绪"
    else
      fail "PI sidecar 健康检查超时，见 $LOG_DIR/pi-sidecar.log"
      exit 1
    fi
  fi

  # agent-server (embeds web; manages localpg when LOCALPG_MODE=managed)
  info "检测后端 agent-server :8080 ..."
  NEED_REBUILD=0
  if [[ ! -f "$BACKEND_BIN_SINGLE" ]]; then
    NEED_REBUILD=1
  elif [[ -n "$(find "$ROOT/cmd" "$ROOT/internal" "$ROOT/web/embed.go" "$ROOT/go.mod" -newer "$BACKEND_BIN_SINGLE" -print -quit 2>/dev/null)" ]]; then
    NEED_REBUILD=1
  fi
  [[ -n "${FORCE_BUILD:-}" ]] && NEED_REBUILD=1

  if [[ "$NEED_REBUILD" -eq 1 ]]; then
    act "构建单二进制（scripts/build-single-binary.sh）..."
    # Prefer quick go build if dist already present
    if [[ -f "$ROOT/web/dist/index.html" ]]; then
      (cd "$ROOT" && CGO_ENABLED=0 go build -o "$BACKEND_BIN_SINGLE" ./cmd/server) || {
        fail "go build 失败"
        exit 1
      }
    else
      bash "$ROOT/scripts/build-single-binary.sh" || { fail "build-single-binary 失败"; exit 1; }
    fi
  fi

  stop_pid_on_port 8080
  # Also stop legacy bin/server if different path
  if [[ -f "$BACKEND_BIN_LEGACY" ]]; then
    pgrep -f "$BACKEND_BIN_LEGACY" >/dev/null 2>&1 && pkill -f "$BACKEND_BIN_LEGACY" || true
  fi

  act "后台启动 agent-server（LOCALPG_MODE=$LOCALPG_MODE）..."
  (
    cd "$ROOT"
    set -a
    # shellcheck disable=SC1091
    [[ -f "$ROOT/.env" ]] && . "$ROOT/.env"
    set +a
    export LOCALPG_MODE PI_AGENT_RUNNER_URL MINIMAX_MODEL_ID PI_AGENT_MODEL DATA_DIR ENV
    nohup "$BACKEND_BIN_SINGLE" >"$LOG_DIR/server.log" 2>&1 &
  )
  if wait_http "$BACKEND_URL" 90; then
    ok "后端已启动并通过健康检查"
  else
    fail "后端健康检查超时，见 $LOG_DIR/server.log"
    exit 1
  fi

  # Vite
  info "检测前端 :$FRONTEND_PORT ..."
  if wait_port "$FRONTEND_PORT" 1; then
    skip "Vite 已运行于 :$FRONTEND_PORT"
  else
    if [[ ! -d "$ROOT/web/node_modules" ]]; then
      act "npm install..."
      (cd "$ROOT/web" && npm install) || { fail "npm install 失败"; exit 1; }
    fi
    act "启动 Vite..."
    (cd "$ROOT/web" && nohup npm run dev >"$LOG_DIR/web.log" 2>&1 &)
    if wait_port "$FRONTEND_PORT" 30; then
      ok "前端已启动"
    else
      fail "前端启动超时，见 $LOG_DIR/web.log"
      exit 1
    fi
  fi

  info "完成（--no-docker）。工作台: http://localhost:$FRONTEND_PORT"
  info "API: http://127.0.0.1:8080  PI: http://127.0.0.1:$PI_SIDECAR_PORT"
  info "若首次切换，请跑: ./scripts/migrate-dev-db.sh 然后 docker compose -p agent-dev stop system-db"
}

if [[ "$NO_DOCKER" -eq 1 ]]; then
  launch_no_docker
  exit 0
fi

# ---------- legacy docker-assisted path (unchanged below) ----------
BACKEND_BIN="$BACKEND_BIN_LEGACY"

info "检测容器组 ($COMPOSE_PROJECT)..."
cd "$ROOT"

if ! docker info >/dev/null 2>&1; then
  fail "Docker 未运行，请先启动 Docker Desktop（或改用 --no-docker）"
  exit 1
fi

SERVICES="$(docker compose -f "$COMPOSE_FILE" config --services 2>/dev/null || true)"
if [[ -z "$SERVICES" ]]; then
  fail "无法解析 $COMPOSE_FILE (docker compose 版本过旧?)"
  exit 1
fi

ENABLED_SERVICES=""
is_excluded() { case " $EXCLUDED_SERVICES " in *" $1 "*) return 0;; *) return 1;; esac; }

CONTAINERS_STALE=""
for svc in $SERVICES; do
  if is_excluded "$svc"; then
    excluded_cid="$(docker compose -f "$COMPOSE_FILE" -p "$COMPOSE_PROJECT" ps -aq "$svc" 2>/dev/null | head -n1)"
    if [[ -n "$excluded_cid" ]]; then
      excluded_state="$(docker inspect --format '{{.State.Status}}' "$excluded_cid" 2>/dev/null || true)"
      if [[ "$excluded_state" != "running" ]]; then
        docker compose -f "$COMPOSE_FILE" -p "$COMPOSE_PROJECT" rm -f "$svc" >/dev/null 2>&1 || true
      fi
    fi
    continue
  fi
  ENABLED_SERVICES="$ENABLED_SERVICES $svc"
  want_hash="$(docker compose -f "$COMPOSE_FILE" config --hash "$svc" 2>/dev/null || true)"
  want_hash="${want_hash##*hash=}"
  want_hash="${want_hash##* }"
  cid="$(docker compose -f "$COMPOSE_FILE" -p "$COMPOSE_PROJECT" ps -q "$svc" 2>/dev/null | head -n1)"
  if [[ -z "$cid" ]]; then
    CONTAINERS_STALE="$CONTAINERS_STALE $svc"
    continue
  fi
  state="$(docker inspect --format '{{.State.Status}}' "$cid" 2>/dev/null || true)"
  have_hash="$(docker inspect --format '{{ index .Config.Labels "com.docker.compose.config-hash" }}' "$cid" 2>/dev/null || true)"
  if [[ "$state" != "running" || ( -n "$want_hash" && "$want_hash" != "$have_hash" ) ]]; then
    CONTAINERS_STALE="$CONTAINERS_STALE $svc"
  fi
done

if [[ -n "${FORCE_BUILD:-}" ]]; then
  act "FORCE_BUILD=1：强制重建容器组 (排除: $EXCLUDED_SERVICES)"
  docker compose -f "$COMPOSE_FILE" -p "$COMPOSE_PROJECT" up -d --build $ENABLED_SERVICES || { fail "容器组重建失败"; exit 1; }
elif [[ -n "$CONTAINERS_STALE" ]]; then
  act "以下服务需要更新/启动：$CONTAINERS_STALE"
  docker compose -f "$COMPOSE_FILE" -p "$COMPOSE_PROJECT" up -d --build $ENABLED_SERVICES || { fail "容器组更新失败"; exit 1; }
else
  skip "容器组全部运行且配置一致，不处理"
fi

act "等待 $PI_AGENT_CONTAINER 健康..."
for ((i = 0; i < 60; i++)); do
  hstate="$(docker inspect --format '{{if .State.Health}}{{.State.Health.Status}}{{else}}none{{end}}' "$PI_AGENT_CONTAINER" 2>/dev/null || true)"
  if [[ "$hstate" == "healthy" || "$hstate" == "none" ]]; then break; fi
  sleep 2
done
if [[ "$hstate" == "healthy" || "$hstate" == "none" ]]; then
  ok "pi-agent 就绪 ($hstate)"
else
  fail "pi-agent 健康检查超时 (当前 $hstate)，可稍后重跑本脚本"
fi

info "检测后端（Go server :8080）..."

NEED_REBUILD=0
if [[ ! -f "$BACKEND_BIN" ]]; then
  NEED_REBUILD=1
elif [[ -n "$(find "$ROOT/cmd" "$ROOT/internal" "$ROOT/go.mod" -name '*.go' -newer "$BACKEND_BIN" -print -quit 2>/dev/null)" ]]; then
  NEED_REBUILD=1
  act "检测到 Go 源码比 bin/server 新"
fi
[[ -n "${FORCE_BUILD:-}" ]] && NEED_REBUILD=1

BACKEND_PID="$(lsof -nP -tiTCP:8080 -sTCP:LISTEN 2>/dev/null | head -n1)"
BACKEND_RUNNING=0
[[ -n "$BACKEND_PID" ]] && BACKEND_RUNNING=1

if [[ "$NEED_REBUILD" -eq 0 && "$BACKEND_RUNNING" -eq 1 ]] && curl -sf -m 3 "$BACKEND_URL" >/dev/null 2>&1; then
  skip "源码无变化且服务健康，保持运行"
else
  if [[ "$BACKEND_RUNNING" -eq 1 ]]; then
    act "停止运行中的后端进程 (PID $BACKEND_PID)..."
    kill "$BACKEND_PID" 2>/dev/null || true
    for ((i = 0; i < 10; i++)); do
      lsof -nP -iTCP:8080 -sTCP:LISTEN >/dev/null 2>&1 || break
      sleep 1
    done
    if lsof -nP -iTCP:8080 -sTCP:LISTEN >/dev/null 2>&1; then
      lsof -nP -tiTCP:8080 -sTCP:LISTEN 2>/dev/null | xargs kill -9 2>/dev/null || true
      sleep 1
    fi
  fi
  if [[ "$NEED_REBUILD" -eq 1 || ! -f "$BACKEND_BIN" ]]; then
    act "重新构建后端（go build ./cmd/server）..."
    (cd "$ROOT" && go build -o "$BACKEND_BIN" ./cmd/server) || { fail "go build 失败"; exit 1; }
  fi
  act "后台启动后端（日志 $LOG_DIR/server.log）..."
  (cd "$ROOT" && nohup "$BACKEND_BIN" >"$LOG_DIR/server.log" 2>&1 &)
  if wait_http "$BACKEND_URL" 30; then
    ok "后端已启动并通过健康检查"
  else
    fail "后端健康检查超时，请查看 $LOG_DIR/server.log"
  fi
fi

info "检测前端 (Vite dev server :$FRONTEND_PORT)..."

if wait_port "$FRONTEND_PORT" 1; then
  skip "Vite 已运行于 :$FRONTEND_PORT (热更新生效，源码改动无需重启)"
else
  if [[ ! -d "$ROOT/web/node_modules" ]]; then
    act "安装前端依赖（npm install）..."
    (cd "$ROOT/web" && npm install) || { fail "npm install 失败"; exit 1; }
  fi
  act "后台启动前端（日志 $LOG_DIR/web.log）..."
  (cd "$ROOT/web" && nohup npm run dev >"$LOG_DIR/web.log" 2>&1 &)
  if wait_port "$FRONTEND_PORT" 30; then
    ok "前端已启动"
  else
    fail "前端启动超时，请查看 $LOG_DIR/web.log"
  fi
fi

info "完成。工作台入口：http://localhost:$FRONTEND_PORT  (/api 代理 → 127.0.0.1:8080)"