#!/usr/bin/env bash
#
# launch.sh — start and supervise the two cc-ci autonomous loops + a phase-aware watchdog.
#
# Model (see plan.md §6 / §6.1): two INDEPENDENT Claude Code sessions —
#   • Builder   (tmux session: cc-ci-builder)   working clone /srv/cc-ci/cc-ci
#   • Adversary (tmux session: cc-ci-adv)        working clone /srv/cc-ci/cc-ci-adv
# coordinating only through the git repo on git.autonomic.zone.
#
# PHASES: the watchdog runs an ordered sequence of sub-phases (default: 1c then 1b). Each phase
# has its own plan + phase-namespaced loop-state files (STATUS-<id>.md etc.). When a phase's
# STATUS-<id>.md shows "## DONE", the watchdog AUTO-TRANSITIONS to the next phase; after the LAST
# phase it STOPS the loops and exits (a manual gate — e.g. check in before Phase 2).
#
# Three jobs: ITERATION (each agent's /loop), RESILIENCE (restart a dead loop), HANDOFF SIGNALLING
# (ping the waiting loop the moment its counterpart hands off), PHASE SEQUENCING (this file).
#
# Usage:
#   ./launch.sh start       # start the sequence at phase 0 + watchdog (stops/relaunches loops)
#   ./launch.sh watchdog    # run only the supervision loop in the foreground
#   ./launch.sh status      # show phase + session + DONE state
#   ./launch.sh logs builder|adversary|watchdog   # tail a session/log
#   ./launch.sh stop        # stop both loops + watchdog

set -euo pipefail

# Absolute path to this script, so the watchdog re-invokes it correctly regardless of cwd.
SELF="$(readlink -f "${BASH_SOURCE[0]}")"

# ----- config -------------------------------------------------------------
PLAN_DIR="${PLAN_DIR:-/srv/cc-ci/cc-ci-plan}"
CLAUDE_BIN="${CLAUDE_BIN:-claude}"
CLAUDE_FLAGS="${CLAUDE_FLAGS:---dangerously-skip-permissions}"
# REMOTE_CONTROL=1 → interactive --remote-control sessions (viewable at claude.ai/code), required
# for /loop. The box must be logged into the claude.ai account. =0 for plain interactive.
REMOTE_CONTROL="${REMOTE_CONTROL:-1}"

BUILDER_DIR="${BUILDER_DIR:-/srv/cc-ci/cc-ci}"        # Builder's repo clone
ADV_DIR="${ADV_DIR:-/srv/cc-ci/cc-ci-adv}"            # Adversary's repo clone
LOG_DIR="${LOG_DIR:-/srv/cc-ci/.cc-ci-logs}"

WATCH_INTERVAL="${WATCH_INTERVAL:-300}"   # seconds between HEAVY checks (phase DONE / restart dead loops)
SIGNAL_INTERVAL="${SIGNAL_INTERVAL:-30}"  # seconds between HANDOFF checks (ping the waiting loop)

BUILDER_SESSION="cc-ci-builder"
ADV_SESSION="cc-ci-adv"
WATCHDOG_SESSION="cc-ci-watchdog"

# Ordered phase sequence: each entry "id|planfile|statusbasename". The watchdog runs them in order,
# auto-transitions on the phase's "## DONE" (in BUILDER_DIR/<statusbasename>), and STOPS after the
# last one (manual gate). Override PHASES_SPEC (semicolon-separated) to change the sequence.
PHASES_SPEC="${PHASES_SPEC:-1c|plan-phase1c-full-reproducibility.md|STATUS-1c.md;1b|plan-phase1b-review-lint.md|STATUS-1b.md}"
IFS=';' read -r -a PHASES <<< "$PHASES_SPEC"
PHASE_IDX_FILE="${PHASE_IDX_FILE:-$LOG_DIR/.phase-idx}"
# --------------------------------------------------------------------------

log() { printf '[launch %(%H:%M:%S)T] %s\n' -1 "$*"; }
die() { log "ERROR: $*"; exit 1; }
need() { command -v "$1" >/dev/null 2>&1 || die "missing dependency: $1"; }

# ----- phase helpers ------------------------------------------------------
cur_idx()      { local i; i="$(cat "$PHASE_IDX_FILE" 2>/dev/null || echo 0)"; [[ "$i" =~ ^[0-9]+$ ]] || i=0; echo "$i"; }
phase_id()     { echo "${PHASES[$1]}" | cut -d'|' -f1; }
phase_plan()   { echo "${PHASES[$1]}" | cut -d'|' -f2; }
phase_status() { echo "${PHASES[$1]}" | cut -d'|' -f3; }
phase_review() { echo "REVIEW-$(phase_id "$1").md"; }
phase_done()   { grep -qE '^##[[:space:]]+DONE' "$BUILDER_DIR/$1" 2>/dev/null; }   # $1 = status basename (read locally)
all_ids()      { local p; for p in "${PHASES[@]}"; do printf '%s ' "$(echo "$p" | cut -d'|' -f1)"; done; }

preflight() {
  need tmux
  command -v "$CLAUDE_BIN" >/dev/null 2>&1 || die "claude CLI not found (set CLAUDE_BIN)"
  local p plan
  for p in "${PHASES[@]}"; do
    plan="$(echo "$p" | cut -d'|' -f2)"
    [[ -f "$PLAN_DIR/$plan" ]] || die "missing phase plan $PLAN_DIR/$plan"
  done
  [[ -f "$PLAN_DIR/prompts/builder.md"   ]] || die "missing $PLAN_DIR/prompts/builder.md"
  [[ -f "$PLAN_DIR/prompts/adversary.md" ]] || die "missing $PLAN_DIR/prompts/adversary.md"
  mkdir -p "$LOG_DIR"
}

session_alive() { tmux has-session -t "$1" 2>/dev/null; }

# Build the per-session kickoff (phase preamble + base role prompt) and launch claude interactively.
# role ∈ {builder, adversary}. Passed as a POSITIONAL arg via inner $(cat ...) — never stdin
# (piping forces print mode and breaks /loop + remote-control).
start_agent() {
  local role="$1" session="$2" workdir="$3"
  if session_alive "$session"; then log "$session already running — leaving it"; return 0; fi
  mkdir -p "$workdir"
  local idx pid plan status kf
  idx="$(cur_idx)"; pid="$(phase_id "$idx")"; plan="$(phase_plan "$idx")"; status="$(phase_status "$idx")"
  kf="$LOG_DIR/.kickoff-$session.txt"
  {
    cat <<PREAMBLE
*** cc-ci SUB-PHASE ${pid} ***
SINGLE SOURCE OF TRUTH for THIS phase: /srv/cc-ci/cc-ci-plan/${plan} — read it in full now; it defines this phase's mission and Definition of Done.
The general loop protocol still applies and lives in /srv/cc-ci/cc-ci-plan/plan.md (§6.1 coordination, §7 pacing, §9 guardrails) — read those sections too.
Track loop state in PHASE-NAMESPACED files in your repo clone: ${status}, BACKLOG-${pid}.md, REVIEW-${pid}.md, JOURNAL-${pid}.md. DECISIONS.md is shared (append).
"Done" for this phase = the Builder writes "## DONE" to ${status} ONLY after every Definition-of-Done item is Adversary-verified with a fresh PASS in REVIEW-${pid}.md (handshake per §6.1).
The repo's Phase-1 STATUS.md / BACKLOG.md / REVIEW.md are HISTORY from the completed Phase 1 — do NOT use them as your state; use the phase-namespaced files above.
Wherever the standing rules below say "plan.md"/"STATUS.md"/"BACKLOG.md"/"REVIEW.md", substitute the phase plan and these phase-namespaced files.

=== standing role & rules ===
PREAMBLE
    cat "$PLAN_DIR/prompts/$role.md"
  } > "$kf"
  log "starting $session (phase=$pid, plan=$plan, cwd=$workdir, rc=$REMOTE_CONTROL)"
  local rc=""
  [[ "$REMOTE_CONTROL" == "1" ]] && rc="--remote-control '$session'"
  tmux new-session -d -s "$session" -c "$workdir" \
    "$CLAUDE_BIN $rc $CLAUDE_FLAGS \"\$(cat '$kf')\""
  tmux pipe-pane -o -t "$session" "cat >> '$LOG_DIR/$session.log'"
}

start_loops() {
  start_agent builder   "$BUILDER_SESSION" "$BUILDER_DIR"
  start_agent adversary "$ADV_SESSION"     "$ADV_DIR"
}

stop_loops() {
  local s
  for s in "$BUILDER_SESSION" "$ADV_SESSION"; do
    if session_alive "$s"; then log "killing $s"; tmux kill-session -t "$s" || true; fi
  done
}

# Wake a loop by typing a one-line message into its tmux session (queues if mid-turn).
ping_session() {
  local s="$1" msg="$2"
  session_alive "$s" || return 0
  tmux send-keys -t "$s" -l -- "$msg" 2>/dev/null && { sleep 0.3; tmux send-keys -t "$s" Enter 2>/dev/null; }
}

# Edge-triggered handoff signalling for the CURRENT phase. Reads the loops' local clones.
# Ping the Adversary only when a gate id NEWLY appears on a "CLAIMED … awaiting" line (never on
# the baseline / restart / a passed-but-kept line). Ping the Builder when the phase REVIEW changes.
_wd_awaiting=""; _wd_baselined=""; _wd_last_review=""
handoff_reset() { _wd_awaiting=""; _wd_baselined=""; _wd_last_review=""; }   # call on phase transition
handoff_check() {
  local idx sf rf cur now added
  idx="$(cur_idx)"
  sf="$BUILDER_DIR/$(phase_status "$idx")"; rf="$ADV_DIR/$(phase_review "$idx")"
  if [[ -f "$sf" ]]; then
    now="$(grep -iE 'CLAIMED.*awaiting' "$sf" 2>/dev/null | grep -oiE 'M[0-9]+(\.[0-9]+)?|[A-Z][0-9]+' | tr '[:lower:]' '[:upper:]' | sort -u || true)"
    if [[ -n "$_wd_baselined" ]]; then
      added="$(comm -13 <(printf '%s\n' "$_wd_awaiting" | sort -u) <(printf '%s\n' "$now" | sort -u) | grep -vE '^$' || true)"
      if [[ -n "$added" ]]; then
        log "handoff: gate(s) newly awaiting verification: $(echo $added) -> pinging Adversary"
        ping_session "$ADV_SESSION" "watchdog ping: the Builder CLAIMED gate(s) [$(echo $added)] in $(phase_status "$idx") and is awaiting your verification. Pull and verify now."
      fi
    fi
    _wd_awaiting="$now"; _wd_baselined=1
  fi
  if [[ -f "$rf" ]]; then
    cur="$(md5sum "$rf" 2>/dev/null | awk '{print $1}' || true)"
    if [[ -n "$cur" && "$cur" != "$_wd_last_review" ]]; then
      [[ -n "$_wd_last_review" ]] && {
        log "handoff: $(phase_review "$idx") changed -> pinging Builder"
        ping_session "$BUILDER_SESSION" "watchdog ping: the Adversary updated $(phase_review "$idx") (a verdict or finding). Pull and act now — if it PASSes your gate, proceed; if it's a finding, address it."
      }
      _wd_last_review="$cur"
    fi
  fi
}

watchdog_loop() {
  local idx pid status next
  idx="$(cur_idx)"; pid="$(phase_id "$idx")"
  log "watchdog up (phase=$pid [$((idx+1))/${#PHASES[@]}], seq='$(all_ids)', signal=${SIGNAL_INTERVAL}s, heavy=${WATCH_INTERVAL}s)"
  local elapsed="$WATCH_INTERVAL"
  while true; do
    handoff_check
    if (( elapsed >= WATCH_INTERVAL )); then
      elapsed=0
      idx="$(cur_idx)"; pid="$(phase_id "$idx")"; status="$(phase_status "$idx")"
      if phase_done "$status"; then
        next=$((idx + 1))
        if (( next < ${#PHASES[@]} )); then
          log "PHASE $pid DONE (## DONE in $status) — auto-transitioning to $(phase_id "$next")."
          stop_loops
          echo "$next" > "$PHASE_IDX_FILE"
          handoff_reset
          start_loops
        else
          log "PHASE SEQUENCE COMPLETE (last phase $pid DONE). Stopping loops — MANUAL CHECK-IN required before Phase 2."
          stop_loops
          printf 'cc-ci phase sequence complete %(%F %T)T. Phases: %s. Loops stopped; manual check-in required before Phase 2.\n' -1 "$(all_ids)" > "$LOG_DIR/SEQUENCE-COMPLETE"
          log "watchdog exiting."
          exit 0
        fi
      else
        session_alive "$BUILDER_SESSION" || { log "builder gone — restarting (phase $pid)"; start_agent builder   "$BUILDER_SESSION" "$BUILDER_DIR"; }
        session_alive "$ADV_SESSION"     || { log "adversary gone — restarting (phase $pid)"; start_agent adversary "$ADV_SESSION"     "$ADV_DIR"; }
      fi
    fi
    sleep "$SIGNAL_INTERVAL"
    elapsed=$(( elapsed + SIGNAL_INTERVAL ))
  done
}

start_watchdog() {
  if session_alive "$WATCHDOG_SESSION"; then log "watchdog already running"; return 0; fi
  log "starting watchdog"
  tmux new-session -d -s "$WATCHDOG_SESSION" -c "$PLAN_DIR" \
    "exec >>'$LOG_DIR/watchdog.log' 2>&1; '$SELF' watchdog"
}

cmd_status() {
  local idx pid; idx="$(cur_idx)"; pid="$(phase_id "$idx")"
  echo "  phase: $pid [$((idx+1))/${#PHASES[@]}]  plan=$(phase_plan "$idx")  status=$(phase_status "$idx")"
  local s
  for s in "$BUILDER_SESSION" "$ADV_SESSION" "$WATCHDOG_SESSION"; do
    if session_alive "$s"; then echo "  $s: RUNNING"; else echo "  $s: stopped"; fi
  done
  if phase_done "$(phase_status "$idx")"; then echo "  phase $pid: ## DONE"; else echo "  phase $pid: in progress"; fi
  [[ -f "$LOG_DIR/SEQUENCE-COMPLETE" ]] && echo "  >>> $(cat "$LOG_DIR/SEQUENCE-COMPLETE")"
}

case "${1:-}" in
  start)
    preflight
    # Fresh sequence: stop any running loops, reset to phase 0 (unless RESUME_PHASE=1 keeps the idx).
    stop_loops
    if [[ "${RESUME_PHASE:-}" != "1" ]]; then echo 0 > "$PHASE_IDX_FILE"; fi
    rm -f "$LOG_DIR/SEQUENCE-COMPLETE"
    start_loops
    start_watchdog
    log "started at phase $(phase_id "$(cur_idx)"). status: ./launch.sh status | attach: tmux attach -t $BUILDER_SESSION"
    ;;
  watchdog)  preflight; watchdog_loop ;;
  status)    cmd_status ;;
  logs)
    case "${2:-}" in
      builder)   tail -f "$LOG_DIR/$BUILDER_SESSION.log" ;;
      adversary) tail -f "$LOG_DIR/$ADV_SESSION.log" ;;
      watchdog)  tail -f "$LOG_DIR/watchdog.log" ;;
      *) die "usage: $0 logs builder|adversary|watchdog" ;;
    esac
    ;;
  stop)
    stop_loops
    if session_alive "$WATCHDOG_SESSION"; then log "killing $WATCHDOG_SESSION"; tmux kill-session -t "$WATCHDOG_SESSION" || true; fi
    log "stopped."
    ;;
  *)
    cat <<EOF
cc-ci loop launcher (phase-aware)

  $0 start    start the phase sequence at phase 0 + watchdog (stops any running loops first)
  $0 status   show phase + session + DONE state
  $0 logs builder|adversary|watchdog   tail a log
  $0 stop     stop both loops + watchdog
  $0 watchdog run supervision loop in foreground

Phase sequence (auto-transition on per-phase ## DONE; STOP after the last = manual gate):
  $(all_ids)
Env: CLAUDE_BIN=$CLAUDE_BIN  REMOTE_CONTROL=$REMOTE_CONTROL  WATCH_INTERVAL=${WATCH_INTERVAL}s  SIGNAL_INTERVAL=${SIGNAL_INTERVAL}s
       PHASES_SPEC='$PHASES_SPEC'
       RESUME_PHASE=1 to keep the current phase index instead of resetting to 0.
EOF
    ;;
esac