@aliou/pi-processes 0.8.0 → 0.9.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aliou/pi-processes",
-  "version": "0.8.0",
+  "version": "0.9.0",
   "license": "MIT",
   "type": "module",
   "private": false,
@@ -36,18 +36,18 @@
     "@aliou/pi-utils-settings": "^0.10.0",
     "@aliou/pi-utils-ui": "^0.1.0",
     "@aliou/sh": "^0.1.0",
-    "@sinclair/typebox": "^0.34.41"
+    "typebox": "^1.0.0"
   },
   "peerDependencies": {
-    "@mariozechner/pi-coding-agent": "0.65.0",
-    "@mariozechner/pi-tui": "0.65.0"
+    "@mariozechner/pi-coding-agent": "0.72.1",
+    "@mariozechner/pi-tui": "0.72.1"
   },
   "devDependencies": {
     "@aliou/biome-plugins": "^0.3.2",
     "@biomejs/biome": "^2.3.13",
     "@changesets/cli": "^2.27.11",
-    "@mariozechner/pi-ai": "0.65.0",
-    "@mariozechner/pi-coding-agent": "0.65.0",
+    "@mariozechner/pi-ai": "0.72.1",
+    "@mariozechner/pi-coding-agent": "0.72.1",
     "@types/node": "^25.0.10",
     "husky": "^9.1.7",
     "typescript": "^5.9.3",

package/src/hooks/process-end.ts

@@ -2,6 +2,7 @@ import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 import { MESSAGE_TYPE_PROCESS_UPDATE, type ProcessInfo } from "../constants";
 import type { ProcessManager } from "../manager";
 import { formatRuntime } from "../utils";
+import { safeSendMessage } from "./utils";
 
 interface ProcessUpdateDetails {
   kind: "lifecycle";
@@ -54,7 +55,8 @@ export function setupProcessEndHook(pi: ExtensionAPI, manager: ProcessManager) {
       runtime,
     };
 
-    pi.sendMessage(
+    safeSendMessage(
+      pi,
       {
         customType: MESSAGE_TYPE_PROCESS_UPDATE,
         content: message,

package/src/hooks/process-watch.ts

@@ -1,6 +1,7 @@
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 import { MESSAGE_TYPE_PROCESS_UPDATE } from "../constants";
 import type { ProcessManager } from "../manager";
+import { safeSendMessage } from "./utils";
 
 interface ProcessWatchUpdateDetails {
   kind: "watch_matched";
@@ -70,7 +71,8 @@ export function setupProcessWatchHook(
       }
     }
 
-    pi.sendMessage(
+    safeSendMessage(
+      pi,
       {
         customType: MESSAGE_TYPE_PROCESS_UPDATE,
         content: message,

package/src/hooks/utils.ts

@@ -0,0 +1,23 @@
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+
+/**
+ * Send a message via pi, swallowing stale-context errors.
+ *
+ * After /new, /resume, or /fork, the pi proxy is invalidated and all
+ * method calls throw. This wrapper catches that specific error so the
+ * extension doesn't crash — the event is simply lost for the old session.
+ */
+export function safeSendMessage(
+  pi: ExtensionAPI,
+  message: Parameters<ExtensionAPI["sendMessage"]>[0],
+  options?: Parameters<ExtensionAPI["sendMessage"]>[1],
+): void {
+  try {
+    pi.sendMessage(message, options);
+  } catch (err: unknown) {
+    if (err instanceof Error && err.message.includes("stale")) {
+      return;
+    }
+    throw err;
+  }
+}

package/src/hooks/widget/setup.ts

@@ -19,7 +19,7 @@ export function setupProcessWidget(
   manager: ProcessManager,
   config: ResolvedProcessesConfig,
 ) {
-  let latestContext: ExtensionContext | null = null;
+  let activeCtx: ExtensionContext | null = null;
   let logDockComponent: LogDockComponent | null = null;
   let logDockComponentTui: { requestRender(): void } | null = null;
 
@@ -30,30 +30,26 @@ export function setupProcessWidget(
   };
 
   function updateWidget() {
-    if (!latestContext?.hasUI) return;
+    if (!activeCtx?.hasUI) return;
 
     if (!configLoader.getConfig().widget.showStatusWidget) {
-      latestContext.ui.setWidget(STATUS_WIDGET_ID, undefined);
+      activeCtx.ui.setWidget(STATUS_WIDGET_ID, undefined);
     } else {
       const processes = manager.list();
       const maxWidth = process.stdout.columns || 120;
-      const lines = renderStatusWidget(
-        processes,
-        latestContext.ui.theme,
-        maxWidth,
-      );
+      const lines = renderStatusWidget(processes, activeCtx.ui.theme, maxWidth);
 
       if (lines.length === 0) {
-        latestContext.ui.setWidget(STATUS_WIDGET_ID, undefined);
+        activeCtx.ui.setWidget(STATUS_WIDGET_ID, undefined);
       } else {
-        latestContext.ui.setWidget(STATUS_WIDGET_ID, lines, {
+        activeCtx.ui.setWidget(STATUS_WIDGET_ID, lines, {
           placement: "belowEditor",
         });
       }
     }
 
     if (dockState.visibility === "hidden") {
-      latestContext.ui.setWidget(LOG_DOCK_WIDGET_ID, undefined);
+      activeCtx.ui.setWidget(LOG_DOCK_WIDGET_ID, undefined);
       if (logDockComponent) {
         logDockComponent.dispose();
         logDockComponent = null;
@@ -72,7 +68,7 @@ export function setupProcessWidget(
         dockHeight: height,
       });
     } else {
-      const ctx = latestContext;
+      const ctx = activeCtx;
       ctx.ui.setWidget(
         LOG_DOCK_WIDGET_ID,
         (tui: { requestRender(): void }, theme: typeof ctx.ui.theme) => {
@@ -152,9 +148,20 @@ export function setupProcessWidget(
       logDockComponent = null;
       logDockComponentTui = null;
     }
-    latestContext = ctx;
+    activeCtx = ctx;
     updateWidget();
   });
 
+  pi.on("session_shutdown", async (_event, ctx) => {
+    activeCtx = null;
+    if (logDockComponent) {
+      logDockComponent.dispose();
+      logDockComponent = null;
+      logDockComponentTui = null;
+    }
+    ctx.ui.setWidget(STATUS_WIDGET_ID, undefined);
+    ctx.ui.setWidget(LOG_DOCK_WIDGET_ID, undefined);
+  });
+
   return { update: updateWidget, dockActions };
 }

package/src/tools/index.ts

@@ -7,7 +7,7 @@ import type {
   ToolRenderResultOptions,
 } from "@mariozechner/pi-coding-agent";
 import { Text } from "@mariozechner/pi-tui";
-import { type Static, Type } from "@sinclair/typebox";
+import { type Static, Type } from "typebox";
 import type { ProcessesDetails } from "../constants";
 import type { ProcessManager } from "../manager";
 import { executeAction, renderActionCall, renderActionResult } from "./actions";
@@ -75,16 +75,12 @@ const ProcessesParams = Type.Object({
         "Get a turn to react when process is killed by external signal (default: false). Note: killing via tool never triggers a turn.",
     }),
   ),
-  ...(DEBUG_PREVIEW_ENABLED
-    ? {
-        preview: Type.Optional(
-          StringEnum(["start", "list", "output", "logs", "error"] as const, {
-            description:
-              "For action=debug_preview only: which rendered result variant to preview (default: start)",
-          }),
-        ),
-      }
-    : {}),
+  preview: Type.Optional(
+    StringEnum(["start", "list", "output", "logs", "error"] as const, {
+      description:
+        "For action=debug_preview only: which rendered result variant to preview (default: start)",
+    }),
+  ),
   logWatches: Type.Optional(
     Type.Array(
       Type.Object(