@staff0rd/assist 0.394.0 → 0.396.0

package/dist/index.js

@@ -6,7 +6,7 @@ import { Command } from "commander";
 // package.json
 var package_default = {
   name: "@staff0rd/assist",
-  version: "0.394.0",
+  version: "0.396.0",
   type: "module",
   main: "dist/index.js",
   bin: {
@@ -6437,6 +6437,7 @@ import { dirname as dirname16, join as join20 } from "path";
 import { z as z4 } from "zod";
 var activitySchema = z4.object({
   kind: z4.enum(["command", "backlog"]),
+  harness: z4.enum(["claude", "codex", "pi"]).optional(),
   name: z4.string().optional(),
   itemId: z4.number().optional(),
   itemName: z4.string().optional(),
@@ -8466,11 +8467,13 @@ async function listUsagePeaks(db, paging) {
     resetsAt: phaseCycleContext.resetsAt,
     avgContextPct: sql11`cast(avg(${phaseCycleContext.peakContextPct}) as double precision)`.as(
       "avg_context_pct"
-    )
+    ),
+    phaseCount: count().as("phase_count")
   }).from(phaseCycleContext).groupBy(phaseCycleContext.window, phaseCycleContext.resetsAt).as("avg_context");
   const query = db.select({
     ...getTableColumns(usagePeaks),
-    avgContextPct: avgContext.avgContextPct
+    avgContextPct: avgContext.avgContextPct,
+    phaseCount: avgContext.phaseCount
   }).from(usagePeaks).leftJoin(
     avgContext,
     and7(
@@ -23666,9 +23669,6 @@ function reportStrays(pids) {
 // src/commands/sessions/daemon/drainDaemon.ts
 import { createInterface as createInterface6 } from "readline";
 
-// src/commands/sessions/daemon/loadPersistedSessions.ts
-import { z as z5 } from "zod";
-
 // src/commands/sessions/daemon/daemonLog.ts
 var RING_CAPACITY = 1e3;
 var ring = [];
@@ -23692,13 +23692,14 @@ function recentDaemonLogLines() {
   return [...ring];
 }
 
-// src/commands/sessions/daemon/loadPersistedSessions.ts
-var SESSIONS_FILE = "sessions.json";
+// src/commands/sessions/daemon/persistedSessionSchema.ts
+import { z as z5 } from "zod";
 var persistedSessionSchema = z5.object({
   name: z5.string(),
   title: z5.string().optional(),
   subtitle: z5.string().optional(),
   commandType: z5.enum(["claude", "run", "assist"]),
+  harness: z5.enum(["claude", "codex", "pi"]).optional(),
   status: z5.enum(["running", "waiting", "done", "error"]).optional(),
   cwd: z5.string(),
   startedAt: z5.number(),
@@ -23711,6 +23712,34 @@ var persistedSessionSchema = z5.object({
   activity: activitySchema.optional(),
   starred: z5.boolean().optional()
 });
+
+// src/commands/sessions/daemon/toPersistedSession.ts
+function toPersistedSession(session) {
+  return {
+    name: session.name,
+    title: session.title,
+    subtitle: session.subtitle,
+    commandType: session.commandType,
+    harness: session.harness,
+    status: session.status,
+    cwd: session.cwd ?? process.cwd(),
+    startedAt: session.startedAt,
+    runningMs: accumulatedRunningMs(session),
+    claudeSessionId: session.claudeSessionId,
+    initialPrompt: session.initialPrompt,
+    runName: session.runName,
+    runArgs: session.runArgs,
+    assistArgs: session.assistArgs,
+    activity: session.activity,
+    starred: session.starred
+  };
+}
+function accumulatedRunningMs(session) {
+  return session.runningSince != null ? session.runningMs + Date.now() - session.runningSince : session.runningMs;
+}
+
+// src/commands/sessions/daemon/loadPersistedSessions.ts
+var SESSIONS_FILE = "sessions.json";
 function loadPersistedSessions() {
   const data = loadJson(SESSIONS_FILE);
   if (!Array.isArray(data)) return [];
@@ -23738,28 +23767,6 @@ function logPersist(live) {
     live.length > 0 ? `persisted ${live.length} session(s): ${live.map((s) => s.name).join(", ")}` : "persisted 0 sessions (sessions.json cleared)"
   );
 }
-function toPersistedSession(session) {
-  return {
-    name: session.name,
-    title: session.title,
-    subtitle: session.subtitle,
-    commandType: session.commandType,
-    status: session.status,
-    cwd: session.cwd ?? process.cwd(),
-    startedAt: session.startedAt,
-    runningMs: accumulatedRunningMs(session),
-    claudeSessionId: session.claudeSessionId,
-    initialPrompt: session.initialPrompt,
-    runName: session.runName,
-    runArgs: session.runArgs,
-    assistArgs: session.assistArgs,
-    activity: session.activity,
-    starred: session.starred
-  };
-}
-function accumulatedRunningMs(session) {
-  return session.runningSince != null ? session.runningMs + Date.now() - session.runningSince : session.runningMs;
-}
 
 // src/commands/sessions/daemon/drainDaemon.ts
 var DRAIN_TIMEOUT_MS = 5e3;
@@ -23870,6 +23877,7 @@ function toSessionInfo({
   title,
   subtitle,
   commandType,
+  harness,
   status: status3,
   startedAt,
   runningMs,
@@ -23893,6 +23901,7 @@ function toSessionInfo({
     title,
     subtitle,
     commandType,
+    harness,
     status: status3,
     startedAt,
     runningMs,
@@ -24100,6 +24109,13 @@ function buildArgs3(opts) {
   return base;
 }
 
+// src/commands/sessions/daemon/spawnPi.ts
+function spawnPi2(opts = {}) {
+  const args = ["pi"];
+  if (opts.prompt) args.push(opts.prompt);
+  return spawnPty(args, opts.cwd, opts.sessionId);
+}
+
 // src/commands/sessions/daemon/spawnRun.ts
 function spawnRun(opts) {
   return spawnPty(["assist", "run", opts.name, ...opts.args ?? []], opts.cwd);
@@ -24119,7 +24135,8 @@ function sessionBase(id, status3) {
     scrollback: ""
   };
 }
-function createSession(id, prompt, cwd, design) {
+function createSession(id, prompt, cwd, design, harness) {
+  if (harness === "pi") return createPiSession(id, prompt, cwd);
   const claudeSessionId = randomUUID4();
   return {
     ...sessionBase(id, prompt ? "running" : "waiting"),
@@ -24132,6 +24149,17 @@ function createSession(id, prompt, cwd, design) {
     design
   };
 }
+function createPiSession(id, prompt, cwd) {
+  return {
+    ...sessionBase(id, prompt ? "running" : "waiting"),
+    name: prompt?.slice(0, 40) || `Session ${id}`,
+    commandType: "claude",
+    harness: "pi",
+    pty: spawnPi2({ prompt, cwd, sessionId: id }),
+    cwd,
+    initialPrompt: prompt
+  };
+}
 function createRunSession(id, runName, runArgs, cwd) {
   return {
     ...sessionBase(id, "running"),
@@ -24796,6 +24824,7 @@ function restoreBase(id, persisted) {
     title: persisted.title,
     subtitle: persisted.subtitle,
     commandType: persisted.commandType,
+    harness: persisted.harness,
     scrollback: "",
     cwd: persisted.cwd,
     assistArgs: persisted.assistArgs,
@@ -25809,8 +25838,10 @@ var SessionManager = class {
     return session.id;
   }
   spawnWith = (create) => this.add(create(sessionLimits.nextId(this.sessions.size, this.idCounter)));
-  spawn(prompt, cwd, design) {
-    return this.spawnWith((id) => createSession(id, prompt, cwd, design));
+  spawn(prompt, cwd, design, harness) {
+    return this.spawnWith(
+      (id) => createSession(id, prompt, cwd, design, harness)
+    );
   }
   spawnRun(runName, runArgs, cwd) {
     return this.spawnWith((id) => createRunSession(id, runName, runArgs, cwd));
@@ -26009,12 +26040,16 @@ var lifecycleHandlers = {
 // src/commands/sessions/daemon/spawnCreate.ts
 function spawnCreate(m, d) {
   const design = d.design === true;
+  const harness = d.harness;
   if (design)
     daemonLog(`create: design session (cwd=${d.cwd ?? ""})`);
+  if (harness && harness !== "claude")
+    daemonLog(`create: ${harness} session (cwd=${d.cwd ?? ""})`);
   return m.spawn(
     d.prompt,
     d.cwd,
-    design
+    design,
+    harness
   );
 }
 
@@ -26850,16 +26885,21 @@ import * as path63 from "path";
 // src/commands/sync/syncPiHooks.ts
 import * as fs43 from "fs";
 import * as path62 from "path";
-var PI_PERMISSION_GATE_FILE = "assist-permission-gate.ts";
-function piPermissionGateTarget() {
-  return path62.join(harnesses.pi.homeDir, "extensions", PI_PERMISSION_GATE_FILE);
-}
-function syncPiHooks(sourcePath) {
-  const target = piPermissionGateTarget();
-  fs43.mkdirSync(path62.dirname(target), { recursive: true });
-  fs43.copyFileSync(sourcePath, target);
+function piExtensionsDir() {
+  return path62.join(harnesses.pi.homeDir, "extensions");
+}
+function syncPiHooks(sourceDir) {
+  const target = piExtensionsDir();
+  fs43.mkdirSync(target, { recursive: true });
+  const files = fs43.readdirSync(sourceDir).filter((f) => f.endsWith(".ts"));
+  for (const file of files) {
+    fs43.copyFileSync(
+      path62.join(sourceDir, file),
+      path62.join(target, `assist-${file}`)
+    );
+  }
   console.log(
-    "Registered assist permission-gate extension in ~/.pi/agent/extensions"
+    `Registered ${files.length} assist extension(s) in ~/.pi/agent/extensions`
   );
 }
 
@@ -26892,7 +26932,7 @@ function syncPi(claudeDir) {
     harnesses.pi,
     commandToPrompt
   );
-  syncPiHooks(path63.join(claudeDir, "..", "pi", "permission-gate.ts"));
+  syncPiHooks(path63.join(claudeDir, "..", "pi"));
   console.log(
     `Synced ${synced} prompt(s) to ~/.pi/agent/prompts and CLAUDE.md to ~/.pi/agent/AGENTS.md`
   );

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@staff0rd/assist",
-	"version": "0.394.0",
+	"version": "0.396.0",
 	"type": "module",
 	"main": "dist/index.js",
 	"bin": {

package/pi/status-driver.ts

@@ -0,0 +1,22 @@
+import { spawn } from "node:child_process";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+
+function setStatus(status: "running" | "waiting"): void {
+	try {
+		const child = spawn("assist", ["sessions", "set-status", status], {
+			stdio: "ignore",
+			detached: true,
+		});
+		child.on("error", () => {});
+		child.unref();
+	} catch {}
+}
+
+export default function (pi: ExtensionAPI) {
+	pi.on("agent_start", () => {
+		setStatus("running");
+	});
+	pi.on("agent_settled", () => {
+		setStatus("waiting");
+	});
+}