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

# Claude Code quick runner for Larson + Tiny (with reproducible env presets)
# Usage:
#   scripts/run_larson_claude.sh [tput|pf|repro|fast0|guard|debug|asan|ubsan|tsan] [DUR] [THREADS]
# Examples:
#   scripts/run_larson_claude.sh tput 2 4
#   scripts/run_larson_claude.sh repro 2 4     # FAST_CAP=0 + TraceRing
#   scripts/run_larson_claude.sh fast0 2 4     # repro + front bypass
#   scripts/run_larson_claude.sh guard 2 4     # repro + remote guard
#   scripts/run_larson_claude.sh debug 10 4    # repro + periodic SIGUSR2 dumps

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

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

# Ensure build (honor 3-layer/route build knobs)
# HAKMEM_BUILD_3LAYER=1    → make larson_hakmem_3layer
# HAKMEM_BUILD_ROUTE=1      → make larson_hakmem_route (implies 3-layer)
if [[ "${HAKMEM_BUILD_ROUTE:-0}" == "1" ]]; then
  make -s larson_hakmem_route >/dev/null
elif [[ "${HAKMEM_BUILD_3LAYER:-0}" == "1" ]]; then
  make -s larson_hakmem_3layer >/dev/null
else
  [[ -x ./larson_hakmem ]] || make -s larson_hakmem >/dev/null
fi

# Common Tiny + Larson envs
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=256

# Defaults used by all modes unless overridden below
export HAKMEM_TINY_FAST_CAP=${HAKMEM_TINY_FAST_CAP:-64}
export HAKMEM_TINY_FAST_SPARE_PERIOD=${HAKMEM_TINY_FAST_SPARE_PERIOD:-8}
export HAKMEM_TINY_TLS_LIST=${HAKMEM_TINY_TLS_LIST:-1}

# Fixed Larson args used across scripts (set early for set -u safety)
MIN=8; MAX=128; CHUNKS=1024; ROUNDS=1; SEED=12345

case "$MODE" in
  tput)
    export HAKMEM_TINY_SS_FORCE_LG=${HAKMEM_TINY_SS_FORCE_LG:-21}
    export HAKMEM_TINY_SS_CACHE=${HAKMEM_TINY_SS_CACHE:-0}
    export HAKMEM_TINY_SS_PRECHARGE=${HAKMEM_TINY_SS_PRECHARGE:-0}
    # Opportunistic background remote drain (lightweight)
    export HAKMEM_TINY_BG_REMOTE=${HAKMEM_TINY_BG_REMOTE:-1}
    export HAKMEM_TINY_BG_REMOTE_TRYRATE=${HAKMEM_TINY_BG_REMOTE_TRYRATE:-16}
    export HAKMEM_TINY_BG_REMOTE_BUDGET=${HAKMEM_TINY_BG_REMOTE_BUDGET:-2}
    ;;
  pf)
    export HAKMEM_TINY_SS_FORCE_LG=${HAKMEM_TINY_SS_FORCE_LG:-20}
    export HAKMEM_TINY_SS_CACHE=${HAKMEM_TINY_SS_CACHE:-4}
    export HAKMEM_TINY_SS_PRECHARGE=${HAKMEM_TINY_SS_PRECHARGE:-1}
    export HAKMEM_TINY_BG_REMOTE=${HAKMEM_TINY_BG_REMOTE:-1}
    export HAKMEM_TINY_BG_REMOTE_TRYRATE=${HAKMEM_TINY_BG_REMOTE_TRYRATE:-8}
    export HAKMEM_TINY_BG_REMOTE_BUDGET=${HAKMEM_TINY_BG_REMOTE_BUDGET:-4}
    ;;
  repro)
    export HAKMEM_TINY_SS_FORCE_LG=${HAKMEM_TINY_SS_FORCE_LG:-21}
    export HAKMEM_TINY_FAST_CAP=0
    export HAKMEM_TINY_TRACE_RING=1
    export HAKMEM_TINY_RF_TRACE=${HAKMEM_TINY_RF_TRACE:-1}
    export HAKMEM_TINY_SS_ADOPT=1
    # Force notify to surface publish even if slab_listed was missed
    export HAKMEM_TINY_RF_FORCE_NOTIFY=${HAKMEM_TINY_RF_FORCE_NOTIFY:-1}
    export HAKMEM_TINY_BG_REMOTE=${HAKMEM_TINY_BG_REMOTE:-1}
    export HAKMEM_TINY_BG_REMOTE_TRYRATE=${HAKMEM_TINY_BG_REMOTE_TRYRATE:-4}
    export HAKMEM_TINY_BG_REMOTE_BUDGET=${HAKMEM_TINY_BG_REMOTE_BUDGET:-2}
    ;;
  fast0)
    export HAKMEM_TINY_SS_FORCE_LG=${HAKMEM_TINY_SS_FORCE_LG:-21}
    export HAKMEM_TINY_FAST_CAP=0
    export HAKMEM_TINY_TRACE_RING=1
    export HAKMEM_TINY_RF_TRACE=${HAKMEM_TINY_RF_TRACE:-1}
    export HAKMEM_TINY_DEBUG_FAST0=1
    export HAKMEM_TINY_SS_ADOPT=1
    export HAKMEM_TINY_RF_FORCE_NOTIFY=${HAKMEM_TINY_RF_FORCE_NOTIFY:-1}
    export HAKMEM_TINY_BG_REMOTE=${HAKMEM_TINY_BG_REMOTE:-1}
    export HAKMEM_TINY_BG_REMOTE_TRYRATE=${HAKMEM_TINY_BG_REMOTE_TRYRATE:-4}
    export HAKMEM_TINY_BG_REMOTE_BUDGET=${HAKMEM_TINY_BG_REMOTE_BUDGET:-2}
    ;;
  guard)
    export HAKMEM_TINY_SS_FORCE_LG=${HAKMEM_TINY_SS_FORCE_LG:-21}
    export HAKMEM_TINY_FAST_CAP=0
    export HAKMEM_TINY_TRACE_RING=1
    export HAKMEM_TINY_RF_TRACE=${HAKMEM_TINY_RF_TRACE:-1}
    export HAKMEM_TINY_DEBUG_REMOTE_GUARD=1
    export HAKMEM_TINY_FORCE_REMOTE=1
    export HAKMEM_TINY_SS_ADOPT=1
    export HAKMEM_TINY_RF_FORCE_NOTIFY=${HAKMEM_TINY_RF_FORCE_NOTIFY:-1}
    export HAKMEM_SAFE_FREE=${HAKMEM_SAFE_FREE:-1}
    export HAKMEM_SAFE_FREE_STRICT=${HAKMEM_SAFE_FREE_STRICT:-1}
    export HAKMEM_TINY_BG_REMOTE=${HAKMEM_TINY_BG_REMOTE:-1}
    export HAKMEM_TINY_BG_REMOTE_TRYRATE=${HAKMEM_TINY_BG_REMOTE_TRYRATE:-4}
    export HAKMEM_TINY_BG_REMOTE_BUDGET=${HAKMEM_TINY_BG_REMOTE_BUDGET:-2}
    ;;
  debug)
    export HAKMEM_TINY_SS_FORCE_LG=${HAKMEM_TINY_SS_FORCE_LG:-21}
    export HAKMEM_TINY_FAST_CAP=0
    export HAKMEM_TINY_TRACE_RING=1
    export HAKMEM_TINY_RF_TRACE=${HAKMEM_TINY_RF_TRACE:-1}
    export HAKMEM_TINY_SS_ADOPT=1
    export HAKMEM_TINY_RF_FORCE_NOTIFY=${HAKMEM_TINY_RF_FORCE_NOTIFY:-1}
    export HAKMEM_SAFE_FREE=${HAKMEM_SAFE_FREE:-1}
    export HAKMEM_SAFE_FREE_STRICT=${HAKMEM_SAFE_FREE_STRICT:-1}
    export HAKMEM_TINY_BG_REMOTE=${HAKMEM_TINY_BG_REMOTE:-1}
    export HAKMEM_TINY_BG_REMOTE_TRYRATE=${HAKMEM_TINY_BG_REMOTE_TRYRATE:-4}
    export HAKMEM_TINY_BG_REMOTE_BUDGET=${HAKMEM_TINY_BG_REMOTE_BUDGET:-2}
    ;;
  asan)
    make -s asan-larson >/dev/null || exit 1
    export HAKMEM_TINY_SS_FORCE_LG=${HAKMEM_TINY_SS_FORCE_LG:-21}
    export HAKMEM_TINY_FAST_CAP=0
    export HAKMEM_TINY_TRACE_RING=1
    export HAKMEM_TINY_RF_TRACE=${HAKMEM_TINY_RF_TRACE:-0}
    export HAKMEM_TINY_SS_ADOPT=1
    export HAKMEM_TINY_RF_FORCE_NOTIFY=${HAKMEM_TINY_RF_FORCE_NOTIFY:-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"
    exit 0
    ;;
  ubsan)
    make -s ubsan-larson >/dev/null || exit 1
    UBSAN_OPTIONS="print_stacktrace=1" ./larson_hakmem_ubsan "$DUR" "$MIN" "$MAX" "$CHUNKS" "$ROUNDS" "$SEED" "$THREADS"
    exit 0
    ;;
  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"
    exit 0
    ;;
  *)
    echo "Unknown mode: $MODE" >&2; exit 1;
    ;;
esac

echo "[Claude Runner] mode=$MODE dur=$DUR threads=$THREADS"
echo "[Claude Runner] SS_LG=$HAKMEM_TINY_SS_FORCE_LG FAST_CAP=$HAKMEM_TINY_FAST_CAP FAST0=${HAKMEM_TINY_DEBUG_FAST0:-0} GUARD=${HAKMEM_TINY_DEBUG_REMOTE_GUARD:-0} TRACE_RING=${HAKMEM_TINY_TRACE_RING:-0}"

if [[ "$MODE" == "debug" ]]; then
  # Run in background and poke SIGUSR2 a few times
  ./larson_hakmem "$DUR" "$MIN" "$MAX" "$CHUNKS" "$ROUNDS" "$SEED" "$THREADS" > /tmp/larson_dbg.out 2> /tmp/larson_dbg.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
  echo "--- stderr (tail) ---"; tail -n 120 /tmp/larson_dbg.err || true
  echo "--- stdout (tail) ---"; tail -n 40 /tmp/larson_dbg.out || true
else
  ./larson_hakmem "$DUR" "$MIN" "$MAX" "$CHUNKS" "$ROUNDS" "$SEED" "$THREADS"
fi