@wrongstack/tui 0.5.5 → 0.5.7

package/dist/index.js

@@ -4,6 +4,7 @@ import * as fs2 from 'fs/promises';
 import * as path2 from 'path';
 import { InputBuilder, DefaultSessionRewinder, formatTodosList, buildChildEnv } from '@wrongstack/core';
 import { routeImagesForModel } from '@wrongstack/runtime/vision';
+import { getProcessRegistry } from '@wrongstack/tools';
 import { readClipboardImage } from '@wrongstack/runtime/clipboard';
 import { jsxs, jsx, Fragment } from 'react/jsx-runtime';
 import { spawn } from 'child_process';
@@ -497,21 +498,13 @@ function History({ entries, streamingText, toolStream }) {
   const toolTail = toolStream?.text ? tailForDisplay(toolStream.text, MAX_STREAM_DISPLAY_CHARS) : "";
   return /* @__PURE__ */ jsxs(Fragment, { children: [
     /* @__PURE__ */ jsx(Static, { items: entries, children: (entry) => /* @__PURE__ */ jsx(Box, { marginBottom: entry.kind === "turn-summary" ? 1 : 0, children: /* @__PURE__ */ jsx(Entry, { entry, termWidth }) }, entry.id) }),
-    tail ? /* @__PURE__ */ jsxs(
-      Box,
-      {
-        flexDirection: "column",
-        borderStyle: "round",
-        borderColor: "blue",
-        paddingX: 2,
-        paddingY: 0,
-        marginY: 1,
-        children: [
-          /* @__PURE__ */ jsx(Box, { flexDirection: "row", marginBottom: 1, children: /* @__PURE__ */ jsx(Text, { bold: true, color: "blue", children: " ASISTANT (streaming...) " }) }),
-          /* @__PURE__ */ jsx(Text, { children: tail })
-        ]
-      }
-    ) : null,
+    tail ? /* @__PURE__ */ jsxs(Box, { flexDirection: "column", marginY: 1, children: [
+      /* @__PURE__ */ jsxs(Box, { flexDirection: "row", children: [
+        /* @__PURE__ */ jsx(Text, { bold: true, color: "cyan", children: "ASSISTANT: " }),
+        /* @__PURE__ */ jsx(Text, { dimColor: true, children: "(streaming...)" })
+      ] }),
+      /* @__PURE__ */ jsx(Text, { color: "white", children: tail })
+    ] }) : null,
     toolTail ? /* @__PURE__ */ jsx(
       ToolStreamBox,
       {
@@ -614,21 +607,10 @@ function Entry({
         entry.queued ? /* @__PURE__ */ jsx(Text, { dimColor: true, children: " (queued)" }) : null
       ] });
     case "assistant":
-      return /* @__PURE__ */ jsxs(
-        Box,
-        {
-          flexDirection: "column",
-          borderStyle: "round",
-          borderColor: "blue",
-          paddingX: 2,
-          paddingY: 0,
-          marginY: 1,
-          children: [
-            /* @__PURE__ */ jsx(Box, { flexDirection: "row", marginBottom: 1, children: /* @__PURE__ */ jsx(Text, { bold: true, color: "blue", children: " ASISTANT " }) }),
-            /* @__PURE__ */ jsx(Text, { children: renderMarkdownTables(entry.text, termWidth) })
-          ]
-        }
-      );
+      return /* @__PURE__ */ jsxs(Box, { flexDirection: "column", marginY: 1, children: [
+        /* @__PURE__ */ jsx(Box, { flexDirection: "row", children: /* @__PURE__ */ jsx(Text, { bold: true, color: "cyan", children: "ASSISTANT: " }) }),
+        /* @__PURE__ */ jsx(Text, { color: "white", children: renderMarkdownTables(entry.text, termWidth) })
+      ] });
     case "tool": {
       const argSummary = formatToolArgs(entry.name, entry.input);
       const outLines = formatToolOutput(
@@ -1649,7 +1631,8 @@ function StatusBar({
   git,
   subagentCount = 0,
   context,
-  projectName
+  projectName,
+  processCount
 }) {
   const usage = tokenCounter?.total();
   const cost = tokenCounter?.estimateCost();
@@ -1714,6 +1697,15 @@ function StatusBar({
               queueCount
             ] })
           ] }) : null,
+          typeof processCount === "number" && processCount > 0 ? /* @__PURE__ */ jsxs(Fragment, { children: [
+            /* @__PURE__ */ jsx(Text, { dimColor: true, children: "\u2502" }),
+            /* @__PURE__ */ jsxs(Text, { color: "red", children: [
+              "\u26A1 ",
+              processCount,
+              " process",
+              processCount === 1 ? "" : "es"
+            ] })
+          ] }) : null,
           hint ? /* @__PURE__ */ jsxs(Fragment, { children: [
             /* @__PURE__ */ jsx(Text, { dimColor: true, children: "\u2502" }),
             /* @__PURE__ */ jsx(Text, { dimColor: true, children: hint })
@@ -2111,6 +2103,101 @@ function oneLine(s, max) {
   const collapsed = s.replace(/\s+/g, " ").trim();
   return collapsed.length <= max ? collapsed : `${collapsed.slice(0, max - 1)}\u2026`;
 }
+var USAGE2 = "Usage:\n  /kill             \u2014 list active processes + breaker state\n  /kill list        \u2014 same as /kill\n  /kill all         \u2014 kill all tracked processes (SIGTERM \u2192 SIGKILL)\n  /kill force       \u2014 kill all with SIGKILL immediately\n  /kill reset       \u2014 reset the circuit breaker to closed\n  /kill <pid>       \u2014 kill a specific process by PID";
+function createKillSlashCommand() {
+  return {
+    name: "kill",
+    description: "List or kill active bash/exec processes managed by the process registry.",
+    async run(args) {
+      const trimmed = args.trim();
+      const parts = trimmed.split(/\s+/);
+      const sub = parts[0]?.toLowerCase() ?? "";
+      if (sub === "" || sub === "list") {
+        return { message: renderList2() };
+      }
+      if (sub === "all") {
+        const pids = getProcessRegistry().killAll();
+        if (pids.length === 0) return { message: "No processes to kill." };
+        return { message: `Killed ${pids.length} process${pids.length === 1 ? "" : "es"}: ${pids.join(", ")}` };
+      }
+      if (sub === "force") {
+        getProcessRegistry().forceBreakerOpen();
+        const pids = getProcessRegistry().killAll({ force: true });
+        if (pids.length === 0) return { message: "Circuit breaker forced open. No processes to kill." };
+        return { message: `Force-killed ${pids.length} process${pids.length === 1 ? "" : "es"}: ${pids.join(", ")}` };
+      }
+      if (sub === "reset") {
+        getProcessRegistry().forceBreakerReset();
+        return { message: "Circuit breaker reset to closed. Bash/exec calls allowed." };
+      }
+      const pid = Number.parseInt(sub, 10);
+      if (!Number.isNaN(pid) && pid > 0) {
+        const found = getProcessRegistry().kill(pid);
+        if (found) return { message: `Killed process ${pid}.` };
+        return { message: `Process ${pid} not found in registry.` };
+      }
+      return { message: `Unknown subcommand "${sub}".
+${USAGE2}` };
+    }
+  };
+}
+function renderList2() {
+  const registry = getProcessRegistry();
+  const stats = registry.stats();
+  const all = registry.list();
+  const breaker = stats.breaker;
+  const stateLabel = breaker.state === "closed" ? "\u{1F7E2} closed" : breaker.state === "half-open" ? "\u{1F7E1} half-open" : `\u{1F534} open (cooldown ${breaker.cooldownRemainingMs !== null ? `${(breaker.cooldownRemainingMs / 1e3).toFixed(0)}s` : "\u2014"})`;
+  const breakerLine = [
+    `  Circuit breaker: ${stateLabel}`,
+    `    consecutive failures: ${breaker.consecutiveFailures}/5`,
+    `    slow calls in window: ${breaker.slowCallsInWindow}/3`,
+    `    calls in window: ${breaker.callsInWindow}/30`
+  ].join("\n");
+  if (all.length === 0) {
+    return `No active processes.
+
+${breakerLine}`;
+  }
+  const now = Date.now();
+  const lines = [`Active processes (${all.length}):`];
+  for (const p of all) {
+    const age = ((now - p.startedAt) / 1e3).toFixed(1);
+    const killedTag = p.killed ? " [killed]" : "";
+    const cmd = p.command.length > 80 ? p.command.slice(0, 77) + "\u2026" : p.command;
+    lines.push(`  ${p.pid}  ${p.name}  ${age}s  ${cmd}${killedTag}`);
+  }
+  return [...lines, "", breakerLine].join("\n");
+}
+function createPsSlashCommand() {
+  return {
+    name: "ps",
+    description: "List all active bash/exec processes tracked by the process registry.",
+    async run(_args) {
+      return { message: renderList3() };
+    }
+  };
+}
+function renderList3() {
+  const registry = getProcessRegistry();
+  const stats = registry.stats();
+  const all = registry.list();
+  if (all.length === 0) return "No active processes.";
+  const breaker = stats.breaker;
+  const stateLabel = breaker.state === "closed" ? "\u{1F7E2} closed" : breaker.state === "half-open" ? "\u{1F7E1} half-open" : `\u{1F534} open`;
+  const now = Date.now();
+  const lines = [
+    `Active processes (${all.length}) \u2014 breaker ${stateLabel}`,
+    `  failure=${breaker.consecutiveFailures}/5  slow=${breaker.slowCallsInWindow}/3  rate=${breaker.callsInWindow}/30`,
+    ""
+  ];
+  for (const p of all) {
+    const age = ((now - p.startedAt) / 1e3).toFixed(1);
+    const killedTag = p.killed ? " [killed]" : "";
+    const cmd = p.command.length > 80 ? p.command.slice(0, 77) + "\u2026" : p.command;
+    lines.push(`  ${p.pid}  ${p.name}  ${age}s  ${cmd}${killedTag}`);
+  }
+  return lines.join("\n");
+}
 function selectedSlashCommandLine(picker) {
   if (!picker.open || picker.matches.length === 0) return null;
   const picked = picker.matches[picker.selected];
@@ -3091,6 +3178,19 @@ function App({
       slashRegistry.unregister("queue");
     };
   }, [slashRegistry]);
+  useEffect(() => {
+    slashRegistry.register(createKillSlashCommand());
+    slashRegistry.register(createPsSlashCommand());
+    return () => {
+      slashRegistry.unregister("kill");
+      slashRegistry.unregister("ps");
+    };
+  }, [slashRegistry]);
+  useEffect(() => {
+    return () => {
+      getProcessRegistry().killAll();
+    };
+  }, []);
   useEffect(() => {
     const ALT_OFF = "\x1B[?1049l";
     const ALT_ON = "\x1B[?1049h";
@@ -3670,6 +3770,7 @@ function App({
     const onSigint = () => {
       const current = stateRef.current;
       if (current.interrupts >= 1) {
+        getProcessRegistry().killAll({ force: true });
         if (current.interrupts >= 2) {
           process.exit(130);
         }
@@ -3707,6 +3808,8 @@ function App({
           });
           void Promise.race([director.terminateAll().catch(() => void 0), cap]);
         }
+        const killed = getProcessRegistry().killAll();
+        const procTag = killed.length > 0 ? ` + killed ${killed.length} process${killed.length === 1 ? "" : "es"}` : "";
         const droppedCount = stateRef.current.queue.length;
         if (droppedCount > 0) {
           dispatch({ type: "queueClear" });
@@ -3714,7 +3817,7 @@ function App({
             type: "addEntry",
             entry: {
               kind: "warn",
-              text: `Iteration cancelled${director ? " + fleet terminated" : ""}. Dropped ${droppedCount} queued message${droppedCount === 1 ? "" : "s"}. Press Ctrl+C again to exit.`
+              text: `Iteration cancelled${director ? " + fleet terminated" : ""}${procTag}. Dropped ${droppedCount} queued message${droppedCount === 1 ? "" : "s"}. Press Ctrl+C again to exit.`
             }
           });
         } else {
@@ -3722,14 +3825,16 @@ function App({
             type: "addEntry",
             entry: {
               kind: "warn",
-              text: `Iteration cancelled${director ? " + fleet terminated" : ""}. Press Ctrl+C again to exit.`
+              text: `Iteration cancelled${director ? " + fleet terminated" : ""}${procTag}. Press Ctrl+C again to exit.`
             }
           });
         }
       } else {
+        const killed = getProcessRegistry().killAll();
+        const procTag = killed.length > 0 ? ` Killed ${killed.length} process${killed.length === 1 ? "" : "es"}.` : "";
         dispatch({
           type: "addEntry",
-          entry: { kind: "warn", text: "Press Ctrl+C again to exit." }
+          entry: { kind: "warn", text: `Press Ctrl+C again to exit.${procTag}` }
         });
       }
     };
@@ -4344,7 +4449,8 @@ User message:
         git: gitInfo,
         context: contextWindow,
         projectName,
-        subagentCount: Object.keys(state.fleet).length
+        subagentCount: Object.keys(state.fleet).length,
+        processCount: getProcessRegistry().activeCount
       }
     ),
     director ? /* @__PURE__ */ jsx(FleetPanel, { entries: state.fleet, totalCost: state.fleetCost, roster: fleetRoster }) : null