@thurstonsand/pi-agent-switchboard 0.0.0 → 0.2.0

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";
@@ -97,18 +101,49 @@ function branch(cwd) {
 }
 
 // packages/pi/src/index.ts
+import { Type } from "typebox";
 var KEY = Symbol.for("agent-switchboard.recorder");
 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;
@@ -126,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 });
@@ -191,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)
@@ -201,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)
@@ -230,6 +263,16 @@ function src_default(pi) {
   pi.on("session_info_changed", (event, ctx) => {
     writeFor(ctx, (db, id) => db.run("UPDATE sessions SET title = ? WHERE session_id = ?", event.name ?? null, id));
   });
+  const markArchived = (ctx) => {
+    let archived = false;
+    writeFor(ctx, (db, id) => {
+      if (!db.get("SELECT 1 FROM sessions WHERE session_id = ?", id))
+        return;
+      db.run("INSERT INTO marks (session_id, archived_at) VALUES (?, ?) ON CONFLICT (session_id) DO UPDATE SET archived_at = excluded.archived_at", id, Date.now());
+      archived = true;
+    });
+    return archived;
+  };
   pi.registerCommand("archive", {
     description: "Archive this session in swb and quit",
     handler: async (_args, ctx) => {
@@ -237,19 +280,33 @@ function src_default(pi) {
         ctx.ui.notify("swb: turn running: wait for it to complete", "error");
         return;
       }
-      let archived = false;
-      writeFor(ctx, (db, id) => {
-        if (!db.get("SELECT 1 FROM sessions WHERE session_id = ?", id))
-          return;
-        db.run("INSERT INTO marks (session_id, archived_at) VALUES (?, ?) ON CONFLICT (session_id) DO UPDATE SET archived_at = excluded.archived_at", id, Date.now());
-        archived = true;
-      });
-      if (archived)
+      if (markArchived(ctx))
         ctx.shutdown();
       else if (!proc.failure)
         ctx.ui.notify("swb: nothing to archive before the first prompt", "error");
     }
   });
+  pi.registerTool({
+    name: "swb_archive",
+    label: "Archive session",
+    description: "Archive this session in Agent Switchboard. It leaves the user's roster of open sessions, and pi quits as soon as this turn ends. The user can restore it with `swb unarchive <id>`.",
+    promptSnippet: "Archive this session when the user asks",
+    promptGuidelines: [
+      "Use swb_archive only when the user asks to archive this session; never archive on your own initiative.",
+      "Call swb_archive as the last action of the turn, then end with a brief final message: pi quits once the turn ends."
+    ],
+    parameters: Type.Object({}),
+    async execute(_toolCallId, _params, _signal, _onUpdate, ctx) {
+      if (!markArchived(ctx)) {
+        throw new Error(proc.failure ? `swb: not recording: ${proc.failure}` : "swb: no session recorded yet");
+      }
+      ctx.shutdown();
+      return {
+        content: [{ type: "text", text: "Archived. pi quits when this turn ends." }],
+        details: undefined
+      };
+    }
+  });
   pi.on("session_shutdown", (event, ctx) => {
     if (event.reason !== "quit")
       return;

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@thurstonsand/pi-agent-switchboard",
-	"version": "0.0.0",
+	"version": "0.2.0",
 	"type": "module",
 	"description": "Records managed pi sessions for Agent Switchboard (swb)",
 	"license": "MIT",
@@ -17,6 +17,7 @@
 		]
 	},
 	"peerDependencies": {
-		"@earendil-works/pi-coding-agent": "*"
+		"@earendil-works/pi-coding-agent": "*",
+		"typebox": "*"
 	}
 }