@wrongstack/tui 0.250.0 → 0.255.0

package/dist/index.d.ts

@@ -338,8 +338,8 @@ interface RunTuiOptions {
      * visible bar without a round-trip. The initial value is loaded from
      * the config file before App mounts.
      */
-    statuslineHiddenItems: Array<'todos' | 'plan' | 'fleet' | 'git' | 'elapsed' | 'context' | 'cost'>;
-    setStatuslineHiddenItems: (items: Array<'todos' | 'plan' | 'fleet' | 'git' | 'elapsed' | 'context' | 'cost'>) => void;
+    statuslineHiddenItems: Array<'todos' | 'plan' | 'tasks' | 'fleet' | 'git' | 'elapsed' | 'context' | 'cost'>;
+    setStatuslineHiddenItems: (items: Array<'todos' | 'plan' | 'tasks' | 'fleet' | 'git' | 'elapsed' | 'context' | 'cost'>) => void;
     /**
      * Controller for the agents monitor overlay. App installs a dispatch-backed
      * setter on mount so the `/agents on|off` slash command can toggle the

package/dist/index.js

@@ -1,6 +1,7 @@
-import { writeErr, resolveWstackPaths, loadGoal, InputBuilder, DefaultSessionRewinder, writeOut, formatTodosList, buildGoalPreamble, expectDefined as expectDefined$1, wstackGlobalRoot, shouldEnhance, enhanceUserPrompt, recentTextTurns, normalizedEqual, buildChildEnv } from '@wrongstack/core';
+import { writeErr, resolveProjectDir, wstackGlobalRoot, GlobalMailbox, resolveWstackPaths, loadGoal, InputBuilder, DefaultSessionRewinder, writeOut, formatTodosList, buildGoalPreamble, expectDefined as expectDefined$1, shouldEnhance, enhanceUserPrompt, recentTextTurns, normalizedEqual, buildChildEnv } from '@wrongstack/core';
 export { buildGoalPreamble } from '@wrongstack/core';
 import { Box as Box$1, useInput, useStdin, useStdout, Text as Text$1, render, useApp, measureElement, Static } from 'ink';
+import { randomUUID } from 'crypto';
 import * as path4 from 'path';
 import React5, { forwardRef, useState, useEffect, useMemo, memo, useRef, useCallback, useReducer, useLayoutEffect } from 'react';
 import * as fs2 from 'fs/promises';
@@ -86,6 +87,27 @@ var Box = forwardRef(function Box2({ borderColor, backgroundColor, ...rest }, re
     }
   );
 });
+function useTokenCounterRefresh(tokenCounter, events) {
+  const [data, setData] = useState(
+    () => tokenCounter ? {
+      usage: tokenCounter.total(),
+      cost: tokenCounter.estimateCost(),
+      cacheStats: tokenCounter.cacheStats()
+    } : void 0
+  );
+  useEffect(() => {
+    if (!tokenCounter || !events) return;
+    const off = events.on("token.accounted", () => {
+      setData({
+        usage: tokenCounter.total(),
+        cost: tokenCounter.estimateCost(),
+        cacheStats: tokenCounter.cacheStats()
+      });
+    });
+    return off;
+  }, [tokenCounter, events]);
+  return data;
+}
 var MODE_ICONS = {
   teach: "\u{1F9D1}\u200D\u{1F3EB}",
   brief: "\u26A1",
@@ -131,6 +153,7 @@ function StatusBar({
   processCount,
   context,
   hiddenItems,
+  events,
   eternalStage,
   goalSummary,
   indexState,
@@ -155,9 +178,10 @@ function StatusBar({
   const isCompact = termWidth < COMPACT_THRESHOLD;
   const isComfortable = termWidth >= COMFORTABLE_THRESHOLD;
   const hiddenSet = new Set(hiddenItems);
-  const usage = tokenCounter?.total();
-  const cost = tokenCounter?.estimateCost();
-  const cache2 = tokenCounter?.cacheStats();
+  const tokenData = useTokenCounterRefresh(tokenCounter, events);
+  const usage = tokenData?.usage;
+  const cost = tokenData?.cost;
+  const cache2 = tokenData?.cacheStats;
   const [elapsedMs, setElapsedMs] = useState(startedAt ? Date.now() - startedAt : 0);
   useEffect(() => {
     if (startedAt == null) return;
@@ -555,7 +579,22 @@ function StatusBar({
           /* @__PURE__ */ jsxs(Text, { color: "cyan", children: [
             "\u{1F465} ",
             mailbox.onlineAgents,
-            " online"
+            " agent",
+            mailbox.onlineAgents === 1 ? "" : "s",
+            mailbox.onlineClients.tui + mailbox.onlineClients.webui + mailbox.onlineClients.repl > 0 ? /* @__PURE__ */ jsxs(Text, { color: "green", children: [
+              mailbox.onlineClients.tui > 0 ? /* @__PURE__ */ jsxs(Text, { color: "green", children: [
+                " \xB7 \u{1F5A5} TUI",
+                mailbox.onlineClients.tui > 1 ? `\xD7${mailbox.onlineClients.tui}` : ""
+              ] }) : null,
+              mailbox.onlineClients.webui > 0 ? /* @__PURE__ */ jsxs(Text, { color: "green", children: [
+                " \xB7 \u{1F310} WebUI",
+                mailbox.onlineClients.webui > 1 ? `\xD7${mailbox.onlineClients.webui}` : ""
+              ] }) : null,
+              mailbox.onlineClients.repl > 0 ? /* @__PURE__ */ jsxs(Text, { color: "green", children: [
+                " \xB7 \u2328 REPL",
+                mailbox.onlineClients.repl > 1 ? `\xD7${mailbox.onlineClients.repl}` : ""
+              ] }) : null
+            ] }) : null
           ] }),
           mailbox.lastSubject ? /* @__PURE__ */ jsxs(Fragment, { children: [
             /* @__PURE__ */ jsx(Text, { dimColor: true, children: "\u2502" }),
@@ -4308,6 +4347,15 @@ function startHeapWatchdog(opts = {}) {
     }).catch(() => void 0);
   };
   const tick = () => {
+    let userTimings = 0;
+    try {
+      userTimings = performance.getEntriesByType("mark").length + performance.getEntriesByType("measure").length;
+      if (userTimings > 0) {
+        performance.clearMarks();
+        performance.clearMeasures();
+      }
+    } catch {
+    }
     const s2 = takeHeapSample();
     if (s2.load >= criticalAt && criticalArmed) {
       criticalArmed = false;
@@ -4336,7 +4384,7 @@ function startHeapWatchdog(opts = {}) {
         extras = opts.collectStats?.() ?? {};
       } catch {
       }
-      append(JSON.stringify({ pid: process.pid, ...s2, ...extras }));
+      append(JSON.stringify({ pid: process.pid, ...s2, userTimings, ...extras }));
     }
   };
   const timer = setInterval(tick, sampleEveryMs);
@@ -6931,6 +6979,7 @@ function useSessionEvents(events, dispatch, onClearHistory) {
     const offRewound = events.on("session.rewound", () => {
       dispatch({ type: "sessionRewound", toPromptIndex: 0 });
       dispatch({ type: "clearHistory" });
+      dispatch({ type: "resetContextChip" });
       onClearHistory?.(dispatch);
     });
     return () => {
@@ -9148,7 +9197,8 @@ function App({
   }, [getLiveSessions]);
   const [mailboxStatus, setMailboxStatus] = useState({
     unread: 0,
-    onlineAgents: 0
+    onlineAgents: 0,
+    onlineClients: { tui: 0, webui: 0, repl: 0 }
   });
   useEffect(() => {
     const seenAgents = /* @__PURE__ */ new Set();
@@ -9174,11 +9224,25 @@ function App({
       if (p?.agentId) seenAgents.add(p.agentId);
       setMailboxStatus((prev) => ({ ...prev, onlineAgents: seenAgents.size }));
     });
+    const unsub5 = events.onPattern("mailbox.sync_clients", (_e, payload) => {
+      const p = payload;
+      if (p) {
+        setMailboxStatus((prev) => ({
+          ...prev,
+          onlineClients: {
+            tui: p.tui ?? 0,
+            webui: p.webui ?? 0,
+            repl: p.repl ?? 0
+          }
+        }));
+      }
+    });
     return () => {
       unsub1();
       unsub2();
       unsub3();
       unsub4();
+      unsub5();
     };
   }, [events]);
   const [mailboxPanelOpen, setMailboxPanelOpen] = useState(false);
@@ -12248,6 +12312,7 @@ User message:
           subagentCount: Object.keys(state.fleet).length,
           processCount: getProcessRegistry().activeCount,
           hiddenItems,
+          events,
           eternalStage: state.eternalStage,
           goalSummary: state.goalSummary,
           indexState,
@@ -12523,6 +12588,7 @@ async function runTui(opts) {
   const cleanup = () => {
     if (cleaned) return;
     cleaned = true;
+    unregisterTuiClient();
     unsilenceTerminal();
     try {
       stopTitle();
@@ -12549,6 +12615,64 @@ async function runTui(opts) {
     }
     process.off("exit", exitHandler);
   };
+  let clientHeartbeatTimer = null;
+  let clientSyncTimer = null;
+  const CLIENT_HEARTBEAT_MS = 15e3;
+  const CLIENT_SYNC_MS = 3e4;
+  const registerTuiClient = async () => {
+    if (!opts.projectRoot) return null;
+    try {
+      const projectDir = resolveProjectDir(opts.projectRoot, wstackGlobalRoot());
+      const mailbox = new GlobalMailbox(projectDir, opts.events);
+      const clientId = `tui@${randomUUID().slice(0, 8)}`;
+      await mailbox.registerClient({
+        clientId,
+        sessionId: opts.projectRoot,
+        name: `TUI [${path4.basename(opts.projectRoot)}]`,
+        source: "tui",
+        pid: process.pid
+      });
+      clientHeartbeatTimer = setInterval(() => {
+        mailbox.clientHeartbeat({ clientId }).catch(() => {
+        });
+      }, CLIENT_HEARTBEAT_MS);
+      clientHeartbeatTimer.unref();
+      const syncClients = async () => {
+        try {
+          const statuses = await mailbox.getClientStatuses();
+          const counts = { tui: 0, webui: 0, repl: 0 };
+          for (const s2 of statuses) {
+            if (s2.online && s2.source in counts) {
+              counts[s2.source]++;
+            }
+          }
+          opts.events.emitCustom("mailbox.sync_clients", counts);
+        } catch {
+        }
+      };
+      setTimeout(() => {
+        void syncClients();
+      }, 5e3);
+      clientSyncTimer = setInterval(() => {
+        void syncClients();
+      }, CLIENT_SYNC_MS);
+      clientSyncTimer.unref();
+      return clientId;
+    } catch {
+      return null;
+    }
+  };
+  const unregisterTuiClient = () => {
+    if (clientHeartbeatTimer) {
+      clearInterval(clientHeartbeatTimer);
+      clientHeartbeatTimer = null;
+    }
+    if (clientSyncTimer) {
+      clearInterval(clientSyncTimer);
+      clientSyncTimer = null;
+    }
+  };
+  registerTuiClient();
   return new Promise((resolve) => {
     let exitCode = 0;
     let hardExitTimer = null;