@wrongstack/tui 0.10.3 → 0.24.0

package/dist/index.d.ts

@@ -234,6 +234,15 @@ interface RunTuiOptions {
         mode: 'off' | 'suggest' | 'auto';
         delayMs: number;
     }) => string | null | Promise<string | null>;
+    /**
+     * Predict likely next steps after a completed turn. The CLI wires this from
+     * the session provider and the `/next` toggle; it returns [] when prediction
+     * is disabled or autonomy isn't 'off'. Display-only — never executed.
+     */
+    predictNext?: (input: {
+        userRequest: string;
+        assistantSummary: string;
+    }) => Promise<string[]>;
 }
 declare function runTui(opts: RunTuiOptions): Promise<number>;
 

package/dist/index.js

@@ -1,6 +1,6 @@
-import { PassThrough } from 'stream';
+import { Readable } from 'stream';
 import { Box, Text, render, useApp, useStdout, measureElement, Static, useInput, useStdin } from 'ink';
-import React4, { useState, useEffect, useReducer, useRef, useMemo, useLayoutEffect } from 'react';
+import React4, { useState, useEffect, useReducer, useRef, useMemo, useCallback, useLayoutEffect } from 'react';
 import * as fs2 from 'fs/promises';
 import * as path2 from 'path';
 import { InputBuilder, DefaultSessionRewinder, formatTodosList, buildGoalPreamble, buildChildEnv } from '@wrongstack/core';
@@ -882,21 +882,58 @@ function AgentsMonitor({
   ] });
 }
 var AUTONOMY_OPTIONS = [
-  { mode: "off", label: "OFF", description: "Agent stops after each turn (normal interactive mode)", color: "green" },
-  { mode: "suggest", label: "SUGGEST", description: "Shows next-step suggestions after each turn", color: "cyan" },
-  { mode: "auto", label: "AUTO", description: "Self-driving \u2014 agent picks next step and continues", color: "yellow" },
-  { mode: "eternal", label: "ETERNAL", description: "Goal-driven loop \u2014 requires /goal set first", color: "red" },
-  { mode: "eternal-parallel", label: "PARALLEL", description: "Fan-out 4\u20138 subagents per tick \u2014 requires /goal", color: "magenta" }
+  {
+    mode: "off",
+    label: "OFF",
+    description: "Agent stops after each turn (normal interactive mode)",
+    color: "green"
+  },
+  {
+    mode: "suggest",
+    label: "SUGGEST",
+    description: "Shows next-step suggestions after each turn",
+    color: "cyan"
+  },
+  {
+    mode: "auto",
+    label: "AUTO",
+    description: "Self-driving \u2014 agent picks next step and continues",
+    color: "yellow"
+  },
+  {
+    mode: "eternal",
+    label: "ETERNAL",
+    description: "Goal-driven loop \u2014 requires /goal set first",
+    color: "red"
+  },
+  {
+    mode: "eternal-parallel",
+    label: "PARALLEL",
+    description: "Fan-out 4\u20138 subagents per tick \u2014 requires /goal",
+    color: "magenta"
+  }
 ];
-function AutonomyPicker({ options, selected, hint }) {
+function AutonomyPicker({
+  options,
+  selected,
+  hint
+}) {
   return /* @__PURE__ */ jsxs(Box, { flexDirection: "column", paddingX: 1, children: [
     /* @__PURE__ */ jsx(Text, { color: "cyan", bold: true, children: "\u2501\u2501 Autonomy Mode \u2501\u2501" }),
     /* @__PURE__ */ jsx(Text, { dimColor: true, children: "\u2191/\u2193 navigate \xB7 Enter select \xB7 Esc cancel \xB7 Ctrl+C exit" }),
-    options.map((opt, i) => /* @__PURE__ */ jsxs(Text, { color: i === selected ? opt.color : void 0, inverse: i === selected, children: [
-      i === selected ? "\u203A " : "  ",
-      /* @__PURE__ */ jsx(Text, { bold: true, children: opt.label.padEnd(12) }),
-      /* @__PURE__ */ jsx(Text, { dimColor: true, children: opt.description })
-    ] }, opt.mode)),
+    options.map((opt, i) => /* @__PURE__ */ jsxs(
+      Text,
+      {
+        color: i === selected ? opt.color : void 0,
+        inverse: i === selected,
+        children: [
+          i === selected ? "\u203A " : "  ",
+          /* @__PURE__ */ jsx(Text, { bold: true, children: opt.label.padEnd(12) }),
+          /* @__PURE__ */ jsx(Text, { dimColor: true, children: opt.description })
+        ]
+      },
+      opt.mode
+    )),
     hint ? /* @__PURE__ */ jsx(Text, { color: "yellow", children: hint }) : null
   ] });
 }
@@ -934,7 +971,8 @@ function CheckpointTimeline({
           new Date(cp.ts).toLocaleTimeString()
         ] }),
         cp.fileCount > 0 && /* @__PURE__ */ jsxs(Text, { dimColor: true, children: [
-          " \xB7 ",
+          " ",
+          "\xB7 ",
           cp.fileCount,
           " file",
           cp.fileCount !== 1 ? "s" : ""
@@ -1055,7 +1093,11 @@ function FilePicker({ query, matches, selected }) {
 function highlight(path3, _query) {
   return path3;
 }
-function FleetPanel({ entries, totalCost, collabSession }) {
+function FleetPanel({
+  entries,
+  totalCost,
+  collabSession
+}) {
   const { stdout } = useStdout();
   const [termWidth, setTermWidth] = useState(stdout?.columns ?? 90);
   useEffect(() => {
@@ -1580,7 +1622,8 @@ function tokenizePython(line, carry) {
 }
 function tokenizeDiff(line) {
   if (line.startsWith("@@")) return [{ text: line, color: C.diffMeta }];
-  if (line.startsWith("+++") || line.startsWith("---")) return [{ text: line, color: C.diffMeta, dim: true }];
+  if (line.startsWith("+++") || line.startsWith("---"))
+    return [{ text: line, color: C.diffMeta, dim: true }];
   if (line.startsWith("+")) return [{ text: line, color: C.diffAdd }];
   if (line.startsWith("-")) return [{ text: line, color: C.diffDel }];
   return [{ text: line, dim: true }];
@@ -1996,7 +2039,7 @@ function MarkdownView({
     if (quote && line.startsWith(">")) {
       rows.push(
         /* @__PURE__ */ jsxs(Box, { flexDirection: "row", children: [
-          /* @__PURE__ */ jsx(Text, { dimColor: true, children: "\u2502 " }),
+          /* @__PURE__ */ jsx(Text, { dimColor: true, children: "  " }),
           /* @__PURE__ */ jsx(InlineLine, { tokens: parseInline(quote[1] ?? ""), dim: true })
         ] }, `q${key++}`)
       );
@@ -2121,13 +2164,13 @@ function splitFencedBlocks(text) {
 function CodeBlock({
   code,
   lang,
-  termWidth
+  contentWidth
 }) {
   let lines = code.replace(/\n+$/, "").split("\n");
   const hidden = Math.max(0, lines.length - MAX_CODE_LINES);
   if (hidden > 0) lines = lines.slice(0, MAX_CODE_LINES);
   const gutterW = String(lines.length).length;
-  const maxW = Math.max(20, Math.min(termWidth - 8 - gutterW - 1, 120));
+  const maxW = Math.max(20, Math.min(contentWidth - 6 - gutterW - 1, 120));
   let carry = {};
   const rows = lines.map((raw) => {
     const display = raw.length > maxW ? `${raw.slice(0, maxW - 1)}\u2026` : raw;
@@ -2135,20 +2178,31 @@ function CodeBlock({
     carry = r.carry;
     return r.tokens;
   });
-  return /* @__PURE__ */ jsxs(Box, { flexDirection: "column", marginLeft: 2, marginY: 0, borderStyle: "round", borderColor: theme.borderDefault, paddingX: 1, children: [
-    lang !== "plain" ? /* @__PURE__ */ jsx(Text, { dimColor: true, children: lang }) : null,
-    rows.map((tokens, i) => (
-      // biome-ignore lint/suspicious/noArrayIndexKey: code lines are positional
-      /* @__PURE__ */ jsxs(Text, { children: [
-        /* @__PURE__ */ jsx(Text, { dimColor: true, children: `${String(i + 1).padStart(gutterW, " ")} ` }),
-        tokens.length === 0 ? " " : tokens.map((t, j) => (
-          // biome-ignore lint/suspicious/noArrayIndexKey: token order is stable per line
-          /* @__PURE__ */ jsx(Text, { color: t.color, dimColor: t.dim, bold: t.bold, children: t.text }, j)
-        ))
-      ] }, i)
-    )),
-    hidden > 0 ? /* @__PURE__ */ jsx(Text, { dimColor: true, italic: true, children: `\u2026 +${hidden} more line${hidden === 1 ? "" : "s"}` }) : null
-  ] });
+  return /* @__PURE__ */ jsxs(
+    Box,
+    {
+      flexDirection: "column",
+      marginLeft: 2,
+      marginY: 0,
+      borderStyle: "round",
+      borderColor: theme.borderDefault,
+      paddingX: 1,
+      children: [
+        lang !== "plain" ? /* @__PURE__ */ jsx(Text, { dimColor: true, children: lang }) : null,
+        rows.map((tokens, i) => (
+          // biome-ignore lint/suspicious/noArrayIndexKey: code lines are positional
+          /* @__PURE__ */ jsxs(Text, { children: [
+            /* @__PURE__ */ jsx(Text, { dimColor: true, children: `${String(i + 1).padStart(gutterW, " ")} ` }),
+            tokens.length === 0 ? " " : tokens.map((t, j) => (
+              // biome-ignore lint/suspicious/noArrayIndexKey: token order is stable per line
+              /* @__PURE__ */ jsx(Text, { color: t.color, dimColor: t.dim, bold: t.bold, children: t.text }, j)
+            ))
+          ] }, i)
+        )),
+        hidden > 0 ? /* @__PURE__ */ jsx(Text, { dimColor: true, italic: true, children: `\u2026 +${hidden} more line${hidden === 1 ? "" : "s"}` }) : null
+      ]
+    }
+  );
 }
 function AssistantBody({
   text,
@@ -2160,7 +2214,7 @@ function AssistantBody({
   return /* @__PURE__ */ jsx(Box, { flexDirection: "column", children: segments.map(
     (seg, i) => seg.type === "code" ? (
       // biome-ignore lint/suspicious/noArrayIndexKey: segment order is stable
-      /* @__PURE__ */ jsx(CodeBlock, { code: seg.text, lang: seg.lang ?? "plain", termWidth }, i)
+      /* @__PURE__ */ jsx(CodeBlock, { code: seg.text, lang: seg.lang ?? "plain", contentWidth: inner }, i)
     ) : (
       // biome-ignore lint/suspicious/noArrayIndexKey: segment order is stable
       /* @__PURE__ */ jsx(MarkdownView, { text: seg.text, termWidth: inner }, i)
@@ -2289,14 +2343,7 @@ var Entry = React4.memo(function Entry2({
           paddingLeft: 1,
           children: [
             /* @__PURE__ */ jsx(Box, { flexDirection: "row", children: /* @__PURE__ */ jsx(Text, { bold: true, color: theme.assistant, children: "ASSISTANT" }) }),
-            /* @__PURE__ */ jsx(
-              AssistantBody,
-              {
-                text: entry.text,
-                termWidth,
-                contentWidth
-              }
-            )
+            /* @__PURE__ */ jsx(AssistantBody, { text: entry.text, termWidth, contentWidth })
           ]
         }
       );
@@ -3411,8 +3458,10 @@ function fmtRecentTool(tool) {
   const name = tool.name.length > 18 ? `${tool.name.slice(0, 17)}...` : tool.name;
   const parts = [status, name];
   if (typeof tool.durationMs === "number") parts.push(fmtElapsed2(tool.durationMs));
-  if (typeof tool.outputBytes === "number" && tool.outputBytes > 0) parts.push(fmtBytes2(tool.outputBytes));
-  if (typeof tool.outputLines === "number" && tool.outputLines > 0) parts.push(`${tool.outputLines}L`);
+  if (typeof tool.outputBytes === "number" && tool.outputBytes > 0)
+    parts.push(fmtBytes2(tool.outputBytes));
+  if (typeof tool.outputLines === "number" && tool.outputLines > 0)
+    parts.push(`${tool.outputLines}L`);
   return parts.join(" ");
 }
 function fmtRecentMessage(message) {
@@ -3545,7 +3594,9 @@ function PhaseMonitor({
     if (key.escape) onClose();
   });
   const phaseList = Object.values(phases);
-  const running = phaseList.filter((p) => runningPhaseIds.includes(Object.keys(phases).find((k) => phases[k] === p) ?? ""));
+  const running = phaseList.filter(
+    (p) => runningPhaseIds.includes(Object.keys(phases).find((k) => phases[k] === p) ?? "")
+  );
   const done = phaseList.filter((p) => p.status === "completed" || p.status === "skipped");
   const failed = phaseList.filter((p) => p.status === "failed");
   return /* @__PURE__ */ jsxs(Box, { flexDirection: "column", borderStyle: "round", borderColor: "cyan", paddingX: 1, children: [
@@ -3756,7 +3807,7 @@ function ScrollableHistory({
       lastReported.current = height;
       onMeasure(height);
     }
-  });
+  }, [onMeasure]);
   const vp = Math.max(1, viewportRows);
   return /* @__PURE__ */ jsxs(Box, { flexDirection: "row", children: [
     /* @__PURE__ */ jsx(
@@ -3992,7 +4043,9 @@ function WorktreePanel({
 }) {
   const list = Object.values(worktrees);
   if (list.length === 0) return null;
-  const active = list.filter((w) => w.status === "active" || w.status === "committing" || w.status === "merging").length;
+  const active = list.filter(
+    (w) => w.status === "active" || w.status === "committing" || w.status === "merging"
+  ).length;
   const merged = list.filter((w) => w.status === "merged").length;
   const failed = list.filter((w) => w.status === "failed" || w.status === "needs-review").length;
   return /* @__PURE__ */ jsxs(
@@ -4036,7 +4089,8 @@ function WorktreePanel({
             /* @__PURE__ */ jsx(Text, { children: w.branch.replace(/^wstack\/ap\//, "").slice(0, 18).padEnd(18) }),
             /* @__PURE__ */ jsx(Text, { dimColor: true, children: w.ownerLabel.slice(0, 12) }),
             conflict ? /* @__PURE__ */ jsx(Text, { color: "magenta", children: " CONFLICT" }) : /* @__PURE__ */ jsxs(Text, { dimColor: true, children: [
-              " +",
+              " ",
+              "+",
               w.insertions,
               "/-",
               w.deletions,
@@ -4200,13 +4254,18 @@ function createKillSlashCommand() {
       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(", ")}` };
+        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 (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();
@@ -5375,6 +5434,7 @@ function App({
   switchProviderAndModel,
   getSettings,
   saveSettings,
+  predictNext,
   switchAutonomy,
   effectiveMaxContext,
   onExit,
@@ -5400,8 +5460,12 @@ function App({
   const [hiddenItems, setHiddenItems] = useState(statuslineHiddenItems);
   const { stdout } = useStdout();
   const [termRows, setTermRows] = useState(stdout?.rows ?? 24);
+  const [termCols, setTermCols] = useState(stdout?.columns ?? 80);
   useEffect(() => {
-    const onResize = () => setTermRows(process.stdout.rows ?? 24);
+    const onResize = () => {
+      setTermRows(process.stdout.rows ?? 24);
+      setTermCols(process.stdout.columns ?? 80);
+    };
     process.stdout.on("resize", onResize);
     return () => {
       process.stdout.off("resize", onResize);
@@ -5551,7 +5615,7 @@ function App({
     if (vp !== s2.viewportRows) {
       dispatch({ type: "setViewportRows", rows: vp });
     }
-  });
+  }, [managedLive, termRows]);
   const handleKeyRef = useRef(null);
   const pendingClickConfirmRef = useRef(null);
   const openModelPickerRef = useRef(null);
@@ -5599,7 +5663,7 @@ function App({
           }
           return;
         }
-        const cols = process.stdout.columns ?? 0;
+        const cols = termCols || 80;
         const onScrollbar = cols > 0 && ev.x >= cols - 2 && ev.y >= 1 && ev.y <= rows;
         if (onScrollbar && s2.totalLines > rows) {
           scrollbarDragRef.current = true;
@@ -5687,7 +5751,7 @@ function App({
       if (!picker || picker.count === 0) {
         const inputDisabled = s2.status === "aborting" && !s2.steeringPending;
         if (!inputDisabled) {
-          const cols = process.stdout.columns ?? 80;
+          const cols = termCols || 80;
           const inputTop = s2.viewportRows + affordance + liveStripRowsRef.current + 1;
           const inputRows = layoutInputRows(INPUT_PROMPT, s2.buffer, s2.cursor, cols).length;
           const rowIdx = ev.y - inputTop;
@@ -5730,7 +5794,8 @@ function App({
       }
     },
     // dispatch is stable (useReducer); refs are mutable — no reactive deps.
-    []
+    // termCols is stable (useState + resize effect).
+    [termCols]
   );
   useEffect(() => {
     if (!subscribeMouse) return;
@@ -5899,12 +5964,12 @@ function App({
   }, [agent.ctx.meta]);
   const prevAnyOverlayOpen = useRef(false);
   const prevEntriesCount = useRef(0);
-  const eraseLiveRegion = () => {
+  const eraseLiveRegion = useCallback(() => {
     try {
       process.stdout.write("\x1B[J");
     } catch {
     }
-  };
+  }, []);
   useEffect(() => {
     const anyOpenNow = state.picker.open || state.slashPicker.open || state.modelPicker.open || state.autonomyPicker.open || state.settingsPicker.open || state.confirmQueue.length > 0;
     const overlayClosed = prevAnyOverlayOpen.current && !anyOpenNow;
@@ -5921,7 +5986,8 @@ function App({
     state.autonomyPicker.open,
     state.settingsPicker.open,
     state.confirmQueue.length,
-    state.entries.length
+    state.entries.length,
+    eraseLiveRegion
   ]);
   useEffect(() => {
     const handleResize = () => eraseLiveRegion();
@@ -5929,7 +5995,7 @@ function App({
     return () => {
       process.stdout.off("resize", handleResize);
     };
-  }, []);
+  }, [eraseLiveRegion]);
   useEffect(() => {
     const detected = detectAtToken(state.buffer, state.cursor);
     if (!detected) {
@@ -7903,6 +7969,22 @@ function App({
           }
         });
       }
+      if (result.status === "done" && predictNext) {
+        try {
+          const userRequest = blocks.filter((b) => b.type === "text").map((b) => b.text).join(" ").trim();
+          const predictions = await predictNext({
+            userRequest,
+            assistantSummary: result.finalText ?? ""
+          });
+          if (predictions.length > 0) {
+            const text = ["\u21B3 likely next:", ...predictions.map((p, i) => `  ${i + 1}. ${p}`)].join(
+              "\n"
+            );
+            dispatch({ type: "addEntry", entry: { kind: "turn-summary", text } });
+          }
+        } catch {
+        }
+      }
     } catch (err) {
       dispatch({
         type: "addEntry",
@@ -8546,8 +8628,48 @@ async function runTui(opts) {
   let inkStdin = stdin;
   let detachMouse = null;
   if (useMouse) {
-    const proxy = new PassThrough();
-    const p = proxy;
+    class KeyboardReadable extends Readable {
+      pendingChunks = [];
+      // eslint-disable-next-line no-useless-constructor
+      constructor() {
+        super({ encoding: "utf8", highWaterMark: 64 * 1024 });
+      }
+      _read(_size) {
+        this.flushPending();
+      }
+      flushPending() {
+        while (this.pendingChunks.length > 0) {
+          const chunk = this.pendingChunks[0];
+          const ok = this.push(chunk);
+          this.pendingChunks.shift();
+          if (!ok) {
+            break;
+          }
+        }
+      }
+      /** Called by the stdin data handler when keyboard bytes are available. */
+      doPush(chunk) {
+        if (chunk.length === 0) return;
+        const ok = this.push(chunk);
+        if (ok) {
+          if (this.pendingChunks.length > 0) {
+            this.flushPending();
+          }
+        } else {
+          if (this.pendingChunks.length >= 100) {
+            this.pendingChunks.shift();
+          }
+          this.pendingChunks.push(chunk);
+        }
+      }
+      /** Called on shutdown so the stream closes cleanly. */
+      doEnd() {
+        this.pendingChunks = [];
+        this.push(null);
+      }
+    }
+    const keyboardStream = new KeyboardReadable();
+    const p = keyboardStream;
     p.isTTY = true;
     p.setRawMode = (mode) => {
       try {
@@ -8578,10 +8700,13 @@ async function runTui(opts) {
         }
       }
       const rest = stripSgrMouse(chunk);
-      if (rest.length > 0) proxy.write(rest);
+      keyboardStream.doPush(rest);
     };
     stdin.on("data", onData);
-    detachMouse = () => stdin.off("data", onData);
+    detachMouse = () => {
+      stdin.off("data", onData);
+      keyboardStream.doEnd();
+    };
     inkStdin = p;
   }
   const subscribeMouse = useMouse ? (fn) => {
@@ -8700,6 +8825,7 @@ async function runTui(opts) {
           projectRoot: opts.projectRoot,
           getSettings: opts.getSettings,
           saveSettings: opts.saveSettings,
+          predictNext: opts.predictNext,
           mouse: useMouse,
           subscribeMouse,
           // Managed viewport (in-app scroll + collapsibility) follows