wave-code 0.5.0 → 0.6.0

package/src/components/MessageList.tsx

@@ -7,9 +7,8 @@ export interface MessageListProps {
   messages: Message[];
   isLoading?: boolean;
   isCommandRunning?: boolean;
-  isCompressing?: boolean;
-  latestTotalTokens?: number;
   isExpanded?: boolean;
+  forceStaticLastMessage?: boolean;
 }
 
 export const MessageList = React.memo(
@@ -17,15 +16,16 @@ export const MessageList = React.memo(
     messages,
     isLoading = false,
     isCommandRunning = false,
-    isCompressing = false,
-    latestTotalTokens = 0,
     isExpanded = false,
+    forceStaticLastMessage = false,
   }: MessageListProps) => {
     // Empty message state
     if (messages.length === 0) {
       return (
-        <Box flexDirection="column" paddingY={1}>
-          <Text color="gray">Welcome to WAVE Code Assistant!</Text>
+        <Box flexDirection="column" gap={1}>
+          <Box flexDirection="column" paddingY={1}>
+            <Text color="gray">Welcome to WAVE Code Assistant!</Text>
+          </Box>
         </Box>
       );
     }
@@ -42,7 +42,8 @@ export const MessageList = React.memo(
       : 0;
 
     // Compute which messages to render statically vs dynamically
-    const shouldRenderLastDynamic = isLoading || isCommandRunning;
+    const shouldRenderLastDynamic =
+      !forceStaticLastMessage && (isLoading || isCommandRunning);
     const staticMessages = shouldRenderLastDynamic
       ? displayMessages.slice(0, -1)
       : displayMessages;
@@ -52,7 +53,7 @@ export const MessageList = React.memo(
         : [];
 
     return (
-      <Box flexDirection="column" gap={1}>
+      <Box flexDirection="column" gap={1} paddingBottom={1}>
         {/* Show omitted message count when limiting */}
         {omittedCount > 0 && (
           <Box>
@@ -86,7 +87,7 @@ export const MessageList = React.memo(
           const previousMessage =
             messageIndex > 0 ? displayMessages[messageIndex - 1] : undefined;
           return (
-            <Box key={`dynamic-${index}`} marginTop={-1}>
+            <Box key={`dynamic-${index}`}>
               <MessageItem
                 message={message}
                 shouldShowHeader={previousMessage?.role !== message.role}
@@ -95,64 +96,6 @@ export const MessageList = React.memo(
             </Box>
           );
         })}
-
-        {(isLoading || isCommandRunning || isCompressing) && (
-          <Box flexDirection="column" gap={1}>
-            {isLoading && (
-              <Box>
-                <Text color="yellow">💭 AI is thinking... </Text>
-                <Text color="gray" dimColor>
-                  |{" "}
-                </Text>
-                <Text color="red" bold>
-                  Esc
-                </Text>
-                <Text color="gray" dimColor>
-                  {" "}
-                  to abort
-                </Text>
-              </Box>
-            )}
-            {isCommandRunning && (
-              <Text color="blue">🚀 Command is running...</Text>
-            )}
-            {isCompressing && (
-              <Text color="magenta">🗜️ Compressing message history...</Text>
-            )}
-          </Box>
-        )}
-
-        {/* Bottom info and shortcut key hints */}
-        {messages.length > 0 && (
-          <Box>
-            <Box justifyContent="space-between" width="100%">
-              <Box>
-                <Text color="gray">
-                  Messages {messages.length}
-                  {latestTotalTokens > 0 && (
-                    <>
-                      <Text color="gray" dimColor>
-                        {" "}
-                        |{" "}
-                      </Text>
-                      <Text color="blue" bold>
-                        {latestTotalTokens.toLocaleString()}
-                      </Text>
-                      <Text color="gray" dimColor>
-                        {" "}
-                        tokens
-                      </Text>
-                    </>
-                  )}
-                </Text>
-              </Box>
-              <Text color="gray" dimColor>
-                <Text color="cyan">Ctrl+O</Text> Toggle{" "}
-                {isExpanded ? "Collapse" : "Expand"}
-              </Text>
-            </Box>
-          </Box>
-        )}
       </Box>
     );
   },

package/src/components/PlanDisplay.tsx

@@ -1,5 +1,5 @@
-import React, { useMemo } from "react";
-import { Box, Text, useStdout } from "ink";
+import React from "react";
+import { Box } from "ink";
 import { Markdown } from "./Markdown.js";
 
 interface PlanDisplayProps {
@@ -7,35 +7,12 @@ interface PlanDisplayProps {
   isExpanded?: boolean;
 }
 
-export const PlanDisplay: React.FC<PlanDisplayProps> = ({
-  plan,
-  isExpanded = false,
-}) => {
-  const { stdout } = useStdout();
-  const maxHeight = useMemo(() => {
-    // Similar to DiffDisplay.tsx maxHeight calculation
-    return Math.max(5, (stdout?.rows || 24) - 25);
-  }, [stdout?.rows]);
-
-  const lines = useMemo(() => plan.split("\n"), [plan]);
-  const isOverflowing = !isExpanded && lines.length > maxHeight;
-
+export const PlanDisplay: React.FC<PlanDisplayProps> = ({ plan }) => {
   return (
     <Box flexDirection="column" marginTop={1}>
-      <Box
-        flexDirection="column"
-        height={isOverflowing ? maxHeight : undefined}
-        overflow="hidden"
-      >
+      <Box flexDirection="column">
         <Markdown>{plan}</Markdown>
       </Box>
-      {isOverflowing && (
-        <Box marginTop={1}>
-          <Text color="yellow" dimColor>
-            ... (plan truncated, {lines.length} lines total)
-          </Text>
-        </Box>
-      )}
     </Box>
   );
 };

package/src/components/SubagentBlock.tsx

@@ -1,4 +1,4 @@
-import React from "react";
+import React, { useMemo } from "react";
 import { Box, Text } from "ink";
 import type { SubagentBlock as SubagentBlockType } from "wave-agent-sdk";
 import { useChat } from "../contexts/useChat.js";
@@ -9,16 +9,22 @@ interface SubagentBlockProps {
 }
 
 export const SubagentBlock: React.FC<SubagentBlockProps> = ({ block }) => {
-  const { subagentMessages } = useChat();
+  const { subagentMessages, subagentLatestTokens } = useChat();
+
+  // Get messages for this subagent from context
+  const messages = useMemo(
+    () => subagentMessages[block.subagentId] || [],
+    [subagentMessages, block.subagentId],
+  );
+
+  // Get latest turn tokens for this subagent
+  const latestTurnTokens = subagentLatestTokens[block.subagentId] || 0;
 
   // If the subagent is running in the background, don't show the block
   if (block.runInBackground) {
     return null;
   }
 
-  // Get messages for this subagent from context
-  const messages = subagentMessages[block.subagentId] || [];
-
   // Status indicator mapping
   const getStatusIndicator = (status: SubagentBlockType["status"]) => {
     switch (status) {
@@ -63,7 +69,7 @@ export const SubagentBlock: React.FC<SubagentBlockProps> = ({ block }) => {
     return { tools: tools.reverse(), totalToolCount }; // Reverse to show oldest first, newest last
   };
 
-  const { tools: lastTwoTools, totalToolCount } = getLastTwoTools();
+  const { tools: lastTwoTools } = getLastTwoTools();
 
   // Get the last text message content if completed
   const getLastTextMessage = () => {
@@ -93,9 +99,7 @@ export const SubagentBlock: React.FC<SubagentBlockProps> = ({ block }) => {
       borderStyle="classic"
       borderColor="magenta"
       paddingX={1}
-      paddingY={0}
       flexDirection="column"
-      marginBottom={1}
     >
       {/* Header Section */}
       <Box flexDirection="row" gap={1}>
@@ -107,26 +111,31 @@ export const SubagentBlock: React.FC<SubagentBlockProps> = ({ block }) => {
           </Text>
           <Text color="gray" dimColor>
             {" "}
-            ({messages.length} messages)
+            ({messages.length} messages
+            {latestTurnTokens > 0 && (
+              <>
+                {" | "}
+                <Text color="blue" bold>
+                  {latestTurnTokens.toLocaleString()}
+                </Text>
+                {" tokens"}
+              </>
+            )}
+            )
           </Text>
         </Box>
       </Box>
 
       {/* Last Text Message Section */}
       {lastTextMessage && (
-        <Box marginTop={1}>
+        <Box>
           <Markdown>{lastTextMessage}</Markdown>
         </Box>
       )}
 
       {/* Tool Names Section - Vertical List */}
       {block.status !== "completed" && lastTwoTools.length > 0 && (
-        <Box flexDirection="column" marginTop={1} gap={1}>
-          {totalToolCount > 2 && (
-            <Text color="gray" dimColor>
-              ...
-            </Text>
-          )}
+        <Box flexDirection="column">
           {lastTwoTools.map((tool, index) => (
             <Box key={index} flexDirection="row">
               <Text color="magenta">🔧 </Text>

package/src/components/TaskList.tsx

@@ -0,0 +1,70 @@
+import React from "react";
+import { useChat } from "../contexts/useChat.js";
+import { Box, Text, useStdout } from "ink";
+import { useTasks } from "../hooks/useTasks.js";
+
+export const TaskList: React.FC = () => {
+  const tasks = useTasks();
+  const { isTaskListVisible } = useChat();
+  const { stdout } = useStdout();
+  const terminalWidth = stdout?.columns ?? 80;
+  const maxSubjectWidth = Math.max(20, terminalWidth - 10);
+
+  if (tasks.length === 0 || !isTaskListVisible) {
+    return null;
+  }
+
+  const getStatusIcon = (status: string, isBlocked: boolean) => {
+    if (isBlocked) {
+      return <Text color="red">🔒</Text>;
+    }
+    switch (status) {
+      case "pending":
+        return <Text color="gray">○</Text>;
+      case "in_progress":
+        return <Text color="yellow">●</Text>;
+      case "completed":
+        return <Text color="green">✓</Text>;
+      case "deleted":
+        return <Text color="red">✕</Text>;
+      default:
+        return <Text color="gray">?</Text>;
+    }
+  };
+
+  const truncate = (text: string, maxWidth: number) => {
+    if (text.length <= maxWidth) {
+      return text;
+    }
+    return text.slice(0, maxWidth - 3) + "...";
+  };
+
+  return (
+    <Box flexDirection="column">
+      {tasks.map((task) => {
+        const isDimmed =
+          task.status === "completed" || task.status === "deleted";
+        const isBlocked = task.blockedBy && task.blockedBy.length > 0;
+        const blockingTaskIds = isBlocked
+          ? task.blockedBy.map((id) => `#${id}`)
+          : [];
+
+        const blockedByText =
+          isBlocked && blockingTaskIds.length > 0
+            ? ` (Blocked by: ${blockingTaskIds.join(", ")})`
+            : "";
+
+        const fullText = `${task.subject}${blockedByText}`;
+
+        return (
+          <Box key={task.id} gap={1}>
+            {getStatusIcon(task.status, isBlocked)}
+            <Text dimColor={isDimmed}>
+              {truncate(fullText, maxSubjectWidth)}
+            </Text>
+          </Box>
+        );
+      })}
+    </Box>
+  );
+};

package/src/components/ToolResultDisplay.tsx

@@ -68,7 +68,7 @@ export const ToolResultDisplay: React.FC<ToolResultDisplayProps> = ({
   const shortResult = getShortResult();
 
   return (
-    <Box flexDirection="column" gap={1}>
+    <Box flexDirection="column">
       <Box>
         <Text color="magenta">🔧 </Text>
         <Text color="white">{toolName}</Text>
@@ -91,7 +91,7 @@ export const ToolResultDisplay: React.FC<ToolResultDisplayProps> = ({
           flexDirection="column"
         >
           {shortResult.split("\n").map((line, index) => (
-            <Text key={index} color="white">
+            <Text key={index} color="gray">
               {line}
             </Text>
           ))}

package/src/contexts/useChat.tsx

@@ -12,6 +12,7 @@ import type {
   Message,
   McpServerStatus,
   BackgroundTask,
+  Task,
   SlashCommand,
   PermissionDecision,
   PermissionMode,
@@ -33,6 +34,8 @@ export interface ChatContextType {
   userInputHistory: string[];
   // Message display state
   isExpanded: boolean;
+  isTaskListVisible: boolean;
+  setIsTaskListVisible: (visible: boolean) => void;
   // AI functionality
   sessionId: string;
   sendMessage: (
@@ -41,14 +44,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 +61,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;
@@ -110,6 +114,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
 
   // Message Display State
   const [isExpanded, setIsExpanded] = useState(false);
+  const [isTaskListVisible, setIsTaskListVisible] = useState(true);
 
   // AI State
   const [messages, setMessages] = useState<Message[]>([]);
@@ -125,6 +130,8 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
 
   // 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 +140,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] =
@@ -209,9 +219,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
           setMcpServers([...servers]);
         },
         onSessionIdChange: (sessionId) => {
-          process.stdout.write("\x1Bc", () => {
-            setSessionId(sessionId);
-          });
+          setSessionId(sessionId);
         },
         onLatestTotalTokensChange: (tokens) => {
           setlatestTotalTokens(tokens);
@@ -225,13 +233,25 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         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);
         },
@@ -338,16 +358,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 +404,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,9 +494,7 @@ 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);
       }
@@ -505,14 +505,16 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   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;
+        return newExpanded;
       });
     }
 
+    if (key.ctrl && input === "t") {
+      setIsTaskListVisible((prev) => !prev);
+    }
+
     // Handle ESC key to cancel confirmation
     if (key.escape && isConfirmationVisible) {
       handleConfirmationCancel();
@@ -525,21 +527,24 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     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,

package/src/hooks/useInputManager.ts

@@ -33,11 +33,8 @@ 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 [permissionMode, setPermissionModeState] =
@@ -61,11 +58,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);
@@ -78,9 +72,6 @@ export const useInputManager = (
           callbacks.onPermissionModeChange?.(mode);
         },
         onImagesStateChange: setAttachedImages,
-        onShowTaskManager: () => setShowTaskManager(true),
-        onShowMcpManager: () => setShowMcpManager(true),
-        onShowRewindManager: () => setShowRewindManager(true),
         ...callbacks,
       });
 
@@ -101,11 +92,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);
@@ -118,9 +106,6 @@ export const useInputManager = (
           callbacks.onPermissionModeChange?.(mode);
         },
         onImagesStateChange: setAttachedImages,
-        onShowTaskManager: () => setShowTaskManager(true),
-        onShowMcpManager: () => setShowMcpManager(true),
-        onShowRewindManager: () => setShowRewindManager(true),
         ...callbacks,
       });
     }
@@ -252,22 +237,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);
@@ -317,9 +286,7 @@ export const useInputManager = (
     slashPosition: commandSelectorState.position,
     showHistorySearch: historySearchState.show,
     historySearchQuery: historySearchState.query,
-    showMemoryTypeSelector: memoryTypeSelectorState.show,
-    memoryMessage: memoryTypeSelectorState.message,
-    showTaskManager,
+    showBackgroundTaskManager,
     showMcpManager,
     showRewindManager,
     permissionMode,
@@ -354,11 +321,6 @@ export const useInputManager = (
     handleHistorySearchSelect,
     handleCancelHistorySearch,
 
-    // Memory type selector
-    activateMemoryTypeSelector,
-    handleMemoryTypeSelect,
-    handleCancelMemoryTypeSelect,
-
     // Input history
     setUserInputHistory,
     navigateHistory,
@@ -368,8 +330,8 @@ export const useInputManager = (
     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);

package/src/hooks/useTasks.ts

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