agent-theatre 1.2.3 → 1.3.1

package/dist/cli.js

@@ -300,9 +300,81 @@ function claimsSuccess(text) {
   if (HEDGE_LEXICON.some((r) => r.test(tail))) return false;
   return SUCCESS_LEXICON.some((r) => r.test(tail));
 }
-var ROW_TYPES = /* @__PURE__ */ new Set(["user", "assistant", "system"]);
 var obj2 = (x) => x && typeof x === "object" ? x : {};
 var str2 = (x) => typeof x === "string" ? x : void 0;
+var TOOL_EVENTS = /* @__PURE__ */ new Set(["PreToolUse", "PostToolUse", "PostToolUseFailure"]);
+function endReason(reason) {
+  switch (reason) {
+    case "clear":
+    case "logout":
+    case "prompt_input_exit":
+      return "user";
+    default:
+      return "stop";
+  }
+}
+function hookPayloadToEvents(payload, opts = {}) {
+  const p = obj2(payload);
+  const name = str2(p.hook_event_name);
+  const sessionId = str2(p.session_id) ?? "";
+  const ts = opts.now ?? Date.now();
+  if (!name) return { events: [], warnings: [{ code: "missing_field", detail: "hook_event_name" }] };
+  const cwd = str2(p.cwd) ?? "";
+  const ctx = { cwd, ...opts.home ? { home: opts.home } : {} };
+  const agentId = TOOL_EVENTS.has(name) ? str2(p.agent_id) : void 0;
+  const extra = agentId ? { agentId } : {};
+  let seq = 0;
+  const mk = (body, id = mintId(ts)) => ({ id, seq: seq++, ts, sessionId, source: "hook", ...extra, ...body });
+  const one = (body) => ({ events: [mk(body)], warnings: [] });
+  switch (name) {
+    case "SessionStart":
+      return one({ kind: "session_start", repoId: basename(cwd) || "repo" });
+    case "SessionEnd":
+      return one({ kind: "session_end", reason: endReason(str2(p.reason)), claimedSuccess: false });
+    case "Stop":
+      return one({ kind: "turn_end", claimedSuccess: claimsSuccess(str2(p.last_assistant_message)), ...noteOf(str2(p.last_assistant_message)) });
+    case "SubagentStart": {
+      const id = str2(p.agent_id) ?? "";
+      const label = str2(p.agent_type);
+      return { events: [mk({ kind: "subagent_start", childId: id, ...label ? { label } : {} })], warnings: [] };
+    }
+    case "SubagentStop":
+      return one({ kind: "subagent_stop", childId: str2(p.agent_id) ?? "", ok: true });
+    case "Notification": {
+      const t = str2(p.notification_type);
+      if (t === "permission_prompt") return one({ kind: "waiting_for_user", reason: "permission" });
+      if (t === "idle_prompt") return one({ kind: "waiting_for_user", reason: "idle" });
+      if (t === "elicitation_dialog") return one({ kind: "waiting_for_user", reason: "question" });
+      return { events: [], warnings: [] };
+    }
+    case "PostToolUseFailure":
+      return one({ kind: "error", where: str2(p.tool_name) ?? "tool", retrying: false });
+    case "PostToolUse": {
+      const tool = str2(p.tool_name);
+      if (!tool) return { events: [], warnings: [{ code: "missing_field", detail: "tool_name" }] };
+      const dur = typeof p.duration_ms === "number" ? { durationMs: p.duration_ms } : {};
+      const out = toolCallToOutcome({ name: tool, input: p.tool_input, response: p.tool_response, ...dur }, ctx);
+      const baseId = str2(p.tool_use_id) ?? mintId(ts);
+      const events = out.bodies.map((body, i) => mk(body, i === 0 ? baseId : `${baseId}:${i}`));
+      return { events, warnings: out.warning ? [out.warning] : [] };
+    }
+    case "PreToolUse":
+    case "UserPromptSubmit":
+    case "PreCompact":
+    case "PostCompact":
+      return { events: [], warnings: [] };
+  }
+  return { events: [], warnings: [{ code: "unknown_row_type", detail: name }] };
+}
+function noteOf(text) {
+  if (!text) return {};
+  const para = text.trim().split(/\n{2,}/).filter((x) => x.trim()).pop()?.replace(/\s+/g, " ").trim() ?? "";
+  if (!para) return {};
+  return { note: para.length > 240 ? "\u2026" + para.slice(-239) : para };
+}
+var ROW_TYPES = /* @__PURE__ */ new Set(["user", "assistant", "system"]);
+var obj3 = (x) => x && typeof x === "object" ? x : {};
+var str3 = (x) => typeof x === "string" ? x : void 0;
 function parseTranscript(text, opts = {}) {
   const rows = [];
   const warnings = [];
@@ -338,21 +410,21 @@ function parseTranscriptRows(rows, opts = {}) {
     events.push({ id, seq: seq++, ts, sessionId, source: "transcript", ...extra, ...body });
   };
   const tsOf = (row) => {
-    const t = str2(row.timestamp);
+    const t = str3(row.timestamp);
     const n = t ? Date.parse(t) : Number.NaN;
     return Number.isFinite(n) ? n : Number.isFinite(lastTs) ? lastTs : 0;
   };
-  const agentOf = (row) => str2(row.agentId) ?? (row.isSidechain === true ? opts.sidechainAgentId ?? "sidechain" : void 0);
+  const agentOf = (row) => str3(row.agentId) ?? (row.isSidechain === true ? opts.sidechainAgentId ?? "sidechain" : void 0);
   for (const raw of rows) {
-    const row = obj2(raw);
-    const type = str2(row.type) ?? "";
+    const row = obj3(raw);
+    const type = str3(row.type) ?? "";
     if (!ROW_TYPES.has(type)) {
       if (type) unknownTypes.add(type);
       continue;
     }
-    if (!sessionId && str2(row.sessionId)) sessionId = str2(row.sessionId);
-    if (!cwd && str2(row.cwd)) cwd = str2(row.cwd);
-    if (!branch && str2(row.gitBranch)) branch = str2(row.gitBranch);
+    if (!sessionId && str3(row.sessionId)) sessionId = str3(row.sessionId);
+    if (!cwd && str3(row.cwd)) cwd = str3(row.cwd);
+    if (!branch && str3(row.gitBranch)) branch = str3(row.gitBranch);
     const ts = tsOf(row);
     if (!Number.isFinite(firstTs)) {
       firstTs = ts;
@@ -360,39 +432,39 @@ function parseTranscriptRows(rows, opts = {}) {
     }
     lastTs = ts;
     const agentId = agentOf(row);
-    const message = obj2(row.message);
+    const message = obj3(row.message);
     const content = message.content;
     if (type === "system") {
       if (row.subtype === "api_error") {
         const attempt = typeof row.retryAttempt === "number" ? row.retryAttempt : 0;
         const max = typeof row.maxRetries === "number" ? row.maxRetries : 0;
-        push(ts, str2(row.uuid) ?? `sys:${seq}`, { kind: "error", where: "api", retrying: attempt < max }, agentId ? { agentId } : {});
+        push(ts, str3(row.uuid) ?? `sys:${seq}`, { kind: "error", where: "api", retrying: attempt < max }, agentId ? { agentId } : {});
       }
       continue;
     }
     if (type === "assistant") {
       const blocks2 = Array.isArray(content) ? content : [];
       for (const b of blocks2) {
-        const block = obj2(b);
-        if (block.type === "tool_use" && str2(block.id) && str2(block.name)) {
-          pending.set(str2(block.id), { name: str2(block.name), input: block.input, ts, ...agentId ? { agentId } : {} });
+        const block = obj3(b);
+        if (block.type === "tool_use" && str3(block.id) && str3(block.name)) {
+          pending.set(str3(block.id), { name: str3(block.name), input: block.input, ts, ...agentId ? { agentId } : {} });
         } else if (block.type === "text" && !agentId) {
-          const t = str2(block.text);
+          const t = str3(block.text);
           if (t) lastAssistantText = t;
         }
       }
       if (typeof content === "string" && !agentId) lastAssistantText = content;
       if (message.stop_reason === "end_turn" && !agentId) {
-        push(ts, `${str2(row.uuid) ?? seq}:turn`, { kind: "turn_end", claimedSuccess: claimsSuccess(lastAssistantText) });
+        push(ts, `${str3(row.uuid) ?? seq}:turn`, { kind: "turn_end", claimedSuccess: claimsSuccess(lastAssistantText), ...noteOf(lastAssistantText) });
       }
       continue;
     }
-    if (row.toolDenialKind !== void 0) push(ts, `${str2(row.uuid) ?? seq}:deny`, { kind: "waiting_for_user", reason: "permission" });
+    if (row.toolDenialKind !== void 0) push(ts, `${str3(row.uuid) ?? seq}:deny`, { kind: "waiting_for_user", reason: "permission" });
     const blocks = Array.isArray(content) ? content : [];
     for (const b of blocks) {
-      const block = obj2(b);
+      const block = obj3(b);
       if (block.type !== "tool_result") continue;
-      const useId = str2(block.tool_use_id);
+      const useId = str3(block.tool_use_id);
       if (!useId) continue;
       const p = pending.get(useId);
       if (!p) {
@@ -413,7 +485,7 @@ function parseTranscriptRows(rows, opts = {}) {
       if (out.pendingSubagent) openSubagents.set(out.pendingSubagent.childId, useId);
     }
     if (blocks.length === 0 && typeof content === "string" && /\[Request interrupted by user/i.test(content)) {
-      push(ts, `${str2(row.uuid) ?? seq}:interrupt`, { kind: "waiting_for_user", reason: "idle" });
+      push(ts, `${str3(row.uuid) ?? seq}:interrupt`, { kind: "waiting_for_user", reason: "idle" });
     }
   }
   for (const [useId, p] of pending) {
@@ -432,72 +504,6 @@ function parseTranscriptRows(rows, opts = {}) {
   }
   return { events, warnings };
 }
-var obj3 = (x) => x && typeof x === "object" ? x : {};
-var str3 = (x) => typeof x === "string" ? x : void 0;
-var TOOL_EVENTS = /* @__PURE__ */ new Set(["PreToolUse", "PostToolUse", "PostToolUseFailure"]);
-function endReason(reason) {
-  switch (reason) {
-    case "clear":
-    case "logout":
-    case "prompt_input_exit":
-      return "user";
-    default:
-      return "stop";
-  }
-}
-function hookPayloadToEvents(payload, opts = {}) {
-  const p = obj3(payload);
-  const name = str3(p.hook_event_name);
-  const sessionId = str3(p.session_id) ?? "";
-  const ts = opts.now ?? Date.now();
-  if (!name) return { events: [], warnings: [{ code: "missing_field", detail: "hook_event_name" }] };
-  const cwd = str3(p.cwd) ?? "";
-  const ctx = { cwd, ...opts.home ? { home: opts.home } : {} };
-  const agentId = TOOL_EVENTS.has(name) ? str3(p.agent_id) : void 0;
-  const extra = agentId ? { agentId } : {};
-  let seq = 0;
-  const mk = (body, id = mintId(ts)) => ({ id, seq: seq++, ts, sessionId, source: "hook", ...extra, ...body });
-  const one = (body) => ({ events: [mk(body)], warnings: [] });
-  switch (name) {
-    case "SessionStart":
-      return one({ kind: "session_start", repoId: basename(cwd) || "repo" });
-    case "SessionEnd":
-      return one({ kind: "session_end", reason: endReason(str3(p.reason)), claimedSuccess: false });
-    case "Stop":
-      return one({ kind: "turn_end", claimedSuccess: claimsSuccess(str3(p.last_assistant_message)) });
-    case "SubagentStart": {
-      const id = str3(p.agent_id) ?? "";
-      const label = str3(p.agent_type);
-      return { events: [mk({ kind: "subagent_start", childId: id, ...label ? { label } : {} })], warnings: [] };
-    }
-    case "SubagentStop":
-      return one({ kind: "subagent_stop", childId: str3(p.agent_id) ?? "", ok: true });
-    case "Notification": {
-      const t = str3(p.notification_type);
-      if (t === "permission_prompt") return one({ kind: "waiting_for_user", reason: "permission" });
-      if (t === "idle_prompt") return one({ kind: "waiting_for_user", reason: "idle" });
-      if (t === "elicitation_dialog") return one({ kind: "waiting_for_user", reason: "question" });
-      return { events: [], warnings: [] };
-    }
-    case "PostToolUseFailure":
-      return one({ kind: "error", where: str3(p.tool_name) ?? "tool", retrying: false });
-    case "PostToolUse": {
-      const tool = str3(p.tool_name);
-      if (!tool) return { events: [], warnings: [{ code: "missing_field", detail: "tool_name" }] };
-      const dur = typeof p.duration_ms === "number" ? { durationMs: p.duration_ms } : {};
-      const out = toolCallToOutcome({ name: tool, input: p.tool_input, response: p.tool_response, ...dur }, ctx);
-      const baseId = str3(p.tool_use_id) ?? mintId(ts);
-      const events = out.bodies.map((body, i) => mk(body, i === 0 ? baseId : `${baseId}:${i}`));
-      return { events, warnings: out.warning ? [out.warning] : [] };
-    }
-    case "PreToolUse":
-    case "UserPromptSubmit":
-    case "PreCompact":
-    case "PostCompact":
-      return { events: [], warnings: [] };
-  }
-  return { events: [], warnings: [{ code: "unknown_row_type", detail: name }] };
-}
 function hashString(s) {
   let h = 2166136261;
   for (let i = 0; i < s.length; i++) {
@@ -1039,7 +1045,7 @@ var Director = class {
           this.emit(e, actor, { kind: "unverified_claim", gremlinId: gid }, 1400, this.mood.onClaimUnverified());
         }
         this.editedSinceTurn = false;
-        this.emit(e, actor, { kind: "your_turn" }, 1e3, this.mood.mood);
+        this.emit(e, actor, { kind: "your_turn", ...e.note ? { note: e.note } : {} }, 1e3, this.mood.mood);
         break;
       }
       case "session_end":
@@ -1309,7 +1315,26 @@ var Theatre = class {
     this.port = typeof addr === "object" && addr ? addr.port : port;
     const home = this.opts.home ?? join2(homedir2(), ".agent-theatre");
     mkdirSync(home, { recursive: true });
+    try {
+      const old = parseInt(readFileSync2(join2(home, "port"), "utf8").trim(), 10);
+      if (old > 0 && old !== this.port && portAlive(old) && !existsSync(join2(home, "ports", String(old)))) {
+        mkdirSync(join2(home, "ports"), { recursive: true });
+        writeFileSync(join2(home, "ports", String(old)), "(started by an older build)");
+      }
+    } catch {
+    }
     writeFileSync(join2(home, "port"), String(this.port));
+    mkdirSync(join2(home, "ports"), { recursive: true });
+    for (const f of readdirSync2(join2(home, "ports"))) {
+      const n = parseInt(f, 10);
+      if (n > 0 && n !== this.port && !portAlive(n)) {
+        try {
+          rmSync(join2(home, "ports", f));
+        } catch {
+        }
+      }
+    }
+    writeFileSync(join2(home, "ports", String(this.port)), this.opts.cwd);
     const lostAfter = this.opts.lostAfterMs ?? 9e4;
     this.timer = setInterval(() => {
       const l = this.live;
@@ -1328,6 +1353,10 @@ var Theatre = class {
       if (readFileSync2(join2(home, "port"), "utf8").trim() === String(this.port)) rmSync(join2(home, "port"));
     } catch {
     }
+    try {
+      rmSync(join2(home, "ports", String(this.port)));
+    } catch {
+    }
     for (const ws of this.wss?.clients ?? []) ws.close();
     this.wss?.close();
     await new Promise((res) => this.server ? this.server.close(() => res()) : res());
@@ -1371,6 +1400,12 @@ function claudeBinary() {
   }
   return null;
 }
+function portAlive(port) {
+  const r = spawnSync("lsof", ["-nP", `-iTCP:${port}`, "-sTCP:LISTEN", "-t"], { encoding: "utf8", timeout: 2e3 });
+  if (r.status === 0 && r.stdout.trim()) return true;
+  if (r.error) return true;
+  return false;
+}
 
 // src/scan.ts
 import { readdirSync as readdirSync3, readFileSync as readFileSync3, statSync as statSync2 } from "fs";
@@ -1505,20 +1540,22 @@ function hasOurHooks(existing) {
 // src/hooks/script.ts
 var HOOK_SCRIPT = `#!/usr/bin/env node
 // Agent Theatre hook \u2014 forwards Claude Code events to the local theatre. Never blocks the agent.
-import { readFileSync } from "node:fs";
+import { readFileSync, readdirSync } from "node:fs";
 import { homedir } from "node:os";
 import { join } from "node:path";
-const portFile = join(process.env.AGENT_THEATRE_HOME || join(homedir(), ".agent-theatre"), "port");
-let port = 0;
-try { port = parseInt(readFileSync(portFile, "utf8").trim(), 10); } catch { process.exit(0); }
-if (!port) process.exit(0);
+// Every running stage registers itself in ports/<port>; the event goes to all of them (browser, VS Code, a second repo\u2026).
+const home = process.env.AGENT_THEATRE_HOME || join(homedir(), ".agent-theatre");
+const ports = new Set();
+try { for (const f of readdirSync(join(home, "ports"))) { const n = parseInt(f, 10); if (n > 0) ports.add(n); } } catch { /* no registry yet */ }
+try { const n = parseInt(readFileSync(join(home, "port"), "utf8").trim(), 10); if (n > 0) ports.add(n); } catch { /* no port file */ }
+if (!ports.size) process.exit(0);
 let body = "";
 process.stdin.setEncoding("utf8");
 process.stdin.on("data", (c) => { body += c; if (body.length > 2_000_000) body = body.slice(0, 2_000_000); });
 process.stdin.on("end", async () => {
   try {
-    await fetch(\`http://127.0.0.1:\${port}/hook\`, { method: "POST", headers: { "content-type": "application/json" }, body, signal: AbortSignal.timeout(300) });
-  } catch { /* theatre not running or slow: that is fine */ }
+    await Promise.allSettled([...ports].map((port) => fetch(\`http://127.0.0.1:\${port}/hook\`, { method: "POST", headers: { "content-type": "application/json" }, body, signal: AbortSignal.timeout(300) })));
+  } catch { /* a theatre not running or slow: that is fine */ }
   process.exit(0);
 });
 setTimeout(() => process.exit(0), 600).unref();