@siuver/omp-debug-mode 0.1.4 → 0.1.5

package/src/debug-mode.ts

@@ -2,8 +2,8 @@ import { Text } from "@oh-my-pi/pi-coding-agent";
 import type { ExtensionAPI, ExtensionContext, MessageRenderer } from "@oh-my-pi/pi-coding-agent";
 import * as fs from "node:fs";
 import * as path from "node:path";
-import { addEvidenceArtifact, describeEvidence, parseEvidencePlan } from "./evidence";
-import { decideGate } from "./gate";
+import { describeEvidence, validateEvidenceArtifact } from "./evidence";
+import { describeOpenReason } from "./gate";
 import {
 	ACTIVE_LOG_FILE,
 	JsonlLineCounter,
@@ -12,30 +12,26 @@ import {
 	readJsonlLines,
 	summarizeHypotheses,
 } from "./log-files";
-import { describeLedger, recordProbes, syncLedger } from "./probes";
-import {
-	CLEANUP_CONTRACT,
-	METHODOLOGY,
-	PROCEED_REMINDER,
-	buildFixedMessage,
-	buildProceedMessage,
-	buildStartMessage,
-	extractAssistantText,
-	extractReproductionSteps,
-} from "./methodology";
+import { type DebugEvent, type Effect, PROMPT_FIXED, PROMPT_PROCEED, PROMPT_START, reduce } from "./machine";
+import { CLEANUP_CONTRACT, METHODOLOGY, extractAssistantText } from "./methodology";
+import { type LedgerScan, describeLedger, probesInInput, scanLedger, survivingProbes } from "./probes";
 import {
+	type DebugSession,
+	type DebugState,
 	DEBUG_CONTEXT_TYPE,
 	DEBUG_ENTRY,
-	type DebugState,
-	type EvidenceObservation,
+	INACTIVE,
+	activeRunId,
 	blackboard,
 	compareRunIds,
+	currentRound,
 	evidenceSummary,
-	freshState,
+	evidenceView,
+	hasNonProbeEvidence,
 	keepLatestCustomType,
 	logFileFor,
-	pendingEvidenceRequests,
-	replaceRoundEvidenceRequests,
+	pendingRequests,
+	reviveState,
 } from "./state";
 import { registerDebugTools } from "./tools";
 import { applyUi } from "./ui";
@@ -50,9 +46,9 @@ const COMMAND_STATUS = "debug-status";
 
 /** Compact transcript lines for the prompts this extension injects. */
 const MESSAGE_SUMMARIES: Record<string, string> = {
-	"debug-mode-start": "debug mode started — hypotheses and instrumentation",
-	"debug-mode-proceed": "proceed — analyzing captured logs",
-	"debug-mode-fixed": "marked fixed — removing probes and summarizing",
+	[PROMPT_START]: "debug mode started — hypotheses and instrumentation",
+	[PROMPT_PROCEED]: "proceed — analyzing captured logs",
+	[PROMPT_FIXED]: "marked fixed — removing probes and summarizing",
 };
 
 interface DebugMessageDetails {
@@ -60,7 +56,7 @@ interface DebugMessageDetails {
 }
 
 export function registerDebugMode(pi: ExtensionAPI): void {
-	const state: DebugState = freshState();
+	let state: DebugState = INACTIVE;
 	let uiCtx: ExtensionContext | null = null;
 	let watchedLogFile: string | null = null;
 	const lineCounter = new JsonlLineCounter();
@@ -73,79 +69,142 @@ export function registerDebugMode(pi: ExtensionAPI): void {
 		pi.registerMessageRenderer<DebugMessageDetails>(customType, renderer);
 	}
 
-	// ============================== log files ==============================
+	// ============================== state plumbing ==============================
 
-	function initializeLogDirectory(ctx: ExtensionContext): boolean {
-		const dir = debugDirFor(ctx.cwd, ctx.sessionManager.getSessionId());
-		try {
-			fs.mkdirSync(dir, { recursive: true });
-			state.debugDir = dir;
-			excludeDebugLogsFromGit(ctx.cwd);
-			return true;
-		} catch (err) {
-			pi.logger.error("debug-mode: cannot create log dir", { dir, err });
-			state.debugDir = null;
-			return false;
+	/**
+	 * Cache-only events carry no effects, so they may run inside a render pass
+	 * without re-entering the UI refresh that asked for them.
+	 */
+	function absorbCache(event: Extract<DebugEvent, { t: "runs_observed" | "ledger_synced" }>): void {
+		state = reduce(state, event).state;
+	}
+
+	/** Run one event through the machine, then realise whatever it asked for. */
+	function dispatch(event: DebugEvent, ctx: ExtensionContext): string | null {
+		const transition = reduce(state, event);
+		// The reducer returns the previous object for a genuine no-op, so a turn
+		// full of tool calls does not append a state snapshot per message.
+		const changed = transition.state !== state;
+		state = transition.state;
+		if (changed) pi.appendEntry(DEBUG_ENTRY, { ...state });
+		refreshUi();
+		watchLogFile();
+		let continueContext: string | null = null;
+		for (const effect of transition.effects) {
+			const context = applyEffect(effect, ctx);
+			if (context) continueContext = context;
+		}
+		return continueContext;
+	}
+
+	function applyEffect(effect: Effect, ctx: ExtensionContext): string | null {
+		switch (effect.kind) {
+			case "notify":
+				ctx.ui.notify(effect.text, effect.level);
+				return null;
+			case "prompt":
+				pi.sendMessage(
+					{
+						customType: effect.customType,
+						content: effect.content,
+						display: true,
+						details: { summary: effect.summary } satisfies DebugMessageDetails,
+					},
+					{ triggerTurn: true },
+				);
+				return null;
+			case "continue":
+				return effect.context;
+			case "teardown": {
+				unwatchLogFile();
+				lineCounter.clear();
+				if (effect.debugDir) {
+					try {
+						fs.rmSync(effect.debugDir, { recursive: true, force: true });
+						pruneDebugRoot(ctx.cwd);
+					} catch (err) {
+						pi.logger.warn("debug-mode: failed to remove debug dir", { err });
+					}
+				}
+				const label = effect.outcome === "finished" ? "finished" : "aborted";
+				if (effect.probesLeft.length > 0) {
+					ctx.ui.notify(
+						`Debug mode ${label}, but ${effect.probesLeft.length} probe(s) remain in code: ${effect.probesLeft.map(p => p.id).join(", ")} — remove manually. Applied fixes remain in the working diff.`,
+						"warning",
+					);
+				} else {
+					ctx.ui.notify(
+						`Debug mode ${label}. Log files removed; applied fixes remain in the working diff for review.`,
+						"info",
+					);
+				}
+				return null;
+			}
 		}
 	}
 
+	function refreshUi(): void {
+		applyUi(uiCtx, state, currentLogCount);
+	}
+
+	// ============================== log files ==============================
+
 	function readRunLines(run: string): string[] {
-		return readJsonlLines(logFileFor(state, run));
+		return state.active ? readJsonlLines(logFileFor(state, run)) : [];
 	}
 
+	/** Re-count every run on disk. Pure cache maintenance — never a decision. */
 	function refreshLogCounts(): void {
-		const runs = new Set(Object.keys(state.logCounts));
-		if (state.debugDir) {
+		if (!state.active) return;
+		const session: DebugSession = state;
+		const active = activeRunId(session);
+		const runs = new Set(Object.keys(session.logCounts));
+		if (session.debugDir) {
 			try {
-				for (const file of fs.readdirSync(state.debugDir)) {
+				for (const file of fs.readdirSync(session.debugDir)) {
 					if (!file.endsWith(".jsonl")) continue;
 					if (file === ACTIVE_LOG_FILE) {
-						if (state.runId) runs.add(state.runId);
+						if (active) runs.add(active);
 					} else {
 						runs.add(file.replace(/\.jsonl$/, ""));
 					}
 				}
 			} catch {}
 		}
-		for (const run of runs) state.logCounts[run] = lineCounter.count(logFileFor(state, run));
-		adoptDiscoveredRuns(runs);
+		const logCounts: Record<string, number> = {};
+		for (const run of runs) logCounts[run] = lineCounter.count(logFileFor(session, run));
+		absorbCache({ t: "runs_observed", runHistory: orderedHistory(session, runs), logCounts });
 	}
 
 	/** Keep the ordered history complete when runs are discovered from disk. */
-	function adoptDiscoveredRuns(runs: Iterable<string>): void {
-		const missing = [...runs].filter(run => run !== state.runId && !state.runHistory.includes(run));
-		if (missing.length === 0) return;
-		const active = state.runId && state.runHistory.includes(state.runId) ? state.runId : null;
-		const completed = state.runHistory.filter(run => run !== active).concat(missing);
+	function orderedHistory(session: DebugSession, runs: Iterable<string>): string[] {
+		const active = activeRunId(session);
+		const missing = [...runs].filter(run => run !== active && !session.runHistory.includes(run));
+		if (missing.length === 0) return session.runHistory;
+		const completed = session.runHistory.filter(run => run !== active).concat(missing);
 		completed.sort(compareRunIds);
-		state.runHistory = active ? [...completed, active] : completed;
+		return active ? [...completed, active] : completed;
 	}
 
 	function currentLogCount(): number {
 		refreshLogCounts();
-		return state.runId ? (state.logCounts[state.runId] ?? 0) : 0;
+		if (!state.active) return 0;
+		const run = activeRunId(state);
+		return run ? (state.logCounts[run] ?? 0) : 0;
 	}
 
-	function newRun(): string | null {
-		if (!state.debugDir) return null;
-		const run = `run${state.round}-${Date.now().toString(36)}`;
+	/** Archive the active log, truncate it for the next reproduction, and name the run. */
+	function createRun(debugDir: string, round: number, previousRun: string | null): string | null {
 		try {
-			prepareRunLog(state.debugDir, state.runId);
+			prepareRunLog(debugDir, previousRun);
 		} catch (err) {
 			pi.logger.error("debug-mode: cannot initialize run log", {
-				file: path.join(state.debugDir, ACTIVE_LOG_FILE),
+				file: path.join(debugDir, ACTIVE_LOG_FILE),
 				err,
 			});
 			return null;
 		}
-		state.runId = run;
-		state.runHistory.push(run);
-		state.logCounts[run] = 0;
-		return run;
-	}
-
-	function refreshUi(): void {
-		applyUi(uiCtx, state, currentLogCount);
+		return `run${round}-${Date.now().toString(36)}`;
 	}
 
 	/**
@@ -154,13 +213,13 @@ export function registerDebugMode(pi: ExtensionAPI): void {
 	 * refresh the widget as observations land.
 	 */
 	function watchLogFile(): void {
-		const file = state.phase === "waiting" ? logFileFor(state) : null;
+		const file = state.active && state.stage === "awaiting_evidence" ? logFileFor(state) : null;
 		if (file === watchedLogFile) return;
 		unwatchLogFile();
 		if (!file || !uiCtx?.hasUI) return;
 		try {
 			fs.watchFile(file, { interval: 1000 }, (curr, prev) => {
-				if (state.phase !== "waiting") return;
+				if (!state.active || state.stage !== "awaiting_evidence") return;
 				if (curr.mtimeMs === prev.mtimeMs && curr.size === prev.size) return;
 				try {
 					refreshUi();
@@ -186,26 +245,35 @@ export function registerDebugMode(pi: ExtensionAPI): void {
 
 	// ============================== probe ledger ==============================
 
+	/** Rescan the recorded files so the ledger matches the code on disk. */
+	async function syncLedger(): Promise<LedgerScan> {
+		const scan = await scanLedger(state.active ? state.probes : []);
+		if (state.active) absorbCache({ t: "ledger_synced", probes: survivingProbes(scan) });
+		return scan;
+	}
+
 	pi.on("tool_call", async (event, ctx) => {
 		if (!state.active) return;
 		if (event.toolName !== "edit" && event.toolName !== "write") return;
-		recordProbes(state.probes, state.round, event.input as Record<string, unknown>, ctx.cwd);
+		const probes = probesInInput(event.input as Record<string, unknown>, ctx.cwd, currentRound(state).index);
+		if (probes.length > 0) dispatch({ t: "probes_found", probes }, ctx);
 	});
 
 	// ============================== prompt injection ==============================
 
 	pi.on("before_agent_start", async () => {
-		if (!state.active || (state.phase !== "round" && state.phase !== "cleanup")) return;
+		if (!state.active || state.stage === "awaiting_evidence") return;
 		// The blackboard claims to be ground truth, so reconcile it with disk first.
-		await syncLedger(state);
+		await syncLedger();
+		if (!state.active) return;
 		// Cleanup has no hypotheses left to form; the full methodology would only
 		// invite another round.
-		const contract = state.phase === "cleanup" ? CLEANUP_CONTRACT : METHODOLOGY;
+		const contract = state.stage === "cleaning_up" ? CLEANUP_CONTRACT : METHODOLOGY;
 		return {
 			message: {
 				customType: DEBUG_CONTEXT_TYPE,
-				content: `${blackboard(state, describeEvidence(state))}\n\n${contract}`,
-                display: false,
+				content: `${blackboard(state, describeEvidence(evidenceView(state)))}\n\n${contract}`,
+				display: false,
 			},
 		};
 	});
@@ -218,304 +286,141 @@ export function registerDebugMode(pi: ExtensionAPI): void {
 		if (filtered.length !== event.messages.length) return { messages: filtered };
 	});
 
-	// ============================== round lifecycle ==============================
+	// ============================== turn lifecycle ==============================
 
-	pi.on("agent_start", async () => {
-		if (state.active) state.hasRoundContent = false;
+	pi.on("agent_start", async (_event, ctx) => {
+		if (state.active) dispatch({ t: "turn_started" }, ctx);
 	});
 
-	pi.on("message_end", async (event) => {
+	pi.on("message_end", async (event, ctx) => {
 		if (!state.active) return;
 		const msg = event.message as { role?: string; content?: unknown };
-		if (msg?.role === "assistant") {
-			if (state.phase === "cleanup") state.cleanupReady = true;
-			state.hasRoundContent = true;
-			const text = extractAssistantText(msg.content);
-			const steps = extractReproductionSteps(text);
-			if (steps.length > 0) state.reproductionSteps = steps;
-			if (state.phase === "round") {
-				const plan = parseEvidencePlan(text, state.round);
-				if (plan.found) {
-					replaceRoundEvidenceRequests(state, state.round, plan.valid ? plan.requests : []);
-					pi.appendEntry(DEBUG_ENTRY, { ...state });
-				}
-			}
-		}
+		if (msg?.role !== "assistant") return;
+		dispatch({ t: "assistant_message", text: extractAssistantText(msg.content) }, ctx);
 	});
 
 	pi.on("session_stop", async (_event, ctx) => {
-		if (!state.active) return;
-		if (state.phase === "cleanup") {
-			if (state.cleanupReady) await teardown(ctx, "finished");
-			return;
-		}
-		// Ordinary editor messages while waiting are conversational only. Only an
-		// explicit debug command advances the run or leaves the gate.
-		if (state.phase !== "round" || !state.hasRoundContent) return;
-		const probesThisRound = state.probes.filter(p => p.round === state.round).length;
-		const hasEvidencePlan = state.evidenceRequests.some(request => request.round === state.round);
-		const decision = decideGate({
-			hasReproductionSteps: state.reproductionSteps.length > 0,
-			hasEvidencePlan,
-			probesThisRound,
-			nudgesUsed: state.gateNudges,
-		});
-		if (decision.kind === "stay") return;
-		if (decision.kind === "nudge") {
-			state.gateNudges += 1;
-			return { continue: true, additionalContext: decision.context };
-		}
-		enterGate(ctx, probesThisRound, decision.missingSteps, decision.missingEvidencePlan);
+		if (!state.active || !state.turnProduced) return;
+		if (state.stage !== "investigating" && state.stage !== "cleaning_up") return;
+		// A declared runtime_probe must be backed by a marker that is really on
+		// disk, so the ledger is reconciled before the round may close.
+		await syncLedger();
+		const context = dispatch({ t: "turn_settled" }, ctx);
+		if (context) return { continue: true, additionalContext: context };
 	});
 
-function enterGate(
-	ctx: ExtensionContext,
-	probesThisRound: number,
-	missingSteps: boolean,
-	missingEvidencePlan: boolean,
-): void {
-	state.phase = "waiting";
-	pi.appendEntry(DEBUG_ENTRY, { ...state });
-	refreshUi();
-	watchLogFile();
-	// Commands are the only interaction at the gate: the widget lists them,
-	// the user picks one. No menu is opened automatically.
-	const pending = pendingEvidenceRequests(state, state.round);
-	const hasValidPlan = state.evidenceRequests.some(request => request.round === state.round);
-	if (probesThisRound === 0 && !hasValidPlan) {
-		ctx.ui.notify(
-			`Debug round ${state.round} paused, but it added no probes and declared no evidence plan — this round cannot produce runtime evidence. Use /${COMMAND_PROCEED} to ask for instrumentation.`,
-			"warning",
-		);
-	} else if (missingSteps) {
-		ctx.ui.notify(
-			`Debug round ${state.round} paused without reproduction steps. Exercise the instrumented path, then ${PROCEED_REMINDER}`,
-			"warning",
-		);
-	} else if (pending.length > 0) {
-		ctx.ui.notify(
-			`Debug round ${state.round} paused. User evidence requested (${pending.length} pending: ${pending.map(r => r.id).join(", ")}) — attach via /${COMMAND_EVIDENCE} <request-id> <path>, then ${PROCEED_REMINDER}`,
-			"info",
-		);
-	} else if (missingEvidencePlan) {
-		ctx.ui.notify(
-			`Debug round ${state.round} paused. The gate is usable, but the next Proceed will ask the model to declare an evidence method.`,
-			"info",
-		);
-	} else {
-		ctx.ui.notify(`Debug round ${state.round} paused. Reproduce the bug, then ${PROCEED_REMINDER}`, "info");
-	}
-}
+	registerDebugTools(pi, { getState: () => state, refreshLogCounts, readRunLines, syncLedger });
 
-	// ============================== round transitions ==============================
+	// ============================== command guards ==============================
 
-	function startDebug(ctx: ExtensionContext, problem: string): void {
-		if (!initializeLogDirectory(ctx)) {
-			ctx.ui.notify("debug-mode: could not create the debug log directory; debug mode was not started", "error");
-			return;
-		}
-		state.active = true;
-		state.phase = "round";
-		state.problem = problem;
-		state.round = 1;
-		state.probes = [];
-		state.runHistory = [];
-		state.logCounts = {};
-		state.hasRoundContent = false;
-		state.cleanupReady = false;
-		state.reproductionSteps = [];
-		state.gateNudges = 0;
-		state.evidenceRequests = [];
-		state.evidenceArtifacts = [];
-		state.evidenceObservations = [];
-		if (!newRun()) {
-			Object.assign(state, freshState());
-			ctx.ui.notify("debug-mode: could not initialize the run log file; debug mode was not started", "error");
-			return;
-		}
-		refreshUi();
-		const logFile = logFileFor(state);
-		if (!logFile) {
-			Object.assign(state, freshState());
-			ctx.ui.notify("debug-mode: could not resolve the run log file; debug mode was not started", "error");
-			return;
+	function activeSession(ctx: ExtensionContext): DebugSession | null {
+		if (!state.active) {
+			ctx.ui.notify("debug-mode: not active", "error");
+			return null;
 		}
-		pi.sendMessage(
-			{
-				customType: "debug-mode-start",
-				content: buildStartMessage(problem, logFile),
-				display: true,
-				details: { summary: "debug mode started — hypotheses and instrumentation" } satisfies DebugMessageDetails,
-			},
-			{ triggerTurn: true },
-		);
+		return state;
 	}
 
-	function ensureWaiting(ctx: ExtensionContext): boolean {
-		if (!state.active) {
-			ctx.ui.notify("debug-mode: not active", "error");
+	function ownsTurn(session: DebugSession): boolean {
+		return session.stage === "awaiting_evidence" || session.stage === "open";
+	}
+
+	/**
+	 * The debug commands belong to the user's turn. An unclosed round is still
+	 * the user's move, so it gets a confirmation rather than the flat "not
+	 * waiting for reproduction" refusal that made an open round look stuck.
+	 */
+	async function claimTurn(ctx: ExtensionContext, action: string): Promise<boolean> {
+		const session = activeSession(ctx);
+		if (!session) return false;
+		if (!ownsTurn(session)) {
+			ctx.ui.notify(
+				`debug-mode: the agent still has round ${currentRound(session).index} (stage: ${session.stage}) — wait for it to stop.`,
+				"error",
+			);
 			return false;
 		}
-		if (state.phase !== "waiting") {
-			ctx.ui.notify(`debug-mode: not waiting for reproduction (phase: ${state.phase})`, "error");
-			return false;
+		if (session.stage === "open" && ctx.hasUI) {
+			const confirmed = await ctx.ui.confirm(
+				`${action} an unclosed round?`,
+				`${describeOpenReason(currentRound(session).openReason ?? "awaiting_reply", currentRound(session).index)} Continue with the evidence that exists?`,
+			);
+			if (!confirmed) return false;
 		}
 		return true;
 	}
 
-    async function markDebugFixed(ctx: ExtensionContext): Promise<void> {
-		if (!ensureWaiting(ctx)) return;
-		const logCount = currentLogCount();
-		const hasNonProbeEvidence = state.evidenceRequests.some(
-			request => request.round === state.round && request.method !== "runtime_probe",
-		);
-		if (logCount === 0 && !hasNonProbeEvidence && ctx.hasUI) {
-			const confirmed = await ctx.ui.confirm(
-				"Mark as fixed without runtime logs?",
-				"No runtime observations were captured for this round. Mark the problem as fixed anyway?",
-			);
-			if (!confirmed) return;
-		}
-		if (!state.active || state.phase !== "waiting") return;
+	// ============================== round transitions ==============================
 
-		state.phase = "cleanup";
-		state.cleanupReady = false;
-		unwatchLogFile();
-		refreshUi();
-		const evidenceJson = JSON.stringify({
-			requests: state.evidenceRequests,
-			observations: state.evidenceObservations,
-			artifacts: state.evidenceArtifacts,
-		});
-		pi.sendMessage(
-			{
-				customType: "debug-mode-fixed",
-				content: buildFixedMessage(JSON.stringify(state.probes), evidenceJson),
-				details: { summary: `marked fixed — removing ${state.probes.length} probe(s)` } satisfies DebugMessageDetails,
-			},
-			{ triggerTurn: true },
-		);
-    }
+	function startDebug(ctx: ExtensionContext, problem: string): void {
+		const debugDir = debugDirFor(ctx.cwd, ctx.sessionManager.getSessionId());
+		try {
+			fs.mkdirSync(debugDir, { recursive: true });
+			excludeDebugLogsFromGit(ctx.cwd);
+		} catch (err) {
+			pi.logger.error("debug-mode: cannot create log dir", { dir: debugDir, err });
+			ctx.ui.notify("debug-mode: could not create the debug log directory; debug mode was not started", "error");
+			return;
+		}
+		const runId = createRun(debugDir, 1, null);
+		if (!runId) {
+			ctx.ui.notify("debug-mode: could not initialize the run log file; debug mode was not started", "error");
+			return;
+		}
+		dispatch({ t: "start", problem, debugDir, runId, logFile: path.join(debugDir, ACTIVE_LOG_FILE) }, ctx);
+	}
 
 	async function advanceDebug(ctx: ExtensionContext, userDetails?: string): Promise<void> {
-		if (!ensureWaiting(ctx)) return;
+		if (!(await claimTurn(ctx, "Proceed from"))) return;
 		refreshLogCounts();
-		const run = state.runId ?? "(none)";
-		const logCount = state.logCounts[run] ?? 0;
-		const hasNonProbeEvidence = state.evidenceRequests.some(
-			request => request.round === state.round && request.method !== "runtime_probe",
-		);
-		if (logCount === 0 && !userDetails && !hasNonProbeEvidence && ctx.hasUI) {
+		if (!state.active) return;
+		const closingRun = activeRunId(state);
+		const logCount = closingRun ? (state.logCounts[closingRun] ?? 0) : 0;
+		if (logCount === 0 && !userDetails && !hasNonProbeEvidence(currentRound(state)) && ctx.hasUI) {
 			const confirmed = await ctx.ui.confirm(
 				"Proceed without runtime logs?",
 				"No runtime observations were captured for this round. Continue to log analysis anyway?",
 			);
 			if (!confirmed) return;
 		}
-		if (!state.active || state.phase !== "waiting") return;
-
-		// Persist optional /debug-proceed details as a batched user observation
-		// tied to every current-round user_report request.
-		if (userDetails && userDetails.trim().length > 0) {
-			const reportIds = state.evidenceRequests
-				.filter(request => request.round === state.round && request.method === "user_report")
-				.map(request => request.id);
-			const observation: EvidenceObservation = {
-				id: `observation-${Date.now().toString(36)}`,
-				requestIds: reportIds,
-				text: userDetails.trim(),
-				round: state.round,
-				addedAt: Date.now(),
-			};
-			state.evidenceObservations = [...state.evidenceObservations, observation];
-		}
-		const stillPending = pendingEvidenceRequests(state, state.round);
+		if (!state.active) return;
+		const stillPending = pendingRequests(state);
 		if (stillPending.length > 0 && ctx.hasUI) {
 			const confirmed = await ctx.ui.confirm(
 				"Proceed without all requested evidence?",
 				`${stillPending.length} evidence request(s) still pending (${stillPending.map(r => r.id).join(", ")}). Continue anyway?`,
 			);
-			if (!confirmed) {
-				pi.appendEntry(DEBUG_ENTRY, { ...state });
-				refreshUi();
-				return;
-			}
+			if (!confirmed) return;
 		}
+		if (!state.active || !ownsTurn(state) || !state.debugDir) return;
 
-		const hypotheses = describeHypotheses(summarizeHypotheses(readRunLines(run)));
-		const previousRound = state.round;
-		const previousSteps = state.reproductionSteps;
-		state.round += 1;
-		state.phase = "round";
-		state.hasRoundContent = false;
-		state.reproductionSteps = [];
-		state.gateNudges = 0;
-		if (!newRun()) {
-			state.round = previousRound;
-			state.phase = "waiting";
-			state.reproductionSteps = previousSteps;
-			ctx.ui.notify("debug-mode: could not initialize the next run log file; staying at the reproduction gate", "error");
-			refreshUi();
+		// The digest must be read before the active log is archived and truncated.
+		const hypotheses = describeHypotheses(summarizeHypotheses(closingRun ? readRunLines(closingRun) : []));
+		const runId = createRun(state.debugDir, currentRound(state).index + 1, closingRun);
+		if (!runId) {
+			ctx.ui.notify("debug-mode: could not initialize the next run log file; staying on this round", "error");
 			return;
 		}
-		unwatchLogFile();
-		refreshUi();
-
-		const summary = userDetails
-			? `proceed with user details — analyzing run ${run} (${logCount} entries)`
-			: `proceed — analyzing run ${run} (${logCount} entries)`;
-		pi.sendMessage(
-			{
-				customType: "debug-mode-proceed",
-				content: buildProceedMessage({
-					run,
-					logCount,
-					userDetails,
-					hypotheses,
-					evidenceSummary: evidenceSummary(state, previousRound),
-				}),
-				display: true,
-				details: { summary } satisfies DebugMessageDetails,
-			},
-			{ triggerTurn: true },
-		);
+		dispatch({ t: "proceed", runId, logCount, hypotheses, details: userDetails, now: Date.now() }, ctx);
 	}
 
-	async function teardown(ctx: ExtensionContext, outcome: "finished" | "aborted"): Promise<void> {
-		unwatchLogFile();
-		lineCounter.clear();
-		if (state.debugDir) {
-			try {
-				fs.rmSync(state.debugDir, { recursive: true, force: true });
-				pruneDebugRoot(ctx.cwd);
-			} catch (err) {
-				pi.logger.warn("debug-mode: failed to remove debug dir", { err });
-			}
-		}
-		const scan = await syncLedger(state);
-		const probesLeft = [...scan.alive, ...scan.unknown];
-		Object.assign(state, freshState());
-		pi.appendEntry(DEBUG_ENTRY, { ...state });
-		refreshUi();
-		const resultLabel = outcome === "finished" ? "finished" : "aborted";
-		if (probesLeft.length > 0) {
-			ctx.ui.notify(
-				`Debug mode ${resultLabel}, but ${probesLeft.length} probe(s) remain in code: ${probesLeft.map(p => p.id).join(", ")} — remove manually. Applied fixes remain in the working diff.`,
-				"warning",
-			);
-		} else {
-			ctx.ui.notify(
-				`Debug mode ${resultLabel}. Log files removed; applied fixes remain in the working diff for review.`,
-				"info",
+	async function markDebugFixed(ctx: ExtensionContext): Promise<void> {
+		if (!(await claimTurn(ctx, "Mark as fixed from"))) return;
+		const logCount = currentLogCount();
+		if (!state.active) return;
+		if (logCount === 0 && !hasNonProbeEvidence(currentRound(state)) && ctx.hasUI) {
+			const confirmed = await ctx.ui.confirm(
+				"Mark as fixed without runtime logs?",
+				"No runtime observations were captured for this round. Mark the problem as fixed anyway?",
 			);
+			if (!confirmed) return;
 		}
+		if (!state.active || !ownsTurn(state)) return;
+		dispatch({ t: "mark_fixed" }, ctx);
 	}
 
 	async function abortDebug(ctx: ExtensionContext): Promise<void> {
-		if (!state.active) {
-			ctx.ui.notify("debug-mode: not active", "error");
-			return;
-		}
+		if (!activeSession(ctx)) return;
 		if (ctx.hasUI) {
 			const confirmed = await ctx.ui.confirm(
 				"Abort debug mode?",
@@ -524,16 +429,21 @@ function enterGate(
 			if (!confirmed) return;
 		}
 		if (!state.active) return;
-		await teardown(ctx, "aborted");
+		dispatch({ t: "abort" }, ctx);
 	}
 
 	/**
-	 * Attach one user-provided evidence file to the current waiting round. The
-	 * file is referenced in place — never copied, moved or deleted — and the
-	 * action stays at the gate: no model-visible message, no agent turn.
+	 * Attach one user-provided evidence file to the current round. The file is
+	 * referenced in place — never copied, moved or deleted — and the action does
+	 * not advance the workflow: no model-visible message, no agent turn.
 	 */
 	async function attachEvidence(ctx: ExtensionContext, rawPath?: string, requestId: string | null = null): Promise<boolean> {
-		if (!ensureWaiting(ctx)) return false;
+		const session = activeSession(ctx);
+		if (!session) return false;
+		if (!ownsTurn(session)) {
+			ctx.ui.notify(`debug-mode: cannot attach evidence while the agent is working (stage: ${session.stage})`, "error");
+			return false;
+		}
 		let input = rawPath?.trim() ?? "";
 		if (!input) {
 			if (!ctx.hasUI) {
@@ -564,20 +474,15 @@ function enterGate(
 			);
 			if (!confirmed) return false;
 		}
-		const result = addEvidenceArtifact(state, trimmed, ctx.cwd, requestId);
-		if ("error" in result) {
-			ctx.ui.notify(`debug-mode: evidence rejected — ${result.error}`, "error");
+		const validation = validateEvidenceArtifact(trimmed, ctx.cwd);
+		if (!validation.ok) {
+			ctx.ui.notify(`debug-mode: evidence rejected — ${validation.reason}`, "error");
 			return false;
 		}
-		Object.assign(state, result.state);
-		pi.appendEntry(DEBUG_ENTRY, { ...state });
-		refreshUi();
-		ctx.ui.notify(
-			`debug-mode: attached ${result.artifact.id} → ${result.artifact.path} (${result.artifact.size} bytes)`,
-			"info",
-		);
+		if (!state.active) return false;
+		dispatch({ t: "attach_artifact", candidate: validation.artifact, requestId, now: Date.now() }, ctx);
 		return true;
-}
+	}
 
 	// ============================== commands ==============================
 
@@ -628,9 +533,9 @@ function enterGate(
 		const trimmed = args.trim();
 		if (!trimmed) return { requestId: null, rawPath: undefined };
 		const firstToken = trimmed.split(/\s+/, 1)[0];
-		const isPending = pendingEvidenceRequests(state, state.round).some(
-			request => request.method === "user_artifact" && request.id === firstToken,
-		);
+		const isPending =
+			state.active &&
+			pendingRequests(state).some(request => request.method === "user_artifact" && request.id === firstToken);
 		if (isPending) {
 			const rest = trimmed.slice(firstToken.length).trim();
 			return rest ? { requestId: firstToken, rawPath: rest } : { requestId: firstToken, rawPath: undefined };
@@ -647,7 +552,6 @@ function enterGate(
 		},
 	});
 
-
 	pi.registerCommand(COMMAND_ABORT, {
 		description: "Abort debug mode: delete logs (fixes stay in the working diff)",
 		handler: async (_args, ctx) => {
@@ -665,11 +569,15 @@ function enterGate(
 				return;
 			}
 			refreshLogCounts();
-			const scan = await syncLedger(state);
-			const tallies = state.runId ? summarizeHypotheses(readRunLines(state.runId)) : [];
-			const roundRequests = state.evidenceRequests.filter(request => request.round === state.round);
-			const pending = pendingEvidenceRequests(state, state.round);
-			const unavailable = state.evidenceArtifacts.filter(artifact => {
+			const scan = await syncLedger();
+			if (!state.active) return;
+			const session: DebugSession = state;
+			const round = currentRound(session);
+			const run = activeRunId(session);
+			const tallies = run ? summarizeHypotheses(readRunLines(run)) : [];
+			const requests = round.plan ?? [];
+			const pending = pendingRequests(session, round);
+			const unavailable = session.artifacts.filter(artifact => {
 				try {
 					const stats = fs.statSync(artifact.path);
 					fs.accessSync(artifact.path, fs.constants.R_OK);
@@ -679,25 +587,22 @@ function enterGate(
 				}
 			});
 			ctx.ui.notify(
-				`debug-mode: phase=${state.phase} round=${state.round} run=${state.runId}\n` +
+				`debug-mode: stage=${session.stage} round=${round.index} run=${run ?? "(none)"}\n` +
+					(round.openReason ? `${describeOpenReason(round.openReason, round.index)}\n` : "") +
 					`${describeLedger(scan)}\n` +
-					`logs: ${state.runHistory.map(r => `${r}=${state.logCounts[r] ?? 0}`).join(", ") || "(none)"}\n` +
+					`logs: ${session.runHistory.map(r => `${r}=${session.logCounts[r] ?? 0}`).join(", ") || "(none)"}\n` +
 					`this run by hypothesis: ${describeHypotheses(tallies)}\n` +
-					`current log file: ${logFileFor(state) ?? "(not initialized)"}\n` +
-					`evidence requests: ${roundRequests.map(r => `${r.id}[${r.method}] ${r.title}`).join("; ") || "(none)"}\n` +
+					`current log file: ${logFileFor(session) ?? "(not initialized)"}\n` +
+					`evidence requests: ${requests.map(r => `${r.id}[${r.method}] ${r.title}`).join("; ") || "(none)"}\n` +
 					`pending: ${pending.map(r => r.id).join(", ") || "(none)"}\n` +
-					`observations: ${state.evidenceObservations.map(o => o.id).join(", ") || "(none)"}\n` +
-					`artifacts: ${state.evidenceArtifacts.map(a => a.id).join(", ") || "(none)"}` +
+					`observations: ${session.observations.map(o => o.id).join(", ") || "(none)"}\n` +
+					`artifacts: ${session.artifacts.map(a => a.id).join(", ") || "(none)"}` +
 					(unavailable.length > 0 ? `\nunavailable artifacts: ${unavailable.map(a => `${a.id} ${a.path}`).join(", ")}` : ""),
 				"info",
 			);
 		},
 	});
 
-	// ============================== tools ==============================
-
-	registerDebugTools(pi, { state, refreshLogCounts, readRunLines });
-
 	// ============================== lifecycle ==============================
 
 	pi.on("session_start", async (_event, ctx) => {
@@ -705,21 +610,24 @@ function enterGate(
 		const entries = ctx.sessionManager.getEntries();
 		const last = entries
 			.filter((e: { type: string; customType?: string }) => e.type === "custom" && e.customType === DEBUG_ENTRY)
-			.pop() as { data?: DebugState } | undefined;
-		if (last?.data?.active) {
-			Object.assign(state, last.data);
-			if (!Array.isArray(state.reproductionSteps)) state.reproductionSteps = [];
-			if (!Array.isArray(state.evidenceRequests)) state.evidenceRequests = [];
-			if (!Array.isArray(state.evidenceArtifacts)) state.evidenceArtifacts = [];
-			if (!Array.isArray(state.evidenceObservations)) state.evidenceObservations = [];
-			if (!Array.isArray(state.runHistory)) state.runHistory = state.runId ? [state.runId] : [];
-			if (typeof state.gateNudges !== "number") state.gateNudges = 0;
-			if (!state.phase) state.phase = "waiting";
-			if (!state.debugDir || !fs.existsSync(state.debugDir)) initializeLogDirectory(ctx);
+			.pop() as { data?: unknown } | undefined;
+		state = reviveState(last?.data);
+		if (state.active) {
+			if (!state.debugDir || !fs.existsSync(state.debugDir)) {
+				const debugDir = debugDirFor(ctx.cwd, ctx.sessionManager.getSessionId());
+				try {
+					fs.mkdirSync(debugDir, { recursive: true });
+					excludeDebugLogsFromGit(ctx.cwd);
+					state = { ...state, debugDir };
+				} catch (err) {
+					pi.logger.error("debug-mode: cannot restore log dir", { dir: debugDir, err });
+				}
+			}
 			const currentFile = logFileFor(state);
 			if (currentFile) {
 				try {
-					const legacyRunFile = state.debugDir && state.runId ? path.join(state.debugDir, `${state.runId}.jsonl`) : null;
+					const run = activeRunId(state);
+					const legacyRunFile = state.debugDir && run ? path.join(state.debugDir, `${run}.jsonl`) : null;
 					if (!fs.existsSync(currentFile) && legacyRunFile && fs.existsSync(legacyRunFile)) {
 						fs.renameSync(legacyRunFile, currentFile);
 					}
@@ -729,19 +637,17 @@ function enterGate(
 				}
 			}
 			refreshLogCounts();
-			ctx.ui.notify(
-				`debug-mode resumed: phase=${state.phase} round=${state.round}, log=${currentFile ?? "unavailable"}; evidence: ${evidenceSummary(state, state.round)}. Use /${COMMAND_STATUS}, /${COMMAND_EVIDENCE}, /${COMMAND_DONE}, /${COMMAND_PROCEED}, or /${COMMAND_ABORT}.`,
-				"info",
-			);
+			if (state.active) {
+				ctx.ui.notify(
+					`debug-mode resumed: stage=${state.stage} round=${currentRound(state).index}, log=${currentFile ?? "unavailable"}; evidence: ${evidenceSummary(state)}. Use /${COMMAND_STATUS}, /${COMMAND_EVIDENCE}, /${COMMAND_DONE}, /${COMMAND_PROCEED}, or /${COMMAND_ABORT}.`,
+					"info",
+				);
+			}
 		}
 		refreshUi();
 		watchLogFile();
 	});
 
-	pi.on("turn_start", async () => {
-		if (state.active) pi.appendEntry(DEBUG_ENTRY, { ...state });
-	});
-
 	pi.on("session_shutdown", async () => {
 		unwatchLogFile();
 	});