@opencode-trace/plugin 0.0.4 → 0.0.5

package/dist/trace.test.js

@@ -1,7 +1,8 @@
 import { describe, test, expect, beforeEach, afterEach, vi } from "vitest";
-import { mkdtempSync, rmSync, existsSync, readFileSync } from "node:fs";
+import { mkdtempSync, rmSync, existsSync, readFileSync, } from "node:fs";
 import { tmpdir } from "node:os";
 import { join } from "node:path";
+import { logger } from "@opencode-trace/core";
 import entrypoint, { _resetForTesting } from "./trace.js";
 vi.mock("node:os", async (importOriginal) => {
     const original = await importOriginal();
@@ -26,13 +27,12 @@ async function waitForFile(filePath, timeoutMs = 5000) {
                     return;
                 }
             }
-            catch {
-            }
+            catch { }
         }
         if (Date.now() - startTime > timeoutMs) {
             throw new Error(`Timeout waiting for valid file ${filePath} after ${timeoutMs}ms`);
         }
-        await new Promise(r => setTimeout(r, 10));
+        await new Promise((r) => setTimeout(r, 10));
     }
 }
 let testDir;
@@ -100,8 +100,8 @@ describe("Plugin - event hook 处理 session.created", () => {
         const traceDir = join(testDir, ".opencode-trace");
         const sessionDir = join(traceDir, sessionId);
         expect(existsSync(sessionDir)).toBe(true);
-        const dbPath = join(traceDir, "state.db");
-        expect(existsSync(dbPath)).toBe(true);
+        const configPath = join(traceDir, "config.json");
+        expect(existsSync(configPath)).toBe(true);
         const metaPath = join(sessionDir, "metadata.json");
         expect(existsSync(metaPath)).toBe(true);
         const meta = JSON.parse(readFileSync(metaPath, "utf-8"));
@@ -162,9 +162,13 @@ describe("tracedFetch stream integration", () => {
         const mockFetch = vi.fn();
         const originalFetch = globalThis.fetch;
         globalThis.fetch = mockFetch;
-        const chunks = ["data: {\"content\": \"Hello\"}\n", "data: {\"content\": \"World\"}\n", "data: [DONE]\n"];
+        const chunks = [
+            'data: {"content": "Hello"}\n',
+            'data: {"content": "World"}\n',
+            "data: [DONE]\n",
+        ];
         const encoder = new TextEncoder();
-        const streamChunks = chunks.map(c => encoder.encode(c));
+        const streamChunks = chunks.map((c) => encoder.encode(c));
         const mockStream = new ReadableStream({
             start(controller) {
                 for (const chunk of streamChunks) {
@@ -208,7 +212,10 @@ describe("tracedFetch stream integration", () => {
                 "content-type": "application/json",
                 "x-opencode-session": sessionId,
             },
-            body: JSON.stringify({ stream: true, messages: [{ role: "user", content: "test" }] }),
+            body: JSON.stringify({
+                stream: true,
+                messages: [{ role: "user", content: "test" }],
+            }),
         });
         const response = await globalThis.fetch(streamRequest);
         expect(response.__latencyMeta).toBeDefined();
@@ -278,7 +285,10 @@ describe("tracedFetch stream integration", () => {
                 "content-type": "application/json",
                 "x-opencode-session": sessionId,
             },
-            body: JSON.stringify({ stream: false, messages: [{ role: "user", content: "test" }] }),
+            body: JSON.stringify({
+                stream: false,
+                messages: [{ role: "user", content: "test" }],
+            }),
         });
         const response = await globalThis.fetch(request);
         expect(response.__latencyMeta).toBeUndefined();
@@ -324,8 +334,8 @@ describe("Plugin - tool.execute.after hook 处理 Task 工具", () => {
             metadata: { session_id: subSessionId },
         });
         const traceDir = join(testDir, ".opencode-trace");
-        const dbPath = join(traceDir, "state.db");
-        expect(existsSync(dbPath)).toBe(true);
+        const configPath = join(traceDir, "config.json");
+        expect(existsSync(configPath)).toBe(true);
     });
     test("tool.execute.after hook 对非 Task 工具不记录 sub session", async () => {
         const hooks = await entrypoint.server({
@@ -364,12 +374,12 @@ describe("Plugin - tool.execute.after hook 处理 Task 工具", () => {
             metadata: {},
         });
         const traceDir = join(testDir, ".opencode-trace");
-        const dbPath = join(traceDir, "state.db");
-        expect(existsSync(dbPath)).toBe(true);
+        const configPath = join(traceDir, "config.json");
+        expect(existsSync(configPath)).toBe(true);
     });
 });
 describe("Plugin - global/local mode", () => {
-    test("trace_use_global tool switches to global mode", async () => {
+    test("trace_on tool enables session scope", async () => {
         const hooks = await entrypoint.server({
             client: {},
             project: {},
@@ -379,7 +389,7 @@ describe("Plugin - global/local mode", () => {
             serverUrl: new URL("http://localhost"),
             $: {},
         });
-        const sessionId = "test-session-global";
+        const sessionId = "test-session-on";
         await hooks.event({
             event: {
                 type: "session.created",
@@ -395,8 +405,10 @@ describe("Plugin - global/local mode", () => {
                 },
             },
         });
-        const result = await hooks.tool.trace_use_global.execute({}, { sessionID: sessionId });
-        expect(result).toContain("global mode");
+        const result = await hooks.tool.trace_on.execute({}, {
+            sessionID: sessionId,
+        });
+        expect(result).toContain("Trace enabled for session");
         const globalDir = join(testDir, ".opencode-trace");
         const sessionDir = join(globalDir, sessionId);
         await hooks.event({
@@ -414,10 +426,10 @@ describe("Plugin - global/local mode", () => {
                 },
             },
         });
-        await new Promise(r => setTimeout(r, 100));
+        await new Promise((r) => setTimeout(r, 100));
         expect(existsSync(sessionDir)).toBe(true);
     });
-    test("trace_use_local tool switches to local mode", async () => {
+    test("trace_off tool disables session scope", async () => {
         const hooks = await entrypoint.server({
             client: {},
             project: {},
@@ -427,7 +439,7 @@ describe("Plugin - global/local mode", () => {
             serverUrl: new URL("http://localhost"),
             $: {},
         });
-        const sessionId = "test-session-local";
+        const sessionId = "test-session-off";
         await hooks.event({
             event: {
                 type: "session.created",
@@ -443,29 +455,12 @@ describe("Plugin - global/local mode", () => {
                 },
             },
         });
-        const result = await hooks.tool.trace_use_local.execute({}, { sessionID: sessionId });
-        expect(result).toContain("local mode");
-        const localDir = join(testDir, ".opencode-trace");
-        const sessionDir = join(localDir, sessionId);
-        await hooks.event({
-            event: {
-                type: "session.updated",
-                properties: {
-                    info: {
-                        id: sessionId,
-                        projectID: "test-project",
-                        directory: testDir,
-                        title: "Test Session Updated",
-                        version: "1.0",
-                        time: { created: Date.now(), updated: Date.now() },
-                    },
-                },
-            },
+        const result = await hooks.tool.trace_off.execute({}, {
+            sessionID: sessionId,
         });
-        await new Promise(r => setTimeout(r, 100));
-        expect(existsSync(sessionDir)).toBe(true);
+        expect(result).toContain("Trace disabled for session");
     });
-    test("trace_storage_status shows current mode", async () => {
+    test("trace_status tool shows current status", async () => {
         const hooks = await entrypoint.server({
             client: {},
             project: {},
@@ -491,11 +486,599 @@ describe("Plugin - global/local mode", () => {
                 },
             },
         });
-        const result = await hooks.tool.trace_storage_status.execute({}, { sessionID: sessionId });
-        const parsed = JSON.parse(result);
-        expect(parsed.mode).toBeDefined();
-        expect(parsed.globalDir).toBeDefined();
-        expect(parsed.localDir).toBeDefined();
+        const result = await hooks.tool.trace_status.execute({}, {
+            sessionID: sessionId,
+        });
+        expect(result).toContain("Trace Status");
+        expect(result).toContain("Global");
+        expect(result).toContain("Local");
+        expect(result).toContain("Session");
+        expect(result).toContain("Storage");
+    });
+});
+async function setupPluginWithMockClient(testDir) {
+    _resetForTesting();
+    const mockPrompt = vi.fn().mockResolvedValue({});
+    const hooks = await entrypoint.server({
+        client: { session: { prompt: mockPrompt } },
+        project: {},
+        directory: testDir,
+        worktree: testDir,
+        experimental_workspace: { register: vi.fn() },
+        serverUrl: new URL("http://localhost"),
+        $: {},
+    });
+    return { hooks, mockPrompt };
+}
+async function runTraceCommand(hooks, mockPrompt, args, sessionId = "test-slash-session") {
+    const output = { parts: [{ type: "text", text: "original" }] };
+    let error = null;
+    try {
+        await hooks["command.execute.before"]({
+            command: "trace",
+            sessionID: sessionId,
+            arguments: args,
+        }, output);
+    }
+    catch (err) {
+        error = err;
+    }
+    expect(error).toBeTruthy();
+    expect(error.message).toBe("__TRACE_HANDLED__");
+    expect(output.parts.length).toBe(0);
+    if (mockPrompt.mock.calls.length > 0) {
+        const call = mockPrompt.mock.calls[0][0];
+        return { text: call.body.parts[0].text, outputParts: output.parts };
+    }
+    return { text: null, outputParts: output.parts };
+}
+async function createSessionViaEvent(hooks, sessionId, testDir, title = "Slash Test Session", parentID) {
+    await hooks.event({
+        event: {
+            type: "session.created",
+            properties: {
+                info: {
+                    id: sessionId,
+                    projectID: "test-project",
+                    directory: testDir,
+                    title,
+                    parentID,
+                    version: "1.0",
+                    time: { created: Date.now(), updated: Date.now() },
+                },
+            },
+        },
+    });
+}
+function readConfig(testDir) {
+    const configPath = join(testDir, ".opencode-trace", "config.json");
+    return JSON.parse(readFileSync(configPath, "utf-8"));
+}
+function readSessionMetadata(testDir, sessionId) {
+    const metaPath = join(testDir, ".opencode-trace", sessionId, "metadata.json");
+    if (!existsSync(metaPath))
+        return null;
+    return JSON.parse(readFileSync(metaPath, "utf-8"));
+}
+describe("Plugin - /trace on (slash command)", () => {
+    test("/trace on with no flags enables global scope by default", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-on-default";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "on", sessionId);
+        expect(text).toContain("Trace enabled");
+        expect(text).toContain("global");
+        const config = readConfig(testDir);
+        expect(config.global_trace_enabled).toBe(true);
+    });
+    test("/trace on -g explicitly enables global scope", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-on-g";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "on -g", sessionId);
+        expect(text).toContain("Trace enabled");
+        expect(text).toContain("global");
+        const config = readConfig(testDir);
+        expect(config.global_trace_enabled).toBe(true);
+    });
+    test("/trace on --global (long form) enables global scope", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-on-long-global";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "on --global", sessionId);
+        expect(text).toContain("Trace enabled");
+        expect(text).toContain("global");
+        const config = readConfig(testDir);
+        expect(config.global_trace_enabled).toBe(true);
+    });
+    test("/trace on -l enables local scope", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-on-l";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "on -l", sessionId);
+        expect(text).toContain("Trace enabled");
+        expect(text).toContain("local");
+        const config = readConfig(testDir);
+        expect(config.global_trace_enabled).toBe(true);
+    });
+    test("/trace on -s enables session scope", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-on-s";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "on -s", sessionId);
+        expect(text).toContain("Trace enabled");
+        expect(text).toContain("session");
+        const meta = readSessionMetadata(testDir, sessionId);
+        expect(meta).toBeTruthy();
+        expect(meta.trace_enabled).toBe(true);
+    });
+    test("/trace on -d local sets storage preference to local", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-on-d-local";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "on -d local", sessionId);
+        expect(text).toContain("Trace enabled");
+        expect(text).toContain("storage: local");
+        const config = readConfig(testDir);
+        expect(config.storage_preference).toBe("local");
+        expect(config.global_trace_enabled).toBe(true);
+    });
+    test("/trace on -d global sets storage preference to global", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-on-d-global";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "on -d global", sessionId);
+        expect(text).toContain("Trace enabled");
+        const config = readConfig(testDir);
+        expect(config.storage_preference).toBe("global");
+    });
+    test("/trace on -g -l enables both global and local scopes", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-on-gl";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "on -g -l", sessionId);
+        expect(text).toContain("Trace enabled");
+        expect(text).toContain("global");
+        expect(text).toContain("local");
+        const config = readConfig(testDir);
+        expect(config.global_trace_enabled).toBe(true);
+    });
+    test("/trace on -g -l -s enables all three scopes", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-on-gls";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "on -g -l -s", sessionId);
+        expect(text).toContain("Trace enabled");
+        expect(text).toContain("global");
+        expect(text).toContain("local");
+        expect(text).toContain("session");
+        const meta = readSessionMetadata(testDir, sessionId);
+        expect(meta.trace_enabled).toBe(true);
+    });
+    test("/trace on -s -d local enables session and sets local storage", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-on-s-d-local";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "on -s -d local", sessionId);
+        expect(text).toContain("Trace enabled");
+        expect(text).toContain("session");
+        expect(text).toContain("storage: local");
+        const meta = readSessionMetadata(testDir, sessionId);
+        expect(meta.trace_enabled).toBe(true);
+        expect(meta.storage_preference).toBe("local");
+    });
+    test("/trace enable (alias) works the same as /trace on", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-enable-alias";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "enable", sessionId);
+        expect(text).toContain("Trace enabled");
+        expect(text).toContain("global");
+    });
+});
+describe("Plugin - /trace off (slash command)", () => {
+    test("/trace off with no flags disables global scope by default", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-off-default";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "off", sessionId);
+        expect(text).toContain("Trace disabled");
+        expect(text).toContain("global");
+        const config = readConfig(testDir);
+        expect(config.global_trace_enabled).toBe(false);
+    });
+    test("/trace off -g explicitly disables global scope", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-off-g";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "off -g", sessionId);
+        expect(text).toContain("Trace disabled");
+        expect(text).toContain("global");
+        const config = readConfig(testDir);
+        expect(config.global_trace_enabled).toBe(false);
+    });
+    test("/trace off -l disables local scope", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-off-l";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "off -l", sessionId);
+        expect(text).toContain("Trace disabled");
+        expect(text).toContain("local");
+        const config = readConfig(testDir);
+        expect(config.global_trace_enabled).toBe(false);
+    });
+    test("/trace off -s disables session scope", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-off-s";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "off -s", sessionId);
+        expect(text).toContain("Trace disabled");
+        expect(text).toContain("session");
+        const meta = readSessionMetadata(testDir, sessionId);
+        expect(meta.trace_enabled).toBe(false);
+    });
+    test("/trace off -g -l -s disables all three scopes", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-off-all";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "off -g -l -s", sessionId);
+        expect(text).toContain("Trace disabled");
+        expect(text).toContain("global");
+        expect(text).toContain("local");
+        expect(text).toContain("session");
+        const config = readConfig(testDir);
+        expect(config.global_trace_enabled).toBe(false);
+        const meta = readSessionMetadata(testDir, sessionId);
+        expect(meta.trace_enabled).toBe(false);
+    });
+    test("/trace disable (alias) works the same as /trace off", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-disable-alias";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "disable", sessionId);
+        expect(text).toContain("Trace disabled");
+        expect(text).toContain("global");
+    });
+});
+describe("Plugin - /trace status (slash command)", () => {
+    test("/trace status with no flags shows full status", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-status-default";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "status", sessionId);
+        expect(text).toContain("Trace Status");
+        expect(text).toContain("Global");
+        expect(text).toContain("Local");
+        expect(text).toContain("Session");
+        expect(text).toContain("Storage");
+        expect(text).toContain("Effective");
+    });
+    test("/trace status -g shows status with global flag", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-status-g";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "status -g", sessionId);
+        expect(text).toContain("Trace Status");
+    });
+    test("/trace status -l shows status with local flag", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-status-l";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "status -l", sessionId);
+        expect(text).toContain("Trace Status");
+    });
+    test("/trace status -s shows status with session flag", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-status-s";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "status -s", sessionId);
+        expect(text).toContain("Trace Status");
+        expect(text).toContain(`Session : ON`);
+    });
+    test("/trace status reflects ON state after /trace on -g", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-status-after-on";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        await runTraceCommand(hooks, mockPrompt, "on -g", sessionId);
+        mockPrompt.mockClear();
+        const { text } = await runTraceCommand(hooks, mockPrompt, "status", sessionId);
+        expect(text).toContain("Trace Status");
+        expect(text).toContain("Global  : ON");
+        expect(text).toContain("Effective: RECORDING");
+    });
+});
+describe("Plugin - /trace help and unknown commands", () => {
+    test("/trace help shows help text", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-help";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "help", sessionId);
+        expect(text).toContain("Usage: /trace");
+        expect(text).toContain("Commands:");
+        expect(text).toContain("on");
+        expect(text).toContain("off");
+        expect(text).toContain("status");
+        expect(text).toContain("-g");
+        expect(text).toContain("-l");
+        expect(text).toContain("-s");
+        expect(text).toContain("-d");
+    });
+    test("/trace with no arguments also shows help", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-no-args";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "", sessionId);
+        expect(text).toContain("Usage: /trace");
+    });
+    test("/trace foo (unknown subcommand) returns error message without crashing", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-unknown";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "foo", sessionId);
+        expect(text).toContain("Unknown command: foo");
+        expect(text).toContain("/trace on");
+        expect(text).toContain("/trace off");
+        expect(text).toContain("/trace status");
+    });
+    test("/trace with mixed-case ON is normalized", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-mixed-case";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const { text } = await runTraceCommand(hooks, mockPrompt, "ON", sessionId);
+        expect(text).toContain("Trace enabled");
+    });
+});
+describe("Plugin - slash command guards", () => {
+    test("non-trace commands are ignored (no prompt sent, no throw)", async () => {
+        const { hooks, mockPrompt } = await setupPluginWithMockClient(testDir);
+        const sessionId = "slash-other-cmd";
+        await createSessionViaEvent(hooks, sessionId, testDir);
+        const output = { parts: [{ type: "text", text: "original" }] };
+        let error = null;
+        try {
+            await hooks["command.execute.before"]({
+                command: "help",
+                sessionID: sessionId,
+                arguments: "",
+            }, output);
+        }
+        catch (err) {
+            error = err;
+        }
+        expect(error).toBeNull();
+        expect(mockPrompt).not.toHaveBeenCalled();
+        expect(output.parts.length).toBe(1);
+    });
+});
+describe("Plugin - tool.execute.after parentID linking", () => {
+    test("tool.execute.after with task tool records sub-session under parent's subSessions", async () => {
+        const { hooks, mockPrompt: _mockPrompt } = await setupPluginWithMockClient(testDir);
+        void _mockPrompt;
+        const parentSessionId = "parent-link-test";
+        const subSessionId = "sub-link-test";
+        await createSessionViaEvent(hooks, parentSessionId, testDir, "Parent");
+        await hooks["tool.execute.after"]({
+            tool: "task",
+            sessionID: parentSessionId,
+            callID: "call-789",
+            args: { description: "do work", prompt: "do work" },
+        }, {
+            title: "Task done",
+            output: "completed",
+            metadata: { session_id: subSessionId },
+        });
+        const parentMeta = readSessionMetadata(testDir, parentSessionId);
+        expect(parentMeta).toBeTruthy();
+        expect(parentMeta.subSessions).toBeDefined();
+        expect(parentMeta.subSessions).toContain(subSessionId);
+    });
+    test("tool.execute.after ignores task output without session_id metadata", async () => {
+        const { hooks, mockPrompt: _mockPrompt } = await setupPluginWithMockClient(testDir);
+        void _mockPrompt;
+        const parentSessionId = "parent-link-no-sid";
+        await createSessionViaEvent(hooks, parentSessionId, testDir, "Parent");
+        await hooks["tool.execute.after"]({
+            tool: "task",
+            sessionID: parentSessionId,
+            callID: "call-no-sid",
+            args: { description: "test" },
+        }, {
+            title: "Task done",
+            output: "completed",
+            metadata: {},
+        });
+        const parentMeta = readSessionMetadata(testDir, parentSessionId);
+        if (parentMeta.subSessions) {
+            expect(parentMeta.subSessions).not.toContain("phantom-sub");
+        }
+    });
+    test("event hook with parentID records sub-session in parent metadata", async () => {
+        const { hooks, mockPrompt: _mockPrompt } = await setupPluginWithMockClient(testDir);
+        void _mockPrompt;
+        const parentSessionId = "parent-event-link";
+        const childSessionId = "child-event-link";
+        await createSessionViaEvent(hooks, parentSessionId, testDir, "Parent");
+        await createSessionViaEvent(hooks, childSessionId, testDir, "Child", parentSessionId);
+        const parentMeta = readSessionMetadata(testDir, parentSessionId);
+        expect(parentMeta).toBeTruthy();
+        expect(parentMeta.subSessions).toBeDefined();
+        expect(parentMeta.subSessions).toContain(childSessionId);
+        const childMeta = readSessionMetadata(testDir, childSessionId);
+        expect(childMeta).toBeTruthy();
+        expect(childMeta.parentID).toBe(parentSessionId);
+    });
+});
+describe("Plugin - new hooks wiring (chat.message, chat.params, tool.execute.before)", () => {
+    test("plugin returns chat.message, chat.params, and tool.execute.before hooks as functions", async () => {
+        const hooks = await entrypoint.server({
+            client: {},
+            project: {},
+            directory: testDir,
+            worktree: testDir,
+            experimental_workspace: { register: vi.fn() },
+            serverUrl: new URL("http://localhost"),
+            $: {},
+        });
+        expect(typeof hooks["chat.message"]).toBe("function");
+        expect(typeof hooks["chat.params"]).toBe("function");
+        expect(typeof hooks["tool.execute.before"]).toBe("function");
+    });
+    test("chat.message hook logs via logger.info and does not throw", async () => {
+        const infoSpy = vi
+            .spyOn(logger, "info")
+            .mockImplementation((() => logger));
+        try {
+            const hooks = await entrypoint.server({
+                client: {},
+                project: {},
+                directory: testDir,
+                worktree: testDir,
+                experimental_workspace: { register: vi.fn() },
+                serverUrl: new URL("http://localhost"),
+                $: {},
+            });
+            const sessionId = "chat-message-test";
+            await createSessionViaEvent(hooks, sessionId, testDir, "ChatMsg");
+            await expect(hooks["chat.message"]({
+                sessionID: sessionId,
+                messageID: "msg-1",
+                agent: "build",
+            }, { message: {}, parts: [] })).resolves.toBeUndefined();
+            const chatMessageCalls = infoSpy.mock.calls.filter((c) => c[0] === "chat.message");
+            expect(chatMessageCalls.length).toBeGreaterThan(0);
+            const payload = chatMessageCalls[0][1];
+            expect(payload.sessionID).toBe(sessionId);
+            expect(payload.messageID).toBe("msg-1");
+            expect(payload.agent).toBe("build");
+        }
+        finally {
+            infoSpy.mockRestore();
+        }
+    });
+    test("chat.params hook logs via logger.info and does not throw", async () => {
+        const infoSpy = vi
+            .spyOn(logger, "info")
+            .mockImplementation((() => logger));
+        try {
+            const hooks = await entrypoint.server({
+                client: {},
+                project: {},
+                directory: testDir,
+                worktree: testDir,
+                experimental_workspace: { register: vi.fn() },
+                serverUrl: new URL("http://localhost"),
+                $: {},
+            });
+            const sessionId = "chat-params-test";
+            await createSessionViaEvent(hooks, sessionId, testDir, "ChatParams");
+            await expect(hooks["chat.params"]({
+                sessionID: sessionId,
+                agent: "build",
+                model: { providerID: "anthropic", modelID: "claude-3-5-sonnet" },
+                provider: { source: "config", info: {}, options: {} },
+                message: {},
+            }, {
+                temperature: 0.7,
+                topP: 0.9,
+                topK: 40,
+                maxOutputTokens: 100,
+                options: {},
+            })).resolves.toBeUndefined();
+            const chatParamsCalls = infoSpy.mock.calls.filter((c) => c[0] === "chat.params");
+            expect(chatParamsCalls.length).toBeGreaterThan(0);
+            const payload = chatParamsCalls[0][1];
+            expect(payload.sessionID).toBe(sessionId);
+            expect(payload.agent).toBe("build");
+        }
+        finally {
+            infoSpy.mockRestore();
+        }
+    });
+    test("tool.execute.before hook logs via logger.info and does not throw", async () => {
+        const infoSpy = vi
+            .spyOn(logger, "info")
+            .mockImplementation((() => logger));
+        try {
+            const hooks = await entrypoint.server({
+                client: {},
+                project: {},
+                directory: testDir,
+                worktree: testDir,
+                experimental_workspace: { register: vi.fn() },
+                serverUrl: new URL("http://localhost"),
+                $: {},
+            });
+            const sessionId = "tool-before-test";
+            await createSessionViaEvent(hooks, sessionId, testDir, "ToolBefore");
+            await expect(hooks["tool.execute.before"]({ tool: "bash", sessionID: sessionId, callID: "call-1" }, { args: { command: "ls" } })).resolves.toBeUndefined();
+            const toolBeforeCalls = infoSpy.mock.calls.filter((c) => c[0] === "tool.execute.before");
+            expect(toolBeforeCalls.length).toBeGreaterThan(0);
+            const payload = toolBeforeCalls[0][1];
+            expect(payload.sessionID).toBe(sessionId);
+            expect(payload.callID).toBe("call-1");
+            expect(payload.tool).toBe("bash");
+        }
+        finally {
+            infoSpy.mockRestore();
+        }
+    });
+});
+describe("Plugin - tool.execute.after parentID eager propagation", () => {
+    test("eagerly writes child session's parentID via updateSessionMetadata when task tool completes", async () => {
+        const { hooks } = await setupPluginWithMockClient(testDir);
+        const parentSessionId = "parent-eager-pid";
+        const childSessionId = "child-eager-pid";
+        await createSessionViaEvent(hooks, parentSessionId, testDir, "Parent");
+        await hooks["tool.execute.after"]({
+            tool: "task",
+            sessionID: parentSessionId,
+            callID: "call-eager",
+            args: { description: "sub-work", prompt: "do it" },
+        }, {
+            title: "Task done",
+            output: "completed",
+            metadata: { session_id: childSessionId },
+        });
+        const childMeta = readSessionMetadata(testDir, childSessionId);
+        expect(childMeta).toBeTruthy();
+        expect(childMeta.parentID).toBe(parentSessionId);
+        const parentMeta = readSessionMetadata(testDir, parentSessionId);
+        expect(parentMeta.subSessions).toContain(childSessionId);
+    });
+    test("does not eagerly write parentID for non-task tools", async () => {
+        const { hooks } = await setupPluginWithMockClient(testDir);
+        const sessionId = "non-task-eager";
+        await createSessionViaEvent(hooks, sessionId, testDir, "Parent");
+        const ghostChild = "ghost-child-eager";
+        await hooks["tool.execute.after"]({
+            tool: "read",
+            sessionID: sessionId,
+            callID: "call-read",
+            args: { filePath: "/foo" },
+        }, {
+            title: "Read done",
+            output: "contents",
+            metadata: { session_id: ghostChild },
+        });
+        const ghostMeta = readSessionMetadata(testDir, ghostChild);
+        if (ghostMeta !== null) {
+            expect(ghostMeta.parentID).toBeUndefined();
+        }
+    });
+    test("does not throw when task metadata has no session_id", async () => {
+        const { hooks } = await setupPluginWithMockClient(testDir);
+        const sessionId = "task-no-sid-eager";
+        await createSessionViaEvent(hooks, sessionId, testDir, "Parent");
+        await expect(hooks["tool.execute.after"]({
+            tool: "task",
+            sessionID: sessionId,
+            callID: "call-no-sid-eager",
+            args: {},
+        }, {
+            title: "Task done",
+            output: "completed",
+            metadata: {},
+        })).resolves.toBeUndefined();
     });
 });
 //# sourceMappingURL=trace.test.js.map