@wrongstack/tui 0.4.1 → 0.5.0

package/dist/index.d.ts

@@ -26,6 +26,10 @@ interface RunTuiOptions {
     queueStore?: QueueStore;
     /** Surfaces the "⚠ YOLO" chip in the status bar. */
     yolo?: boolean;
+    /** Query live YOLO state from the permission policy. */
+    getYolo?: () => boolean;
+    /** Query the live autonomy mode. */
+    getAutonomy?: () => 'off' | 'suggest' | 'auto';
     /** Renders in the startup banner. Read from the CLI's package.json. */
     appVersion?: string;
     /** Provider id for the startup banner ("openai", "anthropic", ...). */
@@ -110,6 +114,16 @@ interface RunTuiOptions {
      * Ignored when `initialGoal` is also set.
      */
     initialAsk?: string;
+    /**
+     * SDD session context getter. When an SDD session is active, returns
+     * the AI prompt context to inject into user messages.
+     */
+    getSDDContext?: () => string | null;
+    /**
+     * Process AI output for SDD auto-detection (spec, tasks, plan).
+     * Returns displayable status messages.
+     */
+    onSDDOutput?: (output: string) => Promise<string[]>;
 }
 declare function runTui(opts: RunTuiOptions): Promise<number>;
 

package/dist/index.js

@@ -1,5 +1,5 @@
 import { render, useApp, Box, useStdout, Static, Text, useInput, useStdin } from 'ink';
-import React4, { useState, useReducer, useRef, useEffect, useMemo } from 'react';
+import React2, { useState, useReducer, useRef, useEffect, useMemo } from 'react';
 import * as fs2 from 'fs/promises';
 import * as path2 from 'path';
 import { InputBuilder, formatTodosList, buildChildEnv } from '@wrongstack/core';
@@ -39,7 +39,7 @@ function ConfirmPrompt({
   suggestedPattern,
   onDecision
 }) {
-  React4.useEffect(() => {
+  React2.useEffect(() => {
     process.stdout.write("\x07");
   }, []);
   useInput((input2, key) => {
@@ -244,6 +244,15 @@ function FleetPanel({ entries, totalCost, roster }) {
               "msg: ",
               message
             ] }) }, `${entry.id}-msg-${index}-${message}`)),
+            entry.budgetWarning ? /* @__PURE__ */ jsx(Box, { paddingLeft: 2, children: /* @__PURE__ */ jsxs(Text, { color: "yellow", children: [
+              "\u26A1 hitting ",
+              entry.budgetWarning.kind,
+              " limit (",
+              entry.budgetWarning.used,
+              "/",
+              entry.budgetWarning.limit,
+              ") \u2014 extending"
+            ] }) }) : null,
             entry.status === "running" && entry.streamingText ? /* @__PURE__ */ jsx(Box, { paddingLeft: 2, children: /* @__PURE__ */ jsxs(Text, { dimColor: true, children: [
               ">",
               " ",
@@ -1563,6 +1572,7 @@ function StatusBar({
   hint,
   queueCount = 0,
   yolo = false,
+  autonomy,
   elapsedMs,
   todos,
   plan,
@@ -1578,7 +1588,7 @@ function StatusBar({
   const cache2 = tokenCounter?.cacheStats();
   const stateColor = state === "idle" ? "cyan" : state === "aborting" ? "yellow" : "green";
   const stateLabel = state === "idle" ? "idle" : state === "aborting" ? "aborting\u2026" : "thinking\u2026";
-  const hasSecondLine = yolo || elapsedMs !== void 0 || git !== null && git !== void 0 || projectName !== void 0 && projectName.length > 0;
+  const hasSecondLine = yolo || autonomy && autonomy !== "off" || elapsedMs !== void 0 || git !== null && git !== void 0 || projectName !== void 0 && projectName.length > 0;
   const fleetHasActivity = fleet && (fleet.running > 0 || fleet.idle > 0 || fleet.pending > 0 || fleet.completed > 0) || subagentCount > 0;
   const hasThirdLine = todos && (todos.pending > 0 || todos.inProgress > 0 || todos.completed > 0) || plan && (plan.open > 0 || plan.inProgress > 0 || plan.done > 0) || fleetHasActivity;
   return /* @__PURE__ */ jsxs(
@@ -1643,6 +1653,13 @@ function StatusBar({
         ] }),
         hasSecondLine ? /* @__PURE__ */ jsxs(Box, { flexDirection: "row", gap: 2, children: [
           yolo ? /* @__PURE__ */ jsx(Text, { color: "red", bold: true, children: "\u26A0 YOLO" }) : null,
+          autonomy && autonomy !== "off" ? /* @__PURE__ */ jsxs(Fragment, { children: [
+            yolo ? /* @__PURE__ */ jsx(Text, { dimColor: true, children: "\u2502" }) : null,
+            /* @__PURE__ */ jsxs(Text, { color: autonomy === "auto" ? "yellow" : "cyan", bold: true, children: [
+              "\u221E ",
+              autonomy.toUpperCase()
+            ] })
+          ] }) : null,
           elapsedMs !== void 0 ? /* @__PURE__ */ jsxs(Fragment, { children: [
             yolo ? /* @__PURE__ */ jsx(Text, { dimColor: true, children: "\u2502" }) : null,
             /* @__PURE__ */ jsxs(Text, { dimColor: true, children: [
@@ -2341,6 +2358,8 @@ function reducer(state, action) {
             ...cur,
             status: "running",
             streamingText: "",
+            budgetWarning: void 0,
+            // clear on restart
             startedAt: Date.now()
           }
         }
@@ -2423,6 +2442,23 @@ function reducer(state, action) {
             toolCalls: action.toolCalls,
             streamingText: "",
             currentTool: void 0,
+            budgetWarning: void 0,
+            // clear on done/restart
+            lastEventAt: Date.now()
+          }
+        }
+      };
+    }
+    case "fleetBudgetWarning": {
+      const cur = state.fleet[action.id];
+      if (!cur) return state;
+      return {
+        ...state,
+        fleet: {
+          ...state.fleet,
+          [action.id]: {
+            ...cur,
+            budgetWarning: { kind: action.kind, used: action.used, limit: action.limit, at: Date.now() },
             lastEventAt: Date.now()
           }
         }
@@ -2552,6 +2588,10 @@ function App({
   banner = true,
   queueStore,
   yolo = false,
+  getYolo,
+  getAutonomy,
+  getSDDContext,
+  onSDDOutput,
   appVersion,
   provider,
   family,
@@ -2570,6 +2610,8 @@ function App({
   const { exit } = useApp();
   const [liveModel, setLiveModel] = useState(model);
   const [liveProvider, setLiveProvider] = useState(provider ?? "agent");
+  const [yoloLive, setYoloLive] = useState(yolo);
+  const [autonomyLive, setAutonomyLive] = useState(getAutonomy?.() ?? "off");
   const [state, dispatch] = useReducer(reducer, {
     entries: banner ? [
       {
@@ -2622,7 +2664,7 @@ function App({
   const inputGateRef = useRef(false);
   const lastEnterAtRef = useRef(0);
   const projectRoot = agent.ctx.projectRoot;
-  const projectName = React4.useMemo(() => {
+  const projectName = React2.useMemo(() => {
     const base = path2.basename(projectRoot);
     return base && base !== path2.sep ? base : void 0;
   }, [projectRoot]);
@@ -2642,13 +2684,13 @@ function App({
     dispatch({ type: "clearInput" });
   };
   const startedAtRef = useRef(Date.now());
-  const [nowTick, setNowTick] = React4.useState(Date.now());
+  const [nowTick, setNowTick] = React2.useState(Date.now());
   useEffect(() => {
     const t = setInterval(() => setNowTick(Date.now()), 1e3);
     return () => clearInterval(t);
   }, []);
   const elapsedMs = nowTick - startedAtRef.current;
-  const [gitInfo, setGitInfo] = React4.useState(null);
+  const [gitInfo, setGitInfo] = React2.useState(null);
   useEffect(() => {
     let cancelled = false;
     const refresh = () => {
@@ -2663,12 +2705,13 @@ function App({
       clearInterval(t);
     };
   }, [agent.ctx.cwd]);
-  const lastInputTokens = tokenCounter?.total().input ?? 0;
+  (tokenCounter?.total().input ?? 0) + (tokenCounter?.total().cacheRead ?? 0) + (tokenCounter?.total().cacheWrite ?? 0);
   const maxContext = effectiveMaxContext ?? agent.ctx.provider.capabilities.maxContext;
+  const currentContextTokens = (tokenCounter?.currentRequestTokens()?.input ?? 0) + (tokenCounter?.currentRequestTokens()?.cacheRead ?? 0);
   const contextWindow = useMemo(() => {
     void state.contextChipVersion;
-    return lastInputTokens > 0 && maxContext > 0 ? { used: lastInputTokens, max: maxContext } : void 0;
-  }, [lastInputTokens, maxContext, state.contextChipVersion]);
+    return currentContextTokens > 0 && maxContext > 0 ? { used: currentContextTokens, max: maxContext } : void 0;
+  }, [currentContextTokens, maxContext, state.contextChipVersion]);
   const todos = useMemo(() => {
     const counts = { pending: 0, inProgress: 0, completed: 0 };
     for (const t of agent.ctx.todos) {
@@ -3264,6 +3307,20 @@ function App({
         }
       });
     });
+    const offBudgetWarning = events.on("subagent.budget_warning", (e) => {
+      const lbl = labelFor(e.subagentId);
+      dispatch({ type: "fleetBudgetWarning", id: e.subagentId, kind: e.kind, used: e.used, limit: e.limit });
+      dispatch({
+        type: "addEntry",
+        entry: {
+          kind: "subagent",
+          agentLabel: lbl.label,
+          agentColor: lbl.color,
+          icon: "\u26A1",
+          text: `hitting ${e.kind} limit (${e.used}/${e.limit}) \u2014 extending`
+        }
+      });
+    });
     const offTool = events.on("subagent.tool_executed", (e) => {
       if (director) return;
       dispatch({
@@ -3280,6 +3337,7 @@ function App({
       offSpawned();
       offStarted();
       offCompleted();
+      offBudgetWarning();
       offTool();
     };
   }, [events, director]);
@@ -3513,7 +3571,7 @@ function App({
     return () => {
       process.off("SIGINT", onSigint);
     };
-  }, [exit, onExit, director]);
+  }, [director]);
   const handleKey = async (input, key) => {
     if (state.status === "aborting" && state.interrupts === 0) return;
     if (state.confirmQueue.length > 0) return;
@@ -3840,6 +3898,15 @@ function App({
           entry: { kind: "warn", text: `Hit max iterations (${result.iterations}).` }
         });
       }
+      if (result.status === "done" && result.finalText && onSDDOutput) {
+        try {
+          const sddMessages = await onSDDOutput(result.finalText);
+          for (const msg of sddMessages) {
+            dispatch({ type: "addEntry", entry: { kind: "info", text: msg } });
+          }
+        } catch {
+        }
+      }
       if (tokenCounter && before) {
         const after = tokenCounter.total();
         const costAfter = tokenCounter.estimateCost().total;
@@ -3895,6 +3962,14 @@ function App({
         if (ctxModel && ctxModel !== liveModel) setLiveModel(ctxModel);
         const ctxProviderId = agent.ctx.provider?.id;
         if (ctxProviderId && ctxProviderId !== liveProvider) setLiveProvider(ctxProviderId);
+        if (getYolo) {
+          const currentYolo = getYolo();
+          if (currentYolo !== yoloLive) setYoloLive(currentYolo);
+        }
+        if (getAutonomy) {
+          const currentAutonomy = getAutonomy();
+          if (currentAutonomy !== autonomyLive) setAutonomyLive(currentAutonomy);
+        }
         if (res?.exit) {
           exit();
           onExit(0);
@@ -3926,6 +4001,15 @@ function App({
     const builder = builderRef.current;
     if (!builder) return;
     const steering = state.steeringPending;
+    const sddContext = getSDDContext?.();
+    if (sddContext && trimmed) {
+      builder.appendText(`[SDD SESSION ACTIVE]
+${sddContext}
+
+---
+User message:
+`);
+    }
     if (trimmed) {
       const toAppend = steering ? buildSteeringPreamble(state.steerSnapshot, trimmed) : trimmed;
       builder.appendText(toAppend);
@@ -4056,7 +4140,8 @@ function App({
         tokenCounter,
         hint: renderRunningTools(state.runningTools) || state.hint,
         queueCount: state.queue.length,
-        yolo,
+        yolo: yoloLive,
+        autonomy: autonomyLive,
         elapsedMs,
         todos,
         plan: planCounts ?? void 0,
@@ -4179,7 +4264,7 @@ async function runTui(opts) {
     let instance;
     try {
       instance = render(
-        React4.createElement(App, {
+        React2.createElement(App, {
           agent: opts.agent,
           slashRegistry: opts.slashRegistry,
           attachments: opts.attachments,
@@ -4191,6 +4276,8 @@ async function runTui(opts) {
           banner: opts.banner ?? true,
           queueStore: opts.queueStore,
           yolo: opts.yolo,
+          getYolo: opts.getYolo,
+          getAutonomy: opts.getAutonomy,
           appVersion: opts.appVersion,
           provider: opts.provider,
           family: opts.family,
@@ -4204,7 +4291,9 @@ async function runTui(opts) {
           onClearHistory: opts.onClearHistory ? (dispatch) => opts.onClearHistory(dispatch) : void 0,
           fleetStreamController: opts.fleetStreamController,
           initialGoal: opts.initialGoal,
-          initialAsk: opts.initialAsk
+          initialAsk: opts.initialAsk,
+          getSDDContext: opts.getSDDContext,
+          onSDDOutput: opts.onSDDOutput
         }),
         { exitOnCtrlC: false }
       );