wave-code 0.7.2 → 0.8.1

package/src/components/StatusCommand.tsx

@@ -0,0 +1,94 @@
+import React from "react";
+import { Box, Text, useInput } from "ink";
+import { useChat } from "../contexts/useChat.js";
+import fs from "fs";
+import path from "path";
+import { fileURLToPath } from "url";
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+export interface StatusCommandProps {
+  onCancel: () => void;
+}
+
+export const StatusCommand: React.FC<StatusCommandProps> = ({ onCancel }) => {
+  const { sessionId, workingDirectory, getGatewayConfig, getModelConfig } =
+    useChat();
+
+  useInput((_, key) => {
+    if (key.escape) {
+      onCancel();
+    }
+  });
+
+  const gatewayConfig = getGatewayConfig();
+  const modelConfig = getModelConfig();
+
+  let version = "unknown";
+  try {
+    const pkgPath = path.resolve(__dirname, "../../package.json");
+    const pkg = JSON.parse(fs.readFileSync(pkgPath, "utf-8"));
+    version = pkg.version;
+  } catch {
+    // Fallback if package.json cannot be read
+  }
+
+  return (
+    <Box
+      flexDirection="column"
+      borderStyle="round"
+      borderColor="cyan"
+      borderLeft={false}
+      borderRight={false}
+      paddingX={1}
+    >
+      <Box marginBottom={1}>
+        <Text color="cyan" bold underline>
+          Agent Status
+        </Text>
+      </Box>
+
+      <Box>
+        <Box width={20}>
+          <Text color="yellow">Version:</Text>
+        </Box>
+        <Text color="white">{version}</Text>
+      </Box>
+
+      <Box>
+        <Box width={20}>
+          <Text color="yellow">Session ID:</Text>
+        </Box>
+        <Text color="white">{sessionId}</Text>
+      </Box>
+
+      <Box>
+        <Box width={20}>
+          <Text color="yellow">cwd:</Text>
+        </Box>
+        <Text color="white" wrap="wrap">
+          {workingDirectory}
+        </Text>
+      </Box>
+
+      <Box>
+        <Box width={20}>
+          <Text color="yellow">Wave base URL:</Text>
+        </Box>
+        <Text color="white">{gatewayConfig.baseURL}</Text>
+      </Box>
+
+      <Box>
+        <Box width={20}>
+          <Text color="yellow">Model:</Text>
+        </Box>
+        <Text color="white">{modelConfig.model}</Text>
+      </Box>
+
+      <Box marginTop={1}>
+        <Text dimColor>Esc to cancel</Text>
+      </Box>
+    </Box>
+  );
+};

package/src/components/WorktreeExitPrompt.tsx

@@ -0,0 +1,86 @@
+import React, { useState } from "react";
+import { Box, Text, useInput } from "ink";
+
+interface WorktreeExitPromptProps {
+  name: string;
+  path: string;
+  hasUncommittedChanges: boolean;
+  hasNewCommits: boolean;
+  onKeep: () => void;
+  onRemove: () => void;
+  onCancel: () => void;
+}
+
+export const WorktreeExitPrompt: React.FC<WorktreeExitPromptProps> = ({
+  name,
+  path: worktreePath,
+  hasUncommittedChanges,
+  hasNewCommits,
+  onKeep,
+  onRemove,
+  onCancel,
+}) => {
+  const [selectedIndex, setSelectedIndex] = useState(0);
+
+  useInput((input, key) => {
+    if (key.upArrow) {
+      setSelectedIndex((prev) => (prev === 0 ? 1 : 0));
+    }
+    if (key.downArrow) {
+      setSelectedIndex((prev) => (prev === 1 ? 0 : 1));
+    }
+    if (key.return) {
+      if (selectedIndex === 0) {
+        onKeep();
+      } else {
+        onRemove();
+      }
+    }
+    if (key.escape) {
+      onCancel();
+    }
+  });
+
+  return (
+    <Box flexDirection="column" paddingX={1} marginY={1}>
+      <Box marginBottom={1}>
+        <Text bold>Exiting worktree session</Text>
+      </Box>
+      <Box marginBottom={1}>
+        {hasUncommittedChanges && hasNewCommits ? (
+          <Text>
+            You have uncommitted changes and new commits. These will be lost if
+            you remove the worktree.
+          </Text>
+        ) : hasUncommittedChanges ? (
+          <Text>
+            You have uncommitted changes. These will be lost if you remove the
+            worktree.
+          </Text>
+        ) : (
+          <Text>
+            You have new commits on worktree-{name}. The branch will be deleted
+            if you remove the worktree.
+          </Text>
+        )}
+      </Box>
+      <Box flexDirection="column">
+        <Box>
+          <Text color={selectedIndex === 0 ? "cyan" : undefined}>
+            {selectedIndex === 0 ? "❯ " : "  "}Keep worktree Stays at{" "}
+            {worktreePath}
+          </Text>
+        </Box>
+        <Box>
+          <Text color={selectedIndex === 1 ? "cyan" : undefined}>
+            {selectedIndex === 1 ? "❯ " : "  "}Remove worktree All changes and
+            commits will be lost.
+          </Text>
+        </Box>
+      </Box>
+      <Box marginTop={1}>
+        <Text dimColor>Enter to confirm · Esc to cancel</Text>
+      </Box>
+    </Box>
+  );
+};

package/src/constants/commands.ts

@@ -0,0 +1,41 @@
+import type { SlashCommand } from "wave-agent-sdk";
+
+export const AVAILABLE_COMMANDS: SlashCommand[] = [
+  {
+    id: "clear",
+    name: "clear",
+    description: "Clear the chat session and terminal",
+    handler: () => {}, // Handler here won't be used, actual processing is in the hook
+  },
+  {
+    id: "tasks",
+    name: "tasks",
+    description: "View and manage background tasks (shells and subagents)",
+    handler: () => {}, // Handler here won't be used, actual processing is in the hook
+  },
+  {
+    id: "mcp",
+    name: "mcp",
+    description: "View and manage MCP servers",
+    handler: () => {}, // Handler here won't be used, actual processing is in the hook
+  },
+  {
+    id: "rewind",
+    name: "rewind",
+    description:
+      "Revert conversation and file changes to a previous checkpoint",
+    handler: () => {}, // Handler here won't be used, actual processing is in the hook
+  },
+  {
+    id: "help",
+    name: "help",
+    description: "Show help and key bindings",
+    handler: () => {}, // Handler here won't be used, actual processing is in the hook
+  },
+  {
+    id: "status",
+    name: "status",
+    description: "Show agent status and configuration",
+    handler: () => {}, // Handler here won't be used, actual processing is in the hook
+  },
+];

package/src/contexts/useChat.tsx

@@ -25,6 +25,8 @@ import {
 import { logger } from "../utils/logger.js";
 import { displayUsageSummary } from "../utils/usageSummary.js";
 
+import { BaseAppProps } from "../types.js";
+
 // Main Chat Context
 export interface ChatContextType {
   messages: Message[];
@@ -43,14 +45,15 @@ export interface ChatContextType {
   ) => Promise<void>;
   abortMessage: () => void;
   latestTotalTokens: number;
+  clearMessages: () => void;
   // MCP functionality
   mcpServers: McpServerStatus[];
   connectMcpServer: (serverName: string) => Promise<boolean>;
   disconnectMcpServer: (serverName: string) => Promise<boolean>;
   // Background tasks
   backgroundTasks: BackgroundTask[];
-  // Session tasks
-  sessionTasks: Task[];
+  // Tasks
+  tasks: Task[];
   getBackgroundTaskOutput: (
     taskId: string,
   ) => { stdout: string; stderr: string; status: string } | null;
@@ -92,6 +95,12 @@ export interface ChatContextType {
   }>;
   wasLastDetailsTooTall: number;
   setWasLastDetailsTooTall: React.Dispatch<React.SetStateAction<number>>;
+  // Status metadata
+  getGatewayConfig: () => import("wave-agent-sdk").GatewayConfig;
+  getModelConfig: () => import("wave-agent-sdk").ModelConfig;
+  workingDirectory: string;
+  version?: string;
+  workdir?: string;
 }
 
 const ChatContext = createContext<ChatContextType | null>(null);
@@ -104,11 +113,8 @@ export const useChat = () => {
   return context;
 };
 
-export interface ChatProviderProps {
+export interface ChatProviderProps extends BaseAppProps {
   children: React.ReactNode;
-  bypassPermissions?: boolean;
-  pluginDirs?: string[];
-  tools?: string[];
 }
 
 export const ChatProvider: React.FC<ChatProviderProps> = ({
@@ -116,6 +122,10 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   bypassPermissions,
   pluginDirs,
   tools,
+  workdir,
+  worktreeSession,
+  version,
+  model,
 }) => {
   const { restoreSessionId, continueLastSession } = useAppConfig();
 
@@ -144,8 +154,8 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
 
   // Background tasks state
   const [backgroundTasks, setBackgroundTasks] = useState<BackgroundTask[]>([]);
-  // Session tasks state
-  const [sessionTasks, setSessionTasks] = useState<Task[]>([]);
+  // Tasks state
+  const [tasks, setTasks] = useState<Task[]>([]);
 
   // Command state
   const [slashCommands, setSlashCommands] = useState<SlashCommand[]>([]);
@@ -195,6 +205,9 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   // Rewind state
   const [rewindId, setRewindId] = useState(0);
 
+  // Status metadata state
+  const [workingDirectory, setWorkingDirectory] = useState("");
+
   // Confirmation too tall state
   const [wasLastDetailsTooTall, setWasLastDetailsTooTall] = useState(0);
 
@@ -237,7 +250,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
                   setMessages([...agentRef.current.messages]);
                 }
                 messagesUpdateTimerRef.current = null;
-              }, 50);
+              }, 100);
             }
           }
         },
@@ -253,11 +266,11 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         onCompressionStateChange: (isCompressingState) => {
           setIsCompressing(isCompressingState);
         },
-        onTasksChange: (tasks) => {
+        onBackgroundTasksChange: (tasks) => {
           setBackgroundTasks([...tasks]);
         },
-        onSessionTasksChange: (tasks) => {
-          setSessionTasks([...tasks]);
+        onTasksChange: (tasks) => {
+          setTasks([...tasks]);
         },
         onSubagentMessagesChange: (subagentId: string, messages: Message[]) => {
           logger.debug("onSubagentMessagesChange", subagentId, messages.length);
@@ -316,6 +329,10 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
           stream: false, // 关闭流式模式
           plugins: pluginDirs?.map((path) => ({ type: "local", path })),
           tools,
+          workdir,
+          worktreeName: worktreeSession?.name,
+          isNewWorktree: worktreeSession?.isNew,
+          model,
         });
 
         agentRef.current = agent;
@@ -328,6 +345,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         setIsCommandRunning(agent.isCommandRunning);
         setIsCompressing(agent.isCompressing);
         setPermissionModeState(agent.getPermissionMode());
+        setWorkingDirectory(agent.workingDirectory);
 
         // Get initial MCP servers state
         const mcpServers = agent.getMcpServers?.() || [];
@@ -349,6 +367,8 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     showConfirmation,
     pluginDirs,
     tools,
+    workdir,
+    worktreeSession,
   ]);
 
   // Cleanup on unmount
@@ -431,6 +451,10 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     agentRef.current?.abortMessage();
   }, []);
 
+  const clearMessages = useCallback(() => {
+    agentRef.current?.clearMessages();
+  }, []);
+
   // Permission management methods
   const setPermissionMode = useCallback((mode: PermissionMode) => {
     setPermissionModeState((prev) => {
@@ -535,6 +559,20 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     return { messages: [], sessionIds: [] };
   }, []);
 
+  const getGatewayConfig = useCallback(() => {
+    if (!agentRef.current) {
+      return { baseURL: "" };
+    }
+    return agentRef.current.getGatewayConfig();
+  }, []);
+
+  const getModelConfig = useCallback(() => {
+    if (!agentRef.current) {
+      return { model: "", fastModel: "" };
+    }
+    return agentRef.current.getModelConfig();
+  }, []);
+
   // Listen for Ctrl+O hotkey to toggle collapse/expand state and ESC to cancel confirmation
   useInput((input, key) => {
     if (key.ctrl && input === "o") {
@@ -584,13 +622,14 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     sessionId,
     sendMessage,
     abortMessage,
+    clearMessages,
     latestTotalTokens,
     isCompressing,
     mcpServers,
     connectMcpServer,
     disconnectMcpServer,
     backgroundTasks,
-    sessionTasks,
+    tasks,
     getBackgroundTaskOutput,
     stopBackgroundTask,
     slashCommands,
@@ -611,6 +650,11 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     getFullMessageThread,
     wasLastDetailsTooTall,
     setWasLastDetailsTooTall,
+    getGatewayConfig,
+    getModelConfig,
+    workingDirectory,
+    version,
+    workdir,
   };
 
   return (

package/src/contracts/status.ts

@@ -0,0 +1,7 @@
+export interface AgentStatus {
+  version: string;
+  sessionId: string;
+  cwd: string;
+  baseURL: string;
+  model: string;
+}

package/src/hooks/useInputManager.ts

@@ -38,6 +38,7 @@ export const useInputManager = (
   const [showMcpManager, setShowMcpManager] = useState(false);
   const [showRewindManager, setShowRewindManager] = useState(false);
   const [showHelp, setShowHelp] = useState(false);
+  const [showStatusCommand, setShowStatusCommand] = useState(false);
   const [permissionMode, setPermissionModeState] =
     useState<PermissionMode>("default");
   const [attachedImages, setAttachedImages] = useState<AttachedImage[]>([]);
@@ -71,6 +72,9 @@ export const useInputManager = (
         onHelpStateChange: (show) => {
           setShowHelp(show);
         },
+        onStatusCommandStateChange: (show) => {
+          setShowStatusCommand(show);
+        },
         onPermissionModeChange: (mode) => {
           setPermissionModeState(mode);
           callbacks.onPermissionModeChange?.(mode);
@@ -108,6 +112,9 @@ export const useInputManager = (
         onHelpStateChange: (show) => {
           setShowHelp(show);
         },
+        onStatusCommandStateChange: (show) => {
+          setShowStatusCommand(show);
+        },
         onPermissionModeChange: (mode) => {
           setPermissionModeState(mode);
           callbacks.onPermissionModeChange?.(mode);
@@ -268,6 +275,7 @@ export const useInputManager = (
     showMcpManager,
     showRewindManager,
     showHelp,
+    showStatusCommand,
     permissionMode,
     attachedImages,
     isManagerReady,
@@ -316,6 +324,10 @@ export const useInputManager = (
       managerRef.current?.setShowHelp(show);
       setShowHelp(show);
     }, []),
+    setShowStatusCommand: useCallback((show: boolean) => {
+      managerRef.current?.setShowStatusCommand(show);
+      setShowStatusCommand(show);
+    }, []),
     setPermissionMode: useCallback((mode: PermissionMode) => {
       setPermissionModeState(mode);
       managerRef.current?.setPermissionMode(mode);

package/src/hooks/usePluginManager.ts

@@ -17,6 +17,7 @@ export function usePluginManager(): PluginManagerContextType {
     selectedId: null,
     isLoading: true,
     error: null,
+    successMessage: null,
     searchQuery: "",
   });
 
@@ -34,11 +35,37 @@ export function usePluginManager(): PluginManagerContextType {
 
   const pluginCore = useMemo(() => new PluginCore(), []);
 
+  const clearPluginFeedback = useCallback(() => {
+    setState((prev: PluginManagerState) => ({
+      ...prev,
+      error: null,
+      successMessage: null,
+    }));
+  }, []);
+
+  const setSuccessMessage = useCallback(
+    (message: string) => {
+      setState((prev: PluginManagerState) => ({
+        ...prev,
+        successMessage: message,
+        error: null,
+      }));
+      setTimeout(() => {
+        setState((prev: PluginManagerState) => ({
+          ...prev,
+          successMessage:
+            prev.successMessage === message ? null : prev.successMessage,
+        }));
+      }, 5000);
+    },
+    [setState],
+  );
+
   const refresh = useCallback(async () => {
+    clearPluginFeedback();
     setState((prev: PluginManagerState) => ({
       ...prev,
       isLoading: true,
-      error: null,
     }));
     try {
       const [mks, installed, enabledMap] = await Promise.all([
@@ -115,14 +142,15 @@ export function usePluginManager(): PluginManagerContextType {
 
   const addMarketplace = useCallback(
     async (source: string) => {
+      clearPluginFeedback();
       setState((prev: PluginManagerState) => ({
         ...prev,
         isLoading: true,
-        error: null,
       }));
       try {
         await pluginCore.addMarketplace(source);
         await refresh();
+        setSuccessMessage(`Marketplace added successfully`);
       } catch (error) {
         setState((prev: PluginManagerState) => ({
           ...prev,
@@ -131,19 +159,20 @@ export function usePluginManager(): PluginManagerContextType {
         }));
       }
     },
-    [pluginCore, refresh],
+    [pluginCore, refresh, clearPluginFeedback, setSuccessMessage],
   );
 
   const removeMarketplace = useCallback(
     async (name: string) => {
+      clearPluginFeedback();
       setState((prev: PluginManagerState) => ({
         ...prev,
         isLoading: true,
-        error: null,
       }));
       try {
         await pluginCore.removeMarketplace(name);
         await refresh();
+        setSuccessMessage(`Marketplace '${name}' removed successfully`);
       } catch (error) {
         setState((prev: PluginManagerState) => ({
           ...prev,
@@ -152,19 +181,20 @@ export function usePluginManager(): PluginManagerContextType {
         }));
       }
     },
-    [pluginCore, refresh],
+    [pluginCore, refresh, clearPluginFeedback, setSuccessMessage],
   );
 
   const updateMarketplace = useCallback(
     async (name: string) => {
+      clearPluginFeedback();
       setState((prev: PluginManagerState) => ({
         ...prev,
         isLoading: true,
-        error: null,
       }));
       try {
         await pluginCore.updateMarketplace(name);
         await refresh();
+        setSuccessMessage(`Marketplace '${name}' updated successfully`);
       } catch (error) {
         setState((prev: PluginManagerState) => ({
           ...prev,
@@ -173,7 +203,7 @@ export function usePluginManager(): PluginManagerContextType {
         }));
       }
     },
-    [pluginCore, refresh],
+    [pluginCore, refresh, clearPluginFeedback, setSuccessMessage],
   );
 
   const installPlugin = useCallback(
@@ -182,15 +212,16 @@ export function usePluginManager(): PluginManagerContextType {
       marketplace: string,
       scope: "user" | "project" | "local" = "project",
     ) => {
+      clearPluginFeedback();
       setState((prev: PluginManagerState) => ({
         ...prev,
         isLoading: true,
-        error: null,
       }));
       try {
         const pluginId = `${name}@${marketplace}`;
         await pluginCore.installPlugin(pluginId, scope);
         await refresh();
+        setSuccessMessage(`Plugin '${name}' installed successfully`);
       } catch (error) {
         setState((prev: PluginManagerState) => ({
           ...prev,
@@ -199,20 +230,21 @@ export function usePluginManager(): PluginManagerContextType {
         }));
       }
     },
-    [pluginCore, refresh],
+    [pluginCore, refresh, clearPluginFeedback, setSuccessMessage],
   );
 
   const uninstallPlugin = useCallback(
     async (name: string, marketplace: string) => {
+      clearPluginFeedback();
       setState((prev: PluginManagerState) => ({
         ...prev,
         isLoading: true,
-        error: null,
       }));
       try {
         const pluginId = `${name}@${marketplace}`;
         await pluginCore.uninstallPlugin(pluginId);
         await refresh();
+        setSuccessMessage(`Plugin '${name}' uninstalled successfully`);
       } catch (error) {
         setState((prev: PluginManagerState) => ({
           ...prev,
@@ -221,20 +253,21 @@ export function usePluginManager(): PluginManagerContextType {
         }));
       }
     },
-    [pluginCore, refresh],
+    [pluginCore, refresh, clearPluginFeedback, setSuccessMessage],
   );
 
   const updatePlugin = useCallback(
     async (name: string, marketplace: string) => {
+      clearPluginFeedback();
       setState((prev: PluginManagerState) => ({
         ...prev,
         isLoading: true,
-        error: null,
       }));
       try {
         const pluginId = `${name}@${marketplace}`;
         await pluginCore.updatePlugin(pluginId);
         await refresh();
+        setSuccessMessage(`Plugin '${name}' updated successfully`);
       } catch (error) {
         setState((prev: PluginManagerState) => ({
           ...prev,
@@ -243,7 +276,7 @@ export function usePluginManager(): PluginManagerContextType {
         }));
       }
     },
-    [pluginCore, refresh],
+    [pluginCore, refresh, clearPluginFeedback, setSuccessMessage],
   );
 
   return {
@@ -261,6 +294,7 @@ export function usePluginManager(): PluginManagerContextType {
       uninstallPlugin,
       updatePlugin,
       refresh,
+      clearPluginFeedback,
     },
   };
 }

package/src/hooks/useTasks.ts

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