#!/usr/bin/env bash
set -euo pipefail

# Claude Code friendly runner for Tiny/Larson with curated env presets.
#
# Usage:
#   scripts/claude_code_debug.sh <mode> [DUR] [THREADS]
# Modes:
#   guard           repro+remote guard (FORCE_REMOTE=1)
#   guard-local     repro+remote guard (FORCE_REMOTE=0)
#   repro           reproducible (TRACE_RING on, FAST_CAP=0)
#   fast0           repro + front bypass
#   debug           repro + periodic SIGUSR2 dumps
#   asan|ubsan|tsan sanitizer builds
#
# Examples:
#   scripts/claude_code_debug.sh guard 2 4
#   scripts/claude_code_debug.sh guard-local 2 4
#   scripts/claude_code_debug.sh asan 2 4

MODE=${1:-guard}
DUR=${2:-2}
THREADS=${3:-4}

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

mkdir -p logs
ts() { date +"%Y%m%d_%H%M%S"; }

# Ensure build for non-sanitizer runs
if [[ "$MODE" != asan && "$MODE" != ubsan && "$MODE" != tsan ]]; then
  [[ -x ./larson_hakmem ]] || make -s larson_hakmem >/dev/null
fi

# Common Tiny + Larson envs (kept minimal; override via env when needed)
export HAKMEM_LARSON_TINY_ONLY=1
export HAKMEM_LARSON=1
export HAKMEM_WRAP_TINY=${HAKMEM_WRAP_TINY:-1}
export HAKMEM_DISABLE_BATCH=1
export HAKMEM_TINY_META_ALLOC=1
export HAKMEM_TINY_META_FREE=1
export HAKMEM_TINY_USE_SUPERSLAB=1
export HAKMEM_TINY_MAILBOX_SLOWDISC=1
export HAKMEM_TINY_MAILBOX_SLOWDISC_PERIOD=${HAKMEM_TINY_MAILBOX_SLOWDISC_PERIOD:-256}

# Conservative defaults suitable for debugging
export HAKMEM_TINY_SS_FORCE_LG=${HAKMEM_TINY_SS_FORCE_LG:-21}
export HAKMEM_TINY_FAST_CAP=${HAKMEM_TINY_FAST_CAP:-0}
export HAKMEM_TINY_TLS_LIST=${HAKMEM_TINY_TLS_LIST:-1}
export HAKMEM_TINY_TRACE_RING=${HAKMEM_TINY_TRACE_RING:-1}
export HAKMEM_TINY_RF_TRACE=${HAKMEM_TINY_RF_TRACE:-0}
export HAKMEM_TINY_SS_ADOPT=${HAKMEM_TINY_SS_ADOPT:-1}
export HAKMEM_TINY_RF_FORCE_NOTIFY=${HAKMEM_TINY_RF_FORCE_NOTIFY:-1}

# Fixed Larson args
MIN=8; MAX=128; CHUNKS=1024; ROUNDS=1; SEED=12345

LOG_TAG="$(ts)_${MODE}_${DUR}s_${THREADS}t"
OUT="logs/${LOG_TAG}.out"
ERR="logs/${LOG_TAG}.err"

case "$MODE" in
  guard)
    export HAKMEM_TINY_DEBUG_REMOTE_GUARD=1
    export HAKMEM_TINY_FORCE_REMOTE=${HAKMEM_TINY_FORCE_REMOTE:-1}
    export HAKMEM_SAFE_FREE=${HAKMEM_SAFE_FREE:-1}
    export HAKMEM_SAFE_FREE_STRICT=${HAKMEM_SAFE_FREE_STRICT:-1}
    ;;
  guard-local)
    export HAKMEM_TINY_DEBUG_REMOTE_GUARD=1
    export HAKMEM_TINY_FORCE_REMOTE=0
    export HAKMEM_SAFE_FREE=${HAKMEM_SAFE_FREE:-1}
    export HAKMEM_SAFE_FREE_STRICT=${HAKMEM_SAFE_FREE_STRICT:-1}
    ;;
  repro)
    : # defaults already set
    ;;
  fast0)
    export HAKMEM_TINY_DEBUG_FAST0=1
    ;;
  debug)
    export HAKMEM_SAFE_FREE=${HAKMEM_SAFE_FREE:-1}
    export HAKMEM_SAFE_FREE_STRICT=${HAKMEM_SAFE_FREE_STRICT:-1}
    ;;
  asan)
    make -s asan-larson >/dev/null || exit 1
    export HAKMEM_SAFE_FREE=${HAKMEM_SAFE_FREE:-1}
    export HAKMEM_SAFE_FREE_STRICT=${HAKMEM_SAFE_FREE_STRICT:-1}
    ASAN_OPTIONS="abort_on_error=1,detect_leaks=0,verify_asan_link_order=0,detect_stack_use_after_return=1,fast_unwind_on_malloc=0" \
    UBSAN_OPTIONS="print_stacktrace=1" \
      ./larson_hakmem_asan "$DUR" "$MIN" "$MAX" "$CHUNKS" "$ROUNDS" "$SEED" "$THREADS" \
      >"$OUT" 2>"$ERR" || true
    ;;
  ubsan)
    make -s ubsan-larson >/dev/null || exit 1
    UBSAN_OPTIONS="print_stacktrace=1" ./larson_hakmem_ubsan "$DUR" "$MIN" "$MAX" "$CHUNKS" "$ROUNDS" "$SEED" "$THREADS" \
      >"$OUT" 2>"$ERR" || true
    ;;
  tsan)
    make -s tsan-larson >/dev/null || exit 1
    TSAN_OPTIONS="history_size=7,halt_on_error=1" ./larson_hakmem_tsan "$DUR" "$MIN" "$MAX" "$CHUNKS" "$ROUNDS" "$SEED" "$THREADS" \
      >"$OUT" 2>"$ERR" || true
    ;;
  best)
    # Source best-known Tiny Hot profile if present
    if [[ -f scripts/profiles/tinyhot_best.env ]]; then
      # shellcheck disable=SC1091
      . scripts/profiles/tinyhot_best.env
      echo "[Claude Runner] loaded scripts/profiles/tinyhot_best.env" >&2
    else
      echo "[Claude Runner] best profile not found; proceeding with defaults" >&2
    fi
    ;;
  *)
    echo "Unknown mode: $MODE" >&2; exit 1;
    ;;
esac

if [[ "$MODE" != asan && "$MODE" != ubsan && "$MODE" != tsan ]]; then
  echo "[Claude Runner] mode=$MODE dur=$DUR threads=$THREADS" | tee "$OUT"
  echo "[Claude Runner] SS_LG=$HAKMEM_TINY_SS_FORCE_LG FAST_CAP=$HAKMEM_TINY_FAST_CAP GUARD=${HAKMEM_TINY_DEBUG_REMOTE_GUARD:-0} TRACE_RING=$HAKMEM_TINY_TRACE_RING" | tee -a "$OUT"
  if [[ "$MODE" == "debug" ]]; then
    ./larson_hakmem "$DUR" "$MIN" "$MAX" "$CHUNKS" "$ROUNDS" "$SEED" "$THREADS" >"$OUT" 2>"$ERR" &
    pid=$!
    sleep 0.08; kill -USR2 $pid || true
    sleep 0.08; kill -USR2 $pid || true
    sleep 0.08; kill -USR2 $pid || true
    wait $pid || true
  else
    ./larson_hakmem "$DUR" "$MIN" "$MAX" "$CHUNKS" "$ROUNDS" "$SEED" "$THREADS" >"$OUT" 2>"$ERR" || true
  fi
fi

echo "--- stderr (tail 120) ---"; tail -n 120 "$ERR" || true
echo "--- stdout (tail 40) ---"; tail -n 40 "$OUT" || true

# Quick greps to surface key signals
echo "--- grep: remote_invalid/REMOTE_SENTINEL/OWNER events ---"
rg -n "remote_invalid|REMOTE_SENTINEL|REMOTE_TRACK_MISMATCH|owner_acq|owner_rel|SIGUSR2 dump" "$ERR" || true

echo "Logs: $OUT $ERR"