@wrongstack/tui 0.4.1 → 0.5.2

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,22 @@ interface RunTuiOptions {
      * Ignored when `initialGoal` is also set.
      */
     initialAsk?: string;
+    /**
+     * Directory containing session JSONL files. Required for rewind
+     * functionality. When provided the TUI can list checkpoints and
+     * trigger a rewind via `/rewind` or Ctrl+R.
+     */
+    sessionsDir?: 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,8 +1,8 @@
 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';
+import { InputBuilder, DefaultSessionRewinder, formatTodosList, buildChildEnv } from '@wrongstack/core';
 import { routeImagesForModel } from '@wrongstack/runtime/vision';
 import { readClipboardImage } from '@wrongstack/runtime/clipboard';
 import { jsxs, jsx, Fragment } from 'react/jsx-runtime';
@@ -39,7 +39,7 @@ function ConfirmPrompt({
   suggestedPattern,
   onDecision
 }) {
-  React4.useEffect(() => {
+  React2.useEffect(() => {
     process.stdout.write("\x07");
   }, []);
   useInput((input2, key) => {
@@ -94,6 +94,49 @@ function ConfirmPrompt({
     }
   );
 }
+function CheckpointTimeline({
+  checkpoints,
+  selected,
+  onSelect,
+  onConfirm,
+  onClose
+}) {
+  useInput((_, key) => {
+    if (key.escape) {
+      onClose();
+    } else if (key.upArrow) {
+      onSelect(Math.max(0, selected - 1));
+    } else if (key.downArrow) {
+      onSelect(Math.min(checkpoints.length - 1, selected + 1));
+    } else if (key.return) {
+      onConfirm(selected);
+    }
+  });
+  return /* @__PURE__ */ jsxs(Box, { flexDirection: "column", padding: 1, children: [
+    /* @__PURE__ */ jsxs(Box, { marginBottom: 1, children: [
+      /* @__PURE__ */ jsx(Text, { bold: true, color: "cyan", children: "\u27F2 Session Rewind" }),
+      /* @__PURE__ */ jsx(Text, { dimColor: true, children: " \u2014 \u2191/\u2193 navigate \xB7 Enter rewind \xB7 Esc cancel" })
+    ] }),
+    checkpoints.length === 0 ? /* @__PURE__ */ jsx(Text, { dimColor: true, children: "No checkpoints in this session." }) : checkpoints.map((cp, i) => {
+      const isSelected = i === selected;
+      const label = `[${cp.promptIndex}] ${cp.promptPreview}`;
+      return /* @__PURE__ */ jsxs(Box, { children: [
+        /* @__PURE__ */ jsx(Text, { color: isSelected ? "cyan" : void 0, bold: isSelected, children: isSelected ? "\u25B8 " : "  " }),
+        /* @__PURE__ */ jsx(Text, { color: isSelected ? "cyan" : void 0, bold: isSelected, children: label }),
+        /* @__PURE__ */ jsxs(Text, { dimColor: true, children: [
+          " ",
+          new Date(cp.ts).toLocaleTimeString()
+        ] }),
+        cp.fileCount > 0 && /* @__PURE__ */ jsxs(Text, { dimColor: true, children: [
+          " \xB7 ",
+          cp.fileCount,
+          " file",
+          cp.fileCount !== 1 ? "s" : ""
+        ] })
+      ] }, cp.promptIndex);
+    })
+  ] });
+}
 function FilePicker({ query, matches, selected }) {
   if (matches.length === 0) {
     return /* @__PURE__ */ jsx(Box, { flexDirection: "column", paddingX: 1, children: /* @__PURE__ */ jsxs(Text, { dimColor: true, children: [
@@ -244,6 +287,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 +1615,7 @@ function StatusBar({
   hint,
   queueCount = 0,
   yolo = false,
+  autonomy,
   elapsedMs,
   todos,
   plan,
@@ -1578,7 +1631,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 +1696,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 +2401,8 @@ function reducer(state, action) {
             ...cur,
             status: "running",
             streamingText: "",
+            budgetWarning: void 0,
+            // clear on restart
             startedAt: Date.now()
           }
         }
@@ -2423,6 +2485,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()
           }
         }
@@ -2434,6 +2513,34 @@ function reducer(state, action) {
     case "setStreamFleet": {
       return { ...state, streamFleet: action.enabled };
     }
+    case "checkpointReceived": {
+      const existing = state.checkpoints.find((c) => c.promptIndex === action.cp.promptIndex);
+      if (existing) return state;
+      return { ...state, checkpoints: [...state.checkpoints, action.cp] };
+    }
+    case "rewindOverlayOpen": {
+      return {
+        ...state,
+        rewindOverlay: { checkpoints: state.checkpoints, selected: state.checkpoints.length - 1 }
+      };
+    }
+    case "rewindOverlayClose": {
+      return { ...state, rewindOverlay: null };
+    }
+    case "rewindOverlayMove": {
+      if (!state.rewindOverlay) return state;
+      const len = state.rewindOverlay.checkpoints.length;
+      if (len === 0) return { ...state, rewindOverlay: null };
+      const selected = Math.max(0, Math.min(len - 1, state.rewindOverlay.selected + action.delta));
+      return { ...state, rewindOverlay: { ...state.rewindOverlay, selected } };
+    }
+    case "sessionRewound": {
+      return {
+        ...state,
+        checkpoints: state.checkpoints.filter((c) => c.promptIndex <= action.toPromptIndex),
+        rewindOverlay: null
+      };
+    }
   }
 }
 var PASTE_THRESHOLD_CHARS = 200;
@@ -2552,6 +2659,10 @@ function App({
   banner = true,
   queueStore,
   yolo = false,
+  getYolo,
+  getAutonomy,
+  getSDDContext,
+  onSDDOutput,
   appVersion,
   provider,
   family,
@@ -2565,11 +2676,14 @@ function App({
   onClearHistory,
   fleetStreamController,
   initialGoal,
-  initialAsk
+  initialAsk,
+  sessionsDir
 }) {
   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 ? [
       {
@@ -2612,7 +2726,9 @@ function App({
     contextChipVersion: 0,
     fleet: {},
     fleetCost: 0,
-    streamFleet: true
+    streamFleet: true,
+    checkpoints: [],
+    rewindOverlay: null
   });
   const builderRef = useRef(null);
   if (builderRef.current === null) {
@@ -2622,7 +2738,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]);
@@ -2633,6 +2749,13 @@ function App({
   stateRef.current = state;
   const draftRef = useRef({ buffer: state.buffer, cursor: state.cursor });
   draftRef.current = { buffer: state.buffer, cursor: state.cursor };
+  const handleRewindTo = React2.useCallback(async (checkpointIndex) => {
+    const sessionId = agent.ctx.session.id;
+    if (!sessionId) return;
+    const rewinder = new DefaultSessionRewinder(sessionsDir ?? "");
+    await rewinder.rewindToCheckpoint(sessionId, checkpointIndex);
+    await agent.ctx.session.truncateToCheckpoint(checkpointIndex);
+  }, [agent.ctx.session, sessionsDir]);
   const setDraft = (buffer, cursor) => {
     draftRef.current = { buffer, cursor };
     dispatch({ type: "setBuffer", buffer, cursor });
@@ -2642,13 +2765,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 +2786,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) {
@@ -3028,6 +3152,39 @@ function App({
       slashRegistry.unregister("steer");
     };
   }, [slashRegistry, director]);
+  useEffect(() => {
+    const cmd = {
+      name: "rewind",
+      description: "Open checkpoint timeline to rewind session: /rewind [checkpoint-index]",
+      help: [
+        "Usage: /rewind [checkpoint-index]",
+        "",
+        "Opens a checkpoint timeline. Use \u2191/\u2193 to navigate, Enter to rewind,",
+        "Esc to cancel. The session is reverted to the selected checkpoint",
+        "and conversation history is truncated \u2014 LLM continues fresh.",
+        "",
+        "If a checkpoint index is provided the timeline is skipped and",
+        "rewind happens immediately."
+      ].join("\n"),
+      async run(args) {
+        const idx = Number.parseInt(args.trim(), 10);
+        if (!Number.isNaN(idx) && idx >= 0) {
+          handleRewindTo(idx);
+          return {};
+        }
+        const s = stateRef.current;
+        if (s.checkpoints.length === 0) {
+          return { message: "No checkpoints in this session yet." };
+        }
+        dispatch({ type: "rewindOverlayOpen" });
+        return {};
+      }
+    };
+    slashRegistry.register(cmd);
+    return () => {
+      slashRegistry.unregister("rewind");
+    };
+  }, [slashRegistry, handleRewindTo]);
   useEffect(() => {
     const cmd = {
       name: "goal",
@@ -3264,6 +3421,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,9 +3451,34 @@ function App({
       offSpawned();
       offStarted();
       offCompleted();
+      offBudgetWarning();
       offTool();
     };
   }, [events, director]);
+  useEffect(() => {
+    const offCheckpoint = events.on("checkpoint.written", (e) => {
+      dispatch({
+        type: "checkpointReceived",
+        cp: {
+          promptIndex: e.promptIndex,
+          promptPreview: e.promptPreview,
+          ts: e.ts,
+          fileCount: e.fileCount
+        }
+      });
+    });
+    const offRewound = events.on("session.rewound", (_e) => {
+      dispatch({ type: "sessionRewound", toPromptIndex: 0 });
+      dispatch({ type: "clearHistory" });
+      if (onClearHistory) {
+        onClearHistory(dispatch);
+      }
+    });
+    return () => {
+      offCheckpoint();
+      offRewound();
+    };
+  }, [events, onClearHistory]);
   useEffect(() => {
     if (!fleetStreamController) return;
     fleetStreamController.enabled = state.streamFleet;
@@ -3513,7 +3709,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 +4036,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 +4100,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 +4139,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);
@@ -4030,6 +4252,16 @@ function App({
         hint: state.modelPicker.hint
       }
     ) : null,
+    state.rewindOverlay ? /* @__PURE__ */ jsx(
+      CheckpointTimeline,
+      {
+        checkpoints: state.rewindOverlay.checkpoints,
+        selected: state.rewindOverlay.selected,
+        onSelect: (i) => dispatch({ type: "rewindOverlayMove", delta: i - state.rewindOverlay.selected }),
+        onConfirm: (i) => handleRewindTo(state.rewindOverlay.checkpoints[i].promptIndex),
+        onClose: () => dispatch({ type: "rewindOverlayClose" })
+      }
+    ) : null,
     state.confirmQueue.length > 0 && (() => {
       const head = state.confirmQueue[0];
       let resolved = false;
@@ -4056,7 +4288,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 +4412,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 +4424,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 +4439,10 @@ 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,
+          sessionsDir: opts.sessionsDir
         }),
         { exitOnCtrlC: false }
       );