wave-code 0.10.0 → 0.10.2

package/src/components/ChatInterface.tsx

@@ -72,7 +72,7 @@ export const ChatInterface: React.FC = () => {
 
     const terminalHeight = stdout?.rows || 24;
     const totalHeight = detailsHeight + selectorHeight + dynamicBlocksHeight;
-    if (totalHeight > terminalHeight) {
+    if (totalHeight > terminalHeight - 3) {
       setIsConfirmationTooTall(true);
     }
   }, [

package/src/components/DiffDisplay.tsx

@@ -1,8 +1,14 @@
 import React, { useMemo } from "react";
 import { Box, Text } from "ink";
-import { WRITE_TOOL_NAME, EDIT_TOOL_NAME } from "wave-agent-sdk";
+import {
+  WRITE_TOOL_NAME,
+  EDIT_TOOL_NAME,
+  type WriteToolParameters,
+} from "wave-agent-sdk";
 import { transformToolBlockToChanges } from "../utils/toolParameterTransforms.js";
 import { diffLines, diffWords } from "diff";
+import path from "path";
+import { Markdown } from "./Markdown.js";
 
 interface DiffDisplayProps {
   toolName?: string;
@@ -97,6 +103,23 @@ export const DiffDisplay: React.FC<DiffDisplayProps> = ({
     }
   };
 
+  // Render highlighted code for Write tool
+  const renderWriteContent = () => {
+    if (!parameters) return null;
+    try {
+      const parsed = JSON.parse(parameters) as WriteToolParameters;
+      const content = parsed.content || "";
+      const filePath = parsed.file_path || "";
+      const ext = path.extname(filePath).slice(1);
+
+      const markdown = `\`\`\`${ext}\n${content}\n\`\`\``;
+      return <Markdown>{markdown}</Markdown>;
+    } catch (error) {
+      console.warn("Error rendering write content:", error);
+      return null;
+    }
+  };
+
   // Render expanded diff display
   const renderExpandedDiff = () => {
     try {
@@ -352,7 +375,9 @@ export const DiffDisplay: React.FC<DiffDisplayProps> = ({
   return (
     <Box flexDirection="column">
       <Box paddingLeft={2} borderLeft borderColor="cyan" flexDirection="column">
-        {renderExpandedDiff()}
+        {toolName === WRITE_TOOL_NAME
+          ? renderWriteContent()
+          : renderExpandedDiff()}
       </Box>
     </Box>
   );

package/src/contexts/useChat.tsx

@@ -223,6 +223,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 +288,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);
@@ -295,9 +309,6 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
         onPermissionModeChange: (mode) => {
           setPermissionModeState(mode);
         },
-        onSlashCommandsChange: (commands) => {
-          setSlashCommands([...commands]);
-        },
       };
 
       try {
@@ -374,11 +385,16 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
     workdir,
     worktreeSession,
     model,
+    debouncedSetTasks,
   ]);
 
   // Cleanup on unmount
   useEffect(() => {
     return () => {
+      if (taskUpdateTimerRef.current) {
+        clearTimeout(taskUpdateTimerRef.current);
+      }
+
       if (agentRef.current) {
         try {
           // Display usage summary before cleanup

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);