@selesai/code 0.13.16 → 0.13.18

package/dist/extensions/pi-subagents/test/unit/index-child-registration.test.ts

@@ -6,14 +6,13 @@ import * as path from "node:path";
 import { fileURLToPath } from "node:url";
 import { describe, it } from "node:test";
 import { WAIT_TOOL_ENABLED_ENV } from "../../src/runs/background/subagent-wait.ts";
-import { SUBAGENT_CHILD_ENV, SUBAGENT_FANOUT_CHILD_ENV } from "../../src/runs/shared/pi-args.ts";
+import { SUBAGENT_CHILD_ENV } from "../../src/runs/shared/child-runtime-config.ts";
 
 const projectRoot = path.resolve(path.dirname(fileURLToPath(import.meta.url)), "..", "..");
 
 function parentToolEnv(agentDir?: string): NodeJS.ProcessEnv {
 	const env = { ...process.env };
 	delete env[SUBAGENT_CHILD_ENV];
-	delete env[SUBAGENT_FANOUT_CHILD_ENV];
 	delete env[WAIT_TOOL_ENABLED_ENV];
 	if (agentDir) env.SELESAI_CODING_AGENT_DIR = agentDir;
 	return env;
@@ -554,7 +553,7 @@ describe("subagent extension child mode", () => {
 				const events = { on(channel, handler) { eventHandlers.set(channel, handler); return () => {}; }, emit() {} };
 				const fakePi = new Proxy({
 					events,
-					on(channel, handler) { handlers.set(channel, handler); },
+					on(channel, handler) { handlers.set(channel, [...(handlers.get(channel) ?? []), handler]); },
 					registerTool() {}, registerCommand() {}, registerShortcut() {}, registerMessageRenderer() {},
 					sendMessage() {}, getSessionName() { return undefined; },
 				}, { get(target, prop) { return prop in target ? target[prop] : () => undefined; } });
@@ -566,13 +565,13 @@ describe("subagent extension child mode", () => {
 					modelRegistry: { getAvailable() { return []; } },
 				};
 				registerSubagentExtension(fakePi);
-				handlers.get("session_start")({}, ctx);
+				for (const handler of handlers.get("session_start")) await handler({}, ctx);
 				widgets.length = 0;
 				eventHandlers.get("subagent:async-started")({ id: "widget-run", pid: 1, sessionId: "session-widget", mode: "single", agent: "worker", asyncDir: "/tmp/widget-run" });
-				handlers.get("tool_result")({ toolName: "subagent" }, ctx);
+				for (const handler of handlers.get("tool_result")) await handler({ toolName: "subagent" }, ctx);
 				const asyncWidgets = widgets.filter((entry) => entry.key === "subagent-async");
 				if (asyncWidgets.length < 2 || asyncWidgets.some((entry) => entry.value !== undefined)) throw new Error("async widget rendered despite disabled config: " + JSON.stringify(asyncWidgets));
-				handlers.get("session_shutdown")();
+				for (const handler of handlers.get("session_shutdown")) await handler();
 			`;
 			const env = parentToolEnv();
 			env.SELESAI_CODING_AGENT_DIR = agentDir;
@@ -595,7 +594,7 @@ describe("subagent extension child mode", () => {
 				const events = { on(channel, handler) { eventHandlers.set(channel, handler); return () => {}; }, emit() {} };
 				const fakePi = new Proxy({
 					events,
-					on(channel, handler) { handlers.set(channel, handler); },
+					on(channel, handler) { handlers.set(channel, [...(handlers.get(channel) ?? []), handler]); },
 					registerTool() {}, registerCommand() {}, registerShortcut() {}, registerMessageRenderer() {},
 					sendMessage() {}, getSessionName() { return undefined; },
 				}, { get(target, prop) { return prop in target ? target[prop] : () => undefined; } });
@@ -607,13 +606,13 @@ describe("subagent extension child mode", () => {
 					modelRegistry: { getAvailable() { return []; } },
 				};
 				registerSubagentExtension(fakePi);
-				handlers.get("session_start")({}, ctx);
+				for (const handler of handlers.get("session_start")) await handler({}, ctx);
 				widgets.length = 0;
 				eventHandlers.get("subagent:async-started")({ id: "widget-run", pid: 1, sessionId: "session-widget", mode: "workflow", agent: "worker", asyncDir: "/tmp/widget-run" });
-				handlers.get("tool_result")({ toolName: "subagent" }, ctx);
+				for (const handler of handlers.get("tool_result")) await handler({ toolName: "subagent" }, ctx);
 				const asyncWidgets = widgets.filter((entry) => entry.key === "subagent-async");
 				if (!asyncWidgets.some((entry) => entry.value !== undefined)) throw new Error("async widget was not rendered with FleetView enabled: " + JSON.stringify(asyncWidgets));
-				handlers.get("session_shutdown")();
+				for (const handler of handlers.get("session_shutdown")) await handler();
 			`;
 			const env = parentToolEnv();
 			env.SELESAI_CODING_AGENT_DIR = agentDir;
@@ -639,7 +638,7 @@ describe("subagent extension child mode", () => {
 			process.env.HERDR_PANE_ID = "w1:p1";
 			const fakePi = new Proxy({
 				events,
-				on(channel, handler) { handlers.set(channel, handler); },
+				on(channel, handler) { handlers.set(channel, [...(handlers.get(channel) ?? []), handler]); },
 				exec(command, args) { herdrCommands.push({ command, args }); return Promise.resolve({ code: 0, stdout: "", stderr: "", killed: false }); },
 				registerTool() {}, registerCommand() {}, registerShortcut() {}, registerMessageRenderer() {},
 				sendMessage() {}, getSessionName() { return undefined; },
@@ -655,7 +654,7 @@ describe("subagent extension child mode", () => {
 			const asyncDir = path.join(DIRS.async, runId);
 			fs.rmSync(asyncDir, { recursive: true, force: true });
 			registerSubagentExtension(fakePi);
-			handlers.get("session_start")({}, ctx);
+			for (const handler of handlers.get("session_start")) await handler({}, ctx);
 			widgets.length = 0;
 			fs.mkdirSync(asyncDir, { recursive: true });
 			fs.writeFileSync(path.join(asyncDir, "status.json"), JSON.stringify({
@@ -663,19 +662,81 @@ describe("subagent extension child mode", () => {
 				startedAt: Date.now(), lastUpdate: Date.now(), cwd: process.cwd(), pid: process.pid,
 			}), "utf-8");
 			updateActiveRunIndex(asyncDir, "running");
-			handlers.get("tool_result")({ toolName: "subagent" }, ctx);
+			for (const handler of handlers.get("tool_result")) await handler({ toolName: "subagent" }, ctx);
 			const fleetWidgets = widgets.filter((entry) => entry.key === "subagent-fleet-status");
 			if (!fleetWidgets.some((entry) => typeof entry.value === "function")) throw new Error("management result did not restore active fleet status: " + JSON.stringify(fleetWidgets));
 			if (!herdrCommands.some(({ args }) => args.includes("summary=⏳ 1 subagent"))) throw new Error("management result did not restore Herdr status: " + JSON.stringify(herdrCommands));
 			const herdrCommandCount = herdrCommands.length;
-			handlers.get("tool_result")({ toolName: "subagent" }, ctx);
+			for (const handler of handlers.get("tool_result")) await handler({ toolName: "subagent" }, ctx);
 			if (herdrCommands.length !== herdrCommandCount) throw new Error("unchanged active jobs redundantly refreshed Herdr status: " + JSON.stringify(herdrCommands));
-			handlers.get("session_shutdown")();
+			for (const handler of handlers.get("session_shutdown")) await handler();
 			fs.rmSync(asyncDir, { recursive: true, force: true });
 		`;
 		execFileSync(process.execPath, ["--experimental-strip-types", "--import", "./test/support/register-loader.mjs", "--input-type=module", "--eval", script], { cwd: projectRoot, env: parentToolEnv(), stdio: "pipe" });
 	});
 
+	it("registers pi-web liveness for the current session and releases it on shutdown", () => {
+		const script = String.raw`
+			import registerSubagentExtension from "./index.ts";
+			import { currentCompletionOwnerId } from "./src/shared/completion-owner.ts";
+			const handlers = new Map();
+			const listeners = new Map();
+			const events = {
+				on(channel, handler) {
+					let set = listeners.get(channel);
+					if (!set) listeners.set(channel, set = new Set());
+					set.add(handler);
+					return () => set.delete(handler);
+				},
+				emit(channel, payload) {
+					for (const handler of [...(listeners.get(channel) ?? [])]) handler(payload);
+				},
+			};
+			const pi = new Proxy({
+				events,
+				on(channel, handler) { handlers.set(channel, [...(handlers.get(channel) ?? []), handler]); },
+				registerTool() {}, registerCommand() {}, registerShortcut() {}, registerMessageRenderer() {},
+				sendMessage() {}, getSessionName() { return undefined; },
+			}, { get(target, prop) { return prop in target ? target[prop] : () => undefined; } });
+			const sessionId = "liveness-" + crypto.randomUUID();
+			const sessionFile = "/tmp/" + sessionId + ".jsonl";
+			let provider;
+			let released = 0;
+			const registryKey = Symbol.for("@agegr/pi-web/session-liveness/v1");
+			globalThis[registryKey] = {
+				version: 1,
+				register(value) {
+					provider = value;
+					return () => { released += 1; };
+				},
+			};
+			const ctx = {
+				cwd: process.cwd(), hasUI: false,
+				ui: { setWidget() {}, requestRender() {}, theme: { fg(_name, text) { return text; }, bg(_name, text) { return text; }, bold(text) { return text; } } },
+				sessionManager: { getSessionId() { return sessionId; }, getSessionFile() { return sessionFile; }, getEntries() { return []; } },
+				modelRegistry: { getAvailable() { return []; } },
+			};
+			registerSubagentExtension(pi);
+			for (const handler of handlers.get("session_start")) await handler({ reason: "startup" }, ctx);
+			if (provider?.name !== "pi-subagents" || provider?.sessionId !== sessionId || provider?.sessionFile !== sessionFile) {
+				throw new Error("liveness provider did not preserve exact session identity: " + JSON.stringify(provider));
+			}
+			if (provider.isActive()) throw new Error("idle runtime reported live work");
+			const completionOwnerId = currentCompletionOwnerId();
+			events.emit("subagent:async-started", { id: "run-1", sessionId: sessionFile, completionOwnerId, mode: "single", agent: "worker", asyncDir: "/tmp/" + sessionId + "-run" });
+			if (!provider.isActive()) throw new Error("queued async work was not reported live");
+			events.emit("subagent:async-complete", { id: "run-1", sessionId: sessionFile, completionOwnerId, success: true, summary: "done" });
+			if (!provider.isActive()) throw new Error("pending completion delivery was not reported live");
+			const deliveryDeadline = Date.now() + 2000;
+			while (provider.isActive() && Date.now() < deliveryDeadline) await new Promise((resolve) => setTimeout(resolve, 10));
+			if (provider.isActive()) throw new Error("retained terminal work was reported live after delivery");
+			for (const handler of handlers.get("session_shutdown")) await handler({ reason: "quit" });
+			if (released !== 1) throw new Error("liveness registration was not released exactly once: " + released);
+			delete globalThis[registryKey];
+		`;
+		execFileSync(process.execPath, ["--experimental-strip-types", "--import", "./test/support/register-loader.mjs", "--input-type=module", "--eval", script], { cwd: projectRoot, env: parentToolEnv(), stdio: "pipe" });
+	});
+
 	it("keeps independent extension runtimes active in one process", () => {
 		const script = String.raw`
 			import registerSubagentExtension from "./index.ts";
@@ -701,7 +762,7 @@ describe("subagent extension child mode", () => {
 				};
 				const pi = new Proxy({
 					events,
-					on(channel, handler) { handlers.set(channel, handler); },
+					on(channel, handler) { handlers.set(channel, [...(handlers.get(channel) ?? []), handler]); },
 					registerTool() {}, registerCommand() {}, registerShortcut() {}, registerMessageRenderer() {},
 					sendMessage() {}, getSessionName() { return undefined; },
 				}, { get(target, prop) { return prop in target ? target[prop] : () => undefined; } });
@@ -721,7 +782,7 @@ describe("subagent extension child mode", () => {
 
 			const first = createRuntime("independent-first");
 			registerSubagentExtension(first.pi);
-			first.handlers.get("session_start")({ reason: "startup" }, first.ctx);
+			for (const handler of first.handlers.get("session_start")) await handler({ reason: "startup" }, first.ctx);
 			first.events.emit("subagent:async-complete", {
 				id: "independent-baseline", agent: "worker", success: true, summary: "Baseline",
 				exitCode: 0, timestamp: Date.now(), sessionId: "independent-first", completionOwnerId,
@@ -730,9 +791,9 @@ describe("subagent extension child mode", () => {
 			if (baselineDeliveries === 0) throw new Error("first runtime did not register async-complete listeners");
 			const second = createRuntime("independent-second");
 			registerSubagentExtension(second.pi);
-			second.handlers.get("session_start")({ reason: "startup" }, second.ctx);
+			for (const handler of second.handlers.get("session_start")) await handler({ reason: "startup" }, second.ctx);
 
-			await first.handlers.get("agent_end")({}, first.ctx);
+			for (const handler of first.handlers.get("agent_end")) await handler({}, first.ctx);
 			first.events.emit("subagent:async-complete", {
 				id: "independent-completion", agent: "worker", success: true, summary: "Done",
 				exitCode: 0, timestamp: Date.now(), sessionId: "independent-first", completionOwnerId,
@@ -741,11 +802,11 @@ describe("subagent extension child mode", () => {
 				throw new Error("first runtime no longer received async-complete after second registration");
 			}
 
-			await first.handlers.get("session_shutdown")();
+			for (const handler of first.handlers.get("session_shutdown")) await handler();
 			if (process.env.SELESAI_SUBAGENT_PARENT_SESSION !== "independent-second") {
 				throw new Error("independent shutdown cleared another runtime's parent session identity");
 			}
-			await second.handlers.get("agent_end")({}, second.ctx);
+			for (const handler of second.handlers.get("agent_end")) await handler({}, second.ctx);
 			second.events.emit("subagent:async-complete", {
 				id: "independent-second-completion", agent: "reviewer", success: true, summary: "Done",
 				exitCode: 0, timestamp: Date.now(), sessionId: "independent-second", completionOwnerId,
@@ -753,7 +814,7 @@ describe("subagent extension child mode", () => {
 			if ((second.eventDeliveries.get("subagent:async-complete") ?? 0) === 0) {
 				throw new Error("first runtime shutdown removed the second runtime's event subscription");
 			}
-			await second.handlers.get("session_shutdown")();
+			for (const handler of second.handlers.get("session_shutdown")) await handler();
 			if (process.env.SELESAI_SUBAGENT_PARENT_SESSION !== undefined) {
 				throw new Error("owning shutdown left its parent session identity active");
 			}
@@ -775,7 +836,7 @@ describe("subagent extension child mode", () => {
 				const events = { on() { return () => {}; }, emit() {} };
 				const pi = new Proxy({
 					events,
-					on(channel, handler) { handlers.set(channel, handler); },
+					on(channel, handler) { handlers.set(channel, [...(handlers.get(channel) ?? []), handler]); },
 					registerTool() {}, registerCommand() {}, registerShortcut() {}, registerMessageRenderer() {},
 					sendMessage() {}, getSessionName() { return undefined; },
 				}, { get(target, prop) { return prop in target ? target[prop] : () => undefined; } });
@@ -791,18 +852,18 @@ describe("subagent extension child mode", () => {
 			const first = createRuntime("slash-first");
 			const second = createRuntime("slash-second");
 			registerSubagentExtension(first.pi);
-			first.handlers.get("session_start")({ reason: "startup" }, first.ctx);
+			for (const handler of first.handlers.get("session_start")) await handler({ reason: "startup" }, first.ctx);
 			registerSubagentExtension(second.pi);
-			second.handlers.get("session_start")({ reason: "startup" }, second.ctx);
+			for (const handler of second.handlers.get("session_start")) await handler({ reason: "startup" }, second.ctx);
 			const details = buildSlashInitialResult("slash-isolation", { agent: "worker", task: "Keep this snapshot" });
 			const liveVersion = getSlashRenderableSnapshot(details).version;
 			if (liveVersion <= 0) throw new Error("slash snapshot was not populated");
 
-			await first.handlers.get("session_shutdown")({ reason: "shutdown" });
+			for (const handler of first.handlers.get("session_shutdown")) await handler({ reason: "shutdown" });
 			if (getSlashRenderableSnapshot(details).version !== liveVersion) {
 				throw new Error("one runtime shutdown cleared another active runtime's slash snapshot");
 			}
-			await second.handlers.get("session_shutdown")({ reason: "shutdown" });
+			for (const handler of second.handlers.get("session_shutdown")) await handler({ reason: "shutdown" });
 			if (getSlashRenderableSnapshot(details).version !== 0) {
 				throw new Error("last runtime shutdown did not clear slash snapshots");
 			}
@@ -849,7 +910,7 @@ describe("subagent extension child mode", () => {
 			const sent = [];
 			const fakePi = new Proxy({
 				events,
-				on(channel, handler) { handlers.set(channel, handler); },
+				on(channel, handler) { handlers.set(channel, [...(handlers.get(channel) ?? []), handler]); },
 				registerTool() {}, registerCommand() {}, registerShortcut() {}, registerMessageRenderer() {},
 				sendMessage(message) { sent.push(message); }, getSessionName() { return undefined; },
 			}, { get(target, prop) { return prop in target ? target[prop] : () => undefined; } });
@@ -860,7 +921,7 @@ describe("subagent extension child mode", () => {
 				modelRegistry: { getAvailable() { return []; } },
 			};
 			registerSubagentExtension(fakePi);
-			handlers.get("session_start")({}, ctx);
+			for (const handler of handlers.get("session_start")) await handler({}, ctx);
 			sent.length = 0;
 			events.emit("subagent:async-complete", {
 				id: "shutdown-held-completion", agent: "worker", success: true, summary: "Done",
@@ -869,7 +930,7 @@ describe("subagent extension child mode", () => {
 			if (sent.length !== 0) throw new Error("completion was not queued before shutdown");
 			const heldTimers = [...pendingTimers.values()];
 			if (heldTimers.length === 0) throw new Error("completion did not schedule a timer");
-			handlers.get("session_shutdown")();
+			for (const handler of handlers.get("session_shutdown")) await handler();
 			if (pendingTimers.size !== 0) throw new Error("shutdown left completion timers pending");
 			for (const handler of heldTimers) handler();
 			if (sent.length !== 0) throw new Error("stale completion sent after shutdown");
@@ -930,7 +991,7 @@ describe("subagent extension child mode", () => {
 				const sent = [];
 				const pi = new Proxy({
 					events,
-					on(channel, handler) { handlers.set(channel, handler); },
+					on(channel, handler) { handlers.set(channel, [...(handlers.get(channel) ?? []), handler]); },
 					registerTool() {}, registerCommand() {}, registerShortcut() {}, registerMessageRenderer() {},
 					sendMessage(message) { sent.push(message); }, getSessionName() { return undefined; },
 				}, { get(target, prop) { return prop in target ? target[prop] : () => undefined; } });
@@ -945,7 +1006,7 @@ describe("subagent extension child mode", () => {
 
 			const oldRuntime = createRuntime();
 			registerSubagentExtension(oldRuntime.pi);
-			oldRuntime.handlers.get("session_start")({ reason: "startup" }, oldRuntime.ctx);
+			for (const handler of oldRuntime.handlers.get("session_start")) await handler({ reason: "startup" }, oldRuntime.ctx);
 			const oldListenerCount = eventListeners.get("subagent:async-complete")?.size ?? 0;
 			if (oldListenerCount === 0) throw new Error("old runtime did not register async-complete listeners");
 			oldRuntime.sent.length = 0;
@@ -960,13 +1021,13 @@ describe("subagent extension child mode", () => {
 
 			const newRuntime = createRuntime();
 			registerSubagentExtension(newRuntime.pi);
-			newRuntime.handlers.get("session_start")({ reason: "reload" }, newRuntime.ctx);
+			for (const handler of newRuntime.handlers.get("session_start")) await handler({ reason: "reload" }, newRuntime.ctx);
 			if (eventListeners.get("subagent:async-complete")?.size !== oldListenerCount) {
 				throw new Error("replacement runtime did not restore its event subscriptions");
 			}
 			for (const [, handler] of oldCompletionTimers) handler();
 			if (oldRuntime.sent.length !== 0) throw new Error("stale completion sent after runtime cleanup");
-			await oldRuntime.handlers.get("session_shutdown")({ reason: "reload" });
+			for (const handler of oldRuntime.handlers.get("session_shutdown")) await handler({ reason: "reload" });
 			if (eventListeners.get("subagent:async-complete")?.size !== oldListenerCount
 				|| process.env.SELESAI_SUBAGENT_PARENT_SESSION !== "notify-reload-session") {
 				throw new Error("stale shutdown changed the replacement runtime");
@@ -984,7 +1045,7 @@ describe("subagent extension child mode", () => {
 				handler();
 			}
 			if (newRuntime.sent.length !== 1) throw new Error("new notifier was not active after reload cleanup");
-			await newRuntime.handlers.get("session_shutdown")({ reason: "shutdown" });
+			for (const handler of newRuntime.handlers.get("session_shutdown")) await handler({ reason: "shutdown" });
 			globalThis.setTimeout = realSetTimeout;
 			globalThis.clearTimeout = realClearTimeout;
 		`;
@@ -1013,7 +1074,7 @@ describe("subagent extension child mode", () => {
 			const events = { on() { return () => {}; }, emit() {} };
 			const fakePi = new Proxy({
 				events,
-				on(channel, handler) { handlers.set(channel, handler); },
+				on(channel, handler) { handlers.set(channel, [...(handlers.get(channel) ?? []), handler]); },
 				registerTool() {}, registerCommand() {}, registerShortcut() {}, registerMessageRenderer() {},
 				sendMessage() {}, getSessionName() { return undefined; },
 			}, { get(target, prop) { return prop in target ? target[prop] : () => undefined; } });
@@ -1032,9 +1093,9 @@ describe("subagent extension child mode", () => {
 				modelRegistry: { getAvailable() { return []; } },
 			};
 			registerSubagentExtension(fakePi);
-			handlers.get("session_start")({ reason: "startup" }, ctx);
+			for (const handler of handlers.get("session_start")) await handler({ reason: "startup" }, ctx);
 			stale = true;
-			handlers.get("session_shutdown")({ reason: "reload" });
+			for (const handler of handlers.get("session_shutdown")) await handler({ reason: "reload" });
 		`;
 
 		try {
@@ -1069,7 +1130,7 @@ describe("subagent extension child mode", () => {
 			const sent = [];
 			const pi = new Proxy({
 				events,
-				on(channel, handler) { handlers.set(channel, handler); },
+				on(channel, handler) { handlers.set(channel, [...(handlers.get(channel) ?? []), handler]); },
 				registerTool() {}, registerCommand() {}, registerShortcut() {}, registerMessageRenderer() {},
 				sendMessage(message) { sent.push(message); }, getSessionName() { return undefined; },
 			}, { get(target, prop) { return prop in target ? target[prop] : () => undefined; } });
@@ -1083,14 +1144,14 @@ describe("subagent extension child mode", () => {
 			const sessionManager = { getSessionId() { return path.basename(currentSession); }, getSessionFile() { return currentSession; }, getEntries() { return []; } };
 			const ctx = { cwd: process.cwd(), hasUI: false, ui: { setWidget() {}, requestRender() {}, theme: { fg(_name, text) { return text; }, bg(_name, text) { return text; }, bold(text) { return text; } } }, sessionManager, modelRegistry: { getAvailable() { return []; } } };
 			registerSubagentExtension(pi);
-			handlers.get("session_start")({ reason: "startup" }, ctx);
+			for (const handler of handlers.get("session_start")) await handler({ reason: "startup" }, ctx);
 			currentSession = newSession;
-			handlers.get("session_start")({ reason: "new", previousSessionFile: oldSession }, ctx);
+			for (const handler of handlers.get("session_start")) await handler({ reason: "new", previousSessionFile: oldSession }, ctx);
 			const owner = currentCompletionOwnerId();
 			events.emit("subagent:async-complete", { id: "old-owned", agent: "worker", success: true, summary: "old", timestamp: 1, sessionId: oldSession, completionOwnerId: owner });
 			events.emit("subagent:async-complete", { id: "foreign", agent: "worker", success: true, summary: "foreign", timestamp: 2, sessionId: path.join(sessions, "foreign.jsonl"), completionOwnerId: owner });
 			if (sent.length !== 1) throw new Error("expected only the claimed predecessor completion, got " + sent.length);
-			await handlers.get("session_shutdown")({ reason: "quit" });
+			for (const handler of handlers.get("session_shutdown")) await handler({ reason: "quit" });
 		`;
 
 		try {
@@ -1108,12 +1169,14 @@ describe("subagent extension child mode", () => {
 			const events = { on() { return () => {}; }, emit() {} };
 			const commands = [];
 			const renderers = [];
+			const entryRenderers = [];
 			const fakePi = new Proxy({
 				events,
 				registerTool() {},
 				registerCommand(name) { commands.push(name); },
 				registerShortcut() {},
 				registerMessageRenderer(type) { renderers.push(type); },
+				registerEntryRenderer(type) { entryRenderers.push(type); },
 				sendMessage() {},
 				getSessionName() { return undefined; },
 			}, {
@@ -1125,6 +1188,8 @@ describe("subagent extension child mode", () => {
 			registerSubagentExtension(fakePi);
 			if (!commands.includes("subagents-watchdog")) throw new Error("watchdog command not registered: " + commands.join(", "));
 			if (!renderers.includes("subagent_watchdog_warning")) throw new Error("watchdog renderer not registered: " + renderers.join(", "));
+			if (!renderers.includes("subagent_supervisor_request")) throw new Error("supervisor request renderer not registered: " + renderers.join(", "));
+			if (!entryRenderers.includes("subagent_supervisor_reply")) throw new Error("supervisor reply entry renderer not registered: " + entryRenderers.join(", "));
 		`;
 
 		execFileSync(
@@ -1141,7 +1206,7 @@ describe("subagent extension child mode", () => {
 		);
 	});
 
-	it("returns before registering anything for non-fanout children", () => {
+	it("returns before registering anything in a child-hosting process", () => {
 		const script = String.raw`
 			import registerSubagentExtension from "./index.ts";
 			const calls = [];
@@ -1169,16 +1234,15 @@ describe("subagent extension child mode", () => {
 				"--eval",
 				script,
 			],
-			{ cwd: projectRoot, env: { ...parentToolEnv(), [SUBAGENT_CHILD_ENV]: "1", [SUBAGENT_FANOUT_CHILD_ENV]: "0" }, stdio: "pipe" },
+			{ cwd: projectRoot, env: { ...parentToolEnv(), [SUBAGENT_CHILD_ENV]: "1" }, stdio: "pipe" },
 		);
 	});
 
-	it("returns before registering anything for fanout children", () => {
+	it("returns before registering anything when the child host flag is set after import", () => {
 		const script = String.raw`
 			import registerSubagentExtension from "./index.ts";
-			import { SUBAGENT_CHILD_ENV, SUBAGENT_FANOUT_CHILD_ENV } from "./src/runs/shared/pi-args.ts";
+			import { SUBAGENT_CHILD_ENV } from "./src/runs/shared/child-runtime-config.ts";
 			process.env[SUBAGENT_CHILD_ENV] = "1";
-			process.env[SUBAGENT_FANOUT_CHILD_ENV] = "1";
 			const calls = [];
 			const fakePi = new Proxy({}, {
 				get(target, prop) {
@@ -1213,9 +1277,9 @@ describe("subagent extension child mode", () => {
 		const script = String.raw`
 			import registerSubagentExtension from "./index.ts";
 			import registerFanoutChildSubagentExtension from "./src/extension/fanout-child.ts";
-			import { SUBAGENT_CHILD_ENV, SUBAGENT_FANOUT_CHILD_ENV } from "./src/runs/shared/pi-args.ts";
+			import { SUBAGENT_CHILD_ENV } from "./src/runs/shared/child-runtime-config.ts";
 			process.env[SUBAGENT_CHILD_ENV] = "1";
-			process.env[SUBAGENT_FANOUT_CHILD_ENV] = "1";
+			const childRuntime = { fanoutChild: true, depth: 1, waitTool: { enabled: true }, fast: false };
 
 			const registeredNames = new Set();
 			const registrations = [];
@@ -1234,7 +1298,7 @@ describe("subagent extension child mode", () => {
 			}
 
 			registerSubagentExtension(makePi("index.ts"));
-			registerFanoutChildSubagentExtension(makePi("fanout-child.ts"));
+			registerFanoutChildSubagentExtension(makePi("fanout-child.ts"), childRuntime);
 			if (registrations.length !== 1 || registrations[0].name !== "subagent" || registrations[0].source !== "fanout-child.ts") {
 				throw new Error("expected only fanout-child.ts to register subagent, got " + JSON.stringify(registrations));
 			}
@@ -1258,16 +1322,13 @@ describe("subagent extension child mode", () => {
 		const script = String.raw`
 			import assert from "node:assert/strict";
 			import registerFanoutChildSubagentExtension from "./src/extension/fanout-child.ts";
-			import { SUBAGENT_CHILD_ENV, SUBAGENT_FANOUT_CHILD_ENV } from "./src/runs/shared/pi-args.ts";
-			process.env[SUBAGENT_CHILD_ENV] = "1";
-			process.env[SUBAGENT_FANOUT_CHILD_ENV] = "1";
 			let registeredTool;
 			const fakePi = {
 				events: { on() { return () => {}; }, emit() {} },
 				registerTool(tool) { registeredTool = tool; },
 				getSessionName() { return undefined; },
 			};
-			registerFanoutChildSubagentExtension(fakePi);
+			registerFanoutChildSubagentExtension(fakePi, { fanoutChild: true, depth: 1, waitTool: { enabled: true }, fast: false });
 			if (!registeredTool) throw new Error("tool not registered");
 			const ctx = {
 				cwd: process.cwd(),
@@ -1304,4 +1365,4 @@ describe("subagent extension child mode", () => {
 			{ cwd: projectRoot, stdio: "pipe" },
 		);
 	});
-});
+});

package/dist/extensions/pi-subagents/test/unit/index-segment.test.ts

@@ -6,6 +6,7 @@ import * as path from "node:path";
 import { describe, it } from "node:test";
 import { ACTIVE_RUN_INDEX_DIR, readActiveRunToolCallIndex, releaseActiveRunIndex, updateActiveRunIndex } from "../../src/runs/background/active-run-index.ts";
 import { encodeIndexSegment, indexSegmentAliases, MAX_INDEX_SEGMENT_BYTES } from "../../src/runs/background/index-segment.ts";
+import { readStatus } from "../../src/shared/utils.ts";
 
 describe("bounded index segments", () => {
 	it("preserves legacy encoding for short values and hashes oversized opaque ids", () => {
@@ -93,4 +94,60 @@ describe("bounded index segments", () => {
 			fs.rmSync(root, { recursive: true, force: true });
 		}
 	});
-});
+
+	it("readStatus safely handles oversized directory paths without throwing ENAMETOOLONG", () => {
+		const root = fs.mkdtempSync(path.join(os.tmpdir(), "pi-read-status-long-path-"));
+		try {
+			const longSegment = "x".repeat(300);
+			const longDir = path.join(root, longSegment);
+			assert.doesNotThrow(() => {
+				const status = readStatus(longDir);
+				assert.equal(status, null);
+			});
+		} finally {
+			fs.rmSync(root, { recursive: true, force: true });
+		}
+	});
+
+	it("surfaces unrelated ENAMETOOLONG errors from status inspection", (t) => {
+		const error = new Error("unrelated status path too long") as NodeJS.ErrnoException;
+		error.code = "ENAMETOOLONG";
+		t.mock.method(fsDefault, "statSync", () => {
+			throw error;
+		});
+		syncBuiltinESMExports();
+		try {
+			assert.throws(() => readStatus(path.join(os.tmpdir(), "short-status-id")), (thrown) =>
+				thrown instanceof Error
+				&& /Failed to inspect async status file/.test(thrown.message)
+				&& (thrown as Error & { cause?: unknown }).cause === error);
+		} finally {
+			t.mock.restoreAll();
+			syncBuiltinESMExports();
+		}
+	});
+
+	it("surfaces ENAMETOOLONG errors from status reads", (t) => {
+		const root = fs.mkdtempSync(path.join(os.tmpdir(), "pi-read-status-read-error-"));
+		const asyncDir = path.join(root, "short-status-id");
+		const statusPath = path.join(asyncDir, "status.json");
+		const error = new Error("unrelated status read path too long") as NodeJS.ErrnoException;
+		error.code = "ENAMETOOLONG";
+		try {
+			fs.mkdirSync(asyncDir);
+			fs.writeFileSync(statusPath, "{}", "utf-8");
+			t.mock.method(fsDefault, "readFileSync", () => {
+				throw error;
+			});
+			syncBuiltinESMExports();
+			assert.throws(() => readStatus(asyncDir), (thrown) =>
+				thrown instanceof Error
+				&& /Failed to read async status file/.test(thrown.message)
+				&& (thrown as Error & { cause?: unknown }).cause === error);
+		} finally {
+			t.mock.restoreAll();
+			syncBuiltinESMExports();
+			fs.rmSync(root, { recursive: true, force: true });
+		}
+	});
+});