wave-code 0.7.0 → 0.7.1

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

@@ -1 +1 @@
-{"version":3,"file":"SessionSelector.d.ts","sourceRoot":"","sources":["../../src/components/SessionSelector.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAmB,MAAM,OAAO,CAAC;AAExC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEtD,MAAM,WAAW,oBAAoB;IACnC,QAAQ,EAAE,CAAC,eAAe,GAAG;QAAE,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,EAAE,CAAC;IAC1D,QAAQ,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,IAAI,CAAC;IACtC,QAAQ,EAAE,MAAM,IAAI,CAAC;CACtB;AAED,eAAO,MAAM,eAAe,EAAE,KAAK,CAAC,EAAE,CAAC,oBAAoB,CAoH1D,CAAC"}
+{"version":3,"file":"SessionSelector.d.ts","sourceRoot":"","sources":["../../src/components/SessionSelector.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAmB,MAAM,OAAO,CAAC;AAExC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEtD,MAAM,WAAW,oBAAoB;IACnC,QAAQ,EAAE,CAAC,eAAe,GAAG;QAAE,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,EAAE,CAAC;IAC1D,QAAQ,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,IAAI,CAAC;IACtC,QAAQ,EAAE,MAAM,IAAI,CAAC;CACtB;AAED,eAAO,MAAM,eAAe,EAAE,KAAK,CAAC,EAAE,CAAC,oBAAoB,CAuH1D,CAAC"}

package/dist/components/SessionSelector.js

@@ -5,9 +5,12 @@ export const SessionSelector = ({ sessions, onSelect, onCancel, }) => {
     const [selectedIndex, setSelectedIndex] = useState(0);
     useInput((input, key) => {
         if (key.return) {
-            if (sessions.length > 0 && selectedIndex < sessions.length) {
-                onSelect(sessions[selectedIndex].id);
-            }
+            setSelectedIndex((prev) => {
+                if (sessions.length > 0 && prev < sessions.length) {
+                    onSelect(sessions[prev].id);
+                }
+                return prev;
+            });
             return;
         }
         if (key.escape) {
@@ -15,11 +18,11 @@ export const SessionSelector = ({ sessions, onSelect, onCancel, }) => {
             return;
         }
         if (key.upArrow) {
-            setSelectedIndex(Math.max(0, selectedIndex - 1));
+            setSelectedIndex((prev) => Math.max(0, prev - 1));
             return;
         }
         if (key.downArrow) {
-            setSelectedIndex(Math.min(sessions.length - 1, selectedIndex + 1));
+            setSelectedIndex((prev) => Math.min(sessions.length - 1, prev + 1));
             return;
         }
     });

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

@@ -1 +1 @@
-{"version":3,"file":"TaskList.d.ts","sourceRoot":"","sources":["../../src/components/TaskList.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAK1B,eAAO,MAAM,QAAQ,EAAE,KAAK,CAAC,EAoD5B,CAAC"}
+{"version":3,"file":"TaskList.d.ts","sourceRoot":"","sources":["../../src/components/TaskList.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAK1B,eAAO,MAAM,QAAQ,EAAE,KAAK,CAAC,EAiD5B,CAAC"}

package/dist/components/TaskList.js

@@ -8,10 +8,7 @@ export const TaskList = () => {
     if (tasks.length === 0 || !isTaskListVisible) {
         return null;
     }
-    const getStatusIcon = (status, isBlocked) => {
-        if (isBlocked) {
-            return _jsx(Text, { color: "red", children: "\uD83D\uDD12" });
-        }
+    const getStatusIcon = (status) => {
         switch (status) {
             case "pending":
                 return _jsx(Text, { color: "gray", children: "\u25A1" });
@@ -35,6 +32,6 @@ export const TaskList = () => {
                 ? ` (Blocked by: ${blockingTaskIds.join(", ")})`
                 : "";
             const fullText = `${task.subject}${blockedByText}`;
-            return (_jsxs(Box, { gap: 1, children: [getStatusIcon(task.status, isBlocked), _jsx(Text, { dimColor: isDimmed, children: fullText })] }, task.id));
+            return (_jsxs(Box, { gap: 1, children: [getStatusIcon(task.status), _jsx(Text, { dimColor: isDimmed, children: fullText })] }, task.id));
         }) }));
 };

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

@@ -1 +1 @@
-{"version":3,"file":"ToolDisplay.d.ts","sourceRoot":"","sources":["../../src/components/ToolDisplay.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAGhD,UAAU,gBAAgB;IACxB,KAAK,EAAE,SAAS,CAAC;IACjB,UAAU,CAAC,EAAE,OAAO,CAAC;CACtB;AAED,eAAO,MAAM,WAAW,EAAE,KAAK,CAAC,EAAE,CAAC,gBAAgB,CA+HlD,CAAC"}
+{"version":3,"file":"ToolDisplay.d.ts","sourceRoot":"","sources":["../../src/components/ToolDisplay.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAGhD,UAAU,gBAAgB;IACxB,KAAK,EAAE,SAAS,CAAC;IACjB,UAAU,CAAC,EAAE,OAAO,CAAC;CACtB;AAED,eAAO,MAAM,WAAW,EAAE,KAAK,CAAC,EAAE,CAAC,gBAAgB,CAmIlD,CAAC"}

package/dist/components/ToolDisplay.js

@@ -40,5 +40,5 @@ export const ToolDisplay = ({ block, isExpanded = false, }) => {
         return null;
     };
     const shortResult = getShortResult();
-    return (_jsxs(Box, { flexDirection: "column", children: [_jsxs(Box, { children: [_jsxs(Box, { flexShrink: 0, children: [_jsx(Text, { color: getStatusColor(), children: "\u25CF " }), _jsx(Text, { color: "white", children: toolName })] }), !isExpanded && compactParams && (_jsxs(Text, { color: "gray", children: [" ", compactParams] })), hasImages() && _jsxs(Text, { color: "blue", children: [" ", getImageIndicator()] })] }), !isExpanded && shortResult && !error && (_jsx(Box, { paddingLeft: 2, borderLeft: true, borderColor: "gray", flexDirection: "column", children: shortResult.split("\n").map((line, index) => (_jsx(Text, { color: "gray", children: line }, index))) })), isExpanded && parameters && (_jsxs(Box, { paddingLeft: 2, borderLeft: true, borderColor: "gray", flexDirection: "column", children: [_jsx(Text, { color: "cyan", bold: true, children: "Parameters:" }), _jsx(Text, { color: "gray", children: parameters })] })), isExpanded && result && (_jsx(Box, { flexDirection: "column", children: _jsxs(Box, { paddingLeft: 2, borderLeft: true, borderColor: "green", flexDirection: "column", children: [_jsx(Text, { color: "cyan", bold: true, children: "Result:" }), _jsx(Text, { color: "white", children: result })] }) })), error && (_jsx(Box, { children: _jsxs(Text, { color: "red", children: ["Error: ", typeof error === "string" ? error : String(error)] }) })), stage === "end" && success && (_jsx(DiffDisplay, { toolName: name, parameters: parameters }))] }));
+    return (_jsxs(Box, { flexDirection: "column", children: [_jsxs(Box, { children: [_jsxs(Box, { flexShrink: 0, children: [_jsx(Text, { color: getStatusColor(), children: "\u25CF " }), _jsx(Text, { color: "white", children: toolName })] }), !isExpanded && compactParams && (_jsxs(Text, { color: "gray", children: [" ", compactParams] })), hasImages() && _jsxs(Text, { color: "blue", children: [" ", getImageIndicator()] })] }), !isExpanded && shortResult && !error && (_jsx(Box, { paddingLeft: 2, borderLeft: true, borderColor: "gray", flexDirection: "column", children: shortResult.split("\n").map((line, index) => (_jsx(Text, { color: "gray", children: line }, index))) })), isExpanded && parameters && (_jsxs(Box, { paddingLeft: 2, borderLeft: true, borderColor: "gray", flexDirection: "column", children: [_jsx(Text, { color: "cyan", bold: true, children: "Parameters:" }), _jsx(Text, { color: "gray", children: parameters })] })), isExpanded && result && (_jsx(Box, { flexDirection: "column", children: _jsxs(Box, { paddingLeft: 2, borderLeft: true, borderColor: "green", flexDirection: "column", children: [_jsx(Text, { color: "cyan", bold: true, children: "Result:" }), _jsx(Text, { color: "white", children: result })] }) })), error && (_jsx(Box, { children: _jsxs(Text, { color: "red", children: ["Error: ", typeof error === "string" ? error : String(error)] }) })), stage === "end" && success && (_jsx(DiffDisplay, { toolName: name, parameters: parameters, startLineNumber: block.startLineNumber }))] }));
 };

package/dist/contexts/useChat.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useChat.d.ts","sourceRoot":"","sources":["../../src/contexts/useChat.tsx"],"names":[],"mappings":"AAAA,OAAO,KAON,MAAM,OAAO,CAAC;AAGf,OAAO,KAAK,EACV,OAAO,EACP,eAAe,EACf,cAAc,EACd,IAAI,EACJ,YAAY,EACZ,kBAAkB,EAClB,cAAc,EACf,MAAM,gBAAgB,CAAC;AAUxB,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,SAAS,EAAE,OAAO,CAAC;IACnB,gBAAgB,EAAE,OAAO,CAAC;IAC1B,aAAa,EAAE,OAAO,CAAC;IAEvB,UAAU,EAAE,OAAO,CAAC;IACpB,iBAAiB,EAAE,OAAO,CAAC;IAC3B,oBAAoB,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IAEjD,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,CACX,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,KAC/C,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,YAAY,EAAE,MAAM,IAAI,CAAC;IACzB,iBAAiB,EAAE,MAAM,CAAC;IAE1B,UAAU,EAAE,eAAe,EAAE,CAAC;IAC9B,gBAAgB,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IAC3D,mBAAmB,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IAE9D,eAAe,EAAE,cAAc,EAAE,CAAC;IAElC,YAAY,EAAE,IAAI,EAAE,CAAC;IACrB,uBAAuB,EAAE,CACvB,MAAM,EAAE,MAAM,KACX;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;IAC/D,kBAAkB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC;IAEhD,aAAa,EAAE,YAAY,EAAE,CAAC;IAC9B,eAAe,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,OAAO,CAAC;IAEhD,gBAAgB,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;IAC5C,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAE7C,cAAc,EAAE,cAAc,CAAC;IAC/B,iBAAiB,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,CAAC;IAElD,qBAAqB,EAAE,OAAO,CAAC;IAC/B,cAAc,CAAC,EAAE;QACf,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAChC,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,oBAAoB,CAAC,EAAE,OAAO,CAAC;KAChC,CAAC;IACF,gBAAgB,EAAE,CAChB,QAAQ,EAAE,MAAM,EAChB,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACnC,eAAe,CAAC,EAAE,MAAM,EACxB,oBAAoB,CAAC,EAAE,OAAO,KAC3B,OAAO,CAAC,kBAAkB,CAAC,CAAC;IACjC,gBAAgB,EAAE,MAAM,IAAI,CAAC;IAC7B,0BAA0B,EAAE,CAAC,QAAQ,EAAE,kBAAkB,KAAK,IAAI,CAAC;IACnE,wBAAwB,EAAE,MAAM,IAAI,CAAC;IAErC,qBAAqB,EAAE,MAAM,IAAI,CAAC;IAElC,QAAQ,EAAE,MAAM,CAAC;IACjB,kBAAkB,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACrD,oBAAoB,EAAE,MAAM,OAAO,CAAC;QAClC,QAAQ,EAAE,OAAO,EAAE,CAAC;QACpB,UAAU,EAAE,MAAM,EAAE,CAAC;KACtB,CAAC,CAAC;IACH,qBAAqB,EAAE,MAAM,CAAC;IAC9B,wBAAwB,EAAE,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;CACxE;AAID,eAAO,MAAM,OAAO,uBAMnB,CAAC;AAEF,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,KAAK,CAAC,SAAS,CAAC;IAC1B,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;CAClB;AAED,eAAO,MAAM,YAAY,EAAE,KAAK,CAAC,EAAE,CAAC,iBAAiB,CAyepD,CAAC"}
+{"version":3,"file":"useChat.d.ts","sourceRoot":"","sources":["../../src/contexts/useChat.tsx"],"names":[],"mappings":"AAAA,OAAO,KAON,MAAM,OAAO,CAAC;AAGf,OAAO,KAAK,EACV,OAAO,EACP,eAAe,EACf,cAAc,EACd,IAAI,EACJ,YAAY,EACZ,kBAAkB,EAClB,cAAc,EACf,MAAM,gBAAgB,CAAC;AAUxB,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,SAAS,EAAE,OAAO,CAAC;IACnB,gBAAgB,EAAE,OAAO,CAAC;IAC1B,aAAa,EAAE,OAAO,CAAC;IAEvB,UAAU,EAAE,OAAO,CAAC;IACpB,iBAAiB,EAAE,OAAO,CAAC;IAC3B,oBAAoB,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IAEjD,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,CACX,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,KAC/C,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,YAAY,EAAE,MAAM,IAAI,CAAC;IACzB,iBAAiB,EAAE,MAAM,CAAC;IAE1B,UAAU,EAAE,eAAe,EAAE,CAAC;IAC9B,gBAAgB,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IAC3D,mBAAmB,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IAE9D,eAAe,EAAE,cAAc,EAAE,CAAC;IAElC,YAAY,EAAE,IAAI,EAAE,CAAC;IACrB,uBAAuB,EAAE,CACvB,MAAM,EAAE,MAAM,KACX;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;IAC/D,kBAAkB,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC;IAEhD,aAAa,EAAE,YAAY,EAAE,CAAC;IAC9B,eAAe,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,OAAO,CAAC;IAEhD,gBAAgB,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;IAC5C,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAE7C,cAAc,EAAE,cAAc,CAAC;IAC/B,iBAAiB,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,CAAC;IAElD,qBAAqB,EAAE,OAAO,CAAC;IAC/B,cAAc,CAAC,EAAE;QACf,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAChC,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,oBAAoB,CAAC,EAAE,OAAO,CAAC;KAChC,CAAC;IACF,gBAAgB,EAAE,CAChB,QAAQ,EAAE,MAAM,EAChB,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACnC,eAAe,CAAC,EAAE,MAAM,EACxB,oBAAoB,CAAC,EAAE,OAAO,KAC3B,OAAO,CAAC,kBAAkB,CAAC,CAAC;IACjC,gBAAgB,EAAE,MAAM,IAAI,CAAC;IAC7B,0BAA0B,EAAE,CAAC,QAAQ,EAAE,kBAAkB,KAAK,IAAI,CAAC;IACnE,wBAAwB,EAAE,MAAM,IAAI,CAAC;IAErC,qBAAqB,EAAE,MAAM,IAAI,CAAC;IAElC,QAAQ,EAAE,MAAM,CAAC;IACjB,kBAAkB,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACrD,oBAAoB,EAAE,MAAM,OAAO,CAAC;QAClC,QAAQ,EAAE,OAAO,EAAE,CAAC;QACpB,UAAU,EAAE,MAAM,EAAE,CAAC;KACtB,CAAC,CAAC;IACH,qBAAqB,EAAE,MAAM,CAAC;IAC9B,wBAAwB,EAAE,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;CACxE;AAID,eAAO,MAAM,OAAO,uBAMnB,CAAC;AAEF,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,KAAK,CAAC,SAAS,CAAC;IAC1B,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;CAClB;AAED,eAAO,MAAM,YAAY,EAAE,KAAK,CAAC,EAAE,CAAC,iBAAiB,CAyfpD,CAAC"}

package/dist/contexts/useChat.js

@@ -24,6 +24,7 @@ export const ChatProvider = ({ children, bypassPermissions, pluginDirs, tools, }
     const [isTaskListVisible, setIsTaskListVisible] = useState(true);
     // AI State
     const [messages, setMessages] = useState([]);
+    const messagesUpdateTimerRef = useRef(null);
     const [isLoading, setIsLoading] = useState(false);
     const [latestTotalTokens, setlatestTotalTokens] = useState(0);
     const [sessionId, setSessionId] = useState("");
@@ -71,9 +72,16 @@ export const ChatProvider = ({ children, bypassPermissions, pluginDirs, tools, }
     useEffect(() => {
         const initializeAgent = async () => {
             const callbacks = {
-                onMessagesChange: (newMessages) => {
+                onMessagesChange: () => {
                     if (!isExpandedRef.current) {
-                        setMessages([...newMessages]);
+                        if (!messagesUpdateTimerRef.current) {
+                            messagesUpdateTimerRef.current = setTimeout(() => {
+                                if (agentRef.current) {
+                                    setMessages([...agentRef.current.messages]);
+                                }
+                                messagesUpdateTimerRef.current = null;
+                            }, 50);
+                        }
                     }
                 },
                 onServersChange: (servers) => {
@@ -173,6 +181,9 @@ export const ChatProvider = ({ children, bypassPermissions, pluginDirs, tools, }
     // Cleanup on unmount
     useEffect(() => {
         return () => {
+            if (messagesUpdateTimerRef.current) {
+                clearTimeout(messagesUpdateTimerRef.current);
+            }
             if (agentRef.current) {
                 try {
                     // Display usage summary before cleanup
@@ -332,6 +343,10 @@ export const ChatProvider = ({ children, bypassPermissions, pluginDirs, tools, }
             setIsExpanded((prev) => {
                 const newExpanded = !prev;
                 if (newExpanded) {
+                    if (messagesUpdateTimerRef.current) {
+                        clearTimeout(messagesUpdateTimerRef.current);
+                        messagesUpdateTimerRef.current = null;
+                    }
                     // Transitioning to EXPANDED: Freeze the current view
                     // Deep copy the last message to ensure it doesn't update if the agent is still writing to it
                     setMessages((currentMessages) => {

package/dist/utils/highlightUtils.d.ts

@@ -0,0 +1,2 @@
+export declare function highlightToAnsi(code: string, language?: string): string;
+//# sourceMappingURL=highlightUtils.d.ts.map

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

@@ -0,0 +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,CAcvE"}

package/dist/utils/highlightUtils.js

@@ -0,0 +1,69 @@
+import hljs from 'highlight.js';
+import { parse, HTMLElement, TextNode } from 'node-html-parser';
+import chalk from 'chalk';
+const theme = {
+    '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) {
+    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, language) {
+    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/dist/utils/toolParameterTransforms.d.ts

@@ -15,5 +15,5 @@ export declare function transformEditParameters(parameters: EditToolParameters):
  * Transform tool block parameters into standardized Change[] array for diff display
  * Forces type judgment based on tool name using type assertions
  */
-export declare function transformToolBlockToChanges(toolName: string, parameters: string): Change[];
+export declare function transformToolBlockToChanges(toolName: string, parameters: string, startLineNumber?: number): Change[];
 //# sourceMappingURL=toolParameterTransforms.d.ts.map

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,GACjB,MAAM,EAAE,CAsBV"}
+{"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"}

package/dist/utils/toolParameterTransforms.js

@@ -44,20 +44,27 @@ export function transformEditParameters(parameters) {
  * Transform tool block parameters into standardized Change[] array for diff display
  * Forces type judgment based on tool name using type assertions
  */
-export function transformToolBlockToChanges(toolName, parameters) {
+export function transformToolBlockToChanges(toolName, parameters, startLineNumber) {
     try {
         if (!toolName) {
             return [];
         }
         const parsedParams = parseToolParameters(parameters);
+        let changes = [];
         switch (toolName) {
             case "Write":
-                return transformWriteParameters(parsedParams);
+                changes = transformWriteParameters(parsedParams);
+                break;
             case "Edit":
-                return transformEditParameters(parsedParams);
+                changes = transformEditParameters(parsedParams);
+                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);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wave-code",
-  "version": "0.7.0",
+  "version": "0.7.1",
   "description": "CLI-based code assistant powered by AI, built with React and Ink",
   "repository": {
     "type": "git",
@@ -32,12 +32,14 @@
     "chalk": "^5.6.2",
     "diff": "^8.0.2",
     "glob": "^13.0.0",
+    "highlight.js": "^11.11.1",
     "ink": "^6.7.0",
     "marked": "^17.0.2",
+    "node-html-parser": "^7.0.2",
     "react": "^19.2.4",
     "react-dom": "19.2.4",
     "yargs": "^17.7.2",
-    "wave-agent-sdk": "0.7.0"
+    "wave-agent-sdk": "0.7.1"
   },
   "devDependencies": {
     "@types/react": "^19.1.8",

package/src/components/BackgroundTaskManager.tsx

@@ -85,11 +85,14 @@ export const BackgroundTaskManager: React.FC<BackgroundTaskManagerProps> = ({
     if (viewMode === "list") {
       // List mode navigation
       if (key.return) {
-        if (tasks.length > 0 && selectedIndex < tasks.length) {
-          const selectedTask = tasks[selectedIndex];
-          setDetailTaskId(selectedTask.id);
-          setViewMode("detail");
-        }
+        setSelectedIndex((prev) => {
+          if (tasks.length > 0 && prev < tasks.length) {
+            const selectedTask = tasks[prev];
+            setDetailTaskId(selectedTask.id);
+            setViewMode("detail");
+          }
+          return prev;
+        });
         return;
       }
 
@@ -99,20 +102,25 @@ export const BackgroundTaskManager: React.FC<BackgroundTaskManagerProps> = ({
       }
 
       if (key.upArrow) {
-        setSelectedIndex(Math.max(0, selectedIndex - 1));
+        setSelectedIndex((prev) => Math.max(0, prev - 1));
         return;
       }
 
       if (key.downArrow) {
-        setSelectedIndex(Math.min(tasks.length - 1, selectedIndex + 1));
+        setSelectedIndex((prev) => Math.min(tasks.length - 1, prev + 1));
         return;
       }
 
-      if (input === "k" && tasks.length > 0 && selectedIndex < tasks.length) {
-        const selectedTask = tasks[selectedIndex];
-        if (selectedTask.status === "running") {
-          stopTask(selectedTask.id);
-        }
+      if (input === "k") {
+        setSelectedIndex((prev) => {
+          if (tasks.length > 0 && prev < tasks.length) {
+            const selectedTask = tasks[prev];
+            if (selectedTask.status === "running") {
+              stopTask(selectedTask.id);
+            }
+          }
+          return prev;
+        });
         return;
       }
     } else if (viewMode === "detail") {

package/src/components/ChatInterface.tsx

@@ -1,4 +1,4 @@
-import React, { useState, useCallback } from "react";
+import React, { useState, useCallback, useLayoutEffect } from "react";
 import { Box, useStdout } from "ink";
 import { MessageList } from "./MessageList.js";
 import { InputBox } from "./InputBox.js";
@@ -12,7 +12,9 @@ import type { PermissionDecision } from "wave-agent-sdk";
 
 export const ChatInterface: React.FC = () => {
   const { stdout } = useStdout();
-  const [isDetailsTooTall, setIsDetailsTooTall] = useState(false);
+  const [detailsHeight, setDetailsHeight] = useState(0);
+  const [selectorHeight, setSelectorHeight] = useState(0);
+  const [isConfirmationTooTall, setIsConfirmationTooTall] = useState(false);
 
   const {
     messages,
@@ -36,39 +38,49 @@ export const ChatInterface: React.FC = () => {
     setWasLastDetailsTooTall,
   } = useChat();
 
-  const handleHeightMeasured = useCallback(
-    (height: number) => {
-      const terminalHeight = stdout?.rows || 24;
-      if (height > terminalHeight - 10) {
-        setIsDetailsTooTall(true);
-      } else {
-        setIsDetailsTooTall(false);
-      }
-    },
-    [stdout?.rows],
-  );
+  const handleDetailsHeightMeasured = useCallback((height: number) => {
+    setDetailsHeight(height);
+  }, []);
+
+  const handleSelectorHeightMeasured = useCallback((height: number) => {
+    setSelectorHeight(height);
+  }, []);
+
+  useLayoutEffect(() => {
+    const terminalHeight = stdout?.rows || 24;
+    const totalHeight = detailsHeight + selectorHeight;
+    if (totalHeight > terminalHeight) {
+      setIsConfirmationTooTall(true);
+    } else {
+      setIsConfirmationTooTall(false);
+    }
+  }, [detailsHeight, selectorHeight, stdout?.rows]);
 
   const handleConfirmationCancel = useCallback(() => {
-    if (isDetailsTooTall) {
+    if (isConfirmationTooTall) {
       setWasLastDetailsTooTall((prev) => prev + 1);
-      setIsDetailsTooTall(false);
+      setIsConfirmationTooTall(false);
     }
     originalHandleConfirmationCancel();
   }, [
-    isDetailsTooTall,
+    isConfirmationTooTall,
     originalHandleConfirmationCancel,
     setWasLastDetailsTooTall,
   ]);
 
   const wrappedHandleConfirmationDecision = useCallback(
     (decision: PermissionDecision) => {
-      if (isDetailsTooTall) {
+      if (isConfirmationTooTall) {
         setWasLastDetailsTooTall((prev) => prev + 1);
-        setIsDetailsTooTall(false);
+        setIsConfirmationTooTall(false);
       }
       handleConfirmationDecision(decision);
     },
-    [isDetailsTooTall, handleConfirmationDecision, setWasLastDetailsTooTall],
+    [
+      isConfirmationTooTall,
+      handleConfirmationDecision,
+      setWasLastDetailsTooTall,
+    ],
   );
 
   if (!sessionId) return null;
@@ -77,10 +89,8 @@ export const ChatInterface: React.FC = () => {
     <Box flexDirection="column">
       <MessageList
         messages={messages}
-        isLoading={isLoading}
-        isCommandRunning={isCommandRunning}
         isExpanded={isExpanded}
-        forceStaticLastMessage={isDetailsTooTall}
+        hideDynamicBlocks={isConfirmationVisible}
       />
 
       {(isLoading || isCommandRunning || isCompressing) &&
@@ -101,7 +111,8 @@ export const ChatInterface: React.FC = () => {
             toolName={confirmingTool!.name}
             toolInput={confirmingTool!.input}
             isExpanded={isExpanded}
-            onHeightMeasured={handleHeightMeasured}
+            onHeightMeasured={handleDetailsHeightMeasured}
+            isStatic={isConfirmationTooTall}
           />
           <ConfirmationSelector
             toolName={confirmingTool!.name}
@@ -112,6 +123,7 @@ export const ChatInterface: React.FC = () => {
             onDecision={wrappedHandleConfirmationDecision}
             onCancel={handleConfirmationCancel}
             onAbort={abortMessage}
+            onHeightMeasured={handleSelectorHeightMeasured}
           />
         </>
       )}

package/src/components/ConfirmationDetails.tsx

@@ -1,4 +1,4 @@
-import React, { useLayoutEffect, useRef, useState } from "react";
+import React, { useLayoutEffect, useRef } from "react";
 import { Box, Text, useStdout, measureElement, Static } from "ink";
 import {
   BASH_TOOL_NAME,
@@ -40,6 +40,7 @@ export interface ConfirmationDetailsProps {
   toolInput?: Record<string, unknown>;
   isExpanded?: boolean;
   onHeightMeasured?: (height: number) => void;
+  isStatic?: boolean;
 }
 
 export const ConfirmationDetails: React.FC<ConfirmationDetailsProps> = ({
@@ -47,21 +48,21 @@ export const ConfirmationDetails: React.FC<ConfirmationDetailsProps> = ({
   toolInput,
   isExpanded = false,
   onHeightMeasured,
+  isStatic = false,
 }) => {
   const { stdout } = useStdout();
-  const [isStatic, setIsStatic] = useState(false);
   const boxRef = useRef(null);
 
+  const startLineNumber =
+    (toolInput?.startLineNumber as number | undefined) ??
+    (toolName === WRITE_TOOL_NAME ? 1 : undefined);
+
   useLayoutEffect(() => {
     if (boxRef.current) {
       const { height } = measureElement(boxRef.current);
-      const terminalHeight = stdout?.rows || 24;
-      if (height > terminalHeight - 10) {
-        setIsStatic(true);
-      }
       onHeightMeasured?.(height);
     }
-  }, [stdout?.rows, onHeightMeasured]);
+  }, [stdout?.rows, onHeightMeasured, toolInput, isExpanded]);
 
   const content = (
     <Box
@@ -72,14 +73,17 @@ export const ConfirmationDetails: React.FC<ConfirmationDetailsProps> = ({
       borderBottom={false}
       borderLeft={false}
       borderRight={false}
-      paddingTop={1}
     >
       <Text color="yellow" bold>
         Tool: {toolName}
       </Text>
       <Text color="yellow">{getActionDescription(toolName, toolInput)}</Text>
 
-      <DiffDisplay toolName={toolName} parameters={JSON.stringify(toolInput)} />
+      <DiffDisplay
+        toolName={toolName}
+        parameters={JSON.stringify(toolInput)}
+        startLineNumber={startLineNumber}
+      />
 
       {toolName !== ASK_USER_QUESTION_TOOL_NAME &&
         toolName === EXIT_PLAN_MODE_TOOL_NAME &&