@siuver/omp-debug-mode 0.1.2 → 0.1.3

package/src/debug-mode.ts

@@ -0,0 +1,638 @@
+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 { decideGate } from "./gate";
+import {
+	ACTIVE_LOG_FILE,
+	JsonlLineCounter,
+	describeHypotheses,
+	prepareRunLog,
+	readJsonlLines,
+	summarizeHypotheses,
+} from "./log-files";
+import {
+	CLEANUP_CONTRACT,
+	METHODOLOGY,
+	PROCEED_REMINDER,
+	buildFixedMessage,
+	buildProceedMessage,
+	buildStartMessage,
+	extractAssistantText,
+	extractReproductionSteps,
+} from "./methodology";
+import { describeLedger, recordProbes, syncLedger } from "./probes";
+import { REVIEW_ABORT, REVIEW_ADD_DETAILS, REVIEW_MARK_FIXED, REVIEW_PROCEED, reviewMenuOptions } from "./review-actions";
+import {
+	DEBUG_CONTEXT_TYPE,
+	DEBUG_ENTRY,
+	type DebugState,
+	blackboard,
+	compareRunIds,
+	freshState,
+	keepLatestCustomType,
+	logFileFor,
+} from "./state";
+import { registerDebugTools } from "./tools";
+import { applyUi, reviewMenuTitle } from "./ui";
+import { debugDirFor, excludeDebugLogsFromGit, pruneDebugRoot } from "./workspace";
+
+const COMMAND_MODE = "debug-mode";
+const COMMAND_MENU = "debug-menu";
+const COMMAND_DONE = "debug-done";
+const COMMAND_PROCEED = "debug-proceed";
+const COMMAND_NOTE = "debug-note";
+const COMMAND_ABORT = "debug-abort";
+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-note": "reproduction details added — analyzing captured logs",
+	"debug-mode-fixed": "marked fixed — removing probes and summarizing",
+};
+
+interface DebugMessageDetails {
+	summary?: string;
+}
+
+export function registerDebugMode(pi: ExtensionAPI): void {
+	const state: DebugState = freshState();
+	let reviewMenuOpen = false;
+	let uiCtx: ExtensionContext | null = null;
+	let watchedLogFile: string | null = null;
+	const lineCounter = new JsonlLineCounter();
+
+	for (const [customType, summary] of Object.entries(MESSAGE_SUMMARIES)) {
+		const renderer: MessageRenderer<DebugMessageDetails> = (message, _options, theme) => {
+			const text = typeof message.details?.summary === "string" ? message.details.summary : summary;
+			return new Text(theme.fg("dim", `🐞 ${text}`), 1, 0);
+		};
+		pi.registerMessageRenderer<DebugMessageDetails>(customType, renderer);
+	}
+
+	// ============================== log files ==============================
+
+	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;
+		}
+	}
+
+	function readRunLines(run: string): string[] {
+		return readJsonlLines(logFileFor(state, run));
+	}
+
+	function refreshLogCounts(): void {
+		const runs = new Set(Object.keys(state.logCounts));
+		if (state.debugDir) {
+			try {
+				for (const file of fs.readdirSync(state.debugDir)) {
+					if (!file.endsWith(".jsonl")) continue;
+					if (file === ACTIVE_LOG_FILE) {
+						if (state.runId) runs.add(state.runId);
+					} else {
+						runs.add(file.replace(/\.jsonl$/, ""));
+					}
+				}
+			} catch {}
+		}
+		for (const run of runs) state.logCounts[run] = lineCounter.count(logFileFor(state, run));
+		adoptDiscoveredRuns(runs);
+	}
+
+	/** 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);
+		completed.sort(compareRunIds);
+		state.runHistory = active ? [...completed, active] : completed;
+	}
+
+	function currentLogCount(): number {
+		refreshLogCounts();
+		return state.runId ? (state.logCounts[state.runId] ?? 0) : 0;
+	}
+
+	function newRun(): string | null {
+		if (!state.debugDir) return null;
+		const run = `run${state.round}-${Date.now().toString(36)}`;
+		try {
+			prepareRunLog(state.debugDir, state.runId);
+		} catch (err) {
+			pi.logger.error("debug-mode: cannot initialize run log", {
+				file: path.join(state.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);
+	}
+
+	/**
+	 * While the user reproduces out-of-band nothing in the session fires, so the
+	 * evidence counter would sit at zero. Poll the active log for appends and
+	 * refresh the widget as observations land.
+	 */
+	function watchLogFile(): void {
+		const file = state.phase === "waiting" ? 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 (curr.mtimeMs === prev.mtimeMs && curr.size === prev.size) return;
+				try {
+					refreshUi();
+				} catch (err) {
+					pi.logger.warn("debug-mode: log watch refresh failed", { err });
+				}
+			});
+			watchedLogFile = file;
+		} catch (err) {
+			pi.logger.warn("debug-mode: cannot watch the run log", { file, err });
+		}
+	}
+
+	function unwatchLogFile(): void {
+		if (!watchedLogFile) return;
+		try {
+			fs.unwatchFile(watchedLogFile);
+		} catch (err) {
+			pi.logger.warn("debug-mode: cannot stop watching the run log", { file: watchedLogFile, err });
+		}
+		watchedLogFile = null;
+	}
+
+	// ============================== probe ledger ==============================
+
+	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);
+	});
+
+	// ============================== prompt injection ==============================
+
+	pi.on("before_agent_start", async () => {
+		if (!state.active || (state.phase !== "round" && state.phase !== "cleanup")) return;
+		// The blackboard claims to be ground truth, so reconcile it with disk first.
+		await syncLedger(state);
+		// Cleanup has no hypotheses left to form; the full methodology would only
+		// invite another round.
+		const contract = state.phase === "cleanup" ? CLEANUP_CONTRACT : METHODOLOGY;
+		return {
+			message: {
+				customType: DEBUG_CONTEXT_TYPE,
+				content: `${blackboard(state)}\n\n${contract}`,
+				display: false,
+			},
+		};
+	});
+
+	// Drop stale blackboard copies. Keep the newest — `context` runs after
+	// `before_agent_start`, so deleting every match would hide the injection
+	// from the model.
+	pi.on("context", async (event) => {
+		const filtered = keepLatestCustomType(event.messages, DEBUG_CONTEXT_TYPE);
+		if (filtered.length !== event.messages.length) return { messages: filtered };
+	});
+
+	// ============================== round lifecycle ==============================
+
+	pi.on("agent_start", async () => {
+		if (state.active) state.hasRoundContent = false;
+	});
+
+	pi.on("message_end", async (event) => {
+		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 steps = extractReproductionSteps(extractAssistantText(msg.content));
+			if (steps.length > 0) state.reproductionSteps = steps;
+		}
+	});
+
+	pi.on("session_stop", async (_event, ctx) => {
+		if (!state.active) return;
+		if (state.phase === "cleanup") {
+			// Cleanup turn settled: keep fixes and remove the temporary logs.
+			if (state.cleanupReady) await teardown(ctx, "finished");
+			return;
+		}
+		if (state.phase !== "round" || !state.hasRoundContent) return;
+
+		const probesThisRound = state.probes.filter(p => p.round === state.round).length;
+		const decision = decideGate({
+			hasReproductionSteps: state.reproductionSteps.length > 0,
+			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);
+	});
+
+	function enterGate(ctx: ExtensionContext, probesThisRound: number, missingSteps: boolean): void {
+		state.phase = "waiting";
+		pi.appendEntry(DEBUG_ENTRY, { ...state });
+		refreshUi();
+		watchLogFile();
+		scheduleReviewMenu(ctx);
+		if (probesThisRound === 0) {
+			ctx.ui.notify(
+				`Debug round ${state.round} paused, but it added no probes — this round cannot produce runtime evidence. Use /${COMMAND_MENU} and 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 {
+			ctx.ui.notify(`Debug round ${state.round} paused. Reproduce the bug, then ${PROCEED_REMINDER}`, "info");
+		}
+	}
+
+	// ============================== round transitions ==============================
+
+	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;
+		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;
+		}
+		pi.sendMessage(
+			{
+				customType: "debug-mode-start",
+				content: buildStartMessage(problem, logFile),
+				display: true,
+				details: { summary: "debug mode started — hypotheses and instrumentation" } satisfies DebugMessageDetails,
+			},
+			{ triggerTurn: true },
+		);
+	}
+
+	function ensureWaiting(ctx: ExtensionContext): boolean {
+		if (!state.active) {
+			ctx.ui.notify("debug-mode: not active", "error");
+			return false;
+		}
+		if (state.phase !== "waiting") {
+			ctx.ui.notify(`debug-mode: not waiting for reproduction (phase: ${state.phase})`, "error");
+			return false;
+		}
+		return true;
+	}
+
+	async function markDebugFixed(ctx: ExtensionContext): Promise<void> {
+		if (!ensureWaiting(ctx)) return;
+		const logCount = currentLogCount();
+		if (logCount === 0 && 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;
+
+		state.phase = "cleanup";
+		state.cleanupReady = false;
+		unwatchLogFile();
+		refreshUi();
+		pi.sendMessage(
+			{
+				customType: "debug-mode-fixed",
+				content: buildFixedMessage(JSON.stringify(state.probes)),
+				display: true,
+				details: { summary: `marked fixed — removing ${state.probes.length} probe(s)` } satisfies DebugMessageDetails,
+			},
+			{ triggerTurn: true },
+		);
+	}
+
+	async function advanceDebug(ctx: ExtensionContext, reproductionDetails?: string): Promise<void> {
+		if (!ensureWaiting(ctx)) return;
+		refreshLogCounts();
+		const run = state.runId ?? "(none)";
+		const logCount = state.logCounts[run] ?? 0;
+		if (logCount === 0 && !reproductionDetails && 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;
+
+		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();
+			return;
+		}
+		unwatchLogFile();
+		refreshUi();
+
+		const summary = reproductionDetails
+			? `reproduction details added — analyzing run ${run} (${logCount} entries)`
+			: `proceed — analyzing run ${run} (${logCount} entries)`;
+		pi.sendMessage(
+			{
+				customType: reproductionDetails ? "debug-mode-note" : "debug-mode-proceed",
+				content: buildProceedMessage({ run, logCount, reproductionDetails, hypotheses }),
+				display: true,
+				details: { summary } satisfies DebugMessageDetails,
+			},
+			{ triggerTurn: true },
+		);
+	}
+
+	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 abortDebug(ctx: ExtensionContext): Promise<void> {
+		if (!state.active) {
+			ctx.ui.notify("debug-mode: not active", "error");
+			return;
+		}
+		if (ctx.hasUI) {
+			const confirmed = await ctx.ui.confirm(
+				"Abort debug mode?",
+				"Delete captured debug logs and stop the workflow? Applied code changes will remain in the working diff.",
+			);
+			if (!confirmed) return;
+		}
+		if (!state.active) return;
+		await teardown(ctx, "aborted");
+	}
+
+	/**
+	 * Open the action menu after `session_stop` has returned. Awaiting it
+	 * inside that handler would hold the agent-loop lock for the whole
+	 * reproduction; the microtask lets the round settle first.
+	 */
+	function scheduleReviewMenu(ctx: ExtensionContext): void {
+		if (!ctx.hasUI) return;
+		queueMicrotask(() => {
+			openReviewMenu(ctx).catch(err => {
+				pi.logger.warn("debug-mode: review menu failed to open", { err });
+			});
+		});
+	}
+
+	async function openReviewMenu(ctx: ExtensionContext): Promise<void> {
+		if (!ensureWaiting(ctx)) return;
+		if (!ctx.hasUI) {
+			ctx.ui.notify(`/${COMMAND_MENU} requires an interactive UI; use /${COMMAND_DONE} or /${COMMAND_PROCEED} instead.`, "warning");
+			return;
+		}
+		if (reviewMenuOpen) {
+			ctx.ui.notify("debug-mode: review menu is already open", "info");
+			return;
+		}
+
+		reviewMenuOpen = true;
+		try {
+			const probesThisRound = state.probes.filter(p => p.round === state.round).length;
+			const title = reviewMenuTitle(state.round, currentLogCount(), probesThisRound);
+			const choice = await ctx.ui.select(title, reviewMenuOptions());
+			if (!choice) return;
+			if (choice === REVIEW_MARK_FIXED) {
+				await markDebugFixed(ctx);
+			} else if (choice === REVIEW_PROCEED) {
+				await advanceDebug(ctx);
+			} else if (choice === REVIEW_ADD_DETAILS) {
+				ctx.ui.setEditorText("/debug-note ");
+				ctx.ui.notify("Add reproduction details in the editor, then submit /debug-note.", "info");
+			} else if (choice === REVIEW_ABORT) {
+				await abortDebug(ctx);
+			}
+		} finally {
+			reviewMenuOpen = false;
+		}
+	}
+
+	// ============================== commands ==============================
+
+	pi.registerCommand(COMMAND_MODE, {
+		description: "Start debug mode: /debug-mode <problem description>",
+		handler: async (args, ctx) => {
+			uiCtx = ctx;
+			if (state.active) {
+				ctx.ui.notify(`debug-mode: already active (use /${COMMAND_MENU}, /${COMMAND_DONE}, /${COMMAND_PROCEED}, or /${COMMAND_ABORT})`, "error");
+				return;
+			}
+			const problem = args.trim();
+			if (!problem) {
+				ctx.ui.notify(
+					"Usage: /debug-mode <problem description — symptoms, expected vs actual, how to reproduce>",
+					"error",
+				);
+				return;
+			}
+			startDebug(ctx, problem);
+		},
+	});
+
+	pi.registerCommand(COMMAND_DONE, {
+		description: "Mark the problem fixed: remove probes and summarize",
+		handler: async (_args, ctx) => {
+			uiCtx = ctx;
+			await markDebugFixed(ctx);
+		},
+	});
+
+	pi.registerCommand(COMMAND_PROCEED, {
+		description: "Continue with captured logs: evaluate hypotheses, fix only with evidence",
+		handler: async (_args, ctx) => {
+			uiCtx = ctx;
+			await advanceDebug(ctx);
+		},
+	});
+
+	pi.registerCommand(COMMAND_MENU, {
+		description: "Open the interactive action menu for a completed debug round",
+		handler: async (_args, ctx) => {
+			uiCtx = ctx;
+			await openReviewMenu(ctx);
+		},
+	});
+
+	pi.registerCommand(COMMAND_NOTE, {
+		description: "Add reproduction details and continue: /debug-note <details>",
+		handler: async (args, ctx) => {
+			uiCtx = ctx;
+			const details = args.trim();
+			if (!details) {
+				ctx.ui.notify("Usage: /debug-note <reproduction details>", "error");
+				return;
+			}
+			await advanceDebug(ctx, details);
+		},
+	});
+
+	pi.registerCommand(COMMAND_ABORT, {
+		description: "Abort debug mode: delete logs (fixes stay in the working diff)",
+		handler: async (_args, ctx) => {
+			uiCtx = ctx;
+			await abortDebug(ctx);
+		},
+	});
+
+	pi.registerCommand(COMMAND_STATUS, {
+		description: "Show debug mode state",
+		handler: async (_args, ctx) => {
+			uiCtx = ctx;
+			if (!state.active) {
+				ctx.ui.notify("debug-mode: idle", "info");
+				return;
+			}
+			refreshLogCounts();
+			const scan = await syncLedger(state);
+			const tallies = state.runId ? summarizeHypotheses(readRunLines(state.runId)) : [];
+			ctx.ui.notify(
+				`debug-mode: phase=${state.phase} round=${state.round} run=${state.runId}\n` +
+					`${describeLedger(scan)}\n` +
+					`logs: ${state.runHistory.map(r => `${r}=${state.logCounts[r] ?? 0}`).join(", ") || "(none)"}\n` +
+					`this run by hypothesis: ${describeHypotheses(tallies)}\n` +
+					`current log file: ${logFileFor(state) ?? "(not initialized)"}`,
+				"info",
+			);
+		},
+	});
+
+	// ============================== tools ==============================
+
+	registerDebugTools(pi, { state, refreshLogCounts, readRunLines });
+
+	// ============================== lifecycle ==============================
+
+	pi.on("session_start", async (_event, ctx) => {
+		uiCtx = ctx;
+		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.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);
+			const currentFile = logFileFor(state);
+			if (currentFile) {
+				try {
+					const legacyRunFile = state.debugDir && state.runId ? path.join(state.debugDir, `${state.runId}.jsonl`) : null;
+					if (!fs.existsSync(currentFile) && legacyRunFile && fs.existsSync(legacyRunFile)) {
+						fs.renameSync(legacyRunFile, currentFile);
+					}
+					fs.writeFileSync(currentFile, "", { flag: "a" });
+				} catch (err) {
+					pi.logger.error("debug-mode: cannot restore run log", { file: currentFile, err });
+				}
+			}
+			refreshLogCounts();
+			ctx.ui.notify(
+				`debug-mode resumed: phase=${state.phase} round=${state.round}, log=${currentFile ?? "unavailable"}. Use /${COMMAND_STATUS}, /${COMMAND_MENU}, /${COMMAND_DONE}, /${COMMAND_PROCEED}, or /${COMMAND_ABORT}.`,
+				"info",
+			);
+		}
+		refreshUi();
+		watchLogFile();
+		if (state.active && state.phase === "waiting") scheduleReviewMenu(ctx);
+	});
+
+	pi.on("turn_start", async () => {
+		if (state.active) pi.appendEntry(DEBUG_ENTRY, { ...state });
+	});
+
+	pi.on("session_shutdown", async () => {
+		unwatchLogFile();
+	});
+}

package/src/gate.ts

@@ -0,0 +1,36 @@
+import { PROCEED_REMINDER } from "./methodology";
+
+/** How many times a single round may be nudged to produce reproduction steps. */
+export const MAX_GATE_NUDGES = 1;
+
+export const GATE_NUDGE =
+	"You instrumented this round but did not end with a <reproduction_steps> block. " +
+	"List the numbered steps the user must perform to exercise the instrumented path, " +
+	`follow them with "${PROCEED_REMINDER}", and then stop. Do not start new work.`;
+
+export type GateDecision =
+	/** Hand control to the user and wait for a reproduction. */
+	| { kind: "gate"; missingSteps: boolean }
+	/** Let the agent finish the round properly before gating. */
+	| { kind: "nudge"; context: string }
+	/** Not a completed round — the agent is mid-conversation with the user. */
+	| { kind: "stay" };
+
+/**
+ * Decide what a settled agent turn means for the reproduction gate.
+ *
+ * Cursor gates on the `<reproduction_steps>` block, so a turn without one is
+ * either a clarifying question (leave the user talking to the agent) or an
+ * instrumented round that forgot to close itself (nudge once, then gate anyway
+ * so the workflow can never stall).
+ */
+export function decideGate(args: {
+	hasReproductionSteps: boolean;
+	probesThisRound: number;
+	nudgesUsed: number;
+}): GateDecision {
+	if (args.hasReproductionSteps) return { kind: "gate", missingSteps: false };
+	if (args.probesThisRound === 0) return { kind: "stay" };
+	if (args.nudgesUsed < MAX_GATE_NUDGES) return { kind: "nudge", context: GATE_NUDGE };
+	return { kind: "gate", missingSteps: true };
+}