#!/bin/bash
# stop-tasks.sh — 随时叫停/重启统一入口 v1.0（2026-08-26）
#
# 用法：
#   bash stop-tasks.sh stop          # 停止全部（hr1+hr2）
#   bash stop-tasks.sh stop hr1      # 只停 hr1
#   bash stop-tasks.sh restart hr2   # 重启 hr2（先干净停，再启动）
#   bash stop-tasks.sh status        # 查询状态（含卡死检测）
#
# 三层停止机制：
#   1. 优雅：创建 STOP.flag 旗标 → runner/scrape 每轮检查自行退出（最多等45秒）
#   2. 强杀：SIGTERM → 3秒 → SIGKILL，父runner和子scrape都杀
#   3. 验证：杀完复查 ps，任何残留即报错；同时清理旗标防止影响下次启动
set -u
STATE_DIR=/opt/data/logs/recruitment/state
LOG_DIR=/opt/data/logs/recruitment
SCRIPTS=/opt/data/skills/recruitment-resume-screener/scripts
STOP_FLAG=$STATE_DIR/STOP.flag
mkdir -p "$STATE_DIR"

# 进程发现：宽匹配（nohup/环境变量/直接参数三种启动方式都能覆盖）
find_pids() {  # $1=acct 或空=全部
  local pat="daily-runner|zhaopin-scrape"
  [ -n "${1:-}" ] && pat="daily-runner.*$1|ACCOUNT=$1"
  ps aux | grep -E "$pat" | grep -v -E "grep|stop-tasks" | awk '{print $2}'
}

kill_wait() {  # $@ = pids
  [ $# -eq 0 ] && return 0
  kill "$@" 2>/dev/null || true
  for i in 1 2 3; do sleep 1; ps -p "$@" >/dev/null 2>&1 || return 0; done
  kill -9 "$@" 2>/dev/null || true
  sleep 1
}

verify_none() {  # $1=acct 或空；返回0=干净
  local left
  left=$(find_pids "${1:-}")
  if [ -n "$left" ]; then
    echo "❌ 仍有残留进程: $(echo $left | tr '\n' ' ')"
    echo "$left" | while read -r p; do
      ps -p "$p" -o pid,stat,etime,cputime,cmd --no-headers 2>/dev/null
    done
    return 1
  fi
  return 0
}

do_stop() {
  local target="${1:-}"
  echo "🛑 停止筛选任务${target:+（$target）}..."
  touch "$STOP_FLAG"                       # 第1层：优雅退出信号
  local pids; pids=$(find_pids "$target")
  if [ -n "$pids" ]; then
    echo "  等待进程优雅退出（最多45秒）: $(echo $pids | tr '\n' ' ')"
    for i in $(seq 1 45); do
      sleep 1
      pids=$(find_pids "$target")
      [ -z "$pids" ] && break
    done
  fi
  pids=$(find_pids "$target")
  if [ -n "$pids" ]; then
    echo "  仍有进程，强制终止: $(echo $pids | tr '\n' ' ')"
    kill_wait $pids                        # 第2层：TERM→KILL
  fi
  rm -f "$STOP_FLAG"                       # 清旗标，不影响下次启动
  if verify_none "$target"; then           # 第3层：验证
    echo "✅ 已全部停止，验证无残留"
    return 0
  else
    echo "⚠️ 建议稍后重跑 status 确认；若仍残留，联系管理员检查"
    return 1
  fi
}

do_start() {  # $1=acct
  local acct="$1" port label
  [ "$acct" = "hr1" ] && { port=9223; label=Chrome; } || { port=9225; label=Edge; }
  # 启动前防重
  local existing; existing=$(find_pids "$acct")
  if [ -n "$existing" ]; then
    echo "⚠️ $acct 已在运行（PID $(echo $existing | tr '\n' ' ')），先停止再重启"
    do_stop "$acct"
  fi
  # 清掉旧状态，标记新运行
  rm -f "$STATE_DIR/state-$acct.json"
  cd "$SCRIPTS"
  ACCOUNT=$acct CDP_PORT=$port nohup node daily-runner.js "$acct" "$port" \
    > "$LOG_DIR/runner-$acct-$(date +%F-%H%M%S).log" 2>&1 &
  local pid=$!
  sleep 3
  if ps -p "$pid" >/dev/null 2>&1; then
    echo "✅ $acct($label) 已启动 PID=$pid 日志=$(ls -t $LOG_DIR/runner-$acct-*.log | head -1)"
    return 0
  else
    echo "❌ $acct 启动失败，查看日志: $(ls -t $LOG_DIR/runner-$acct-*.log | head -1)"
    tail -5 "$(ls -t $LOG_DIR/runner-$acct-*.log | head -1)"
    return 1
  fi
}

do_restart() {
  do_stop "$1" && do_start "$1"
}

do_status() {
  echo "📋 筛选任务状态（$(TZ=Asia/Shanghai date '+%F %T')）"
  local found=0
  for acct in hr1 hr2; do
    local pids; pids=$(find_pids "$acct")
    if [ -n "$pids" ]; then
      found=1
      echo "  $acct: ▶️ 运行中"
      echo "$pids" | while read -r p; do
        local info
        info=$(ps -p "$p" -o pid,stat,etime,cputime,cmd --no-headers 2>/dev/null | cut -c1-120)
        # 卡死检测：cputime 为 00:00 且 etime>10min
        local et cpu
        et=$(echo "$info" | awk '{print $3}'); cpu=$(echo "$info" | awk '{print $4}')
        local mark=""
        [ "$cpu" = "00:00:00" ] && case "$et" in *:*:*) mark=" ⚠️疑似卡死(cputime=0)";; esac
        echo "    $info$mark"
      done
      local st; st=$(cat "$STATE_DIR/state-$acct.json" 2>/dev/null || echo "无状态文件")
      echo "    状态: $(echo "$st" | head -c 200)"
    else
      echo "  $acct: ⏹️ 未运行（状态: $(cat "$STATE_DIR/state-$acct.json" 2>/dev/null | head -c 120 || echo '无')）"
    fi
  done
  [ "$found" = "0" ] && echo "  （全部停止）"
  [ -f "$STOP_FLAG" ] && echo "  ⚠️ 存在停止旗标 STOP.flag（新批次将被阻止启动）"
}

case "${1:-status}" in
  stop)    do_stop "${2:-}" ;;
  start)   do_start "${2:?用法: start hr1|hr2}" ;;
  restart) do_restart "${2:?用法: restart hr1|hr2}" ;;
  status)  do_status ;;
  *) echo "用法: $0 stop [hr1|hr2] | start hr1|hr2 | restart hr1|hr2 | status"; exit 1 ;;
esac