wave-code 0.6.4 → 0.7.0

package/src/components/App.tsx

@@ -9,14 +9,20 @@ interface AppProps {
   continueLastSession?: boolean;
   bypassPermissions?: boolean;
   pluginDirs?: string[];
+  tools?: string[];
 }
 
 const AppWithProviders: React.FC<{
   bypassPermissions?: boolean;
   pluginDirs?: string[];
-}> = ({ bypassPermissions, pluginDirs }) => {
+  tools?: string[];
+}> = ({ bypassPermissions, pluginDirs, tools }) => {
   return (
-    <ChatProvider bypassPermissions={bypassPermissions} pluginDirs={pluginDirs}>
+    <ChatProvider
+      bypassPermissions={bypassPermissions}
+      pluginDirs={pluginDirs}
+      tools={tools}
+    >
       <ChatInterfaceWithRemount />
     </ChatProvider>
   );
@@ -74,6 +80,7 @@ export const App: React.FC<AppProps> = ({
   continueLastSession,
   bypassPermissions,
   pluginDirs,
+  tools,
 }) => {
   return (
     <AppProvider
@@ -83,6 +90,7 @@ export const App: React.FC<AppProps> = ({
       <AppWithProviders
         bypassPermissions={bypassPermissions}
         pluginDirs={pluginDirs}
+        tools={tools}
       />
     </AppProvider>
   );

package/src/components/BackgroundTaskManager.tsx

@@ -23,6 +23,7 @@ export const BackgroundTaskManager: React.FC<BackgroundTaskManagerProps> = ({
     useChat();
   const [tasks, setTasks] = useState<Task[]>([]);
   const [selectedIndex, setSelectedIndex] = useState(0);
+  const MAX_VISIBLE_ITEMS = 3;
   const [viewMode, setViewMode] = useState<"list" | "detail">("list");
   const [detailTaskId, setDetailTaskId] = useState<string | null>(null);
   const [detailOutput, setDetailOutput] = useState<{
@@ -70,6 +71,16 @@ export const BackgroundTaskManager: React.FC<BackgroundTaskManagerProps> = ({
     stopBackgroundTask(taskId);
   };
 
+  // Calculate visible window
+  const startIndex = Math.max(
+    0,
+    Math.min(
+      selectedIndex - Math.floor(MAX_VISIBLE_ITEMS / 2),
+      Math.max(0, tasks.length - MAX_VISIBLE_ITEMS),
+    ),
+  );
+  const visibleTasks = tasks.slice(startIndex, startIndex + MAX_VISIBLE_ITEMS);
+
   useInput((input, key) => {
     if (viewMode === "list") {
       // List mode navigation
@@ -273,40 +284,46 @@ export const BackgroundTaskManager: React.FC<BackgroundTaskManagerProps> = ({
       </Box>
       <Text dimColor>Select a task to view details</Text>
 
-      {tasks.map((task, index) => (
-        <Box key={task.id} flexDirection="column">
-          <Text
-            color={index === selectedIndex ? "black" : "white"}
-            backgroundColor={index === selectedIndex ? "cyan" : undefined}
-          >
-            {index === selectedIndex ? "▶ " : "  "}
-            {index + 1}. [{task.id}] {task.type}
-            {task.description ? `: ${task.description}` : ""}
-            <Text
-              color={
-                task.status === "running"
-                  ? "green"
-                  : task.status === "completed"
-                    ? "blue"
-                    : "red"
-              }
-            >
-              {" "}
-              ({task.status})
-            </Text>
-          </Text>
-          {index === selectedIndex && (
-            <Box marginLeft={4} flexDirection="column">
-              <Text color="gray" dimColor>
-                Started: {formatTime(task.startTime)}
-                {task.runtime !== undefined &&
-                  ` | Runtime: ${formatDuration(task.runtime)}`}
-                {task.exitCode !== undefined && ` | Exit: ${task.exitCode}`}
+      <Box flexDirection="column">
+        {visibleTasks.map((task, index) => {
+          const actualIndex = startIndex + index;
+          const isSelected = actualIndex === selectedIndex;
+          return (
+            <Box key={task.id} flexDirection="column">
+              <Text
+                color={isSelected ? "black" : "white"}
+                backgroundColor={isSelected ? "cyan" : undefined}
+              >
+                {isSelected ? "▶ " : "  "}
+                {actualIndex + 1}. [{task.id}] {task.type}
+                {task.description ? `: ${task.description}` : ""}
+                <Text
+                  color={
+                    task.status === "running"
+                      ? "green"
+                      : task.status === "completed"
+                        ? "blue"
+                        : "red"
+                  }
+                >
+                  {" "}
+                  ({task.status})
+                </Text>
               </Text>
+              {isSelected && (
+                <Box marginLeft={4} flexDirection="column">
+                  <Text color="gray" dimColor>
+                    Started: {formatTime(task.startTime)}
+                    {task.runtime !== undefined &&
+                      ` | Runtime: ${formatDuration(task.runtime)}`}
+                    {task.exitCode !== undefined && ` | Exit: ${task.exitCode}`}
+                  </Text>
+                </Box>
+              )}
             </Box>
-          )}
-        </Box>
-      ))}
+          );
+        })}
+      </Box>
 
       <Box marginTop={1}>
         <Text dimColor>

package/src/components/ConfirmationDetails.tsx

@@ -3,8 +3,6 @@ import { Box, Text, useStdout, measureElement, Static } from "ink";
 import {
   BASH_TOOL_NAME,
   EDIT_TOOL_NAME,
-  MULTI_EDIT_TOOL_NAME,
-  DELETE_FILE_TOOL_NAME,
   WRITE_TOOL_NAME,
   EXIT_PLAN_MODE_TOOL_NAME,
   ASK_USER_QUESTION_TOOL_NAME,
@@ -26,10 +24,6 @@ const getActionDescription = (
       return `Execute command: ${toolInput.command || "unknown command"}`;
     case EDIT_TOOL_NAME:
       return `Edit file: ${toolInput.file_path || "unknown file"}`;
-    case MULTI_EDIT_TOOL_NAME:
-      return `Edit multiple sections in: ${toolInput.file_path || "unknown file"}`;
-    case DELETE_FILE_TOOL_NAME:
-      return `Delete file: ${toolInput.target_file || "unknown file"}`;
     case WRITE_TOOL_NAME:
       return `Write to file: ${toolInput.file_path || "unknown file"}`;
     case EXIT_PLAN_MODE_TOOL_NAME:

package/src/components/DiffDisplay.tsx

@@ -1,10 +1,6 @@
 import React, { useMemo } from "react";
 import { Box, Text } from "ink";
-import {
-  WRITE_TOOL_NAME,
-  EDIT_TOOL_NAME,
-  MULTI_EDIT_TOOL_NAME,
-} from "wave-agent-sdk";
+import { WRITE_TOOL_NAME, EDIT_TOOL_NAME } from "wave-agent-sdk";
 import { transformToolBlockToChanges } from "../utils/toolParameterTransforms.js";
 import { diffLines, diffWords } from "diff";
 
@@ -18,8 +14,7 @@ export const DiffDisplay: React.FC<DiffDisplayProps> = ({
   parameters,
 }) => {
   const showDiff =
-    toolName &&
-    [WRITE_TOOL_NAME, EDIT_TOOL_NAME, MULTI_EDIT_TOOL_NAME].includes(toolName);
+    toolName && [WRITE_TOOL_NAME, EDIT_TOOL_NAME].includes(toolName);
 
   // Diff detection and transformation using typed parameters
   const changes = useMemo(() => {
@@ -109,15 +104,6 @@ export const DiffDisplay: React.FC<DiffDisplayProps> = ({
 
       changes.forEach((change, changeIndex) => {
         try {
-          // Add ellipsis between non-contiguous edits in MultiEdit
-          if (toolName === MULTI_EDIT_TOOL_NAME && changeIndex > 0) {
-            allElements.push(
-              <Box key={`multi-edit-separator-${changeIndex}`}>
-                <Text color="gray">...</Text>
-              </Box>,
-            );
-          }
-
           // Get line-level diff to understand the structure
           const lineDiffs = diffLines(
             change.oldContent || "",

package/src/components/MessageItem.tsx

@@ -25,14 +25,10 @@ export const MessageItem = ({ message, isExpanded }: MessageItemProps) => {
             {block.type === "text" && block.content.trim() && (
               <Box>
                 {block.customCommandContent && (
-                  <Text color="cyan" bold>
-                    $
-                  </Text>
+                  <Text color="cyan" bold>$ </Text>
                 )}
                 {block.source === MessageSource.HOOK && (
-                  <Text color="magenta" bold>
-                    ~
-                  </Text>
+                  <Text color="magenta" bold>~ </Text>
                 )}
                 {message.role === "user" ? (
                   <Text backgroundColor="gray" color="white">

package/src/components/MessageList.tsx

@@ -1,12 +1,6 @@
 import React from "react";
 import { Box, Text, Static } from "ink";
 import type { Message } from "wave-agent-sdk";
-import {
-  TASK_CREATE_TOOL_NAME,
-  TASK_GET_TOOL_NAME,
-  TASK_UPDATE_TOOL_NAME,
-  TASK_LIST_TOOL_NAME,
-} from "wave-agent-sdk";
 import { MessageItem } from "./MessageItem.js";
 
 export interface MessageListProps {
@@ -40,31 +34,9 @@ export const MessageList = React.memo(
     const maxExpandedMessages = 20;
     const shouldLimitMessages =
       isExpanded && messages.length > maxExpandedMessages;
-
-    // Filter out task management tools and empty messages
-    const taskMgmtTools = [
-      TASK_CREATE_TOOL_NAME,
-      TASK_GET_TOOL_NAME,
-      TASK_UPDATE_TOOL_NAME,
-      TASK_LIST_TOOL_NAME,
-    ];
-    const filteredMessages = messages
-      .map((message) => ({
-        ...message,
-        blocks: message.blocks.filter(
-          (block) =>
-            !(
-              block.type === "tool" &&
-              typeof block.name === "string" &&
-              taskMgmtTools.includes(block.name)
-            ),
-        ),
-      }))
-      .filter((message) => message.blocks.length > 0);
-
     const displayMessages = shouldLimitMessages
-      ? filteredMessages.slice(-maxExpandedMessages)
-      : filteredMessages;
+      ? messages.slice(-maxExpandedMessages)
+      : messages;
 
     // Compute which messages to render statically vs dynamically
     const lastMessage = displayMessages[displayMessages.length - 1];

package/src/contexts/useChat.tsx

@@ -108,12 +108,14 @@ export interface ChatProviderProps {
   children: React.ReactNode;
   bypassPermissions?: boolean;
   pluginDirs?: string[];
+  tools?: string[];
 }
 
 export const ChatProvider: React.FC<ChatProviderProps> = ({
   children,
   bypassPermissions,
   pluginDirs,
+  tools,
 }) => {
   const { restoreSessionId, continueLastSession } = useAppConfig();
 
@@ -304,6 +306,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
           canUseTool: permissionCallback,
           stream: false, // 关闭流式模式
           plugins: pluginDirs?.map((path) => ({ type: "local", path })),
+          tools,
         });
 
         agentRef.current = agent;
@@ -336,6 +339,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     bypassPermissions,
     showConfirmation,
     pluginDirs,
+    tools,
   ]);
 
   // Cleanup on unmount

package/src/hooks/usePluginManager.ts

@@ -1,9 +1,6 @@
 import { useState, useCallback, useEffect, useMemo } from "react";
 import {
-  MarketplaceService,
-  PluginScopeManager,
-  ConfigurationService,
-  PluginManager,
+  PluginCore,
   KnownMarketplace,
   InstalledPlugin,
   MarketplacePluginEntry,
@@ -35,25 +32,7 @@ export function usePluginManager(): PluginManagerContextType {
     })[]
   >([]);
 
-  const marketplaceService = useMemo(() => new MarketplaceService(), []);
-  const configurationService = useMemo(() => new ConfigurationService(), []);
-  const pluginManager = useMemo(
-    () =>
-      new PluginManager({
-        workdir: process.cwd(),
-        configurationService,
-      }),
-    [configurationService],
-  );
-  const pluginScopeManager = useMemo(
-    () =>
-      new PluginScopeManager({
-        workdir: process.cwd(),
-        configurationService,
-        pluginManager,
-      }),
-    [configurationService, pluginManager],
-  );
+  const pluginCore = useMemo(() => new PluginCore(), []);
 
   const refresh = useCallback(async () => {
     setState((prev: PluginManagerState) => ({
@@ -63,9 +42,9 @@ export function usePluginManager(): PluginManagerContextType {
     }));
     try {
       const [mks, installed, enabledMap] = await Promise.all([
-        marketplaceService.listMarketplaces(),
-        marketplaceService.getInstalledPlugins(),
-        Promise.resolve(pluginScopeManager.getMergedEnabledPlugins()),
+        pluginCore.listMarketplaces(),
+        pluginCore.getInstalledPlugins(),
+        Promise.resolve(pluginCore.getMergedEnabledPlugins()),
       ]);
 
       setMarketplaces(mks);
@@ -74,7 +53,7 @@ export function usePluginManager(): PluginManagerContextType {
         return {
           ...p,
           enabled: !!enabledMap[pluginId],
-          scope: pluginScopeManager.findPluginScope(pluginId) || undefined,
+          scope: pluginCore.findPluginScope(pluginId) || undefined,
         };
       });
 
@@ -88,8 +67,8 @@ export function usePluginManager(): PluginManagerContextType {
       })[] = [];
       for (const mk of mks) {
         try {
-          const manifest = await marketplaceService.loadMarketplaceManifest(
-            marketplaceService.getMarketplacePath(mk),
+          const manifest = await pluginCore.loadMarketplaceManifest(
+            pluginCore.getMarketplacePath(mk),
           );
           manifest.plugins.forEach((p) => {
             const pluginId = `${p.name}@${mk.name}`;
@@ -120,7 +99,7 @@ export function usePluginManager(): PluginManagerContextType {
         error: error instanceof Error ? error.message : String(error),
       }));
     }
-  }, [marketplaceService, pluginScopeManager]);
+  }, [pluginCore]);
 
   useEffect(() => {
     refresh();
@@ -142,7 +121,7 @@ export function usePluginManager(): PluginManagerContextType {
         error: null,
       }));
       try {
-        await marketplaceService.addMarketplace(source);
+        await pluginCore.addMarketplace(source);
         await refresh();
       } catch (error) {
         setState((prev: PluginManagerState) => ({
@@ -152,7 +131,7 @@ export function usePluginManager(): PluginManagerContextType {
         }));
       }
     },
-    [marketplaceService, refresh],
+    [pluginCore, refresh],
   );
 
   const removeMarketplace = useCallback(
@@ -163,7 +142,7 @@ export function usePluginManager(): PluginManagerContextType {
         error: null,
       }));
       try {
-        await marketplaceService.removeMarketplace(name);
+        await pluginCore.removeMarketplace(name);
         await refresh();
       } catch (error) {
         setState((prev: PluginManagerState) => ({
@@ -173,7 +152,7 @@ export function usePluginManager(): PluginManagerContextType {
         }));
       }
     },
-    [marketplaceService, refresh],
+    [pluginCore, refresh],
   );
 
   const updateMarketplace = useCallback(
@@ -184,7 +163,7 @@ export function usePluginManager(): PluginManagerContextType {
         error: null,
       }));
       try {
-        await marketplaceService.updateMarketplace(name);
+        await pluginCore.updateMarketplace(name);
         await refresh();
       } catch (error) {
         setState((prev: PluginManagerState) => ({
@@ -194,7 +173,7 @@ export function usePluginManager(): PluginManagerContextType {
         }));
       }
     },
-    [marketplaceService, refresh],
+    [pluginCore, refresh],
   );
 
   const installPlugin = useCallback(
@@ -210,9 +189,7 @@ export function usePluginManager(): PluginManagerContextType {
       }));
       try {
         const pluginId = `${name}@${marketplace}`;
-        const workdir = process.cwd();
-        await marketplaceService.installPlugin(pluginId, workdir);
-        await pluginScopeManager.enablePlugin(scope, pluginId);
+        await pluginCore.installPlugin(pluginId, scope);
         await refresh();
       } catch (error) {
         setState((prev: PluginManagerState) => ({
@@ -222,7 +199,7 @@ export function usePluginManager(): PluginManagerContextType {
         }));
       }
     },
-    [marketplaceService, pluginScopeManager, refresh],
+    [pluginCore, refresh],
   );
 
   const uninstallPlugin = useCallback(
@@ -234,20 +211,7 @@ export function usePluginManager(): PluginManagerContextType {
       }));
       try {
         const pluginId = `${name}@${marketplace}`;
-        const workdir = process.cwd();
-
-        // 1. Remove from global registry and potentially clean up cache
-        await marketplaceService.uninstallPlugin(pluginId, workdir);
-
-        // 2. Find the scope where it's currently enabled and remove it from there
-        const scope = pluginScopeManager.findPluginScope(pluginId);
-        if (scope) {
-          await configurationService.removeEnabledPlugin(
-            workdir,
-            scope,
-            pluginId,
-          );
-        }
+        await pluginCore.uninstallPlugin(pluginId);
         await refresh();
       } catch (error) {
         setState((prev: PluginManagerState) => ({
@@ -257,7 +221,7 @@ export function usePluginManager(): PluginManagerContextType {
         }));
       }
     },
-    [configurationService, marketplaceService, pluginScopeManager, refresh],
+    [pluginCore, refresh],
   );
 
   const updatePlugin = useCallback(
@@ -269,7 +233,7 @@ export function usePluginManager(): PluginManagerContextType {
       }));
       try {
         const pluginId = `${name}@${marketplace}`;
-        await marketplaceService.updatePlugin(pluginId);
+        await pluginCore.updatePlugin(pluginId);
         await refresh();
       } catch (error) {
         setState((prev: PluginManagerState) => ({
@@ -279,7 +243,7 @@ export function usePluginManager(): PluginManagerContextType {
         }));
       }
     },
-    [marketplaceService, refresh],
+    [pluginCore, refresh],
   );
 
   return {

package/src/index.ts

@@ -43,6 +43,12 @@ export async function main() {
         string: true,
         global: false,
       })
+      .option("tools", {
+        description:
+          'Specify a comma-separated list of tools to enable (e.g., \'Bash,Read,Write\'). Use "" to disable all, "default" for all.',
+        type: "string",
+        global: false,
+      })
       .command(
         "plugin",
         "Manage plugins and marketplaces",
@@ -217,6 +223,14 @@ export async function main() {
       .strict()
       .parseAsync();
 
+    const parseTools = (tools: string | undefined): string[] | undefined => {
+      if (tools === undefined || tools === "default") return undefined;
+      if (tools === "") return [];
+      return tools.split(",").map((t) => t.trim());
+    };
+
+    const tools = parseTools(argv.tools as string | undefined);
+
     // Handle restore session command
     if (
       argv.restore === "" ||
@@ -236,6 +250,7 @@ export async function main() {
         restoreSessionId: selectedSessionId,
         bypassPermissions: argv.dangerouslySkipPermissions,
         pluginDirs: argv.pluginDir as string[],
+        tools,
       });
     }
 
@@ -249,6 +264,7 @@ export async function main() {
         showStats: argv.showStats,
         bypassPermissions: argv.dangerouslySkipPermissions,
         pluginDirs: argv.pluginDir as string[],
+        tools,
       });
     }
 
@@ -257,6 +273,7 @@ export async function main() {
       continueLastSession: argv.continue,
       bypassPermissions: argv.dangerouslySkipPermissions,
       pluginDirs: argv.pluginDir as string[],
+      tools,
     });
   } catch (error) {
     console.error("Failed to start WAVE Code:", error);

package/src/print-cli.ts

@@ -8,6 +8,7 @@ export interface PrintCliOptions {
   showStats?: boolean;
   bypassPermissions?: boolean;
   pluginDirs?: string[];
+  tools?: string[];
 }
 
 function displayTimingInfo(startTime: Date, showStats: boolean): void {
@@ -33,6 +34,7 @@ export async function startPrintCli(options: PrintCliOptions): Promise<void> {
     showStats = false,
     bypassPermissions,
     pluginDirs,
+    tools,
   } = options;
 
   if (
@@ -136,6 +138,7 @@ export async function startPrintCli(options: PrintCliOptions): Promise<void> {
       continueLastSession,
       permissionMode: bypassPermissions ? "bypassPermissions" : undefined,
       plugins: pluginDirs?.map((path) => ({ type: "local", path })),
+      tools,
       // 保持流式模式以获得更好的命令行用户体验
     });
 

package/src/utils/toolParameterTransforms.ts

@@ -7,7 +7,6 @@ import {
   type Change,
   type WriteToolParameters,
   type EditToolParameters,
-  type MultiEditToolParameters,
 } from "wave-agent-sdk";
 import { logger } from "./logger.js";
 
@@ -55,18 +54,6 @@ export function transformEditParameters(
   ];
 }
 
-/**
- * Transform MultiEdit tool parameters to changes
- */
-export function transformMultiEditParameters(
-  parameters: MultiEditToolParameters,
-): Change[] {
-  return parameters.edits.map((edit) => ({
-    oldContent: edit.old_string,
-    newContent: edit.new_string,
-  }));
-}
-
 /**
  * Transform tool block parameters into standardized Change[] array for diff display
  * Forces type judgment based on tool name using type assertions
@@ -89,11 +76,6 @@ export function transformToolBlockToChanges(
       case "Edit":
         return transformEditParameters(parsedParams as EditToolParameters);
 
-      case "MultiEdit":
-        return transformMultiEditParameters(
-          parsedParams as MultiEditToolParameters,
-        );
-
       default:
         return [];
     }