@aliou/pi-processes 0.1.1 → 0.2.1

package/hooks/widget.ts

@@ -2,44 +2,35 @@ import type {
   ExtensionAPI,
   ExtensionContext,
 } from "@mariozechner/pi-coding-agent";
-import type { ProcessInfo, ProcessManager } from "../manager";
+import type { ProcessInfo } from "../constants";
+import type { ProcessManager } from "../manager";
 
 const WIDGET_ID = "processes-status";
 
-function formatRuntime(startTime: number, endTime: number | null): string {
-  const end = endTime ?? Date.now();
-  const ms = end - startTime;
-  const seconds = Math.floor(ms / 1000);
-  const minutes = Math.floor(seconds / 60);
-  const hours = Math.floor(minutes / 60);
-
-  if (hours > 0) {
-    return `${hours}h${minutes % 60}m`;
-  }
-  if (minutes > 0) {
-    return `${minutes}m${seconds % 60}s`;
-  }
-  return `${seconds}s`;
-}
-
 function formatProcessStatus(
   proc: ProcessInfo,
   theme: ExtensionContext["ui"]["theme"],
 ): string {
-  const runtime = formatRuntime(proc.startTime, proc.endTime);
   const name =
     proc.name.length > 20 ? `${proc.name.slice(0, 17)}...` : proc.name;
 
-  if (proc.status === "running") {
-    return `${theme.fg("accent", name)} ${theme.fg("dim", runtime)}`;
-  }
-  if (proc.status === "killed") {
-    return `${theme.fg("warning", name)} ${theme.fg("dim", "killed")}`;
-  }
-  if (proc.success) {
-    return `${theme.fg("dim", name)} ${theme.fg("success", "done")}`;
+  switch (proc.status) {
+    case "running":
+      return `${theme.fg("accent", name)} ${theme.fg("dim", "running")}`;
+    case "terminating":
+      return `${theme.fg("warning", name)} ${theme.fg("dim", "terminating")}`;
+    case "terminate_timeout":
+      return `${theme.fg("error", name)} ${theme.fg("error", "terminate_timeout")}`;
+    case "killed":
+      return `${theme.fg("warning", name)} ${theme.fg("dim", "killed")}`;
+    case "exited":
+      if (proc.success) {
+        return `${theme.fg("dim", name)} ${theme.fg("success", "done")}`;
+      }
+      return `${theme.fg("error", name)} ${theme.fg("error", `exit(${proc.exitCode ?? "?"})`)}`;
+    default:
+      return `${theme.fg("dim", name)} ${theme.fg("dim", proc.status)}`;
   }
-  return `${theme.fg("error", name)} ${theme.fg("error", `exit(${proc.exitCode ?? "?"})`)}`;
 }
 
 function renderWidget(
@@ -50,17 +41,25 @@ function renderWidget(
     return [];
   }
 
-  const running = processes.filter((p) => p.status === "running");
-  const finished = processes.filter((p) => p.status !== "running");
+  const aliveish = processes.filter(
+    (p) =>
+      p.status === "running" ||
+      p.status === "terminating" ||
+      p.status === "terminate_timeout",
+  );
+  const finished = processes.filter(
+    (p) =>
+      p.status !== "running" &&
+      p.status !== "terminating" &&
+      p.status !== "terminate_timeout",
+  );
 
   const parts: string[] = [];
 
-  // Show running processes first
-  for (const proc of running) {
+  for (const proc of aliveish) {
     parts.push(formatProcessStatus(proc, theme));
   }
 
-  // Show finished processes (most recent first, limit to 3)
   const recentFinished = finished
     .sort((a, b) => (b.endTime ?? 0) - (a.endTime ?? 0))
     .slice(0, 3);
@@ -69,7 +68,6 @@ function renderWidget(
     parts.push(formatProcessStatus(proc, theme));
   }
 
-  // If there are more finished processes, show count
   const hiddenCount = finished.length - recentFinished.length;
   if (hiddenCount > 0) {
     parts.push(theme.fg("dim", `+${hiddenCount} more`));
@@ -81,7 +79,6 @@ function renderWidget(
 
 export function setupProcessWidget(pi: ExtensionAPI, manager: ProcessManager) {
   let latestContext: ExtensionContext | null = null;
-  let refreshInterval: ReturnType<typeof setInterval> | null = null;
 
   function updateWidget() {
     if (!latestContext?.hasUI) return;
@@ -98,28 +95,9 @@ export function setupProcessWidget(pi: ExtensionAPI, manager: ProcessManager) {
     }
   }
 
-  function startRefresh() {
-    if (refreshInterval) return;
-    refreshInterval = setInterval(() => {
-      const hasRunning = manager.list().some((p) => p.status === "running");
-      if (hasRunning) {
-        updateWidget();
-      }
-    }, 1000);
-  }
-
-  function stopRefresh() {
-    if (refreshInterval) {
-      clearInterval(refreshInterval);
-      refreshInterval = null;
-    }
-  }
-
-  // Capture context and update widget
   pi.on("session_start", async (_event, ctx) => {
     latestContext = ctx;
     updateWidget();
-    startRefresh();
   });
 
   pi.on("session_switch", async (_event, ctx) => {
@@ -127,16 +105,9 @@ export function setupProcessWidget(pi: ExtensionAPI, manager: ProcessManager) {
     updateWidget();
   });
 
-  pi.on("session_shutdown", async () => {
-    stopRefresh();
-  });
-
-  // Chain into process end callback
-  const originalOnProcessEnd = manager.onProcessEnd;
-  manager.onProcessEnd = (info) => {
-    originalOnProcessEnd?.call(manager, info);
+  manager.onEvent(() => {
     updateWidget();
-  };
+  });
 
   return {
     update: updateWidget,

package/index.ts

@@ -5,6 +5,14 @@ import { ProcessManager } from "./manager";
 import { setupProcessesTools } from "./tools";
 
 export default function (pi: ExtensionAPI) {
+  if (process.platform === "win32") {
+    pi.on("session_start", async (_event, ctx) => {
+      if (!ctx.hasUI) return;
+      ctx.ui.notify("processes extension not available on Windows", "warning");
+    });
+    return;
+  }
+
   const manager = new ProcessManager();
 
   setupProcessesHooks(pi, manager);