wave-code 0.10.1 → 0.10.3

package/dist/print-cli.js

@@ -14,7 +14,7 @@ function displayTimingInfo(startTime, showStats) {
 }
 export async function startPrintCli(options) {
     const startTime = new Date();
-    const { restoreSessionId, continueLastSession, message, showStats = false, bypassPermissions, pluginDirs, tools, worktreeSession, workdir, model, } = options;
+    const { restoreSessionId, continueLastSession, message, showStats = false, bypassPermissions, permissionMode, pluginDirs, tools, worktreeSession, workdir, model, } = options;
     if ((!message || message.trim() === "") &&
         !continueLastSession &&
         !restoreSessionId) {
@@ -96,7 +96,7 @@ export async function startPrintCli(options) {
             callbacks,
             restoreSessionId,
             continueLastSession,
-            permissionMode: bypassPermissions ? "bypassPermissions" : undefined,
+            permissionMode: permissionMode || (bypassPermissions ? "bypassPermissions" : undefined),
             plugins: pluginDirs?.map((path) => ({ type: "local", path })),
             tools,
             workdir,

package/dist/types.d.ts

@@ -1,6 +1,8 @@
 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/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAEtD,MAAM,WAAW,YAAY;IAC3B,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;IACjB,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAEhD,MAAM,WAAW,YAAY;IAC3B,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;IACjB,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB"}

package/dist/utils/highlightUtils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"highlightUtils.d.ts","sourceRoot":"","sources":["../../src/utils/highlightUtils.ts"],"names":[],"mappings":"AA6DA,wBAAgB,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,MAAM,CAsCvE"}
+{"version":3,"file":"highlightUtils.d.ts","sourceRoot":"","sources":["../../src/utils/highlightUtils.ts"],"names":[],"mappings":"AAqEA,wBAAgB,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,MAAM,CAsCvE"}

package/dist/utils/highlightUtils.js

@@ -45,6 +45,14 @@ function nodeToAnsi(node) {
         const classes = node.getAttribute("class")?.split(/\s+/) || [];
         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/dist/utils/toolParameterTransforms.d.ts

@@ -2,11 +2,7 @@
  * Tool parameter transformation utilities for UI rendering
  * Forces type judgment based on tool name using type assertions
  */
-import { type Change, type WriteToolParameters, type EditToolParameters } from "wave-agent-sdk";
-/**
- * Transform Write tool parameters to changes
- */
-export declare function transformWriteParameters(parameters: WriteToolParameters): Change[];
+import { type Change, type EditToolParameters } from "wave-agent-sdk";
 /**
  * Transform Edit tool parameters to changes
  */

package/dist/utils/toolParameterTransforms.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"toolParameterTransforms.d.ts","sourceRoot":"","sources":["../../src/utils/toolParameterTransforms.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EACL,KAAK,MAAM,EACX,KAAK,mBAAmB,EACxB,KAAK,kBAAkB,EACxB,MAAM,gBAAgB,CAAC;AAmBxB;;GAEG;AACH,wBAAgB,wBAAwB,CACtC,UAAU,EAAE,mBAAmB,GAC9B,MAAM,EAAE,CAOV;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CACrC,UAAU,EAAE,kBAAkB,GAC7B,MAAM,EAAE,CAOV;AAED;;;GAGG;AACH,wBAAgB,2BAA2B,CACzC,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,eAAe,CAAC,EAAE,MAAM,GACvB,MAAM,EAAE,CA8BV"}
+{"version":3,"file":"toolParameterTransforms.d.ts","sourceRoot":"","sources":["../../src/utils/toolParameterTransforms.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,KAAK,MAAM,EAAE,KAAK,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAmBtE;;GAEG;AACH,wBAAgB,uBAAuB,CACrC,UAAU,EAAE,kBAAkB,GAC7B,MAAM,EAAE,CAOV;AAED;;;GAGG;AACH,wBAAgB,2BAA2B,CACzC,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,eAAe,CAAC,EAAE,MAAM,GACvB,MAAM,EAAE,CA0BV"}

package/dist/utils/toolParameterTransforms.js

@@ -18,17 +18,6 @@ function parseToolParameters(parameters) {
         return {};
     }
 }
-/**
- * Transform Write tool parameters to changes
- */
-export function transformWriteParameters(parameters) {
-    return [
-        {
-            oldContent: "", // No previous content for write operations
-            newContent: parameters.content,
-        },
-    ];
-}
 /**
  * Transform Edit tool parameters to changes
  */
@@ -52,9 +41,6 @@ export function transformToolBlockToChanges(toolName, parameters, startLineNumbe
         const parsedParams = parseToolParameters(parameters);
         let changes = [];
         switch (toolName) {
-            case "Write":
-                changes = transformWriteParameters(parsedParams);
-                break;
             case "Edit":
                 changes = transformEditParameters(parsedParams);
                 break;

package/dist/utils/worktree.js

@@ -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 const WORKTREE_DIR = ".wave/worktrees";
 /**
  * Create a new git worktree
@@ -10,7 +10,7 @@ export const WORKTREE_DIR = ".wave/worktrees";
  * @returns Worktree session details
  */
 export function createWorktree(name, cwd) {
-    const repoRoot = getGitRepoRoot(cwd);
+    const repoRoot = getGitMainRepoRoot(cwd);
     const worktreePath = path.join(repoRoot, WORKTREE_DIR, name);
     const branchName = `worktree-${name}`;
     const baseBranch = getDefaultRemoteBranch(cwd);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wave-code",
-  "version": "0.10.1",
+  "version": "0.10.3",
   "description": "CLI-based code assistant powered by AI, built with React and Ink",
   "repository": {
     "type": "git",
@@ -41,7 +41,7 @@
     "yargs": "^17.7.2",
     "@agentclientprotocol/sdk": "0.16.1",
     "zod": "^3.23.8",
-    "wave-agent-sdk": "0.10.1"
+    "wave-agent-sdk": "0.10.3"
   },
   "devDependencies": {
     "@types/react": "^19.1.8",

package/src/acp/agent.ts

@@ -6,7 +6,9 @@ import {
   AgentToolBlockUpdateParams,
   Task,
   listSessions as listWaveSessions,
+  listAllSessions as listAllWaveSessions,
   deleteSession as deleteWaveSession,
+  truncateContent,
 } from "wave-agent-sdk";
 import * as fs from "node:fs/promises";
 import * as path from "node:path";
@@ -137,6 +139,7 @@ export class WaveAcpAgent implements AcpAgent {
     const agent = await WaveAgent.create({
       workdir: cwd,
       restoreSessionId: sessionId,
+      stream: false,
       canUseTool: (context) => {
         if (!agentRef.instance) {
           throw new Error("Agent instance not yet initialized");
@@ -170,19 +173,11 @@ export class WaveAcpAgent implements AcpAgent {
     // Update the callbacks object with the correct sessionId
     Object.assign(callbacks, this.createCallbacks(actualSessionId));
 
-    return agent;
-  }
-
-  async newSession(params: NewSessionRequest): Promise<NewSessionResponse> {
-    const { cwd } = params;
-    logger.info(`Creating new session in ${cwd}`);
-    const agent = await this.createAgent(undefined, cwd);
-    logger.info(`New session created with ID: ${agent.sessionId}`);
-
     // Send initial available commands after agent creation
+    // Use setImmediate to ensure the client receives the session response before the update
     setImmediate(() => {
       this.connection.sessionUpdate({
-        sessionId: agent.sessionId as AcpSessionId,
+        sessionId: actualSessionId as AcpSessionId,
         update: {
           sessionUpdate: "available_commands_update",
           availableCommands: agent.getSlashCommands().map((cmd) => ({
@@ -196,6 +191,15 @@ export class WaveAcpAgent implements AcpAgent {
       });
     });
 
+    return agent;
+  }
+
+  async newSession(params: NewSessionRequest): Promise<NewSessionResponse> {
+    const { cwd } = params;
+    logger.info(`Creating new session in ${cwd}`);
+    const agent = await this.createAgent(undefined, cwd);
+    logger.info(`New session created with ID: ${agent.sessionId}`);
+
     return {
       sessionId: agent.sessionId as AcpSessionId,
       modes: this.getSessionModeState(agent),
@@ -208,23 +212,6 @@ export class WaveAcpAgent implements AcpAgent {
     logger.info(`Loading session: ${sessionId} in ${cwd}`);
     const agent = await this.createAgent(sessionId, cwd);
 
-    // Send initial available commands after agent creation
-    setImmediate(() => {
-      this.connection.sessionUpdate({
-        sessionId: agent.sessionId as AcpSessionId,
-        update: {
-          sessionUpdate: "available_commands_update",
-          availableCommands: agent.getSlashCommands().map((cmd) => ({
-            name: cmd.name,
-            description: cmd.description,
-            input: {
-              hint: "Enter arguments...",
-            },
-          })),
-        },
-      });
-    });
-
     return {
       modes: this.getSessionModeState(agent),
       configOptions: this.getSessionConfigOptions(agent),
@@ -236,17 +223,21 @@ export class WaveAcpAgent implements AcpAgent {
   ): Promise<ListSessionsResponse> {
     const { cwd } = params;
     logger.info(`listSessions called with params: ${JSON.stringify(params)}`);
+
+    let waveSessions;
     if (!cwd) {
-      logger.warn("listSessions called without cwd, returning empty list");
-      return { sessions: [] };
+      logger.info("listSessions called without cwd, listing all sessions");
+      waveSessions = await listAllWaveSessions();
+    } else {
+      logger.info(`Listing sessions for ${cwd}`);
+      waveSessions = await listWaveSessions(cwd);
     }
 
-    logger.info(`Listing sessions for ${cwd}`);
-    const waveSessions = await listWaveSessions(cwd);
-    logger.info(`Found ${waveSessions.length} sessions for ${cwd}`);
+    logger.info(`Found ${waveSessions.length} sessions`);
     const sessions: SessionInfo[] = waveSessions.map((meta) => ({
       sessionId: meta.id as AcpSessionId,
       cwd: meta.workdir,
+      title: meta.firstMessage ? truncateContent(meta.firstMessage) : undefined,
       updatedAt: meta.lastActiveAt.toISOString(),
     }));
     return { sessions };
@@ -378,6 +369,27 @@ export class WaveAcpAgent implements AcpAgent {
       context.toolCallId ||
       "perm-" + Math.random().toString(36).substring(2, 9);
 
+    let effectiveName = context.toolName;
+    let effectiveCompactParams: string | undefined = undefined;
+
+    if (agent?.messages && context.toolCallId) {
+      const toolBlock = agent.messages
+        .flatMap((m) => m.blocks)
+        .find((b) => b.type === "tool" && b.id === context.toolCallId) as
+        | import("wave-agent-sdk").ToolBlock
+        | undefined;
+      if (toolBlock) {
+        effectiveName = toolBlock.name || effectiveName;
+        effectiveCompactParams =
+          toolBlock.compactParams || effectiveCompactParams;
+      }
+    }
+
+    const displayTitle =
+      effectiveName && effectiveCompactParams
+        ? `${effectiveName}: ${effectiveCompactParams}`
+        : effectiveName || "Tool Call";
+
     const options: PermissionOption[] = [
       {
         optionId: "allow_once",
@@ -394,11 +406,6 @@ export class WaveAcpAgent implements AcpAgent {
         name: "Reject Once",
         kind: "reject_once",
       },
-      {
-        optionId: "reject_always",
-        name: "Reject Always",
-        kind: "reject_always",
-      },
     ];
 
     const content = context.toolName
@@ -420,7 +427,7 @@ export class WaveAcpAgent implements AcpAgent {
         sessionId: sessionId as AcpSessionId,
         toolCall: {
           toolCallId,
-          title: `Permission for ${context.toolName}`,
+          title: displayTitle,
           status: "pending",
           rawInput: context.toolInput,
           content,
@@ -438,21 +445,15 @@ export class WaveAcpAgent implements AcpAgent {
       logger.info(`User selected permission option: ${selectedOptionId}`);
 
       switch (selectedOptionId) {
-        case "allow_once":
-          return { behavior: "allow" };
         case "allow_always":
           return {
             behavior: "allow",
             newPermissionRule: `${context.toolName}(*)`,
           };
+        case "allow_once":
+          return { behavior: "allow" };
         case "reject_once":
           return { behavior: "deny", message: "Rejected by user" };
-        case "reject_always":
-          return {
-            behavior: "deny",
-            message: "Rejected by user",
-            newPermissionRule: `!${context.toolName}(*)`,
-          };
         default:
           return { behavior: "deny", message: "Unknown option selected" };
       }
@@ -474,7 +475,8 @@ export class WaveAcpAgent implements AcpAgent {
     if (name === "Write") {
       let oldText: string | null = null;
       try {
-        const filePath = parameters.file_path as string;
+        const filePath = (parameters.file_path ||
+          parameters.filePath) as string;
         const fullPath = path.isAbsolute(filePath)
           ? filePath
           : path.join(workdir, filePath);
@@ -485,7 +487,7 @@ export class WaveAcpAgent implements AcpAgent {
       return [
         {
           type: "diff",
-          path: parameters.file_path as string,
+          path: (parameters.file_path || parameters.filePath) as string,
           oldText,
           newText: parameters.content as string,
         },
@@ -495,7 +497,8 @@ export class WaveAcpAgent implements AcpAgent {
       let oldText: string | null = null;
       let newText: string | null = null;
       try {
-        const filePath = parameters.file_path as string;
+        const filePath = (parameters.file_path ||
+          parameters.filePath) as string;
         const fullPath = path.isAbsolute(filePath)
           ? filePath
           : path.join(workdir, filePath);
@@ -520,7 +523,7 @@ export class WaveAcpAgent implements AcpAgent {
         return [
           {
             type: "diff",
-            path: parameters.file_path as string,
+            path: (parameters.file_path || parameters.filePath) as string,
             oldText,
             newText,
           },
@@ -531,55 +534,83 @@ export class WaveAcpAgent implements AcpAgent {
       return [
         {
           type: "diff",
-          path: parameters.file_path as string,
+          path: (parameters.file_path || parameters.filePath) as string,
           oldText: parameters.old_string as string,
           newText: parameters.new_string as string,
         },
       ];
     }
-    return this.getToolContent(name, parameters);
+    return this.getToolContent(name, parameters, undefined);
   }
 
   private getToolContent(
     name: string,
     parameters: Record<string, unknown> | undefined,
+    shortResult: string | undefined,
   ): ToolCallContent[] | undefined {
-    if (!parameters) return undefined;
-    if (name === "Write") {
-      return [
-        {
+    const contents: ToolCallContent[] = [];
+    if (parameters) {
+      if (name === "Write") {
+        contents.push({
           type: "diff",
-          path: parameters.file_path as string,
+          path: (parameters.file_path || parameters.filePath) as string,
           oldText: null,
           newText: parameters.content as string,
-        },
-      ];
-    }
-    if (name === "Edit") {
-      return [
-        {
+        });
+      } else if (name === "Edit") {
+        contents.push({
           type: "diff",
-          path: parameters.file_path as string,
+          path: (parameters.file_path || parameters.filePath) as string,
           oldText: parameters.old_string as string,
           newText: parameters.new_string as string,
+        });
+      }
+    }
+
+    if (shortResult) {
+      contents.push({
+        type: "content",
+        content: {
+          type: "text",
+          text: shortResult,
         },
-      ];
+      });
     }
-    return undefined;
+
+    return contents.length > 0 ? contents : undefined;
   }
 
   private getToolLocations(
     name: string,
     parameters: Record<string, unknown> | undefined,
+    extraStartLineNumber?: number,
   ): ToolCallLocation[] | undefined {
     if (!parameters) return undefined;
-    if (name === "Write" || name === "Edit" || name === "Read") {
-      return [
-        {
-          path: parameters.file_path as string,
-          line: parameters.offset as number,
-        },
-      ];
+    if (
+      name === "Write" ||
+      name === "Edit" ||
+      name === "Read" ||
+      name === "LSP"
+    ) {
+      const filePath = (parameters.file_path || parameters.filePath) as string;
+      let line =
+        extraStartLineNumber ??
+        (parameters.startLineNumber as number) ??
+        (parameters.line as number) ??
+        (parameters.offset as number);
+
+      if (name === "Write" && line === undefined) {
+        line = 1;
+      }
+
+      if (filePath) {
+        return [
+          {
+            path: filePath,
+            line: line,
+          },
+        ];
+      }
     }
     return undefined;
   }
@@ -605,6 +636,15 @@ export class WaveAcpAgent implements AcpAgent {
 
   private createCallbacks(sessionId: string): AgentOptions["callbacks"] {
     const getAgent = () => this.agents.get(sessionId);
+    const toolStates = new Map<
+      string,
+      {
+        name?: string;
+        compactParams?: string;
+        shortResult?: string;
+        startLineNumber?: number;
+      }
+    >();
     return {
       onAssistantContentUpdated: (chunk: string) => {
         this.connection.sessionUpdate({
@@ -631,7 +671,42 @@ export class WaveAcpAgent implements AcpAgent {
         });
       },
       onToolBlockUpdated: (params: AgentToolBlockUpdateParams) => {
-        const { id, name, stage, success, error, result, parameters } = params;
+        const {
+          id,
+          name,
+          stage,
+          success,
+          error,
+          result,
+          parameters,
+          compactParams,
+          shortResult,
+          startLineNumber,
+        } = params;
+
+        let state = toolStates.get(id);
+        if (!state) {
+          state = {};
+          toolStates.set(id, state);
+        }
+        if (name) state.name = name;
+        if (compactParams) state.compactParams = compactParams;
+        if (shortResult) state.shortResult = shortResult;
+        if (startLineNumber !== undefined)
+          state.startLineNumber = startLineNumber;
+
+        const effectiveName = state.name || name;
+        const effectiveCompactParams = state.compactParams || compactParams;
+        const effectiveShortResult = state.shortResult || shortResult;
+        const effectiveStartLineNumber =
+          state.startLineNumber !== undefined
+            ? state.startLineNumber
+            : startLineNumber;
+
+        const displayTitle =
+          effectiveName && effectiveCompactParams
+            ? `${effectiveName}: ${effectiveCompactParams}`
+            : effectiveName || "Tool Call";
 
         let parsedParameters: Record<string, unknown> | undefined = undefined;
         if (parameters) {
@@ -643,14 +718,24 @@ export class WaveAcpAgent implements AcpAgent {
         }
 
         const content =
-          name && parsedParameters
-            ? this.getToolContent(name, parsedParameters)
+          effectiveName && (parsedParameters || effectiveShortResult)
+            ? this.getToolContent(
+                effectiveName,
+                parsedParameters,
+                effectiveShortResult,
+              )
             : undefined;
         const locations =
-          name && parsedParameters
-            ? this.getToolLocations(name, parsedParameters)
+          effectiveName && parsedParameters
+            ? this.getToolLocations(
+                effectiveName,
+                parsedParameters,
+                effectiveStartLineNumber,
+              )
             : undefined;
-        const kind = name ? this.getToolKind(name) : undefined;
+        const kind = effectiveName
+          ? this.getToolKind(effectiveName)
+          : undefined;
 
         if (stage === "start") {
           this.connection.sessionUpdate({
@@ -658,7 +743,7 @@ export class WaveAcpAgent implements AcpAgent {
             update: {
               sessionUpdate: "tool_call",
               toolCallId: id,
-              title: name || "Tool Call",
+              title: displayTitle,
               status: "pending",
               content,
               locations,
@@ -689,7 +774,7 @@ export class WaveAcpAgent implements AcpAgent {
             sessionUpdate: "tool_call_update",
             toolCallId: id,
             status,
-            title: name || "Tool Call",
+            title: displayTitle,
             rawOutput: result || error,
             content,
             locations,
@@ -697,6 +782,10 @@ export class WaveAcpAgent implements AcpAgent {
             rawInput: parsedParameters,
           },
         });
+
+        if (stage === "end") {
+          toolStates.delete(id);
+        }
       },
       onTasksChange: (tasks) => {
         this.connection.sessionUpdate({

package/src/cli.tsx

@@ -15,6 +15,7 @@ export async function startCli(options: CliOptions): Promise<void> {
     restoreSessionId,
     continueLastSession,
     bypassPermissions,
+    permissionMode,
     pluginDirs,
     tools,
     worktreeSession,
@@ -37,6 +38,7 @@ export async function startCli(options: CliOptions): Promise<void> {
       restoreSessionId={restoreSessionId}
       continueLastSession={continueLastSession}
       bypassPermissions={bypassPermissions}
+      permissionMode={permissionMode}
       pluginDirs={pluginDirs}
       tools={tools}
       worktreeSession={worktreeSession}

package/src/components/App.tsx

@@ -23,6 +23,7 @@ interface AppWithProvidersProps extends BaseAppProps {
 
 const AppWithProviders: React.FC<AppWithProvidersProps> = ({
   bypassPermissions,
+  permissionMode,
   pluginDirs,
   tools,
   worktreeSession,
@@ -94,6 +95,7 @@ const AppWithProviders: React.FC<AppWithProvidersProps> = ({
   return (
     <ChatProvider
       bypassPermissions={bypassPermissions}
+      permissionMode={permissionMode}
       pluginDirs={pluginDirs}
       tools={tools}
       workdir={workdir}
@@ -157,6 +159,7 @@ export const App: React.FC<AppProps> = ({
   restoreSessionId,
   continueLastSession,
   bypassPermissions,
+  permissionMode,
   pluginDirs,
   tools,
   worktreeSession,
@@ -172,6 +175,7 @@ export const App: React.FC<AppProps> = ({
     >
       <AppWithProviders
         bypassPermissions={bypassPermissions}
+        permissionMode={permissionMode}
         pluginDirs={pluginDirs}
         tools={tools}
         worktreeSession={worktreeSession}

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