fix: make qmd memory search opt-in (#6)

Co-authored-by: luckyyzh <luckyyzh@users.noreply.github.com>
2026-10-03 02:59:35 +00:00 · 2026-07-31 23:37:06 +08:00
parent 3b48681852
commit 551c59413e
9 changed files with 289 additions and 3 deletions
@@ -0,0 +1,45 @@
+import assert from "node:assert/strict";
+import { existsSync, mkdirSync, mkdtempSync, readFileSync, writeFileSync } from "node:fs";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import test from "node:test";
+import {
+  configurePiMemory,
+  isPiMemoryLiteConfigured,
+  patchPiMemorySource,
+} from "./configure-pi-memory.mjs";
+import { agentDir, managedEnvironment } from "./profile.mjs";
+
+test("managed profile defaults pi-memory to lightweight mode", () => {
+  const defaultEnvironment = managedEnvironment({ ...process.env, PI_MEMORY_NO_SEARCH: undefined });
+  const searchEnvironment = managedEnvironment({ ...process.env, PI_MEMORY_NO_SEARCH: "0" });
+
+  assert.equal(defaultEnvironment.PI_MEMORY_NO_SEARCH, "1");
+  assert.equal(searchEnvironment.PI_MEMORY_NO_SEARCH, "0");
+});
+
+test("installed pi-memory integration is patched and idempotent", (context) => {
+  const sourcePath = join(agentDir, "npm", "node_modules", "pi-memory", "index.ts");
+  if (!existsSync(sourcePath)) {
+    context.skip("managed pi-memory is not installed; npm run setup installs it before CI tests");
+    return;
+  }
+
+  configurePiMemory({ quiet: true });
+  const source = readFileSync(sourcePath, "utf8");
+  assert.equal(isPiMemoryLiteConfigured(source), true);
+  assert.equal(patchPiMemorySource(source), source);
+});
+
+test("integration rejects an unreviewed pi-memory version", () => {
+  const temporaryAgentDir = mkdtempSync(join(tmpdir(), "pi-memory-version-"));
+  const packageDir = join(temporaryAgentDir, "npm", "node_modules", "pi-memory");
+  mkdirSync(packageDir, { recursive: true });
+  writeFileSync(join(packageDir, "package.json"), JSON.stringify({ version: "9.9.9" }), "utf8");
+  writeFileSync(join(packageDir, "index.ts"), "export default function () {}\n", "utf8");
+
+  assert.throws(
+    () => configurePiMemory({ agentDir: temporaryAgentDir, quiet: true }),
+    /Unsupported pi-memory version 9\.9\.9/,
+  );
+});