feat: integrate Pi backend and Pi Web

2026-10-03 02:59:35 +00:00 · 2026-07-30 19:37:53 +08:00
commit 7392ab9dd7
1390 changed files with 337197 additions and 0 deletions
@@ -0,0 +1,88 @@
+import * as fs from "node:fs";
+import * as os from "node:os";
+import * as path from "node:path";
+import { afterEach, describe, expect, it } from "vitest";
+import { ENV_AGENT_DIR } from "../src/config.ts";
+import { KeybindingsManager } from "../src/core/keybindings.ts";
+import { runMigrations } from "../src/migrations.ts";
+
+describe("keybindings migration", () => {
+	const tempDirs: string[] = [];
+
+	afterEach(() => {
+		for (const dir of tempDirs.splice(0)) {
+			fs.rmSync(dir, { recursive: true, force: true });
+		}
+	});
+
+	function createAgentDir(config: Record<string, unknown>): string {
+		const agentDir = fs.mkdtempSync(path.join(os.tmpdir(), "pi-keybindings-test-"));
+		tempDirs.push(agentDir);
+		fs.writeFileSync(path.join(agentDir, "keybindings.json"), `${JSON.stringify(config, null, 2)}\n`, "utf-8");
+		return agentDir;
+	}
+
+	it("rewrites old key names to namespaced ids", () => {
+		const agentDir = createAgentDir({
+			cursorUp: ["up", "ctrl+p"],
+			expandTools: "ctrl+x",
+		});
+		const previousAgentDir = process.env[ENV_AGENT_DIR];
+		process.env[ENV_AGENT_DIR] = agentDir;
+		runMigrations(agentDir);
+		if (previousAgentDir === undefined) {
+			delete process.env[ENV_AGENT_DIR];
+		} else {
+			process.env[ENV_AGENT_DIR] = previousAgentDir;
+		}
+
+		const migrated = JSON.parse(fs.readFileSync(path.join(agentDir, "keybindings.json"), "utf-8")) as Record<
+			string,
+			unknown
+		>;
+		expect(migrated).toEqual({
+			"tui.editor.cursorUp": ["up", "ctrl+p"],
+			"app.tools.expand": "ctrl+x",
+		});
+	});
+
+	it("keeps the namespaced value when old and new names both exist", () => {
+		const agentDir = createAgentDir({
+			expandTools: "ctrl+x",
+			"app.tools.expand": "ctrl+y",
+		});
+		const previousAgentDir = process.env[ENV_AGENT_DIR];
+		process.env[ENV_AGENT_DIR] = agentDir;
+		runMigrations(agentDir);
+		if (previousAgentDir === undefined) {
+			delete process.env[ENV_AGENT_DIR];
+		} else {
+			process.env[ENV_AGENT_DIR] = previousAgentDir;
+		}
+
+		const migrated = JSON.parse(fs.readFileSync(path.join(agentDir, "keybindings.json"), "utf-8")) as Record<
+			string,
+			unknown
+		>;
+		expect(migrated).toEqual({
+			"app.tools.expand": "ctrl+y",
+		});
+	});
+
+	it("loads old key names in memory before the file is rewritten", () => {
+		const agentDir = createAgentDir({
+			selectConfirm: "enter",
+			interrupt: "ctrl+x",
+		});
+
+		const keybindings = KeybindingsManager.create(agentDir);
+
+		expect(keybindings.getUserBindings()).toEqual({
+			"tui.select.confirm": "enter",
+			"app.interrupt": "ctrl+x",
+		});
+		const effective = keybindings.getEffectiveConfig();
+		expect(effective["tui.select.confirm"]).toBe("enter");
+		expect(effective["app.interrupt"]).toBe("ctrl+x");
+	});
+});