wave-code 0.10.1 → 0.10.3

package/src/contexts/useChat.tsx

@@ -125,6 +125,7 @@ export interface ChatProviderProps extends BaseAppProps {
 export const ChatProvider: React.FC<ChatProviderProps> = ({
   children,
   bypassPermissions,
+  permissionMode: initialPermissionMode,
   pluginDirs,
   tools,
   workdir,
@@ -180,8 +181,10 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   >({});
 
   // Permission state
-  const [permissionMode, setPermissionModeState] =
-    useState<PermissionMode>("default");
+  const [permissionMode, setPermissionModeState] = useState<PermissionMode>(
+    initialPermissionMode ||
+      (bypassPermissions ? "bypassPermissions" : "default"),
+  );
 
   // Confirmation state with queue-based architecture
   const [isConfirmationVisible, setIsConfirmationVisible] = useState(false);
@@ -223,6 +226,20 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   const [wasLastDetailsTooTall, setWasLastDetailsTooTall] = useState(0);
 
   const agentRef = useRef<Agent | null>(null);
+  const taskUpdateTimerRef = useRef<NodeJS.Timeout | null>(null);
+
+  const debouncedSetTasks = useCallback(
+    (newTasks: Task[]) => {
+      if (taskUpdateTimerRef.current) {
+        clearTimeout(taskUpdateTimerRef.current);
+      }
+      taskUpdateTimerRef.current = setTimeout(() => {
+        setTasks([...newTasks]);
+        taskUpdateTimerRef.current = null;
+      }, 100);
+    },
+    [setTasks],
+  );
 
   // Permission confirmation methods with queue support
   const showConfirmation = useCallback(
@@ -274,7 +291,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
           setBackgroundTasks([...tasks]);
         },
         onTasksChange: (tasks) => {
-          setTasks([...tasks]);
+          debouncedSetTasks(tasks);
         },
         onSubagentMessagesChange: (subagentId: string, messages: Message[]) => {
           logger.debug("onSubagentMessagesChange", subagentId, messages.length);
@@ -325,7 +342,9 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
           restoreSessionId,
           continueLastSession,
           logger,
-          permissionMode: bypassPermissions ? "bypassPermissions" : undefined,
+          permissionMode:
+            initialPermissionMode ||
+            (bypassPermissions ? "bypassPermissions" : undefined),
           canUseTool: permissionCallback,
           stream: false, // 关闭流式模式
           plugins: pluginDirs?.map((path) => ({ type: "local", path })),
@@ -371,11 +390,17 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     workdir,
     worktreeSession,
     model,
+    debouncedSetTasks,
+    initialPermissionMode,
   ]);
 
   // Cleanup on unmount
   useEffect(() => {
     return () => {
+      if (taskUpdateTimerRef.current) {
+        clearTimeout(taskUpdateTimerRef.current);
+      }
+
       if (agentRef.current) {
         try {
           // Display usage summary before cleanup

package/src/index.ts

@@ -1,7 +1,7 @@
 import yargs from "yargs";
 import { hideBin } from "yargs/helpers";
 import { startCli } from "./cli.js";
-import { Scope, generateRandomName } from "wave-agent-sdk";
+import { Scope, generateRandomName, type PermissionMode } from "wave-agent-sdk";
 import { createWorktree, type WorktreeSession } from "./utils/worktree.js";
 import path from "path";
 import { readFileSync } from "fs";
@@ -53,6 +53,18 @@ export async function main() {
         default: false,
         global: false,
       })
+      .option("permission-mode", {
+        description: "Permission mode to use for the session",
+        choices: [
+          "acceptEdits",
+          "bypassPermissions",
+          "default",
+          "dontAsk",
+          "plan",
+        ],
+        type: "string",
+        global: false,
+      })
       .option("plugin-dir", {
         description: "Load a plugin from a specific directory",
         type: "array",
@@ -274,7 +286,8 @@ export async function main() {
       // Continue with the selected session
       return startCli({
         restoreSessionId: selectedSessionId,
-        bypassPermissions: argv.dangerouslySkipPermissions,
+        bypassPermissions: argv.dangerouslySkipPermissions as boolean,
+        permissionMode: argv.permissionMode as PermissionMode | undefined,
         pluginDirs,
         tools,
         worktreeSession,
@@ -295,6 +308,7 @@ export async function main() {
         bypassPermissions: argv.dangerouslySkipPermissions as
           | boolean
           | undefined,
+        permissionMode: argv.permissionMode as PermissionMode | undefined,
         pluginDirs,
         tools,
         worktreeSession,
@@ -308,6 +322,7 @@ export async function main() {
       restoreSessionId: argv.restore as string | undefined,
       continueLastSession: argv.continue as boolean | undefined,
       bypassPermissions: argv.dangerouslySkipPermissions as boolean | undefined,
+      permissionMode: argv.permissionMode as PermissionMode | undefined,
       pluginDirs,
       tools,
       worktreeSession,

package/src/print-cli.ts

@@ -38,6 +38,7 @@ export async function startPrintCli(options: PrintCliOptions): Promise<void> {
     message,
     showStats = false,
     bypassPermissions,
+    permissionMode,
     pluginDirs,
     tools,
     worktreeSession,
@@ -144,7 +145,8 @@ export async function startPrintCli(options: PrintCliOptions): Promise<void> {
       callbacks,
       restoreSessionId,
       continueLastSession,
-      permissionMode: bypassPermissions ? "bypassPermissions" : undefined,
+      permissionMode:
+        permissionMode || (bypassPermissions ? "bypassPermissions" : undefined),
       plugins: pluginDirs?.map((path) => ({ type: "local", path })),
       tools,
       workdir,

package/src/types.ts

@@ -1,7 +1,9 @@
 import { WorktreeSession } from "./utils/worktree.js";
+import { PermissionMode } from "wave-agent-sdk";
 
 export interface BaseAppProps {
   bypassPermissions?: boolean;
+  permissionMode?: PermissionMode;
   pluginDirs?: string[];
   tools?: string[];
   worktreeSession?: WorktreeSession;

package/src/utils/highlightUtils.ts

@@ -49,6 +49,14 @@ function nodeToAnsi(node: Node): string {
 
     for (const className of classes) {
       if (theme[className]) {
+        // If content has newlines, split it and apply style to each line
+        // to ensure ANSI codes are correctly applied when splitting the final string by lines.
+        if (content.includes("\n")) {
+          return content
+            .split("\n")
+            .map((line) => theme[className](line))
+            .join("\n");
+        }
         return theme[className](content);
       }
     }

package/src/utils/toolParameterTransforms.ts

@@ -3,11 +3,7 @@
  * Forces type judgment based on tool name using type assertions
  */
 
-import {
-  type Change,
-  type WriteToolParameters,
-  type EditToolParameters,
-} from "wave-agent-sdk";
+import { type Change, type EditToolParameters } from "wave-agent-sdk";
 import { logger } from "./logger.js";
 
 /**
@@ -26,20 +22,6 @@ function parseToolParameters(parameters: string): unknown {
   }
 }
 
-/**
- * Transform Write tool parameters to changes
- */
-export function transformWriteParameters(
-  parameters: WriteToolParameters,
-): Change[] {
-  return [
-    {
-      oldContent: "", // No previous content for write operations
-      newContent: parameters.content,
-    },
-  ];
-}
-
 /**
  * Transform Edit tool parameters to changes
  */
@@ -72,10 +54,6 @@ export function transformToolBlockToChanges(
 
     let changes: Change[] = [];
     switch (toolName) {
-      case "Write":
-        changes = transformWriteParameters(parsedParams as WriteToolParameters);
-        break;
-
       case "Edit":
         changes = transformEditParameters(parsedParams as EditToolParameters);
         break;

package/src/utils/worktree.ts

@@ -1,7 +1,7 @@
 import { execSync } from "node:child_process";
 import * as path from "node:path";
 import * as fs from "node:fs";
-import { getGitRepoRoot, getDefaultRemoteBranch } from "wave-agent-sdk";
+import { getDefaultRemoteBranch, getGitMainRepoRoot } from "wave-agent-sdk";
 
 export interface WorktreeSession {
   name: string;
@@ -22,7 +22,7 @@ export const WORKTREE_DIR = ".wave/worktrees";
  * @returns Worktree session details
  */
 export function createWorktree(name: string, cwd: string): WorktreeSession {
-  const repoRoot = getGitRepoRoot(cwd);
+  const repoRoot = getGitMainRepoRoot(cwd);
   const worktreePath = path.join(repoRoot, WORKTREE_DIR, name);
   const branchName = `worktree-${name}`;
   const baseBranch = getDefaultRemoteBranch(cwd);