wave-code 0.5.0 → 0.6.2

package/src/contexts/useChat.tsx

@@ -12,6 +12,7 @@ import type {
   Message,
   McpServerStatus,
   BackgroundTask,
+  Task,
   SlashCommand,
   PermissionDecision,
   PermissionMode,
@@ -30,9 +31,10 @@ export interface ChatContextType {
   isLoading: boolean;
   isCommandRunning: boolean;
   isCompressing: boolean;
-  userInputHistory: string[];
   // Message display state
   isExpanded: boolean;
+  isTaskListVisible: boolean;
+  setIsTaskListVisible: (visible: boolean) => void;
   // AI functionality
   sessionId: string;
   sendMessage: (
@@ -41,14 +43,14 @@ export interface ChatContextType {
   ) => Promise<void>;
   abortMessage: () => void;
   latestTotalTokens: number;
-  // Memory functionality
-  saveMemory: (message: string, type: "project" | "user") => Promise<void>;
   // MCP functionality
   mcpServers: McpServerStatus[];
   connectMcpServer: (serverName: string) => Promise<boolean>;
   disconnectMcpServer: (serverName: string) => Promise<boolean>;
   // Background tasks
   backgroundTasks: BackgroundTask[];
+  // Session tasks
+  sessionTasks: Task[];
   getBackgroundTaskOutput: (
     taskId: string,
   ) => { stdout: string; stderr: string; status: string } | null;
@@ -58,6 +60,7 @@ export interface ChatContextType {
   hasSlashCommand: (commandId: string) => boolean;
   // Subagent messages
   subagentMessages: Record<string, Message[]>;
+  subagentLatestTokens: Record<string, number>;
   // Permission functionality
   permissionMode: PermissionMode;
   setPermissionMode: (mode: PermissionMode) => void;
@@ -83,6 +86,12 @@ export interface ChatContextType {
   // Rewind functionality
   rewindId: number;
   handleRewindSelect: (index: number) => Promise<void>;
+  getFullMessageThread: () => Promise<{
+    messages: Message[];
+    sessionIds: string[];
+  }>;
+  wasLastDetailsTooTall: number;
+  setWasLastDetailsTooTall: React.Dispatch<React.SetStateAction<number>>;
 }
 
 const ChatContext = createContext<ChatContextType | null>(null);
@@ -110,6 +119,14 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
 
   // Message Display State
   const [isExpanded, setIsExpanded] = useState(false);
+  const isExpandedRef = useRef(isExpanded);
+  const frozenMessagesRef = useRef<Message[] | null>(null);
+
+  useEffect(() => {
+    isExpandedRef.current = isExpanded;
+  }, [isExpanded]);
+
+  const [isTaskListVisible, setIsTaskListVisible] = useState(true);
 
   // AI State
   const [messages, setMessages] = useState<Message[]>([]);
@@ -118,13 +135,14 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   const [sessionId, setSessionId] = useState("");
   const [isCommandRunning, setIsCommandRunning] = useState(false);
   const [isCompressing, setIsCompressing] = useState(false);
-  const [userInputHistory, setUserInputHistory] = useState<string[]>([]);
 
   // MCP State
   const [mcpServers, setMcpServers] = useState<McpServerStatus[]>([]);
 
   // Background tasks state
   const [backgroundTasks, setBackgroundTasks] = useState<BackgroundTask[]>([]);
+  // Session tasks state
+  const [sessionTasks, setSessionTasks] = useState<Task[]>([]);
 
   // Command state
   const [slashCommands, setSlashCommands] = useState<SlashCommand[]>([]);
@@ -133,6 +151,9 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   const [subagentMessages, setSubagentMessages] = useState<
     Record<string, Message[]>
   >({});
+  const [subagentLatestTokens, setSubagentLatestTokens] = useState<
+    Record<string, number>
+  >({});
 
   // Permission state
   const [permissionMode, setPermissionModeState] =
@@ -171,6 +192,9 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   // Rewind state
   const [rewindId, setRewindId] = useState(0);
 
+  // Confirmation too tall state
+  const [wasLastDetailsTooTall, setWasLastDetailsTooTall] = useState(0);
+
   const agentRef = useRef<Agent | null>(null);
 
   // Permission confirmation methods with queue support
@@ -203,35 +227,46 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     const initializeAgent = async () => {
       const callbacks: AgentCallbacks = {
         onMessagesChange: (newMessages) => {
-          setMessages([...newMessages]);
+          if (isExpandedRef.current) {
+            frozenMessagesRef.current = [...newMessages];
+          } else {
+            setMessages([...newMessages]);
+          }
         },
         onServersChange: (servers) => {
           setMcpServers([...servers]);
         },
         onSessionIdChange: (sessionId) => {
-          process.stdout.write("\x1Bc", () => {
-            setSessionId(sessionId);
-          });
+          setSessionId(sessionId);
         },
         onLatestTotalTokensChange: (tokens) => {
           setlatestTotalTokens(tokens);
         },
-        onUserInputHistoryChange: (history) => {
-          setUserInputHistory([...history]);
-        },
         onCompressionStateChange: (isCompressingState) => {
           setIsCompressing(isCompressingState);
         },
         onTasksChange: (tasks) => {
           setBackgroundTasks([...tasks]);
         },
-        onSubagentMessagesChange: (subagentId, messages) => {
+        onSessionTasksChange: (tasks) => {
+          setSessionTasks([...tasks]);
+        },
+        onSubagentMessagesChange: (subagentId: string, messages: Message[]) => {
           logger.debug("onSubagentMessagesChange", subagentId, messages.length);
           setSubagentMessages((prev) => ({
             ...prev,
             [subagentId]: [...messages],
           }));
         },
+        onSubagentLatestTotalTokensChange: (
+          subagentId: string,
+          tokens: number,
+        ) => {
+          setSubagentLatestTokens((prev) => ({
+            ...prev,
+            [subagentId]: tokens,
+          }));
+        },
         onPermissionModeChange: (mode) => {
           setPermissionModeState(mode);
         },
@@ -283,7 +318,6 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         setlatestTotalTokens(agent.latestTotalTokens);
         setIsCommandRunning(agent.isCommandRunning);
         setIsCompressing(agent.isCompressing);
-        setUserInputHistory(agent.userInputHistory);
         setPermissionModeState(agent.getPermissionMode());
 
         // Get initial MCP servers state
@@ -338,16 +372,6 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
       if (!hasTextContent && !hasImageAttachments) return;
 
       try {
-        // Handle memory mode - check if it's a memory message (starts with # and only one line)
-        if (content.startsWith("#") && !content.includes("\n")) {
-          const memoryText = content.substring(1).trim();
-          if (!memoryText) return;
-
-          // In memory mode, don't add user message, only wait for user to choose memory type then add assistant message
-          // Don't auto-save, wait for user to choose memory type
-          return;
-        }
-
         // Handle bash mode - check if it's a bash command (starts with ! and only one line)
         if (content.startsWith("!") && !content.includes("\n")) {
           const command = content.substring(1).trim();
@@ -394,14 +418,6 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     agentRef.current?.abortMessage();
   }, []);
 
-  // Memory save function - delegate to Agent
-  const saveMemory = useCallback(
-    async (message: string, type: "project" | "user") => {
-      await agentRef.current?.saveMemory(message, type);
-    },
-    [],
-  );
-
   // Permission management methods
   const setPermissionMode = useCallback((mode: PermissionMode) => {
     setPermissionModeState((prev) => {
@@ -492,27 +508,38 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         await agentRef.current.truncateHistory(index);
 
         // Clear terminal screen after rewind
-        process.stdout.write("\x1Bc", () => {
-          setRewindId((prev) => prev + 1);
-        });
+        setRewindId((prev) => prev + 1);
       } catch (error) {
         logger.error("Failed to rewind:", error);
       }
     }
   }, []);
 
+  const getFullMessageThread = useCallback(async () => {
+    if (agentRef.current) {
+      return await agentRef.current.getFullMessageThread();
+    }
+    return { messages: [], sessionIds: [] };
+  }, []);
+
   // Listen for Ctrl+O hotkey to toggle collapse/expand state and ESC to cancel confirmation
   useInput((input, key) => {
     if (key.ctrl && input === "o") {
       // Clear terminal screen when expanded state changes
-      process.stdout.write("\x1Bc", () => {
-        setIsExpanded((prev) => {
-          const newExpanded = !prev;
-          return newExpanded;
-        });
+      setIsExpanded((prev) => {
+        const newExpanded = !prev;
+        if (!newExpanded && frozenMessagesRef.current) {
+          setMessages(frozenMessagesRef.current);
+          frozenMessagesRef.current = null;
+        }
+        return newExpanded;
       });
     }
 
+    if (key.ctrl && input === "t") {
+      setIsTaskListVisible((prev) => !prev);
+    }
+
     // Handle ESC key to cancel confirmation
     if (key.escape && isConfirmationVisible) {
       handleConfirmationCancel();
@@ -523,23 +550,25 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     messages,
     isLoading,
     isCommandRunning,
-    userInputHistory,
     isExpanded,
+    isTaskListVisible,
+    setIsTaskListVisible,
     sessionId,
     sendMessage,
     abortMessage,
     latestTotalTokens,
     isCompressing,
-    saveMemory,
     mcpServers,
     connectMcpServer,
     disconnectMcpServer,
     backgroundTasks,
+    sessionTasks,
     getBackgroundTaskOutput,
     stopBackgroundTask,
     slashCommands,
     hasSlashCommand,
     subagentMessages,
+    subagentLatestTokens,
     permissionMode,
     setPermissionMode,
     isConfirmationVisible,
@@ -551,6 +580,9 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     backgroundCurrentTask,
     rewindId,
     handleRewindSelect,
+    getFullMessageThread,
+    wasLastDetailsTooTall,
+    setWasLastDetailsTooTall,
   };
 
   return (

package/src/hooks/useInputManager.ts

@@ -33,13 +33,11 @@ export const useInputManager = (
     show: false,
     query: "",
   });
-  const [memoryTypeSelectorState, setMemoryTypeSelectorState] = useState({
-    show: false,
-    message: "",
-  });
-  const [showTaskManager, setShowTaskManager] = useState(false);
+  const [showBackgroundTaskManager, setShowBackgroundTaskManager] =
+    useState(false);
   const [showMcpManager, setShowMcpManager] = useState(false);
   const [showRewindManager, setShowRewindManager] = useState(false);
+  const [showHelp, setShowHelp] = useState(false);
   const [permissionMode, setPermissionModeState] =
     useState<PermissionMode>("default");
   const [attachedImages, setAttachedImages] = useState<AttachedImage[]>([]);
@@ -61,11 +59,8 @@ export const useInputManager = (
         onHistorySearchStateChange: (show, query) => {
           setHistorySearchState({ show, query });
         },
-        onMemoryTypeSelectorStateChange: (show, message) => {
-          setMemoryTypeSelectorState({ show, message });
-        },
-        onTaskManagerStateChange: (show) => {
-          setShowTaskManager(show);
+        onBackgroundTaskManagerStateChange: (show) => {
+          setShowBackgroundTaskManager(show);
         },
         onMcpManagerStateChange: (show) => {
           setShowMcpManager(show);
@@ -73,14 +68,14 @@ export const useInputManager = (
         onRewindManagerStateChange: (show) => {
           setShowRewindManager(show);
         },
+        onHelpStateChange: (show) => {
+          setShowHelp(show);
+        },
         onPermissionModeChange: (mode) => {
           setPermissionModeState(mode);
           callbacks.onPermissionModeChange?.(mode);
         },
         onImagesStateChange: setAttachedImages,
-        onShowTaskManager: () => setShowTaskManager(true),
-        onShowMcpManager: () => setShowMcpManager(true),
-        onShowRewindManager: () => setShowRewindManager(true),
         ...callbacks,
       });
 
@@ -101,11 +96,8 @@ export const useInputManager = (
         onHistorySearchStateChange: (show, query) => {
           setHistorySearchState({ show, query });
         },
-        onMemoryTypeSelectorStateChange: (show, message) => {
-          setMemoryTypeSelectorState({ show, message });
-        },
-        onTaskManagerStateChange: (show) => {
-          setShowTaskManager(show);
+        onBackgroundTaskManagerStateChange: (show) => {
+          setShowBackgroundTaskManager(show);
         },
         onMcpManagerStateChange: (show) => {
           setShowMcpManager(show);
@@ -113,14 +105,14 @@ export const useInputManager = (
         onRewindManagerStateChange: (show) => {
           setShowRewindManager(show);
         },
+        onHelpStateChange: (show) => {
+          setShowHelp(show);
+        },
         onPermissionModeChange: (mode) => {
           setPermissionModeState(mode);
           callbacks.onPermissionModeChange?.(mode);
         },
         onImagesStateChange: setAttachedImages,
-        onShowTaskManager: () => setShowTaskManager(true),
-        onShowMcpManager: () => setShowMcpManager(true),
-        onShowRewindManager: () => setShowRewindManager(true),
         ...callbacks,
       });
     }
@@ -165,14 +157,6 @@ export const useInputManager = (
     managerRef.current?.moveCursorRight();
   }, []);
 
-  const moveCursorToStart = useCallback(() => {
-    managerRef.current?.moveCursorToStart();
-  }, []);
-
-  const moveCursorToEnd = useCallback(() => {
-    managerRef.current?.moveCursorToEnd();
-  }, []);
-
   // File selector methods
   const activateFileSelector = useCallback((position: number) => {
     managerRef.current?.activateFileSelector(position);
@@ -252,43 +236,6 @@ export const useInputManager = (
     managerRef.current?.handleCancelHistorySearch();
   }, []);
 
-  // Memory type selector methods
-  const activateMemoryTypeSelector = useCallback((message: string) => {
-    managerRef.current?.activateMemoryTypeSelector(message);
-  }, []);
-
-  const handleMemoryTypeSelect = useCallback(
-    async (type: "project" | "user") => {
-      await managerRef.current?.handleMemoryTypeSelect(type);
-    },
-    [],
-  );
-
-  const handleCancelMemoryTypeSelect = useCallback(() => {
-    managerRef.current?.handleCancelMemoryTypeSelect();
-  }, []);
-
-  // Input history methods
-  const setUserInputHistory = useCallback((history: string[]) => {
-    managerRef.current?.setUserInputHistory(history);
-  }, []);
-
-  const navigateHistory = useCallback(
-    (direction: "up" | "down", currentInput: string) => {
-      return (
-        managerRef.current?.navigateHistory(direction, currentInput) || {
-          newInput: currentInput,
-          newCursorPosition: currentInput.length,
-        }
-      );
-    },
-    [],
-  );
-
-  const resetHistoryNavigation = useCallback(() => {
-    managerRef.current?.resetHistoryNavigation();
-  }, []);
-
   // Special character handling
   const handleSpecialCharInput = useCallback((char: string) => {
     managerRef.current?.handleSpecialCharInput(char);
@@ -317,11 +264,10 @@ export const useInputManager = (
     slashPosition: commandSelectorState.position,
     showHistorySearch: historySearchState.show,
     historySearchQuery: historySearchState.query,
-    showMemoryTypeSelector: memoryTypeSelectorState.show,
-    memoryMessage: memoryTypeSelectorState.message,
-    showTaskManager,
+    showBackgroundTaskManager,
     showMcpManager,
     showRewindManager,
+    showHelp,
     permissionMode,
     attachedImages,
     isManagerReady,
@@ -332,8 +278,6 @@ export const useInputManager = (
     clearInput,
     moveCursorLeft,
     moveCursorRight,
-    moveCursorToStart,
-    moveCursorToEnd,
 
     // File selector
     activateFileSelector,
@@ -354,22 +298,12 @@ export const useInputManager = (
     handleHistorySearchSelect,
     handleCancelHistorySearch,
 
-    // Memory type selector
-    activateMemoryTypeSelector,
-    handleMemoryTypeSelect,
-    handleCancelMemoryTypeSelect,
-
-    // Input history
-    setUserInputHistory,
-    navigateHistory,
-    resetHistoryNavigation,
-
     // Special handling
     handleSpecialCharInput,
 
     // Bash/MCP Manager
-    setShowTaskManager: useCallback((show: boolean) => {
-      managerRef.current?.setShowTaskManager(show);
+    setShowBackgroundTaskManager: useCallback((show: boolean) => {
+      managerRef.current?.setShowBackgroundTaskManager(show);
     }, []),
     setShowMcpManager: useCallback((show: boolean) => {
       managerRef.current?.setShowMcpManager(show);
@@ -378,6 +312,10 @@ export const useInputManager = (
       managerRef.current?.setShowRewindManager(show);
       setShowRewindManager(show);
     }, []),
+    setShowHelp: useCallback((show: boolean) => {
+      managerRef.current?.setShowHelp(show);
+      setShowHelp(show);
+    }, []),
     setPermissionMode: useCallback((mode: PermissionMode) => {
       setPermissionModeState(mode);
       managerRef.current?.setPermissionMode(mode);

package/src/hooks/useTasks.ts

@@ -0,0 +1,6 @@
+import { useChat } from "../contexts/useChat.js";
+
+export const useTasks = () => {
+  const { sessionTasks } = useChat();
+  return sessionTasks;
+};