wave-code 0.6.5 → 0.7.1

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/highlightUtils.ts

@@ -0,0 +1,76 @@
+import hljs from 'highlight.js';
+import { parse, Node, HTMLElement, TextNode } from 'node-html-parser';
+import chalk from 'chalk';
+
+const theme: Record<string, (text: string) => string> = {
+  'hljs-keyword': chalk.blue,
+  'hljs-built_in': chalk.cyan,
+  'hljs-type': chalk.cyan,
+  'hljs-literal': chalk.magenta,
+  'hljs-number': chalk.magenta,
+  'hljs-operator': chalk.white,
+  'hljs-punctuation': chalk.white,
+  'hljs-property': chalk.yellow,
+  'hljs-attr': chalk.yellow,
+  'hljs-variable': chalk.white,
+  'hljs-template-variable': chalk.white,
+  'hljs-string': chalk.green,
+  'hljs-char': chalk.green,
+  'hljs-comment': chalk.gray,
+  'hljs-doctag': chalk.gray,
+  'hljs-function': chalk.yellow,
+  'hljs-title': chalk.yellow,
+  'hljs-params': chalk.white,
+  'hljs-tag': chalk.blue,
+  'hljs-name': chalk.blue,
+  'hljs-selector-tag': chalk.blue,
+  'hljs-selector-id': chalk.blue,
+  'hljs-selector-class': chalk.blue,
+  'hljs-selector-attr': chalk.blue,
+  'hljs-selector-pseudo': chalk.blue,
+  'hljs-subst': chalk.white,
+  'hljs-section': chalk.blue.bold,
+  'hljs-bullet': chalk.magenta,
+  'hljs-emphasis': chalk.italic,
+  'hljs-strong': chalk.bold,
+  'hljs-addition': chalk.green,
+  'hljs-deletion': chalk.red,
+  'hljs-link': chalk.blue.underline,
+};
+
+function nodeToAnsi(node: Node): string {
+  if (node instanceof TextNode) {
+    return node.text;
+  }
+
+  if (node instanceof HTMLElement) {
+    const content = node.childNodes.map(nodeToAnsi).join('');
+    const classes = node.getAttribute('class')?.split(/\s+/) || [];
+
+    for (const className of classes) {
+      if (theme[className]) {
+        return theme[className](content);
+      }
+    }
+
+    return content;
+  }
+
+  return '';
+}
+
+export function highlightToAnsi(code: string, language?: string): string {
+  if (!code) {
+    return '';
+  }
+  try {
+    const highlighted = language
+      ? hljs.highlight(code, { language }).value
+      : hljs.highlightAuto(code).value;
+
+    const root = parse(highlighted);
+    return root.childNodes.map(nodeToAnsi).join('');
+  } catch {
+    return code;
+  }
+}

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
@@ -74,6 +61,7 @@ export function transformMultiEditParameters(
 export function transformToolBlockToChanges(
   toolName: string,
   parameters: string,
+  startLineNumber?: number,
 ): Change[] {
   try {
     if (!toolName) {
@@ -82,21 +70,24 @@ export function transformToolBlockToChanges(
 
     const parsedParams = parseToolParameters(parameters);
 
+    let changes: Change[] = [];
     switch (toolName) {
       case "Write":
-        return transformWriteParameters(parsedParams as WriteToolParameters);
+        changes = transformWriteParameters(parsedParams as WriteToolParameters);
+        break;
 
       case "Edit":
-        return transformEditParameters(parsedParams as EditToolParameters);
-
-      case "MultiEdit":
-        return transformMultiEditParameters(
-          parsedParams as MultiEditToolParameters,
-        );
+        changes = transformEditParameters(parsedParams as EditToolParameters);
+        break;
 
       default:
         return [];
     }
+
+    if (changes.length > 0 && startLineNumber !== undefined) {
+      changes[0].startLineNumber = startLineNumber;
+    }
+    return changes;
   } catch (error) {
     logger.warn("Failed to transform tool block to changes:", error);
     return [];

package/dist/components/MessageItem.d.ts

@@ -1,8 +0,0 @@
-import type { Message } from "wave-agent-sdk";
-export interface MessageItemProps {
-    message: Message;
-    isExpanded: boolean;
-    shouldShowHeader: boolean;
-}
-export declare const MessageItem: ({ message, isExpanded }: MessageItemProps) => import("react/jsx-runtime").JSX.Element | null;
-//# sourceMappingURL=MessageItem.d.ts.map

package/dist/components/MessageItem.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"MessageItem.d.ts","sourceRoot":"","sources":["../../src/components/MessageItem.tsx"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAQ9C,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,EAAE,OAAO,CAAC;IACpB,gBAAgB,EAAE,OAAO,CAAC;CAC3B;AAED,eAAO,MAAM,WAAW,GAAI,yBAAyB,gBAAgB,mDAgEpE,CAAC"}

package/dist/components/MessageItem.js

@@ -1,13 +0,0 @@
-import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
-import { Box, Text } from "ink";
-import { MessageSource } from "wave-agent-sdk";
-import { CommandOutputDisplay } from "./CommandOutputDisplay.js";
-import { ToolDisplay } from "./ToolDisplay.js";
-import { CompressDisplay } from "./CompressDisplay.js";
-import { ReasoningDisplay } from "./ReasoningDisplay.js";
-import { Markdown } from "./Markdown.js";
-export const MessageItem = ({ message, isExpanded }) => {
-    if (message.blocks.length === 0)
-        return null;
-    return (_jsx(Box, { flexDirection: "column", gap: 1, marginTop: 1, children: _jsx(Box, { flexDirection: "column", gap: 1, children: message.blocks.map((block, blockIndex) => (_jsxs(Box, { children: [block.type === "text" && block.content.trim() && (_jsxs(Box, { children: [block.customCommandContent && (_jsx(Text, { color: "cyan", bold: true, children: "$ " })), block.source === MessageSource.HOOK && (_jsx(Text, { color: "magenta", bold: true, children: "~ " })), message.role === "user" ? (_jsx(Text, { backgroundColor: "gray", color: "white", children: block.content })) : (_jsx(Markdown, { children: block.content }))] })), block.type === "error" && (_jsx(Box, { children: _jsxs(Text, { color: "red", children: ["Error: ", block.content] }) })), block.type === "command_output" && (_jsx(CommandOutputDisplay, { block: block, isExpanded: isExpanded })), block.type === "tool" && (_jsx(ToolDisplay, { block: block, isExpanded: isExpanded })), block.type === "image" && (_jsxs(Box, { children: [_jsx(Text, { color: "magenta", bold: true, children: "# Image" }), block.imageUrls && block.imageUrls.length > 0 && (_jsxs(Text, { color: "gray", dimColor: true, children: [" ", "(", block.imageUrls.length, ")"] }))] })), block.type === "compress" && (_jsx(CompressDisplay, { block: block, isExpanded: isExpanded })), block.type === "reasoning" && _jsx(ReasoningDisplay, { block: block })] }, blockIndex))) }) }));
-};

package/src/components/MessageItem.tsx

@@ -1,81 +0,0 @@
-import React from "react";
-import { Box, Text } from "ink";
-import type { Message } from "wave-agent-sdk";
-import { MessageSource } from "wave-agent-sdk";
-import { CommandOutputDisplay } from "./CommandOutputDisplay.js";
-import { ToolDisplay } from "./ToolDisplay.js";
-import { CompressDisplay } from "./CompressDisplay.js";
-import { ReasoningDisplay } from "./ReasoningDisplay.js";
-import { Markdown } from "./Markdown.js";
-
-export interface MessageItemProps {
-  message: Message;
-  isExpanded: boolean;
-  shouldShowHeader: boolean;
-}
-
-export const MessageItem = ({ message, isExpanded }: MessageItemProps) => {
-  if (message.blocks.length === 0) return null;
-
-  return (
-    <Box flexDirection="column" gap={1} marginTop={1}>
-      <Box flexDirection="column" gap={1}>
-        {message.blocks.map((block, blockIndex) => (
-          <Box key={blockIndex}>
-            {block.type === "text" && block.content.trim() && (
-              <Box>
-                {block.customCommandContent && (
-                  <Text color="cyan" bold>$ </Text>
-                )}
-                {block.source === MessageSource.HOOK && (
-                  <Text color="magenta" bold>~ </Text>
-                )}
-                {message.role === "user" ? (
-                  <Text backgroundColor="gray" color="white">
-                    {block.content}
-                  </Text>
-                ) : (
-                  <Markdown>{block.content}</Markdown>
-                )}
-              </Box>
-            )}
-
-            {block.type === "error" && (
-              <Box>
-                <Text color="red">Error: {block.content}</Text>
-              </Box>
-            )}
-
-            {block.type === "command_output" && (
-              <CommandOutputDisplay block={block} isExpanded={isExpanded} />
-            )}
-
-            {block.type === "tool" && (
-              <ToolDisplay block={block} isExpanded={isExpanded} />
-            )}
-
-            {block.type === "image" && (
-              <Box>
-                <Text color="magenta" bold>
-                  # Image
-                </Text>
-                {block.imageUrls && block.imageUrls.length > 0 && (
-                  <Text color="gray" dimColor>
-                    {" "}
-                    ({block.imageUrls.length})
-                  </Text>
-                )}
-              </Box>
-            )}
-
-            {block.type === "compress" && (
-              <CompressDisplay block={block} isExpanded={isExpanded} />
-            )}
-
-            {block.type === "reasoning" && <ReasoningDisplay block={block} />}
-          </Box>
-        ))}
-      </Box>
-    </Box>
-  );
-};