@wrongstack/tui 0.5.0 → 0.5.3

package/dist/index.d.ts

@@ -114,6 +114,12 @@ 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.

package/dist/index.js

@@ -2,7 +2,7 @@ import { render, useApp, Box, useStdout, Static, Text, useInput, useStdin } from
 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';
@@ -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: [
@@ -2470,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;
@@ -2605,7 +2676,8 @@ function App({
   onClearHistory,
   fleetStreamController,
   initialGoal,
-  initialAsk
+  initialAsk,
+  sessionsDir
 }) {
   const { exit } = useApp();
   const [liveModel, setLiveModel] = useState(model);
@@ -2654,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) {
@@ -2675,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 });
@@ -3071,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",
@@ -3341,6 +3455,30 @@ function App({
       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;
@@ -4114,6 +4252,16 @@ User message:
         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;
@@ -4293,7 +4441,8 @@ async function runTui(opts) {
           initialGoal: opts.initialGoal,
           initialAsk: opts.initialAsk,
           getSDDContext: opts.getSDDContext,
-          onSDDOutput: opts.onSDDOutput
+          onSDDOutput: opts.onSDDOutput,
+          sessionsDir: opts.sessionsDir
         }),
         { exitOnCtrlC: false }
       );