wave-code 0.2.0 → 0.4.0

package/src/components/SessionSelector.tsx

@@ -0,0 +1,127 @@
+import React, { useState } from "react";
+import { Box, Text, useInput } from "ink";
+import type { SessionMetadata } from "wave-agent-sdk";
+
+export interface SessionSelectorProps {
+  sessions: (SessionMetadata & { firstMessage?: string })[];
+  onSelect: (sessionId: string) => void;
+  onCancel: () => void;
+}
+
+export const SessionSelector: React.FC<SessionSelectorProps> = ({
+  sessions,
+  onSelect,
+  onCancel,
+}) => {
+  const [selectedIndex, setSelectedIndex] = useState(0);
+
+  useInput((input, key) => {
+    if (key.return) {
+      if (sessions.length > 0 && selectedIndex < sessions.length) {
+        onSelect(sessions[selectedIndex].id);
+      }
+      return;
+    }
+
+    if (key.escape) {
+      onCancel();
+      return;
+    }
+
+    if (key.upArrow) {
+      setSelectedIndex(Math.max(0, selectedIndex - 1));
+      return;
+    }
+
+    if (key.downArrow) {
+      setSelectedIndex(Math.min(sessions.length - 1, selectedIndex + 1));
+      return;
+    }
+  });
+
+  if (sessions.length === 0) {
+    return (
+      <Box
+        flexDirection="column"
+        borderStyle="single"
+        borderColor="yellow"
+        borderLeft={false}
+        borderRight={false}
+        paddingX={1}
+        width="100%"
+      >
+        <Text color="yellow">No sessions found.</Text>
+        <Text dimColor>Press Escape to cancel</Text>
+      </Box>
+    );
+  }
+
+  const maxDisplay = 10;
+  const startIndex = Math.max(
+    0,
+    Math.min(
+      selectedIndex - Math.floor(maxDisplay / 2),
+      sessions.length - maxDisplay,
+    ),
+  );
+  const displaySessions = sessions.slice(startIndex, startIndex + maxDisplay);
+
+  return (
+    <Box
+      flexDirection="column"
+      paddingX={1}
+      gap={1}
+      borderStyle="single"
+      borderColor="cyan"
+      borderLeft={false}
+      borderRight={false}
+      width="100%"
+    >
+      <Box>
+        <Text color="cyan" bold>
+          Select a session to resume
+        </Text>
+      </Box>
+
+      <Box flexDirection="column">
+        {displaySessions.map((session, index) => {
+          const actualIndex = startIndex + index;
+          const isSelected = actualIndex === selectedIndex;
+          const lastActiveAt = new Date(session.lastActiveAt).toLocaleString();
+
+          return (
+            <Box key={session.id} flexDirection="column" width="100%">
+              <Box width="100%">
+                <Text
+                  color={isSelected ? "black" : "white"}
+                  backgroundColor={isSelected ? "cyan" : undefined}
+                >
+                  {isSelected ? "▶ " : "  "}
+                  {session.id} | {lastActiveAt} | {session.latestTotalTokens}{" "}
+                  tokens
+                </Text>
+              </Box>
+              <Box marginLeft={4} width="100%">
+                <Text dimColor italic>
+                  {session.firstMessage}
+                </Text>
+              </Box>
+            </Box>
+          );
+        })}
+      </Box>
+
+      {sessions.length > maxDisplay && (
+        <Box>
+          <Text dimColor>
+            ... showing {displaySessions.length} of {sessions.length} sessions
+          </Text>
+        </Box>
+      )}
+
+      <Box>
+        <Text dimColor>↑↓ navigate • Enter to select • Esc to cancel</Text>
+      </Box>
+    </Box>
+  );
+};

package/src/components/SubagentBlock.tsx

@@ -2,6 +2,7 @@ import React from "react";
 import { Box, Text } from "ink";
 import type { SubagentBlock as SubagentBlockType } from "wave-agent-sdk";
 import { useChat } from "../contexts/useChat.js";
+import { Markdown } from "./Markdown.js";
 
 interface SubagentBlockProps {
   block: SubagentBlockType;
@@ -59,6 +60,26 @@ export const SubagentBlock: React.FC<SubagentBlockProps> = ({ block }) => {
 
   const { tools: lastTwoTools, totalToolCount } = getLastTwoTools();
 
+  // Get the last text message content if completed
+  const getLastTextMessage = () => {
+    if (block.status !== "completed") return null;
+
+    for (let i = messages.length - 1; i >= 0; i--) {
+      const message = messages[i];
+      if (message.role === "assistant") {
+        for (let j = message.blocks.length - 1; j >= 0; j--) {
+          const messageBlock = message.blocks[j];
+          if (messageBlock.type === "text" && messageBlock.content) {
+            return messageBlock.content;
+          }
+        }
+      }
+    }
+    return null;
+  };
+
+  const lastTextMessage = getLastTextMessage();
+
   return (
     <Box
       borderRight={false}
@@ -86,8 +107,15 @@ export const SubagentBlock: React.FC<SubagentBlockProps> = ({ block }) => {
         </Box>
       </Box>
 
+      {/* Last Text Message Section */}
+      {lastTextMessage && (
+        <Box marginTop={1}>
+          <Markdown>{lastTextMessage}</Markdown>
+        </Box>
+      )}
+
       {/* Tool Names Section - Vertical List */}
-      {lastTwoTools.length > 0 && (
+      {block.status !== "completed" && lastTwoTools.length > 0 && (
         <Box flexDirection="column" marginTop={1} gap={1}>
           {totalToolCount > 2 && (
             <Text color="gray" dimColor>

package/src/components/ToolResultDisplay.tsx

@@ -135,8 +135,8 @@ export const ToolResultDisplay: React.FC<ToolResultDisplayProps> = ({
         </Box>
       )}
 
-      {/* Diff display - only show after tool execution completes */}
-      {stage === "end" && (
+      {/* Diff display - only show after tool execution completes and was successful */}
+      {stage === "end" && success && (
         <DiffDisplay toolName={name} parameters={parameters} />
       )}
     </Box>

package/src/contexts/PluginManagerContext.ts

@@ -0,0 +1,15 @@
+import { createContext, useContext } from "react";
+import { PluginManagerContextType } from "../components/PluginManagerTypes.js";
+
+export const PluginManagerContext =
+  createContext<PluginManagerContextType | null>(null);
+
+export const usePluginManagerContext = () => {
+  const context = useContext(PluginManagerContext);
+  if (!context) {
+    throw new Error(
+      "usePluginManagerContext must be used within a PluginManagerProvider",
+    );
+  }
+  return context;
+};

package/src/contexts/useChat.tsx

@@ -78,6 +78,9 @@ export interface ChatContextType {
   hideConfirmation: () => void;
   handleConfirmationDecision: (decision: PermissionDecision) => void;
   handleConfirmationCancel: () => void;
+  // Rewind functionality
+  rewindId: number;
+  handleRewindSelect: (index: number) => Promise<void>;
 }
 
 const ChatContext = createContext<ChatContextType | null>(null);
@@ -165,6 +168,9 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     reject: () => void;
   } | null>(null);
 
+  // Rewind state
+  const [rewindId, setRewindId] = useState(0);
+
   const agentRef = useRef<Agent | null>(null);
 
   // Permission confirmation methods with queue support
@@ -229,6 +235,9 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         onPermissionModeChange: (mode) => {
           setPermissionModeState(mode);
         },
+        onSlashCommandsChange: (commands) => {
+          setSlashCommands([...commands]);
+        },
       };
 
       try {
@@ -473,6 +482,21 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     hideConfirmation();
   }, [currentConfirmation, hideConfirmation]);
 
+  const handleRewindSelect = useCallback(async (index: number) => {
+    if (agentRef.current) {
+      try {
+        await agentRef.current.truncateHistory(index);
+
+        // Clear terminal screen after rewind
+        process.stdout.write("\x1Bc", () => {
+          setRewindId((prev) => prev + 1);
+        });
+      } catch (error) {
+        logger.error("Failed to rewind:", error);
+      }
+    }
+  }, []);
+
   // Listen for Ctrl+O hotkey to toggle collapse/expand state and ESC to cancel confirmation
   useInput((input, key) => {
     if (key.ctrl && input === "o") {
@@ -520,6 +544,8 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     hideConfirmation,
     handleConfirmationDecision,
     handleConfirmationCancel,
+    rewindId,
+    handleRewindSelect,
   };
 
   return (

package/src/hooks/useInputManager.ts

@@ -29,10 +29,9 @@ export const useInputManager = (
     query: "",
     position: -1,
   });
-  const [bashHistorySelectorState, setBashHistorySelectorState] = useState({
+  const [historySearchState, setHistorySearchState] = useState({
     show: false,
     query: "",
-    position: -1,
   });
   const [memoryTypeSelectorState, setMemoryTypeSelectorState] = useState({
     show: false,
@@ -40,6 +39,7 @@ export const useInputManager = (
   });
   const [showBashManager, setShowBashManager] = useState(false);
   const [showMcpManager, setShowMcpManager] = useState(false);
+  const [showRewindManager, setShowRewindManager] = useState(false);
   const [permissionMode, setPermissionModeState] =
     useState<PermissionMode>("default");
   const [attachedImages, setAttachedImages] = useState<AttachedImage[]>([]);
@@ -58,8 +58,8 @@ export const useInputManager = (
         onCommandSelectorStateChange: (show, query, position) => {
           setCommandSelectorState({ show, query, position });
         },
-        onBashHistorySelectorStateChange: (show, query, position) => {
-          setBashHistorySelectorState({ show, query, position });
+        onHistorySearchStateChange: (show, query) => {
+          setHistorySearchState({ show, query });
         },
         onMemoryTypeSelectorStateChange: (show, message) => {
           setMemoryTypeSelectorState({ show, message });
@@ -70,6 +70,9 @@ export const useInputManager = (
         onMcpManagerStateChange: (show) => {
           setShowMcpManager(show);
         },
+        onRewindManagerStateChange: (show) => {
+          setShowRewindManager(show);
+        },
         onPermissionModeChange: (mode) => {
           setPermissionModeState(mode);
           callbacks.onPermissionModeChange?.(mode);
@@ -77,6 +80,7 @@ export const useInputManager = (
         onImagesStateChange: setAttachedImages,
         onShowBashManager: () => setShowBashManager(true),
         onShowMcpManager: () => setShowMcpManager(true),
+        onShowRewindManager: () => setShowRewindManager(true),
         ...callbacks,
       });
 
@@ -94,8 +98,8 @@ export const useInputManager = (
         onCommandSelectorStateChange: (show, query, position) => {
           setCommandSelectorState({ show, query, position });
         },
-        onBashHistorySelectorStateChange: (show, query, position) => {
-          setBashHistorySelectorState({ show, query, position });
+        onHistorySearchStateChange: (show, query) => {
+          setHistorySearchState({ show, query });
         },
         onMemoryTypeSelectorStateChange: (show, message) => {
           setMemoryTypeSelectorState({ show, message });
@@ -106,6 +110,9 @@ export const useInputManager = (
         onMcpManagerStateChange: (show) => {
           setShowMcpManager(show);
         },
+        onRewindManagerStateChange: (show) => {
+          setShowRewindManager(show);
+        },
         onPermissionModeChange: (mode) => {
           setPermissionModeState(mode);
           callbacks.onPermissionModeChange?.(mode);
@@ -113,6 +120,7 @@ export const useInputManager = (
         onImagesStateChange: setAttachedImages,
         onShowBashManager: () => setShowBashManager(true),
         onShowMcpManager: () => setShowMcpManager(true),
+        onShowRewindManager: () => setShowRewindManager(true),
         ...callbacks,
       });
     }
@@ -235,37 +243,13 @@ export const useInputManager = (
     return managerRef.current?.checkForSlashDeletion(cursorPos) || false;
   }, []);
 
-  // Bash history selector methods
-  const activateBashHistorySelector = useCallback((position: number) => {
-    managerRef.current?.activateBashHistorySelector(position);
-  }, []);
-
-  const handleBashHistorySelect = useCallback(
-    (command: string) => {
-      return (
-        managerRef.current?.handleBashHistorySelect(command) || {
-          newInput: inputText,
-          newCursorPosition: cursorPosition,
-        }
-      );
-    },
-    [inputText, cursorPosition],
-  );
-
-  const handleCancelBashHistorySelect = useCallback(() => {
-    managerRef.current?.handleCancelBashHistorySelect();
-  }, []);
-
-  const updateBashHistorySearchQuery = useCallback((query: string) => {
-    managerRef.current?.updateBashHistorySearchQuery(query);
+  // History search methods
+  const handleHistorySearchSelect = useCallback((prompt: string) => {
+    managerRef.current?.handleHistorySearchSelect(prompt);
   }, []);
 
-  const handleBashHistoryExecute = useCallback((command: string) => {
-    return managerRef.current?.handleBashHistoryExecute(command) || command;
-  }, []);
-
-  const checkForExclamationDeletion = useCallback((cursorPos: number) => {
-    return managerRef.current?.checkForExclamationDeletion(cursorPos) || false;
+  const handleCancelHistorySearch = useCallback(() => {
+    managerRef.current?.handleCancelHistorySearch();
   }, []);
 
   // Memory type selector methods
@@ -320,17 +304,6 @@ export const useInputManager = (
   }, []);
 
   // Complex handlers that combine multiple operations
-  const handleBashHistoryExecuteAndSend = useCallback((command: string) => {
-    managerRef.current?.handleBashHistoryExecuteAndSend(command);
-  }, []);
-
-  const handleBashHistoryDelete = useCallback(
-    (command: string, workdir?: string) => {
-      managerRef.current?.handleBashHistoryDelete(command, workdir);
-    },
-    [],
-  );
-
   return {
     // State
     inputText,
@@ -342,13 +315,13 @@ export const useInputManager = (
     showCommandSelector: commandSelectorState.show,
     commandSearchQuery: commandSelectorState.query,
     slashPosition: commandSelectorState.position,
-    showBashHistorySelector: bashHistorySelectorState.show,
-    bashHistorySearchQuery: bashHistorySelectorState.query,
-    exclamationPosition: bashHistorySelectorState.position,
+    showHistorySearch: historySearchState.show,
+    historySearchQuery: historySearchState.query,
     showMemoryTypeSelector: memoryTypeSelectorState.show,
     memoryMessage: memoryTypeSelectorState.message,
     showBashManager,
     showMcpManager,
+    showRewindManager,
     permissionMode,
     attachedImages,
     isManagerReady,
@@ -377,14 +350,9 @@ export const useInputManager = (
     updateCommandSearchQuery,
     checkForSlashDeletion,
 
-    // Bash history selector
-    activateBashHistorySelector,
-    handleBashHistorySelect,
-    handleCancelBashHistorySelect,
-    updateBashHistorySearchQuery,
-
-    handleBashHistoryExecute,
-    checkForExclamationDeletion,
+    // History search
+    handleHistorySearchSelect,
+    handleCancelHistorySearch,
 
     // Memory type selector
     activateMemoryTypeSelector,
@@ -406,6 +374,10 @@ export const useInputManager = (
     setShowMcpManager: useCallback((show: boolean) => {
       managerRef.current?.setShowMcpManager(show);
     }, []),
+    setShowRewindManager: useCallback((show: boolean) => {
+      managerRef.current?.setShowRewindManager(show);
+      setShowRewindManager(show);
+    }, []),
     setPermissionMode: useCallback((mode: PermissionMode) => {
       setPermissionModeState(mode);
       managerRef.current?.setPermissionMode(mode);
@@ -450,10 +422,6 @@ export const useInputManager = (
       managerRef.current?.clearLongTextMap();
     }, []),
 
-    // Complex handlers combining multiple operations
-    handleBashHistoryExecuteAndSend,
-    handleBashHistoryDelete,
-
     // Main input handler
     handleInput: useCallback(
       async (