#!/bin/bash

# P4 finding: `oak switch B` SIGKILLed mid-materialization leaves a half-switched

# worktree. The store still says branch A (sqlite3 metadata), status reports the

# other branch's files as the user's uncommitted changes, and re-running

# `oak switch B` refuses ("Working directory has uncommitted changes", exit 4)

# with no recovery guidance. Committing now would record B's partial tree on A.

# Usage: switch-kill.sh <oak> <outdir> [kill_after_seconds]

set -u

OAK=$1; OUT=$2; DELAY=${3:-0.4}

rm -rf "$OUT"; mkdir -p "$OUT/home" "$OUT/a"

export HOME=$OUT/home OAK_NO_UPDATE_CHECK=1 OAK_API_KEY=p4tok OAK_AUTHOR=p4

unset OAK_REMOTE OAK_REPO OAK_TOKEN

PORT=$(python3 -c 'import socket;s=socket.socket();s.bind(("127.0.0.1",0));print(s.getsockname()[1])')

"$OAK" serve --dir "$OUT/srv" --port $PORT --token p4tok >"$OUT/serve.log" 2>&1 & SP=$!

trap 'kill $SP 2>/dev/null' EXIT

for i in $(seq 1 100); do curl -s -o /dev/null http://127.0.0.1:$PORT/api/capabilities && break; sleep 0.05; done

cd "$OUT/a"; "$OAK" init . >/dev/null 2>&1

for i in $(seq 0 199); do mkdir -p d$((i%10)); printf 'base %s\n%.0s' $i $(seq 1 40) > d$((i%10))/f$i.txt; done

"$OAK" commit >/dev/null 2>&1; "$OAK" push --repo p4/r -r http://127.0.0.1:$PORT >/dev/null 2>&1 || { echo push failed; exit 2; }

A=$(sqlite3 .oak/oak.db "select value from metadata where key='current_branch'")

"$OAK" switch -c side >/dev/null 2>&1

for i in $(seq 0 199); do printf 'side %s\n%.0s' $i $(seq 1 40) > d$((i%10))/f$i.txt; done

"$OAK" commit >/dev/null 2>&1; "$OAK" switch "$A" >/dev/null 2>&1

"$OAK" switch side >/dev/null 2>&1 & P=$!

python3 -c "import time;time.sleep($DELAY)"; kill -9 $P 2>/dev/null && echo "killed switch after ${DELAY}s"; wait $P 2>/dev/null

echo "store current_branch: $(sqlite3 .oak/oak.db "select value from metadata where key='current_branch'")"

echo "files with side content: $(grep -l '^side' d*/f*.txt | wc -l | tr -d ' ') / 200"

"$OAK" status --json | python3 -c 'import sys,json;d=json.load(sys.stdin);print("status branch:",d["branch"],"changes:",len(d["working_changes"]["changes"]))'

"$OAK" switch side; echo "re-run switch exit=$?"