wave-code 0.2.0 → 0.4.0

package/src/managers/InputManager.ts

@@ -1,9 +1,9 @@
 import { FileItem } from "../components/FileSelector.js";
 import {
   searchFiles as searchFilesUtil,
-  deleteBashCommandFromHistory,
   PermissionMode,
   Logger,
+  PromptHistoryManager,
 } from "wave-agent-sdk";
 import { readClipboardImage } from "../utils/clipboard.js";
 import type { Key } from "ink";
@@ -28,16 +28,14 @@ export interface InputManagerCallbacks {
     query: string,
     position: number,
   ) => void;
-  onBashHistorySelectorStateChange?: (
-    show: boolean,
-    query: string,
-    position: number,
-  ) => void;
+  onHistorySearchStateChange?: (show: boolean, query: string) => void;
   onMemoryTypeSelectorStateChange?: (show: boolean, message: string) => void;
   onShowBashManager?: () => void;
   onShowMcpManager?: () => void;
+  onShowRewindManager?: () => void;
   onBashManagerStateChange?: (show: boolean) => void;
   onMcpManagerStateChange?: (show: boolean) => void;
+  onRewindManagerStateChange?: (show: boolean) => void;
   onImagesStateChange?: (images: AttachedImage[]) => void;
   onSendMessage?: (
     content: string,
@@ -68,10 +66,9 @@ export class InputManager {
   private slashPosition: number = -1;
   private commandSearchQuery: string = "";
 
-  // Bash history selector state
-  private showBashHistorySelector: boolean = false;
-  private exclamationPosition: number = -1;
-  private bashHistorySearchQuery: string = "";
+  // History search state
+  private showHistorySearch: boolean = false;
+  private historySearchQuery: string = "";
 
   // Memory type selector state
   private showMemoryTypeSelector: boolean = false;
@@ -99,6 +96,7 @@ export class InputManager {
   // Additional UI state
   private showBashManager: boolean = false;
   private showMcpManager: boolean = false;
+  private showRewindManager: boolean = false;
 
   // Permission mode state
   private permissionMode: PermissionMode = "default";
@@ -369,6 +367,12 @@ export class InputManager {
           } else if (command === "mcp" && this.callbacks.onShowMcpManager) {
             this.callbacks.onShowMcpManager();
             commandExecuted = true;
+          } else if (
+            command === "rewind" &&
+            this.callbacks.onShowRewindManager
+          ) {
+            this.callbacks.onShowRewindManager();
+            commandExecuted = true;
           }
         }
       })();
@@ -434,105 +438,6 @@ export class InputManager {
     return false;
   }
 
-  // Bash history selector methods
-  activateBashHistorySelector(position: number): void {
-    this.showBashHistorySelector = true;
-    this.exclamationPosition = position;
-    this.bashHistorySearchQuery = "";
-
-    this.callbacks.onBashHistorySelectorStateChange?.(true, "", position);
-  }
-
-  updateBashHistorySearchQuery(query: string): void {
-    this.bashHistorySearchQuery = query;
-    this.callbacks.onBashHistorySelectorStateChange?.(
-      this.showBashHistorySelector,
-      query,
-      this.exclamationPosition,
-    );
-  }
-
-  handleBashHistorySelect(command: string): {
-    newInput: string;
-    newCursorPosition: number;
-  } {
-    if (this.exclamationPosition >= 0) {
-      const beforeExclamation = this.inputText.substring(
-        0,
-        this.exclamationPosition,
-      );
-      const afterQuery = this.inputText.substring(this.cursorPosition);
-      const newInput = beforeExclamation + `!${command}` + afterQuery;
-      const newCursorPosition = beforeExclamation.length + command.length + 1;
-
-      this.inputText = newInput;
-      this.cursorPosition = newCursorPosition;
-
-      this.handleCancelBashHistorySelect();
-
-      // Set flag to prevent handleInput from processing the same Enter key
-      this.selectorJustUsed = true;
-      setTimeout(() => {
-        this.selectorJustUsed = false;
-      }, 0);
-
-      this.callbacks.onInputTextChange?.(newInput);
-      this.callbacks.onCursorPositionChange?.(newCursorPosition);
-
-      return { newInput, newCursorPosition };
-    }
-    return { newInput: this.inputText, newCursorPosition: this.cursorPosition };
-  }
-
-  handleCancelBashHistorySelect(): void {
-    this.showBashHistorySelector = false;
-    this.exclamationPosition = -1;
-    this.bashHistorySearchQuery = "";
-
-    this.callbacks.onBashHistorySelectorStateChange?.(false, "", -1);
-  }
-
-  handleBashHistoryExecute(command: string): string {
-    this.showBashHistorySelector = false;
-    this.exclamationPosition = -1;
-    this.bashHistorySearchQuery = "";
-
-    this.callbacks.onBashHistorySelectorStateChange?.(false, "", -1);
-
-    return command; // Return command to execute
-  }
-
-  handleBashHistoryExecuteAndSend(command: string): void {
-    const commandToExecute = this.handleBashHistoryExecute(command);
-    // Clear input box and execute command, ensure command starts with !
-    const bashCommand = commandToExecute.startsWith("!")
-      ? commandToExecute
-      : `!${commandToExecute}`;
-    this.clearInput();
-    this.callbacks.onSendMessage?.(bashCommand);
-  }
-
-  handleBashHistoryDelete(command: string, workdir?: string): void {
-    deleteBashCommandFromHistory(command, workdir);
-    // Trigger a refresh of the selector state to force re-render of BashHistorySelector
-    this.callbacks.onBashHistorySelectorStateChange?.(
-      this.showBashHistorySelector,
-      this.bashHistorySearchQuery,
-      this.exclamationPosition,
-    );
-  }
-
-  checkForExclamationDeletion(cursorPosition: number): boolean {
-    if (
-      this.showBashHistorySelector &&
-      cursorPosition <= this.exclamationPosition
-    ) {
-      this.handleCancelBashHistorySelect();
-      return true;
-    }
-    return false;
-  }
-
   // Memory type selector methods
   activateMemoryTypeSelector(message: string): void {
     this.showMemoryTypeSelector = true;
@@ -631,10 +536,6 @@ export class InputManager {
     return this.showCommandSelector;
   }
 
-  isBashHistorySelectorActive(): boolean {
-    return this.showBashHistorySelector;
-  }
-
   isMemoryTypeSelectorActive(): boolean {
     return this.showMemoryTypeSelector;
   }
@@ -656,14 +557,6 @@ export class InputManager {
     };
   }
 
-  getBashHistorySelectorState() {
-    return {
-      show: this.showBashHistorySelector,
-      query: this.bashHistorySearchQuery,
-      position: this.exclamationPosition,
-    };
-  }
-
   getMemoryTypeSelectorState() {
     return {
       show: this.showMemoryTypeSelector,
@@ -686,11 +579,6 @@ export class InputManager {
       const queryEnd = cursorPosition;
       const newQuery = inputText.substring(queryStart, queryEnd);
       this.updateCommandSearchQuery(newQuery);
-    } else if (this.showBashHistorySelector && this.exclamationPosition >= 0) {
-      const queryStart = this.exclamationPosition + 1;
-      const queryEnd = cursorPosition;
-      const newQuery = inputText.substring(queryStart, queryEnd);
-      this.updateBashHistorySearchQuery(newQuery);
     }
   }
 
@@ -698,11 +586,14 @@ export class InputManager {
   handleSpecialCharInput(char: string): void {
     if (char === "@") {
       this.activateFileSelector(this.cursorPosition - 1);
-    } else if (char === "/" && !this.showFileSelector) {
+    } else if (
+      char === "/" &&
+      !this.showFileSelector &&
+      this.cursorPosition === 1
+    ) {
       // Don't activate command selector when file selector is active
+      // Only activate command selector if '/' is at the start of input
       this.activateCommandSelector(this.cursorPosition - 1);
-    } else if (char === "!" && this.cursorPosition === 1) {
-      this.activateBashHistorySelector(0);
     } else if (char === "#" && this.cursorPosition === 1) {
       // Memory message detection will be handled in submit
     } else {
@@ -879,6 +770,15 @@ export class InputManager {
     this.callbacks.onMcpManagerStateChange?.(show);
   }
 
+  getShowRewindManager(): boolean {
+    return this.showRewindManager;
+  }
+
+  setShowRewindManager(show: boolean): void {
+    this.showRewindManager = show;
+    this.callbacks.onRewindManagerStateChange?.(show);
+  }
+
   // Permission mode methods
   getPermissionMode(): PermissionMode {
     return this.permissionMode;
@@ -941,6 +841,11 @@ export class InputManager {
       let cleanContent = this.inputText.replace(imageRegex, "").trim();
       cleanContent = this.expandLongTextPlaceholders(cleanContent);
 
+      // Save to prompt history
+      PromptHistoryManager.addEntry(cleanContent).catch((err: unknown) => {
+        this.logger?.error("Failed to save prompt history", err);
+      });
+
       this.callbacks.onSendMessage?.(
         cleanContent,
         referencedImages.length > 0 ? referencedImages : undefined,
@@ -961,7 +866,6 @@ export class InputManager {
           // Check for special character deletion
           this.checkForAtDeletion(newCursorPosition);
           this.checkForSlashDeletion(newCursorPosition);
-          this.checkForExclamationDeletion(newCursorPosition);
 
           // Update search queries using the same logic as character input
           this.updateSearchQueriesForActiveSelectors(
@@ -1004,6 +908,32 @@ export class InputManager {
     return false;
   }
 
+  // History search methods
+  activateHistorySearch(): void {
+    this.showHistorySearch = true;
+    this.historySearchQuery = "";
+    this.callbacks.onHistorySearchStateChange?.(true, "");
+  }
+
+  updateHistorySearchQuery(query: string): void {
+    this.historySearchQuery = query;
+    this.callbacks.onHistorySearchStateChange?.(true, query);
+  }
+
+  handleHistorySearchSelect(prompt: string): void {
+    this.inputText = prompt;
+    this.cursorPosition = prompt.length;
+    this.callbacks.onInputTextChange?.(prompt);
+    this.callbacks.onCursorPositionChange?.(prompt.length);
+    this.handleCancelHistorySearch();
+  }
+
+  handleCancelHistorySearch(): void {
+    this.showHistorySearch = false;
+    this.historySearchQuery = "";
+    this.callbacks.onHistorySearchStateChange?.(false, "");
+  }
+
   // Handle normal input (when no selector is active)
   async handleNormalInput(
     input: string,
@@ -1024,8 +954,6 @@ export class InputManager {
         this.handleCancelFileSelect();
       } else if (this.showCommandSelector) {
         this.handleCancelCommandSelect();
-      } else if (this.showBashHistorySelector) {
-        this.handleCancelBashHistorySelect();
       }
       return true;
     }
@@ -1039,7 +967,6 @@ export class InputManager {
         const newCursorPosition = this.cursorPosition - 1;
         this.checkForAtDeletion(newCursorPosition);
         this.checkForSlashDeletion(newCursorPosition);
-        this.checkForExclamationDeletion(newCursorPosition);
       }
       return true;
     }
@@ -1072,23 +999,19 @@ export class InputManager {
       return true;
     }
 
+    // Handle Ctrl+R for history search
+    if (key.ctrl && input === "r") {
+      this.activateHistorySearch();
+      return true;
+    }
+
     // Handle up/down keys for history navigation (only when no selector is active)
-    if (
-      key.upArrow &&
-      !this.showFileSelector &&
-      !this.showCommandSelector &&
-      !this.showBashHistorySelector
-    ) {
+    if (key.upArrow && !this.showFileSelector && !this.showCommandSelector) {
       this.navigateHistory("up", this.inputText);
       return true;
     }
 
-    if (
-      key.downArrow &&
-      !this.showFileSelector &&
-      !this.showCommandSelector &&
-      !this.showBashHistorySelector
-    ) {
+    if (key.downArrow && !this.showFileSelector && !this.showCommandSelector) {
       this.navigateHistory("down", this.inputText);
       return true;
     }
@@ -1147,19 +1070,40 @@ export class InputManager {
     if (
       this.showFileSelector ||
       this.showCommandSelector ||
-      this.showBashHistorySelector ||
+      this.showHistorySearch ||
       this.showMemoryTypeSelector ||
       this.showBashManager ||
-      this.showMcpManager
+      this.showMcpManager ||
+      this.showRewindManager
     ) {
       if (
         this.showMemoryTypeSelector ||
         this.showBashManager ||
-        this.showMcpManager
+        this.showMcpManager ||
+        this.showRewindManager
       ) {
-        // Memory type selector, bash manager and MCP manager don't need to handle input, handled by component itself
+        // Memory type selector, bash manager, MCP manager and Rewind don't need to handle input, handled by component itself
         return false;
       }
+
+      if (this.showHistorySearch) {
+        if (key.escape) {
+          this.handleCancelHistorySearch();
+          return true;
+        }
+        if (key.backspace || key.delete) {
+          if (this.historySearchQuery.length > 0) {
+            this.updateHistorySearchQuery(this.historySearchQuery.slice(0, -1));
+          }
+          return true;
+        }
+        if (input && !key.ctrl && !key.meta && !key.return && !key.tab) {
+          this.updateHistorySearchQuery(this.historySearchQuery + input);
+          return true;
+        }
+        return true; // Let HistorySearch component handle arrows and Enter
+      }
+
       return this.handleSelectorInput(input, key);
     } else {
       return await this.handleNormalInput(

package/src/plugin-manager-cli.tsx

@@ -0,0 +1,13 @@
+import React from "react";
+import { render } from "ink";
+import { PluginManagerShell } from "./components/PluginManagerShell.js";
+
+/**
+ * Entry point for the Plugin Manager CLI.
+ * Renders the Ink component and handles the lifecycle.
+ */
+export async function startPluginManagerCli(): Promise<boolean> {
+  const { waitUntilExit } = render(<PluginManagerShell />);
+  await waitUntilExit();
+  return true;
+}

package/src/session-selector-cli.tsx

@@ -0,0 +1,37 @@
+import React from "react";
+import { render, Box } from "ink";
+import { listSessions, truncateContent } from "wave-agent-sdk";
+import { SessionSelector } from "./components/SessionSelector.js";
+
+export async function startSessionSelectorCli(): Promise<string | null> {
+  const currentWorkdir = process.cwd();
+  const sessions = await listSessions(currentWorkdir);
+
+  if (sessions.length === 0) {
+    console.log(`No sessions found for workdir: ${currentWorkdir}`);
+    return null;
+  }
+
+  const sessionsWithContent = sessions.map((s) => ({
+    ...s,
+    firstMessage: truncateContent(s.firstMessage || "No content", 80),
+  }));
+
+  return new Promise((resolve) => {
+    const { unmount } = render(
+      <Box padding={1}>
+        <SessionSelector
+          sessions={sessionsWithContent}
+          onSelect={(sessionId) => {
+            unmount();
+            resolve(sessionId);
+          }}
+          onCancel={() => {
+            unmount();
+            resolve(null);
+          }}
+        />
+      </Box>,
+    );
+  });
+}

package/dist/components/BashHistorySelector.d.ts

@@ -1,11 +0,0 @@
-import React from "react";
-export interface BashHistorySelectorProps {
-    searchQuery: string;
-    workdir: string;
-    onSelect: (command: string) => void;
-    onExecute: (command: string) => void;
-    onDelete: (command: string, workdir?: string) => void;
-    onCancel: () => void;
-}
-export declare const BashHistorySelector: React.FC<BashHistorySelectorProps>;
-//# sourceMappingURL=BashHistorySelector.d.ts.map

package/dist/components/BashHistorySelector.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"BashHistorySelector.d.ts","sourceRoot":"","sources":["../../src/components/BashHistorySelector.tsx"],"names":[],"mappings":"AAAA,OAAO,KAA8B,MAAM,OAAO,CAAC;AAKnD,MAAM,WAAW,wBAAwB;IACvC,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACpC,SAAS,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACrC,QAAQ,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,KAAK,IAAI,CAAC;IACtD,QAAQ,EAAE,MAAM,IAAI,CAAC;CACtB;AAED,eAAO,MAAM,mBAAmB,EAAE,KAAK,CAAC,EAAE,CAAC,wBAAwB,CAsKlE,CAAC"}

package/dist/components/BashHistorySelector.js

@@ -1,93 +0,0 @@
-import { jsxs as _jsxs, jsx as _jsx } from "react/jsx-runtime";
-import { useState, useEffect } from "react";
-import { Box, Text, useInput } from "ink";
-import { searchBashHistory } from "wave-agent-sdk";
-import { logger } from "../utils/logger.js";
-export const BashHistorySelector = ({ searchQuery, workdir, onSelect, onExecute, onDelete, onCancel, }) => {
-    const [selectedIndex, setSelectedIndex] = useState(0);
-    const [commands, setCommands] = useState([]);
-    const [refreshCounter, setRefreshCounter] = useState(0);
-    // Search bash history
-    useEffect(() => {
-        const results = searchBashHistory(searchQuery, 10);
-        setCommands(results);
-        setSelectedIndex(0);
-        logger.debug("Bash history search:", {
-            searchQuery,
-            workdir,
-            resultCount: results.length,
-            refreshCounter,
-        });
-    }, [searchQuery, workdir, refreshCounter]);
-    useInput((input, key) => {
-        logger.debug("BashHistorySelector useInput:", {
-            input,
-            key,
-            commandsLength: commands.length,
-            selectedIndex,
-        });
-        if (key.return) {
-            if (commands.length > 0 && selectedIndex < commands.length) {
-                const selectedCommand = commands[selectedIndex];
-                onExecute(selectedCommand.command);
-            }
-            else if (commands.length === 0 && searchQuery.trim()) {
-                // When no history records match, execute the search query as a new command
-                onExecute(searchQuery.trim());
-            }
-            return;
-        }
-        if (key.tab) {
-            if (commands.length > 0 && selectedIndex < commands.length) {
-                const selectedCommand = commands[selectedIndex];
-                onSelect(selectedCommand.command);
-            }
-            else if (commands.length === 0 && searchQuery.trim()) {
-                // When no history records match, insert the search query
-                onSelect(searchQuery.trim());
-            }
-            return;
-        }
-        if (key.escape) {
-            onCancel();
-            return;
-        }
-        if (key.ctrl && input === "d") {
-            if (commands.length > 0 && selectedIndex < commands.length) {
-                const selectedCommand = commands[selectedIndex];
-                onDelete(selectedCommand.command, selectedCommand.workdir);
-                setRefreshCounter((prev) => prev + 1);
-            }
-            return;
-        }
-        if (key.upArrow) {
-            setSelectedIndex(Math.max(0, selectedIndex - 1));
-            return;
-        }
-        if (key.downArrow) {
-            setSelectedIndex(Math.min(commands.length - 1, selectedIndex + 1));
-            return;
-        }
-    });
-    if (commands.length === 0) {
-        return (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: "yellow", borderBottom: false, borderLeft: false, borderRight: false, paddingTop: 1, children: [_jsxs(Text, { color: "yellow", children: ["No bash history found ", searchQuery && `for "${searchQuery}"`] }), searchQuery.trim() && (_jsxs(Text, { color: "green", children: ["Press Enter to execute: ", searchQuery] })), searchQuery.trim() && (_jsxs(Text, { color: "blue", children: ["Press Tab to insert: ", searchQuery] })), _jsx(Text, { dimColor: true, children: "Press Escape to cancel" })] }));
-    }
-    const formatTimestamp = (timestamp) => {
-        const date = new Date(timestamp);
-        const now = new Date();
-        const diffMs = now.getTime() - date.getTime();
-        const diffHours = Math.floor(diffMs / (1000 * 60 * 60));
-        const diffDays = Math.floor(diffHours / 24);
-        if (diffDays > 0) {
-            return `${diffDays}d ago`;
-        }
-        else if (diffHours > 0) {
-            return `${diffHours}h ago`;
-        }
-        else {
-            const diffMinutes = Math.floor(diffMs / (1000 * 60));
-            return diffMinutes > 0 ? `${diffMinutes}m ago` : "just now";
-        }
-    };
-    return (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: "blue", borderBottom: false, borderLeft: false, borderRight: false, paddingTop: 1, gap: 1, children: [_jsx(Box, { children: _jsxs(Text, { color: "blue", bold: true, children: ["Bash History ", searchQuery && `(filtering: "${searchQuery}")`] }) }), commands.map((cmd, index) => (_jsxs(Box, { flexDirection: "column", children: [_jsx(Text, { color: index === selectedIndex ? "black" : "white", backgroundColor: index === selectedIndex ? "blue" : undefined, children: cmd.command }), index === selectedIndex && (_jsx(Box, { marginLeft: 4, flexDirection: "column", children: _jsxs(Text, { color: "gray", dimColor: true, children: [formatTimestamp(cmd.timestamp), cmd.workdir !== workdir && ` • in ${cmd.workdir}`] }) }))] }, index))), _jsx(Box, { children: _jsx(Text, { dimColor: true, children: "Use \u2191\u2193 to navigate, Enter to execute, Tab to insert, Ctrl+d to remove, Escape to cancel" }) })] }));
-};

package/dist/hooks/usePagination.d.ts

@@ -1,20 +0,0 @@
-import type { Message } from "wave-agent-sdk";
-interface PaginationInfo {
-    currentPage: number;
-    totalPages: number;
-    startIndex: number;
-    endIndex: number;
-    messagesPerPage: number;
-}
-export declare const usePagination: (messages: Message[]) => {
-    displayInfo: PaginationInfo;
-    manualPage: number | null;
-    setManualPage: import("react").Dispatch<import("react").SetStateAction<number | null>>;
-    goToPage: (page: number | null) => void;
-    goToPrevPage: () => void;
-    goToNextPage: () => void;
-    goToFirstPage: () => void;
-    goToLastPage: () => void;
-};
-export {};
-//# sourceMappingURL=usePagination.d.ts.map

package/dist/hooks/usePagination.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"usePagination.d.ts","sourceRoot":"","sources":["../../src/hooks/usePagination.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAG9C,UAAU,cAAc;IACtB,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,eAAe,EAAE,MAAM,CAAC;CACzB;AAED,eAAO,MAAM,aAAa,GAAI,UAAU,OAAO,EAAE;;;;qBAwIvB,MAAM,GAAG,IAAI;;;;;CAqDtC,CAAC"}