chore: import upstream snapshot with attribution

2026-07-13 13:01:18 +08:00
commit 979fb22d7c
613 changed files with 113494 additions and 0 deletions
@@ -0,0 +1,137 @@
+import { describe, it, expect } from "vitest";
+import { writeFileSync, readFileSync, mkdirSync, rmSync } from "node:fs";
+import { join, resolve } from "node:path";
+import { tmpdir } from "node:os";
+import {
+  buildMergedHooks,
+  findPluginRoot,
+  type HookManifest,
+} from "../src/cli/connect/codex-hooks.js";
+
+const PLUGIN_ROOT = resolve(__dirname, "..", "plugin");
+
+describe("findPluginRoot", () => {
+  it("locates the bundled plugin/ directory from src/cli/connect/", () => {
+    const root = findPluginRoot();
+    expect(root).toBe(PLUGIN_ROOT);
+  });
+});
+
+describe("buildMergedHooks", () => {
+  it("rewrites ${CLAUDE_PLUGIN_ROOT} to absolute pluginRoot in every command", () => {
+    const merged = buildMergedHooks(null, PLUGIN_ROOT);
+    for (const entries of Object.values(merged.hooks)) {
+      for (const entry of entries) {
+        for (const handler of entry.hooks) {
+          expect(handler.command).not.toContain("${CLAUDE_PLUGIN_ROOT}");
+          expect(handler.command).toContain(`${PLUGIN_ROOT}/scripts/`);
+        }
+      }
+    }
+  });
+
+  it("preserves matchers from the bundled manifest (e.g. PreToolUse)", () => {
+    const merged = buildMergedHooks(null, PLUGIN_ROOT);
+    const preToolUse = merged.hooks["PreToolUse"];
+    expect(preToolUse).toBeDefined();
+    expect(preToolUse!.length).toBeGreaterThan(0);
+    expect(preToolUse![0].matcher).toBe("Edit|Write|Read|Glob|Grep");
+  });
+
+  it("includes all six expected lifecycle events", () => {
+    const merged = buildMergedHooks(null, PLUGIN_ROOT);
+    for (const event of [
+      "SessionStart",
+      "UserPromptSubmit",
+      "PreToolUse",
+      "PostToolUse",
+      "PreCompact",
+      "Stop",
+    ]) {
+      expect(Object.keys(merged.hooks)).toContain(event);
+    }
+  });
+
+  it("appends to existing user hooks without dropping them", () => {
+    const existing: HookManifest = {
+      hooks: {
+        SessionStart: [
+          {
+            hooks: [{ type: "command", command: "echo user-custom" }],
+          },
+        ],
+        UserPromptSubmit: [
+          {
+            hooks: [{ type: "command", command: "echo another-user-hook" }],
+          },
+        ],
+      },
+    };
+    const merged = buildMergedHooks(existing, PLUGIN_ROOT);
+    const sessionStart = merged.hooks["SessionStart"]!;
+    const userHook = sessionStart.find((e) =>
+      e.hooks.some((h) => h.command === "echo user-custom"),
+    );
+    expect(userHook, "user's SessionStart hook should survive").toBeDefined();
+    const ours = sessionStart.find((e) =>
+      e.hooks.some((h) => h.command.includes(`${PLUGIN_ROOT}/scripts/session-start.mjs`)),
+    );
+    expect(ours, "agentmemory SessionStart hook should be appended").toBeDefined();
+  });
+
+  it("re-install strips previous agentmemory entries (idempotent by script path)", () => {
+    const first = buildMergedHooks(null, PLUGIN_ROOT);
+    const second = buildMergedHooks(first, PLUGIN_ROOT);
+    for (const event of Object.keys(first.hooks)) {
+      expect(
+        second.hooks[event]!.length,
+        `${event} should not double after second install`,
+      ).toBe(first.hooks[event]!.length);
+    }
+  });
+
+  it("re-install preserves unrelated user entries", () => {
+    const userEntry = {
+      hooks: [{ type: "command", command: "echo user-untouchable" }],
+    };
+    const withUser: HookManifest = {
+      hooks: {
+        SessionStart: [userEntry],
+        Stop: [{ hooks: [{ type: "command", command: "echo also-user" }] }],
+      },
+    };
+    const installed = buildMergedHooks(withUser, PLUGIN_ROOT);
+    const reinstalled = buildMergedHooks(installed, PLUGIN_ROOT);
+    expect(
+      reinstalled.hooks["SessionStart"]!.some((e) =>
+        e.hooks.some((h) => h.command === "echo user-untouchable"),
+      ),
+    ).toBe(true);
+    expect(
+      reinstalled.hooks["Stop"]!.some((e) =>
+        e.hooks.some((h) => h.command === "echo also-user"),
+      ),
+    ).toBe(true);
+  });
+
+  it("handles empty existing manifest object", () => {
+    const merged = buildMergedHooks({ hooks: {} }, PLUGIN_ROOT);
+    expect(Object.keys(merged.hooks).length).toBeGreaterThan(0);
+  });
+});
+
+describe("buildMergedHooks file round-trip", () => {
+  it("produces JSON that parses back to a structurally equivalent manifest", () => {
+    const dir = join(tmpdir(), `agentmemory-codex-hooks-${process.pid}-${Date.now()}`);
+    mkdirSync(dir, { recursive: true });
+    const path = join(dir, "hooks.json");
+    try {
+      const merged = buildMergedHooks(null, PLUGIN_ROOT);
+      writeFileSync(path, `${JSON.stringify(merged, null, 2)}\n`, "utf-8");
+      const reread = JSON.parse(readFileSync(path, "utf-8")) as HookManifest;
+      expect(Object.keys(reread.hooks).sort()).toEqual(Object.keys(merged.hooks).sort());
+    } finally {
+      rmSync(dir, { recursive: true, force: true });
+    }
+  });
+});