agent-theatre 1.1.4 → 1.2.1

package/dist/cli.js

@@ -9,9 +9,10 @@ import pc from "picocolors";
 
 // src/server.ts
 import { createServer } from "http";
-import { existsSync, mkdirSync, readFileSync as readFileSync2, rmSync, writeFileSync, realpathSync } from "fs";
+import { existsSync, mkdirSync, readFileSync as readFileSync2, readdirSync as readdirSync2, rmSync, writeFileSync, realpathSync } from "fs";
 import { homedir as homedir2 } from "os";
 import { join as join2, basename as basename2, resolve, sep } from "path";
+import { spawn, spawnSync } from "child_process";
 import { Hono } from "hono";
 import { getRequestListener } from "@hono/node-server";
 import { WebSocketServer } from "ws";
@@ -1195,6 +1196,8 @@ var Theatre = class {
   wss = null;
   live;
   signal = "ok";
+  /** The most recent Claude Code session that spoke to this stage; what a prompt from the stage resumes. */
+  lastSessionId = null;
   timer = null;
   port = 0;
   routes() {
@@ -1208,6 +1211,8 @@ var Theatre = class {
       }
       const cwd = typeof payload?.cwd === "string" ? payload.cwd : "";
       if (cwd && !isWithin(cwd, this.opts.cwd)) return c.body(null, 204);
+      const sid = payload.session_id;
+      if (typeof sid === "string" && sid) this.lastSessionId = sid;
       const { events } = hookPayloadToEvents(payload, { home: homedir2() });
       if (events.length) this.ingest(events);
       return c.body(null, 204);
@@ -1232,6 +1237,18 @@ var Theatre = class {
       const beats = direct(events, world);
       return c.json({ mode: "replay", repoId: this.opts.repoId, title: s.title, world, events, beats, warnings: warnings.length });
     });
+    a.get("/api/prompt", (c) => c.json({ available: !!claudeBinary() && !!this.lastSessionId, session: this.lastSessionId }));
+    a.post("/api/prompt", async (c) => {
+      const body = await c.req.json().catch(() => ({}));
+      const text = typeof body.text === "string" ? body.text.trim() : "";
+      const bin = claudeBinary();
+      if (!text) return c.json({ ok: false, error: "empty prompt" }, 400);
+      if (!bin) return c.json({ ok: false, error: "Claude Code 2.x not found on PATH" }, 503);
+      if (!this.lastSessionId) return c.json({ ok: false, error: "no session to resume yet" }, 409);
+      const child = spawn(bin, ["-p", "--resume", this.lastSessionId, text], { cwd: this.opts.cwd, stdio: "ignore", detached: true, env: process.env });
+      child.unref();
+      return c.json({ ok: true, session: this.lastSessionId });
+    });
     a.get("/healthz", (c) => c.json({ ok: true, port: this.port, events: this.live.events.length }));
     if (this.opts.webDir) {
       const dir = this.opts.webDir;
@@ -1264,7 +1281,15 @@ var Theatre = class {
     this.broadcast({ type: "beats", beats: out, events });
     return out;
   }
+  /** In-process subscribers (the VS Code extension relays these to its webview). */
+  listeners = /* @__PURE__ */ new Set();
   broadcast(msg) {
+    for (const l of this.listeners) {
+      try {
+        l(msg);
+      } catch {
+      }
+    }
     const data = JSON.stringify(msg);
     for (const ws of this.wss?.clients ?? []) if (ws.readyState === 1) ws.send(data);
   }
@@ -1307,6 +1332,10 @@ var Theatre = class {
     this.wss?.close();
     await new Promise((res) => this.server ? this.server.close(() => res()) : res());
   }
+  /** The live scene as /api/scene returns it. */
+  scene() {
+    return { mode: "live", repoId: this.opts.repoId, world: this.live.world, events: this.live.events, beats: this.live.beats, signal: this.signal };
+  }
   get url() {
     return `http://127.0.0.1:${this.port}/`;
   }
@@ -1325,9 +1354,26 @@ function isWithin(dir, root) {
   const d = norm3(dir), r = norm3(root);
   return d === r || d.startsWith(r.endsWith(sep) ? r : r + sep);
 }
+function claudeBinary() {
+  const cands = [process.env.AGENT_THEATRE_CLAUDE, "claude"].filter((x) => !!x);
+  try {
+    const ext = join2(homedir2(), ".vscode", "extensions");
+    for (const d of (existsSync(ext) ? readdirSync2(ext) : []).filter((n) => n.startsWith("anthropic.claude-code-")).sort().reverse()) cands.push(join2(ext, d, "resources", "native-binary", "claude"));
+  } catch {
+  }
+  for (const c of cands) {
+    try {
+      const r = spawnSync(c, ["--version"], { encoding: "utf8", timeout: 4e3 });
+      const m = /^(\d+)\./.exec(r.stdout.trim());
+      if (r.status === 0 && m && Number(m[1]) >= 2) return c;
+    } catch {
+    }
+  }
+  return null;
+}
 
 // src/scan.ts
-import { readdirSync as readdirSync2, readFileSync as readFileSync3, statSync as statSync2 } from "fs";
+import { readdirSync as readdirSync3, readFileSync as readFileSync3, statSync as statSync2 } from "fs";
 import { join as join3, relative } from "path";
 function scanRepo(root, opts = {}) {
   const maxFiles = opts.maxFiles ?? 2e4, maxBytes = opts.maxBytesForLoc ?? 512e3;
@@ -1336,7 +1382,7 @@ function scanRepo(root, opts = {}) {
     if (files.length >= maxFiles) return;
     let ents;
     try {
-      ents = readdirSync2(dir, { withFileTypes: true });
+      ents = readdirSync3(dir, { withFileTypes: true });
     } catch {
       return;
     }