@thurstonsand/pi-agent-switchboard 0.1.0 → 0.2.1

package/index.js

@@ -1,3 +1,7 @@
+// packages/pi/src/index.ts
+import { execFile } from "node:child_process";
+import { promisify } from "node:util";
+
 // packages/shared/src/index.ts
 import { execFileSync } from "node:child_process";
 import { readFileSync } from "node:fs";
@@ -103,13 +107,43 @@ var store = globalThis;
 function claim() {
   const env = process.env;
   if (env.SWB_MANAGED === "1" && env.SWB_DB && env.SWB_TMUX_SESSION) {
-    store[KEY] = { launch: Object.freeze({ db: env.SWB_DB, tmuxSession: env.SWB_TMUX_SESSION }), attention: new Set };
+    store[KEY] = {
+      launch: Object.freeze({ db: env.SWB_DB, tmuxSession: env.SWB_TMUX_SESSION, bin: env.SWB_BIN ?? "" }),
+      attention: new Set,
+      ...env.SWB_BIN ? {} : { failure: "started by an swb older than this recorder; upgrade swb" }
+    };
   }
   delete env.SWB_MANAGED;
   delete env.SWB_DB;
   delete env.SWB_TMUX_SESSION;
+  delete env.SWB_BIN;
   return store[KEY];
 }
+var run = promisify(execFile);
+function host(bin) {
+  const swb = async (...args) => {
+    try {
+      return (await run(bin, args, { timeout: 15000 })).stdout;
+    } catch (error) {
+      const { stderr, message } = error;
+      throw new Error(stderr?.trim() || message);
+    }
+  };
+  return {
+    name: "swb",
+    async launch(input) {
+      await swb("launch", "--cwd", input.cwd, "--session-id", input.sessionId, "--model", input.model);
+      return { success: true };
+    },
+    async listSessions() {
+      const sessions = JSON.parse(await swb("ls", "--json"));
+      return sessions.filter((s) => s.open).map((s) => ({ sessionId: s.id, cwd: s.cwd, title: s.title ?? s.id }));
+    },
+    async wake(sessionId) {
+      await swb("wake", sessionId);
+    }
+  };
+}
 function liveHost(row) {
   if (row && row.boot_id === bootId() && pidAlive(row.pid))
     return row.tmux_session;
@@ -127,6 +161,8 @@ function src_default(pi) {
     return;
   const proc = claimed;
   let ui;
+  if (proc.launch.bin)
+    pi.events.on("pi-sessions:hosts:v1", (request) => request.register(host(proc.launch.bin)));
   const showFailure = () => ui?.setStatus("swb", ui.theme.fg("error", `swb: not recording: ${proc.failure}`));
   const connect = () => {
     proc.db ??= new Db(proc.launch.db, { create: false });
@@ -192,9 +228,7 @@ function src_default(pi) {
       db.run("UPDATE sessions SET phase = 'idle', cwd = ?, project_root = ?, branch = ?, transcript = ? WHERE session_id = ?", ctx.cwd, projectRoot(ctx.cwd), branch(ctx.cwd), transcript, id);
     });
   });
-  pi.on("message_start", (event, ctx) => {
-    if (event.message.role !== "user")
-      return;
+  pi.on("agent_start", (_event, ctx) => {
     writeFor(ctx, (db, id) => {
       const transcript = ctx.sessionManager.getSessionFile();
       if (!transcript)
@@ -202,12 +236,10 @@ function src_default(pi) {
       const now = Date.now();
       db.run(`INSERT INTO sessions (session_id, tool, cwd, project_root, transcript, title, branch, phase, created_at, last_prompt_at)
 				 VALUES (?, 'pi', ?, ?, ?, ?, ?, 'working', ?, ?)
-				 ON CONFLICT (session_id) DO UPDATE SET phase = 'working', last_prompt_at = excluded.last_prompt_at`, id, ctx.cwd, projectRoot(ctx.cwd), transcript, ctx.sessionManager.getSessionName() ?? null, branch(ctx.cwd), now, now);
+				 ON CONFLICT (session_id) DO UPDATE SET phase = 'working',
+				   last_prompt_at = CASE WHEN sessions.phase = 'idle' THEN excluded.last_prompt_at ELSE sessions.last_prompt_at END`, id, ctx.cwd, projectRoot(ctx.cwd), transcript, ctx.sessionManager.getSessionName() ?? null, branch(ctx.cwd), now, now);
     });
   });
-  pi.on("agent_start", (_event, ctx) => {
-    writeFor(ctx, (db, id) => db.run("UPDATE sessions SET phase = 'working' WHERE session_id = ?", id));
-  });
   const attention = (open) => (data) => {
     const attentionId = data.attentionId;
     if (typeof attentionId !== "string" || !current)

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@thurstonsand/pi-agent-switchboard",
-	"version": "0.1.0",
+	"version": "0.2.1",
 	"type": "module",
 	"description": "Records managed pi sessions for Agent Switchboard (swb)",
 	"license": "MIT",