#!/usr/bin/env bash
# 盯住某个 tmux 窗口里的 Claude，停下来就追问，直到它把活干透。
#
# 用法：bash keep-going.sh <tmux窗口> [追问清单文件|单条催促语]
#   例：bash keep-going.sh hub:1 ~/my-probes/training.txt
#       bash keep-going.sh hub:1 "继续，不要停"
#
# 清单文件放仓库外。**这是刻意的**：追问清单会写满「这个项目该做什么、
# 卡在哪、下一步是什么」——那是项目内情，不是工具的一部分。
# 放进仓库就等于把它跟着工具一起发出去了。
# 格式很简单：一条一行，# 开头是注释，空行忽略。
#
# ── 为什么是「追问清单」而不是一句「继续」 ──────────────────────
#
# 「继续」只能防它**停**，防不了它**做浅**。它完全可以每次都继续、
# 每次都做表面功夫，最后给一份看起来很完整的报告。
#
# 观察者的价值在于问它不想被问的那些：测了几轮、调过什么参数、
# 哪些卡点没解决、有没有落盘。所以清单里的每一条都是一个角度，
# 轮着问，到底了从头再来。
#
# ── 怎么判断"停了" ────────────────────────────────────────────────
#
# **不靠截屏猜。** 屏幕上有没有转圈、输入框空不空，都会因为终端宽度、
# 主题、正在滚动而变，猜错的代价是往一个正在干活的会话里插一句话。
#
# 靠 Claude Code 自己写的状态文件 `~/.claude/sessions/<pid>.json`，
# 里面的 status 是权威的（Hub 的会话列表也读它）。
#
# pid 从 tmux pane 反查，不是写死的 —— 会话中途重启（resume）会换 pid，
# 写死的话盯梢就断了而且毫无征兆。
set -uo pipefail

TARGET="${1:?用法：keep-going.sh <tmux窗口> [追问清单文件|催促语]}"
SOURCE="${2:-继续，不要停。遇到问题自己想办法解决，把活干完再停。}"

# 追问清单：一条一行，# 开头和空行忽略。给的不是文件就当成单条催促语。
PROBES=()
if [ -f "$SOURCE" ]; then
  while IFS= read -r line; do
    case "$line" in ''|'#'*) continue;; esac
    PROBES+=("$line")
  done < "$SOURCE"
fi
[ ${#PROBES[@]} -eq 0 ] && PROBES=("$SOURCE")
probe_index=0

INTERVAL=45          # 多久看一眼
NEED_STREAK=2        # 连续几次非工作态才认定"真停了"（≈90 秒）
COOLDOWN=180         # 催完之后至少隔这么久才再催
LOG=/tmp/keep-going-$(echo "$TARGET" | tr ':' '-').log

streak=0
last_nudge=0

log() { echo "[$(date '+%H:%M:%S')] $*" | tee -a "$LOG"; }

log "开始盯 $TARGET —— 每 ${INTERVAL}s 看一次，连续 ${NEED_STREAK} 次不在干活才追问；清单共 ${#PROBES[@]} 条"

while true; do
  if ! tmux list-panes -t "$TARGET" >/dev/null 2>&1; then
    log "窗口没了，退出"
    exit 0
  fi

  pane_pid=$(tmux list-panes -t "$TARGET" -F '#{pane_pid}' 2>/dev/null | head -1)
  state=$(python3 - "$pane_pid" <<'PY' 2>/dev/null || echo unknown
import json, os, sys
pid = sys.argv[1]
path = os.path.expanduser(f"~/.claude/sessions/{pid}.json")
try:
    with open(path) as handle:
        print(json.load(handle).get("status") or "unknown")
except Exception:
    print("unknown")
PY
)

  # 屏幕上有没有转圈计时（"✻ Boondoggling… (17s · ↓ 201 tokens)"）。
  #
  # 这一条是**必要的补充**，不是冗余：状态文件里的 shell 表示"挂着一个后台
  # shell"，和 Claude 自己在不在思考无关 —— 实测它空在提示符上等输入时，
  # 状态照样是 shell。只信状态文件的话这种停顿永远催不到。
  spinning=0
  tmux capture-pane -t "$TARGET" -p -S -12 2>/dev/null \
    | grep -qE '\([0-9]+(m [0-9]+)?s · ' && spinning=1

  # 输入框里有没有没发完的字。
  #
  # **有就绝对不能发。** send-keys 是往当前输入位置敲字，用户正在打字时
  # 会直接拼到他那句后面 —— 实测发生过，把用户的话和追问接成了一句。
  # 轻则语义变了，重则改变他本来要说的意思。
  typing=0
  draft=$(tmux capture-pane -t "$TARGET" -p -S -12 2>/dev/null \
    | grep '^❯' | tail -1 | sed 's/^❯[[:space:]]*//')
  [ -n "$(echo "$draft" | tr -d '[:space:]')" ] && typing=1

  now=$(date +%s)
  if [ "$spinning" = 1 ]; then
    [ "$streak" -gt 0 ] && log "屏幕在转圈，计数清零"
    streak=0
    sleep "$INTERVAL"
    continue
  fi
  if [ "$typing" = 1 ]; then
    log "输入框里有你没发完的字，这轮不打扰"
    sleep "$INTERVAL"
    continue
  fi

  case "$state" in
    busy)
      # 真的在思考。streak 清零 —— 中途喘口气不算停。
      [ "$streak" -gt 0 ] && log "又开始干活了，计数清零"
      streak=0
      ;;
    idle|waiting|shell)
      streak=$((streak + 1))
      if [ "$streak" -ge "$NEED_STREAK" ] && [ $((now - last_nudge)) -ge "$COOLDOWN" ]; then
        message="${PROBES[$probe_index]}"
        probe_index=$(( (probe_index + 1) % ${#PROBES[@]} ))

        # 分两次发：先文字，停一下再回车。
        # 合在一起发时，终端偶尔会把回车吃进上一段输入里，消息就卡在输入框不发出去 ——
        # 看起来催过了，其实一个字都没送出去。
        tmux send-keys -t "$TARGET" "$message"
        sleep 1
        tmux send-keys -t "$TARGET" Enter
        log "状态 $state → 追问第 $((probe_index == 0 ? ${#PROBES[@]} : probe_index))/${#PROBES[@]} 条：${message:0:40}…"
        last_nudge=$now
        streak=0
      else
        log "状态 $state（第 $streak 次）"
      fi
      ;;
    unknown)
      # 状态文件读不到：可能刚重启、也可能 claude 已经退了。
      # **这种情况不催** —— 往一个不确定的窗口里敲字比不敲危险。
      log "状态读不出来（pane pid=$pane_pid），先不动"
      streak=0
      ;;
  esac

  sleep "$INTERVAL"
done