feat: add answer extraction replay commands

tests/ask-payload-store.test.ts

@@ -0,0 +1,95 @@
+import assert from "node:assert/strict";
+import test from "node:test";
+import { findLatestPayloadInCurrentBranch } from "../src/ask-payload-store.ts";
+import type { AskParams } from "../src/types.ts";
+
+const params: AskParams = {
+	questions: [
+		{
+			id: "goal",
+			prompt: "Goal?",
+			options: [{ value: "a", label: "A" }],
+		},
+	],
+};
+
+function custom(data: unknown) {
+	return { type: "custom", customType: "ask:payload", data };
+}
+
+function ctx(branch: unknown[]) {
+	return { sessionManager: { getBranch: () => branch } } as never;
+}
+
+test("branch payload lookup returns the latest valid matching source", () => {
+	const older = {
+		version: 1,
+		source: "tool",
+		params,
+		timestamp: 1,
+	};
+	const newer = {
+		version: 1,
+		source: "tool",
+		params: { ...params, title: "Newer" },
+		timestamp: 2,
+	};
+
+	const result = findLatestPayloadInCurrentBranch(
+		ctx([custom(older), custom(newer)]),
+		"tool"
+	);
+
+	assert.equal(result.data, newer);
+	assert.equal(result.invalidMatchFound, false);
+});
+
+test("branch payload lookup allows freeform only for answer extraction payloads", () => {
+	const freeformParams: AskParams = {
+		questions: [
+			{
+				id: "language",
+				prompt: "Which language?",
+				options: [{ value: "freeform", label: "Type answer", freeform: true }],
+			},
+		],
+	};
+
+	const answerPayload = {
+		version: 1,
+		source: "answer-extraction",
+		params: freeformParams,
+		timestamp: 1,
+	};
+	const toolPayload = {
+		version: 1,
+		source: "tool",
+		params: freeformParams,
+		timestamp: 2,
+	};
+
+	assert.equal(
+		findLatestPayloadInCurrentBranch(
+			ctx([custom(answerPayload)]),
+			"answer-extraction"
+		).data,
+		answerPayload
+	);
+	assert.equal(
+		findLatestPayloadInCurrentBranch(ctx([custom(toolPayload)]), "tool").data,
+		undefined
+	);
+});
+
+test("branch payload lookup ignores invalid stored payloads", () => {
+	const result = findLatestPayloadInCurrentBranch(
+		ctx([
+			custom({ version: 1, source: "tool", params: { questions: [] } }),
+			custom({ version: 1, source: "answer-extraction", params }),
+		]),
+		"tool"
+	);
+
+	assert.equal(result.data, undefined);
+	assert.equal(result.invalidMatchFound, true);
+});