feat(chrome): hand snapshots to context mode

2026-08-28 08:35:57 +00:00 · 2026-08-28 14:45:37 +08:00
parent 221e978622
commit f3a2abe1e4
96 changed files with 12800 additions and 26 deletions
@@ -0,0 +1,100 @@
+import assert from "node:assert/strict";
+import { mkdtemp, readFile, stat } from "node:fs/promises";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import test from "node:test";
+import { SessionSnapshotStore, snapshotTargetKey } from "../src/session-snapshot-store.ts";
+import { CHROME_TOOL_NAMES, formatChromeFindMatches, formatSnapshotCapture } from "../extensions/chrome-profile-bridge/index.ts";
+
+test("snapshotTargetKey prefers explicit target selectors", () => {
+	assert.equal(snapshotTargetKey(), "default");
+	assert.equal(snapshotTargetKey({ titleIncludes: "Inbox" }), "title:Inbox");
+	assert.equal(snapshotTargetKey({ urlIncludes: "example.com" }), "url:example.com");
+	assert.equal(snapshotTargetKey({ targetId: "42", urlIncludes: "ignored" }), "target:42");
+});
+
+test("SessionSnapshotStore atomically keeps only the latest capture per target", async () => {
+	const workspace = await mkdtemp(join(tmpdir(), "pi-chrome-snapshot-"));
+	const store = new SessionSnapshotStore(workspace, "session/private/id");
+	const first = await store.save({ mode: "interactive", targetId: 42, elements: [{ uid: "el-1" }], matches: [] });
+	const second = await store.save({ mode: "changes", targetId: 42, elements: [{ uid: "el-2" }], matches: [{ uid: "el-2" }] });
+
+	assert.equal(first.metadata.path, second.metadata.path);
+	assert.notEqual(first.metadata.captureId, second.metadata.captureId);
+	assert.equal(second.metadata.elementCount, 1);
+	assert.equal(second.metadata.matchCount, 1);
+	assert.equal(second.metadata.targetId, 42);
+	assert.ok(!second.metadata.path.includes("private"));
+
+	const stored = JSON.parse(await readFile(second.absolutePath, "utf8"));
+	assert.equal(stored.version, 1);
+	assert.equal(stored.captureId, second.metadata.captureId);
+	assert.equal(stored.snapshot.elements[0].uid, "el-2");
+
+	if (process.platform !== "win32") {
+		assert.equal((await stat(store.sessionRoot)).mode & 0o777, 0o700);
+		assert.equal((await stat(second.absolutePath)).mode & 0o777, 0o600);
+	}
+
+	await store.clear();
+	await assert.rejects(stat(second.absolutePath), { code: "ENOENT" });
+});
+
+test("target invalidation and session cleanup do not affect another session", async () => {
+	const workspace = await mkdtemp(join(tmpdir(), "pi-chrome-isolation-"));
+	const firstSession = new SessionSnapshotStore(workspace, "session-a");
+	const secondSession = new SessionSnapshotStore(workspace, "session-b");
+	const firstTarget = { targetId: "101" };
+	const secondTarget = { targetId: "202" };
+	const first = await firstSession.save({ elements: [] }, firstTarget);
+	const otherTarget = await firstSession.save({ elements: [] }, secondTarget);
+	const otherSession = await secondSession.save({ elements: [] }, firstTarget);
+
+	await firstSession.invalidate(firstTarget);
+	await assert.rejects(stat(first.absolutePath), { code: "ENOENT" });
+	await stat(otherTarget.absolutePath);
+	await stat(otherSession.absolutePath);
+
+	await firstSession.clear();
+	await assert.rejects(stat(otherTarget.absolutePath), { code: "ENOENT" });
+	await stat(otherSession.absolutePath);
+	await secondSession.clear();
+});
+
+test("snapshot result formatting stays bounded and points to Context Mode", () => {
+	const text = formatSnapshotCapture({
+		version: 1,
+		captureId: "capture-1",
+		capturedAt: "2026-08-28T00:00:00.000Z",
+		path: ".pi/chrome-context/session/target/snapshot.json",
+		bytes: 80_000,
+		targetKey: "target:42",
+		targetId: 42,
+		mode: "interactive",
+		elementCount: 80,
+		matchCount: 3,
+		staleAfterInteraction: true,
+	});
+
+	assert.match(text, /ctx_execute_file/);
+	assert.match(text, /\.pi\/chrome-context\/session\/target\/snapshot\.json/);
+	assert.doesNotMatch(text, /raw-secret-page-content/);
+});
+
+test("chrome_find formatting is capped and authorization tracks all 21 tools", () => {
+	const matches = Array.from({ length: 25 }, (_, index) => ({
+		kind: "element",
+		uid: `el-${index}`,
+		role: "button",
+		label: `Button ${index}`,
+		rect: { x: index, y: index, width: 10, height: 10 },
+	}));
+	const text = formatChromeFindMatches({ query: "button", matches });
+
+	assert.match(text, /el-0/);
+	assert.match(text, /el-19/);
+	assert.doesNotMatch(text, /el-20/);
+	assert.equal(CHROME_TOOL_NAMES.length, 21);
+	assert.ok(CHROME_TOOL_NAMES.includes("chrome_find"));
+	assert.ok(CHROME_TOOL_NAMES.includes("chrome_inspect"));
+});