wave-code 0.4.0 → 0.6.1

package/dist/components/{Confirmation.js → ConfirmationSelector.js}

@@ -1,33 +1,7 @@
-import { jsxs as _jsxs, jsx as _jsx, Fragment as _Fragment } from "react/jsx-runtime";
+import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
 import { useState } from "react";
 import { Box, Text, useInput } from "ink";
-import { BASH_TOOL_NAME, EDIT_TOOL_NAME, MULTI_EDIT_TOOL_NAME, DELETE_FILE_TOOL_NAME, WRITE_TOOL_NAME, EXIT_PLAN_MODE_TOOL_NAME, ASK_USER_QUESTION_TOOL_NAME, } from "wave-agent-sdk";
-import { DiffDisplay } from "./DiffDisplay.js";
-import { PlanDisplay } from "./PlanDisplay.js";
-// Helper function to generate descriptive action text
-const getActionDescription = (toolName, toolInput) => {
-    if (!toolInput) {
-        return "Execute operation";
-    }
-    switch (toolName) {
-        case BASH_TOOL_NAME:
-            return `Execute command: ${toolInput.command || "unknown command"}`;
-        case EDIT_TOOL_NAME:
-            return `Edit file: ${toolInput.file_path || "unknown file"}`;
-        case MULTI_EDIT_TOOL_NAME:
-            return `Edit multiple sections in: ${toolInput.file_path || "unknown file"}`;
-        case DELETE_FILE_TOOL_NAME:
-            return `Delete file: ${toolInput.target_file || "unknown file"}`;
-        case WRITE_TOOL_NAME:
-            return `Write to file: ${toolInput.file_path || "unknown file"}`;
-        case EXIT_PLAN_MODE_TOOL_NAME:
-            return "Review and approve the plan";
-        case ASK_USER_QUESTION_TOOL_NAME:
-            return "Answer questions to clarify intent";
-        default:
-            return "Execute operation";
-    }
-};
+import { BASH_TOOL_NAME, EXIT_PLAN_MODE_TOOL_NAME, ASK_USER_QUESTION_TOOL_NAME, } from "wave-agent-sdk";
 const getHeaderColor = (header) => {
     const colors = ["red", "green", "blue", "magenta", "cyan"];
     let hash = 0;
@@ -36,18 +10,19 @@ const getHeaderColor = (header) => {
     }
     return colors[Math.abs(hash) % colors.length];
 };
-export const Confirmation = ({ toolName, toolInput, suggestedPrefix, hidePersistentOption, isExpanded = false, onDecision, onCancel, onAbort, }) => {
+export const ConfirmationSelector = ({ toolName, toolInput, suggestedPrefix, hidePersistentOption, isExpanded = false, onDecision, onCancel, onAbort, }) => {
     const [state, setState] = useState({
         selectedOption: "allow",
         alternativeText: "",
+        alternativeCursorPosition: 0,
         hasUserInput: false,
     });
-    // Specialized state for AskUserQuestion
     const [currentQuestionIndex, setCurrentQuestionIndex] = useState(0);
     const [selectedOptionIndex, setSelectedOptionIndex] = useState(0);
     const [selectedOptionIndices, setSelectedOptionIndices] = useState(new Set());
     const [userAnswers, setUserAnswers] = useState({});
     const [otherText, setOtherText] = useState("");
+    const [otherCursorPosition, setOtherCursorPosition] = useState(0);
     const questions = toolInput?.questions || [];
     const currentQuestion = questions[currentQuestionIndex];
     const getAutoOptionText = () => {
@@ -63,7 +38,6 @@ export const Confirmation = ({ toolName, toolInput, suggestedPrefix, hidePersist
         return "Yes, and auto-accept edits";
     };
     useInput((input, key) => {
-        // Handle ESC to cancel and abort
         if (key.escape) {
             onCancel();
             onAbort();
@@ -73,7 +47,7 @@ export const Confirmation = ({ toolName, toolInput, suggestedPrefix, hidePersist
             if (!currentQuestion)
                 return;
             const options = [...currentQuestion.options, { label: "Other" }];
-            const isMultiSelect = !!currentQuestion.multiSelect;
+            const isMultiSelect = currentQuestion.multiSelect;
             const isOtherFocused = selectedOptionIndex === options.length - 1;
             if (key.return) {
                 let answer = "";
@@ -107,9 +81,9 @@ export const Confirmation = ({ toolName, toolInput, suggestedPrefix, hidePersist
                     setSelectedOptionIndex(0);
                     setSelectedOptionIndices(new Set());
                     setOtherText("");
+                    setOtherCursorPosition(0);
                 }
                 else {
-                    // All questions answered
                     onDecision({
                         behavior: "allow",
                         message: JSON.stringify(newAnswers),
@@ -122,45 +96,54 @@ export const Confirmation = ({ toolName, toolInput, suggestedPrefix, hidePersist
                     (!isOtherFocused || !selectedOptionIndices.has(selectedOptionIndex))) {
                     setSelectedOptionIndices((prev) => {
                         const next = new Set(prev);
-                        if (next.has(selectedOptionIndex)) {
+                        if (next.has(selectedOptionIndex))
                             next.delete(selectedOptionIndex);
-                        }
-                        else {
+                        else
                             next.add(selectedOptionIndex);
-                        }
                         return next;
                     });
                     return;
                 }
-                if (!isOtherFocused) {
+                if (!isOtherFocused)
                     return;
-                }
-                // If isOtherFocused is true, fall through to handle space as text input
             }
             if (key.upArrow) {
-                if (selectedOptionIndex > 0) {
+                if (selectedOptionIndex > 0)
                     setSelectedOptionIndex(selectedOptionIndex - 1);
-                }
                 return;
             }
             if (key.downArrow) {
-                if (selectedOptionIndex < options.length - 1) {
+                if (selectedOptionIndex < options.length - 1)
                     setSelectedOptionIndex(selectedOptionIndex + 1);
-                }
                 return;
             }
             if (isOtherFocused) {
+                if (key.leftArrow) {
+                    setOtherCursorPosition((prev) => Math.max(0, prev - 1));
+                    return;
+                }
+                if (key.rightArrow) {
+                    setOtherCursorPosition((prev) => Math.min(otherText.length, prev + 1));
+                    return;
+                }
                 if (key.backspace || key.delete) {
-                    setOtherText((prev) => prev.slice(0, -1));
+                    if (otherCursorPosition > 0) {
+                        setOtherText((prev) => prev.slice(0, otherCursorPosition - 1) +
+                            prev.slice(otherCursorPosition));
+                        setOtherCursorPosition((prev) => prev - 1);
+                    }
+                    return;
                 }
-                else if (input && !key.ctrl && !key.meta) {
-                    setOtherText((prev) => prev + input);
+                if (input && !key.ctrl && !key.meta) {
+                    setOtherText((prev) => prev.slice(0, otherCursorPosition) +
+                        input +
+                        prev.slice(otherCursorPosition));
+                    setOtherCursorPosition((prev) => prev + input.length);
+                    return;
                 }
-                return;
             }
             return;
         }
-        // Handle Enter to confirm selection
         if (key.return) {
             if (state.selectedOption === "allow") {
                 if (toolName === EXIT_PLAN_MODE_TOOL_NAME) {
@@ -175,38 +158,40 @@ export const Confirmation = ({ toolName, toolInput, suggestedPrefix, hidePersist
                     const rule = suggestedPrefix
                         ? `Bash(${suggestedPrefix}*)`
                         : `Bash(${toolInput?.command})`;
-                    onDecision({
-                        behavior: "allow",
-                        newPermissionRule: rule,
-                    });
+                    onDecision({ behavior: "allow", newPermissionRule: rule });
                 }
                 else {
-                    onDecision({
-                        behavior: "allow",
-                        newPermissionMode: "acceptEdits",
-                    });
+                    onDecision({ behavior: "allow", newPermissionMode: "acceptEdits" });
                 }
             }
-            else {
-                // For alternative option, require text input
-                if (state.alternativeText.trim()) {
-                    onDecision({
-                        behavior: "deny",
-                        message: state.alternativeText.trim(),
-                    });
-                }
+            else if (state.alternativeText.trim()) {
+                onDecision({ behavior: "deny", message: state.alternativeText.trim() });
             }
             return;
         }
-        // Handle arrow keys for navigation
+        if (state.selectedOption === "alternative") {
+            if (key.leftArrow) {
+                setState((prev) => ({
+                    ...prev,
+                    alternativeCursorPosition: Math.max(0, prev.alternativeCursorPosition - 1),
+                }));
+                return;
+            }
+            if (key.rightArrow) {
+                setState((prev) => ({
+                    ...prev,
+                    alternativeCursorPosition: Math.min(prev.alternativeText.length, prev.alternativeCursorPosition + 1),
+                }));
+                return;
+            }
+        }
         if (key.upArrow) {
             setState((prev) => {
-                if (prev.selectedOption === "alternative") {
+                if (prev.selectedOption === "alternative")
                     return {
                         ...prev,
                         selectedOption: hidePersistentOption ? "allow" : "auto",
                     };
-                }
                 if (prev.selectedOption === "auto")
                     return { ...prev, selectedOption: "allow" };
                 return prev;
@@ -215,61 +200,67 @@ export const Confirmation = ({ toolName, toolInput, suggestedPrefix, hidePersist
         }
         if (key.downArrow) {
             setState((prev) => {
-                if (prev.selectedOption === "allow") {
+                if (prev.selectedOption === "allow")
                     return {
                         ...prev,
                         selectedOption: hidePersistentOption ? "alternative" : "auto",
                     };
-                }
                 if (prev.selectedOption === "auto")
                     return { ...prev, selectedOption: "alternative" };
                 return prev;
             });
             return;
         }
-        // Handle text input for alternative option
         if (input && !key.ctrl && !key.meta && !("alt" in key && key.alt)) {
-            // Focus on alternative option when user starts typing
-            setState((prev) => ({
-                selectedOption: "alternative",
-                alternativeText: prev.alternativeText + input,
-                hasUserInput: true,
-            }));
-            return;
-        }
-        // Handle backspace and delete (same behavior - delete one character)
-        if (key.backspace || key.delete) {
             setState((prev) => {
-                const newText = prev.alternativeText.slice(0, -1);
+                const nextText = prev.alternativeText.slice(0, prev.alternativeCursorPosition) +
+                    input +
+                    prev.alternativeText.slice(prev.alternativeCursorPosition);
                 return {
                     ...prev,
                     selectedOption: "alternative",
-                    alternativeText: newText,
-                    hasUserInput: newText.length > 0,
+                    alternativeText: nextText,
+                    alternativeCursorPosition: prev.alternativeCursorPosition + input.length,
+                    hasUserInput: true,
                 };
             });
             return;
         }
+        if (key.backspace || key.delete) {
+            setState((prev) => {
+                if (prev.alternativeCursorPosition > 0) {
+                    const nextText = prev.alternativeText.slice(0, prev.alternativeCursorPosition - 1) +
+                        prev.alternativeText.slice(prev.alternativeCursorPosition);
+                    return {
+                        ...prev,
+                        selectedOption: "alternative",
+                        alternativeText: nextText,
+                        alternativeCursorPosition: prev.alternativeCursorPosition - 1,
+                        hasUserInput: nextText.length > 0,
+                    };
+                }
+                return prev;
+            });
+            return;
+        }
     });
     const placeholderText = "Type here to tell Wave what to do differently";
     const showPlaceholder = state.selectedOption === "alternative" && !state.hasUserInput;
-    return (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: "yellow", borderBottom: false, borderLeft: false, borderRight: false, paddingTop: 1, children: [_jsxs(Text, { color: "yellow", bold: true, children: ["Tool: ", toolName] }), _jsx(Text, { color: "yellow", children: getActionDescription(toolName, toolInput) }), _jsx(DiffDisplay, { toolName: toolName, parameters: JSON.stringify(toolInput), isExpanded: isExpanded }), toolName === ASK_USER_QUESTION_TOOL_NAME &&
+    return (_jsxs(Box, { flexDirection: "column", children: [toolName === ASK_USER_QUESTION_TOOL_NAME &&
                 currentQuestion &&
-                !isExpanded && (_jsxs(Box, { flexDirection: "column", marginTop: 1, children: [_jsxs(Box, { marginBottom: 1, children: [_jsx(Box, { backgroundColor: getHeaderColor(currentQuestion.header), paddingX: 1, marginRight: 1, children: _jsx(Text, { color: "black", bold: true, children: currentQuestion.header.slice(0, 12).toUpperCase() }) }), _jsx(Text, { bold: true, children: currentQuestion.question })] }), _jsx(Box, { flexDirection: "column", children: (() => {
-                            const isMultiSelect = !!currentQuestion.multiSelect;
-                            return [...currentQuestion.options, { label: "Other" }].map((option, index) => {
-                                const isSelected = selectedOptionIndex === index;
-                                const isChecked = isMultiSelect
-                                    ? selectedOptionIndices.has(index)
-                                    : isSelected;
-                                const isOther = index === currentQuestion.options.length;
-                                const isRecommended = !isOther && option.isRecommended;
-                                return (_jsx(Box, { children: _jsxs(Text, { color: isSelected ? "black" : "white", backgroundColor: isSelected ? "yellow" : undefined, children: [isSelected ? "> " : "  ", isMultiSelect ? (isChecked ? "[x] " : "[ ] ") : "", option.label, isRecommended && (_jsxs(Text, { color: "green", bold: true, children: [" ", "(Recommended)"] })), option.description ? ` - ${option.description}` : "", isOther && isSelected && (_jsxs(Text, { children: [":", " ", otherText || (_jsx(Text, { color: "gray", dimColor: true, children: "[Type your answer...]" }))] }))] }) }, index));
-                            });
-                        })() }), _jsx(Box, { marginTop: 1, children: _jsxs(Text, { dimColor: true, children: ["Question ", currentQuestionIndex + 1, " of ", questions.length, " \u2022", currentQuestion.multiSelect ? " Space to toggle •" : "", " Use \u2191\u2193 to navigate \u2022 Enter to confirm"] }) })] })), toolName !== ASK_USER_QUESTION_TOOL_NAME &&
-                toolName === EXIT_PLAN_MODE_TOOL_NAME &&
-                !!toolInput?.plan_content && (_jsx(PlanDisplay, { plan: toolInput.plan_content, isExpanded: isExpanded })), toolName !== ASK_USER_QUESTION_TOOL_NAME && !isExpanded && (_jsxs(_Fragment, { children: [_jsx(Box, { marginTop: 1, children: _jsx(Text, { children: "Do you want to proceed?" }) }), _jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsx(Box, { children: _jsxs(Text, { color: state.selectedOption === "allow" ? "black" : "white", backgroundColor: state.selectedOption === "allow" ? "yellow" : undefined, bold: state.selectedOption === "allow", children: [state.selectedOption === "allow" ? "> " : "  ", toolName === EXIT_PLAN_MODE_TOOL_NAME
+                !isExpanded && (_jsxs(Box, { flexDirection: "column", marginTop: 1, children: [_jsxs(Box, { marginBottom: 1, children: [_jsx(Box, { backgroundColor: getHeaderColor(currentQuestion.header), paddingX: 1, marginRight: 1, children: _jsx(Text, { color: "black", bold: true, children: currentQuestion.header.slice(0, 12).toUpperCase() }) }), _jsx(Text, { bold: true, children: currentQuestion.question })] }), _jsx(Box, { flexDirection: "column", children: [...currentQuestion.options, { label: "Other" }].map((option, index) => {
+                            const isSelected = selectedOptionIndex === index;
+                            const isChecked = currentQuestion.multiSelect
+                                ? selectedOptionIndices.has(index)
+                                : isSelected;
+                            const isOther = index === currentQuestion.options.length;
+                            return (_jsx(Box, { children: _jsxs(Text, { color: isSelected ? "black" : "white", backgroundColor: isSelected ? "yellow" : undefined, children: [isSelected ? "> " : "  ", currentQuestion.multiSelect
+                                            ? isChecked
+                                                ? "[x] "
+                                                : "[ ] "
+                                            : "", option.label, option.description ? ` - ${option.description}` : "", isOther && isSelected && (_jsxs(Text, { children: [":", " ", otherText ? (_jsxs(_Fragment, { children: [otherText.slice(0, otherCursorPosition), _jsx(Text, { backgroundColor: "white", color: "black", children: otherText[otherCursorPosition] || " " }), otherText.slice(otherCursorPosition + 1)] })) : (_jsx(Text, { color: "gray", dimColor: true, children: "[Type your answer...]" }))] }))] }) }, index));
+                        }) }), _jsx(Box, { marginTop: 1, children: _jsxs(Text, { dimColor: true, children: ["Question ", currentQuestionIndex + 1, " of ", questions.length, " \u2022", currentQuestion.multiSelect ? " Space to toggle •" : "", " Use \u2191\u2193 to navigate \u2022 Enter to confirm"] }) })] })), toolName !== ASK_USER_QUESTION_TOOL_NAME && !isExpanded && (_jsxs(_Fragment, { children: [_jsx(Box, { marginTop: 1, children: _jsx(Text, { children: "Do you want to proceed?" }) }), _jsxs(Box, { marginTop: 1, flexDirection: "column", children: [_jsx(Box, { children: _jsxs(Text, { color: state.selectedOption === "allow" ? "black" : "white", backgroundColor: state.selectedOption === "allow" ? "yellow" : undefined, bold: state.selectedOption === "allow", children: [state.selectedOption === "allow" ? "> " : "  ", toolName === EXIT_PLAN_MODE_TOOL_NAME
                                             ? "Yes, proceed with default mode"
-                                            : "Yes"] }) }, "allow-option"), !hidePersistentOption && (_jsx(Box, { children: _jsxs(Text, { color: state.selectedOption === "auto" ? "black" : "white", backgroundColor: state.selectedOption === "auto" ? "yellow" : undefined, bold: state.selectedOption === "auto", children: [state.selectedOption === "auto" ? "> " : "  ", getAutoOptionText()] }) }, "auto-option")), _jsx(Box, { children: _jsxs(Text, { color: state.selectedOption === "alternative" ? "black" : "white", backgroundColor: state.selectedOption === "alternative" ? "yellow" : undefined, bold: state.selectedOption === "alternative", children: [state.selectedOption === "alternative" ? "> " : "  ", showPlaceholder ? (_jsx(Text, { color: "gray", dimColor: true, children: placeholderText })) : (_jsx(Text, { children: state.alternativeText ||
-                                                "Type here to tell Wave what to do differently" }))] }) }, "alternative-option")] }), _jsx(Box, { marginTop: 1, children: _jsx(Text, { dimColor: true, children: "Use \u2191\u2193 to navigate \u2022 ESC to cancel" }) })] }))] }));
+                                            : "Yes"] }) }, "allow-option"), !hidePersistentOption && (_jsx(Box, { children: _jsxs(Text, { color: state.selectedOption === "auto" ? "black" : "white", backgroundColor: state.selectedOption === "auto" ? "yellow" : undefined, bold: state.selectedOption === "auto", children: [state.selectedOption === "auto" ? "> " : "  ", getAutoOptionText()] }) }, "auto-option")), _jsx(Box, { children: _jsxs(Text, { color: state.selectedOption === "alternative" ? "black" : "white", backgroundColor: state.selectedOption === "alternative" ? "yellow" : undefined, bold: state.selectedOption === "alternative", children: [state.selectedOption === "alternative" ? "> " : "  ", showPlaceholder ? (_jsx(Text, { color: "gray", dimColor: true, children: placeholderText })) : (_jsx(Text, { children: state.alternativeText ? (_jsxs(_Fragment, { children: [state.alternativeText.slice(0, state.alternativeCursorPosition), _jsx(Text, { backgroundColor: "white", color: "black", children: state.alternativeText[state.alternativeCursorPosition] || " " }), state.alternativeText.slice(state.alternativeCursorPosition + 1)] })) : ("Type here to tell Wave what to do differently") }))] }) }, "alternative-option")] }), _jsx(Box, { marginTop: 1, children: _jsx(Text, { dimColor: true, children: "Use \u2191\u2193 to navigate \u2022 ESC to cancel" }) })] }))] }));
 };
+ConfirmationSelector.displayName = "ConfirmationSelector";

package/dist/components/DiffDisplay.d.ts

@@ -2,7 +2,6 @@ import React from "react";
 interface DiffDisplayProps {
     toolName?: string;
     parameters?: string;
-    isExpanded?: boolean;
 }
 export declare const DiffDisplay: React.FC<DiffDisplayProps>;
 export {};

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

@@ -1 +1 @@
-{"version":3,"file":"DiffDisplay.d.ts","sourceRoot":"","sources":["../../src/components/DiffDisplay.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAkB,MAAM,OAAO,CAAC;AAUvC,UAAU,gBAAgB;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,OAAO,CAAC;CACtB;AAED,eAAO,MAAM,WAAW,EAAE,KAAK,CAAC,EAAE,CAAC,gBAAgB,CA6SlD,CAAC"}
+{"version":3,"file":"DiffDisplay.d.ts","sourceRoot":"","sources":["../../src/components/DiffDisplay.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAkB,MAAM,OAAO,CAAC;AAUvC,UAAU,gBAAgB;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,eAAO,MAAM,WAAW,EAAE,KAAK,CAAC,EAAE,CAAC,gBAAgB,CAySlD,CAAC"}

package/dist/components/DiffDisplay.js

@@ -1,14 +1,10 @@
 import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
 import React, { useMemo } from "react";
-import { Box, Text, useStdout } from "ink";
+import { Box, Text } from "ink";
 import { WRITE_TOOL_NAME, EDIT_TOOL_NAME, MULTI_EDIT_TOOL_NAME, } from "wave-agent-sdk";
 import { transformToolBlockToChanges } from "../utils/toolParameterTransforms.js";
 import { diffLines, diffWords } from "diff";
-export const DiffDisplay = ({ toolName, parameters, isExpanded = false, }) => {
-    const { stdout } = useStdout();
-    const maxHeight = useMemo(() => {
-        return Math.max(5, (stdout?.rows || 24) - 20);
-    }, [stdout?.rows]);
+export const DiffDisplay = ({ toolName, parameters, }) => {
     const showDiff = toolName &&
         [WRITE_TOOL_NAME, EDIT_TOOL_NAME, MULTI_EDIT_TOOL_NAME].includes(toolName);
     // Diff detection and transformation using typed parameters
@@ -58,7 +54,7 @@ export const DiffDisplay = ({ toolName, parameters, isExpanded = false, }) => {
             };
         }
     };
-    // Render expanded diff display using word-level diff for all changes
+    // Render expanded diff display
     const renderExpandedDiff = () => {
         try {
             if (changes.length === 0)
@@ -68,8 +64,8 @@ export const DiffDisplay = ({ toolName, parameters, isExpanded = false, }) => {
                 try {
                     // Get line-level diff to understand the structure
                     const lineDiffs = diffLines(change.oldContent || "", change.newContent || "");
+                    // Process line diffs
                     const diffElements = [];
-                    // Process line diffs and apply word-level diff to changed lines
                     lineDiffs.forEach((part, partIndex) => {
                         if (part.added) {
                             const lines = part.value
@@ -92,46 +88,76 @@ export const DiffDisplay = ({ toolName, parameters, isExpanded = false, }) => {
                             const lines = part.value
                                 .split("\n")
                                 .filter((line) => line !== "");
-                            lines.forEach((line, lineIndex) => {
+                            const isFirstBlock = partIndex === 0;
+                            const isLastBlock = partIndex === lineDiffs.length - 1;
+                            let linesToDisplay = lines;
+                            let showEllipsisTop = false;
+                            let showEllipsisBottom = false;
+                            if (isFirstBlock && !isLastBlock) {
+                                // First block: keep last 3
+                                if (lines.length > 3) {
+                                    linesToDisplay = lines.slice(-3);
+                                    showEllipsisTop = true;
+                                }
+                            }
+                            else if (isLastBlock && !isFirstBlock) {
+                                // Last block: keep first 3
+                                if (lines.length > 3) {
+                                    linesToDisplay = lines.slice(0, 3);
+                                    showEllipsisBottom = true;
+                                }
+                            }
+                            else if (!isFirstBlock && !isLastBlock) {
+                                // Middle block: keep first 3 and last 3
+                                if (lines.length > 6) {
+                                    linesToDisplay = [...lines.slice(0, 3), ...lines.slice(-3)];
+                                    showEllipsisTop = false; // We'll put ellipsis in the middle
+                                }
+                            }
+                            else if (isFirstBlock && isLastBlock) {
+                                // Only one block (no changes?) - keep all or apply a general limit
+                                // For now, let's keep all if it's the only block
+                            }
+                            if (showEllipsisTop) {
+                                diffElements.push(_jsx(Box, { children: _jsx(Text, { color: "gray", children: " ..." }) }, `ellipsis-top-${changeIndex}-${partIndex}`));
+                            }
+                            linesToDisplay.forEach((line, lineIndex) => {
+                                // If it's a middle block and we are at the split point
+                                if (!isFirstBlock &&
+                                    !isLastBlock &&
+                                    lines.length > 6 &&
+                                    lineIndex === 3) {
+                                    diffElements.push(_jsx(Box, { children: _jsx(Text, { color: "gray", children: " ..." }) }, `ellipsis-mid-${changeIndex}-${partIndex}`));
+                                }
                                 diffElements.push(_jsxs(Box, { flexDirection: "row", children: [_jsx(Text, { color: "white", children: " " }), _jsx(Text, { color: "white", children: line })] }, `context-${changeIndex}-${partIndex}-${lineIndex}`));
                             });
+                            if (showEllipsisBottom) {
+                                diffElements.push(_jsx(Box, { children: _jsx(Text, { color: "gray", children: " ..." }) }, `ellipsis-bottom-${changeIndex}-${partIndex}`));
+                            }
                         }
                     });
-                    // Now look for pairs of removed/added lines that can be word-diffed
-                    let i = 0;
-                    while (i < diffElements.length) {
-                        const current = diffElements[i];
-                        const next = i + 1 < diffElements.length ? diffElements[i + 1] : null;
-                        // Check if we have a removed line followed by an added line
-                        const currentKey = React.isValidElement(current) ? current.key : "";
-                        const nextKey = React.isValidElement(next) ? next.key : "";
-                        const isCurrentRemoved = typeof currentKey === "string" && currentKey.includes("remove-");
-                        const isNextAdded = typeof nextKey === "string" && nextKey.includes("add-");
-                        if (isCurrentRemoved &&
-                            isNextAdded &&
-                            React.isValidElement(current) &&
-                            React.isValidElement(next)) {
-                            // Extract the text content from the removed and added lines
-                            const removedText = extractTextFromElement(current);
-                            const addedText = extractTextFromElement(next);
-                            if (removedText && addedText) {
-                                // Apply word-level diff
-                                const { removedParts, addedParts } = renderWordLevelDiff(removedText, addedText, `word-${changeIndex}-${i}`);
-                                allElements.push(_jsxs(Box, { flexDirection: "row", children: [_jsx(Text, { color: "red", children: "-" }), removedParts] }, `word-diff-removed-${changeIndex}-${i}`));
-                                allElements.push(_jsxs(Box, { flexDirection: "row", children: [_jsx(Text, { color: "green", children: "+" }), addedParts] }, `word-diff-added-${changeIndex}-${i}`));
-                                i += 2; // Skip the next element since we processed it
-                            }
-                            else {
-                                // Fallback to original elements
-                                allElements.push(current);
-                                i += 1;
-                            }
+                    // If it's a single line change (one removed, one added), use word-level diff
+                    if (diffElements.length === 2 &&
+                        React.isValidElement(diffElements[0]) &&
+                        React.isValidElement(diffElements[1]) &&
+                        typeof diffElements[0].key === "string" &&
+                        diffElements[0].key.includes("remove-") &&
+                        typeof diffElements[1].key === "string" &&
+                        diffElements[1].key.includes("add-")) {
+                        const removedText = extractTextFromElement(diffElements[0]);
+                        const addedText = extractTextFromElement(diffElements[1]);
+                        if (removedText && addedText) {
+                            const { removedParts, addedParts } = renderWordLevelDiff(removedText, addedText, `word-${changeIndex}`);
+                            allElements.push(_jsxs(Box, { flexDirection: "row", children: [_jsx(Text, { color: "red", children: "-" }), removedParts] }, `word-diff-removed-${changeIndex}`));
+                            allElements.push(_jsxs(Box, { flexDirection: "row", children: [_jsx(Text, { color: "green", children: "+" }), addedParts] }, `word-diff-added-${changeIndex}`));
                         }
                         else {
-                            allElements.push(current);
-                            i += 1;
+                            allElements.push(...diffElements);
                         }
                     }
+                    else {
+                        allElements.push(...diffElements);
+                    }
                 }
                 catch (error) {
                     console.warn(`Error rendering diff for change ${changeIndex}:`, error);
@@ -139,35 +165,31 @@ export const DiffDisplay = ({ toolName, parameters, isExpanded = false, }) => {
                     allElements.push(_jsxs(Box, { flexDirection: "column", children: [_jsxs(Text, { color: "red", children: ["-", change.oldContent || ""] }), _jsxs(Text, { color: "green", children: ["+", change.newContent || ""] })] }, `fallback-${changeIndex}`));
                 }
             });
-            const isTruncated = !isExpanded && allElements.length > maxHeight;
-            const displayElements = isTruncated
-                ? allElements.slice(0, maxHeight - 1)
-                : allElements;
-            return (_jsxs(Box, { flexDirection: "column", children: [displayElements, isTruncated && (_jsxs(Text, { color: "yellow", dimColor: true, children: ["... (truncated ", allElements.length - (maxHeight - 1), " more lines)"] }))] }));
+            return _jsx(Box, { flexDirection: "column", children: allElements });
         }
         catch (error) {
             console.warn("Error rendering expanded diff:", error);
             return (_jsx(Box, { children: _jsx(Text, { color: "gray", children: "Error rendering diff display" }) }));
         }
     };
-    // Helper function to extract text content from a React element
-    const extractTextFromElement = (element) => {
-        if (!React.isValidElement(element))
-            return null;
-        // Navigate through Box -> Text structure
-        const children = element.props.children;
-        if (Array.isArray(children) && children.length >= 2) {
-            const textElement = children[1]; // Second child should be the Text with content
-            if (React.isValidElement(textElement) &&
-                textElement.props.children) {
-                return textElement.props.children;
-            }
-        }
-        return null;
-    };
     // Don't render anything if no diff should be shown
     if (!showDiff) {
         return null;
     }
     return (_jsx(Box, { flexDirection: "column", children: _jsxs(Box, { paddingLeft: 2, borderLeft: true, borderColor: "cyan", flexDirection: "column", children: [_jsx(Text, { color: "cyan", bold: true, children: "Diff:" }), renderExpandedDiff()] }) }));
 };
+// Helper function to extract text content from a React element
+const extractTextFromElement = (element) => {
+    if (!React.isValidElement(element))
+        return null;
+    // Navigate through Box -> Text structure
+    const children = element.props.children;
+    if (Array.isArray(children) && children.length >= 2) {
+        const textElement = children[1]; // Second child should be the Text with content
+        if (React.isValidElement(textElement) &&
+            textElement.props.children) {
+            return textElement.props.children;
+        }
+    }
+    return null;
+};

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

@@ -1 +1 @@
-{"version":3,"file":"FileSelector.d.ts","sourceRoot":"","sources":["../../src/components/FileSelector.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAmB,MAAM,OAAO,CAAC;AAExC,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE/C,OAAO,EAAE,KAAK,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE/C,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,QAAQ,EAAE,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,IAAI,CAAC;IACrC,QAAQ,EAAE,MAAM,IAAI,CAAC;CACtB;AAED,eAAO,MAAM,YAAY,EAAE,KAAK,CAAC,EAAE,CAAC,iBAAiB,CA6HpD,CAAC"}
+{"version":3,"file":"FileSelector.d.ts","sourceRoot":"","sources":["../../src/components/FileSelector.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAmB,MAAM,OAAO,CAAC;AAExC,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE/C,OAAO,EAAE,KAAK,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE/C,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,QAAQ,EAAE,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,IAAI,CAAC;IACrC,QAAQ,EAAE,MAAM,IAAI,CAAC;CACtB;AAED,eAAO,MAAM,YAAY,EAAE,KAAK,CAAC,EAAE,CAAC,iBAAiB,CA2HpD,CAAC"}

package/dist/components/FileSelector.js

@@ -24,7 +24,7 @@ export const FileSelector = ({ files, searchQuery, onSelect, onCancel, }) => {
         }
     });
     if (files.length === 0) {
-        return (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: "yellow", borderBottom: false, borderLeft: false, borderRight: false, paddingTop: 1, children: [_jsxs(Text, { color: "yellow", children: ["\uD83D\uDCC1 No files found for \"", searchQuery, "\""] }), _jsx(Text, { dimColor: true, children: "Press Escape to cancel" })] }));
+        return (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: "yellow", borderBottom: false, borderLeft: false, borderRight: false, children: [_jsxs(Text, { color: "yellow", children: ["\uD83D\uDCC1 No files found for \"", searchQuery, "\""] }), _jsx(Text, { dimColor: true, children: "Press Escape to cancel" })] }));
     }
     const maxDisplay = 10;
     // Calculate display window start and end positions
@@ -39,7 +39,7 @@ export const FileSelector = ({ files, searchQuery, onSelect, onCancel, }) => {
         };
     };
     const { startIndex, endIndex, displayFiles } = getDisplayWindow();
-    return (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: "cyan", borderBottom: false, borderLeft: false, borderRight: false, paddingTop: 1, children: [_jsxs(Text, { color: "cyan", bold: true, children: ["\uD83D\uDCC1 Select File/Directory", " ", searchQuery && `(filtering: "${searchQuery}")`] }), startIndex > 0 && (_jsxs(Text, { dimColor: true, children: ["... ", startIndex, " more files above"] })), displayFiles.map((fileItem, displayIndex) => {
+    return (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: "cyan", borderBottom: false, borderLeft: false, borderRight: false, children: [_jsxs(Text, { color: "cyan", bold: true, children: ["\uD83D\uDCC1 Select File/Directory", " ", searchQuery && `(filtering: "${searchQuery}")`] }), startIndex > 0 && (_jsxs(Text, { dimColor: true, children: ["... ", startIndex, " more files above"] })), displayFiles.map((fileItem, displayIndex) => {
                 const actualIndex = startIndex + displayIndex;
                 const isSelected = actualIndex === selectedIndex;
                 const icon = fileItem.type === "directory" ? "📁" : "📄";

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

@@ -1 +1 @@
-{"version":3,"file":"HistorySearch.d.ts","sourceRoot":"","sources":["../../src/components/HistorySearch.tsx"],"names":[],"mappings":"AAAA,OAAO,KAA8B,MAAM,OAAO,CAAC;AAInD,MAAM,WAAW,kBAAkB;IACjC,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;IACnC,QAAQ,EAAE,MAAM,IAAI,CAAC;CACtB;AAED,eAAO,MAAM,aAAa,EAAE,KAAK,CAAC,EAAE,CAAC,kBAAkB,CAyItD,CAAC"}
+{"version":3,"file":"HistorySearch.d.ts","sourceRoot":"","sources":["../../src/components/HistorySearch.tsx"],"names":[],"mappings":"AAAA,OAAO,KAA8B,MAAM,OAAO,CAAC;AAInD,MAAM,WAAW,kBAAkB;IACjC,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;IACnC,QAAQ,EAAE,MAAM,IAAI,CAAC;CACtB;AAED,eAAO,MAAM,aAAa,EAAE,KAAK,CAAC,EAAE,CAAC,kBAAkB,CAiKtD,CAAC"}

package/dist/components/HistorySearch.js

@@ -3,6 +3,7 @@ import React, { useState, useEffect } from "react";
 import { Box, Text, useInput } from "ink";
 import { PromptHistoryManager } from "wave-agent-sdk";
 export const HistorySearch = ({ searchQuery, onSelect, onCancel, }) => {
+    const MAX_VISIBLE_ITEMS = 5;
     const [selectedIndex, setSelectedIndex] = useState(0);
     const [entries, setEntries] = useState([]);
     const entriesRef = React.useRef([]);
@@ -16,8 +17,8 @@ export const HistorySearch = ({ searchQuery, onSelect, onCancel, }) => {
     useEffect(() => {
         const fetchHistory = async () => {
             const results = await PromptHistoryManager.searchHistory(searchQuery);
-            const limitedResults = results.slice(0, 10);
-            setEntries(limitedResults); // Limit to 10 results
+            const limitedResults = results.slice(0, 20);
+            setEntries(limitedResults); // Limit to 20 results
             setSelectedIndex(0);
         };
         fetchHistory();
@@ -44,7 +45,7 @@ export const HistorySearch = ({ searchQuery, onSelect, onCancel, }) => {
         }
     });
     if (entries.length === 0) {
-        return (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: "yellow", borderBottom: false, borderLeft: false, borderRight: false, paddingTop: 1, children: [_jsxs(Text, { color: "yellow", children: ["No history found ", searchQuery && `for "${searchQuery}"`] }), _jsx(Text, { dimColor: true, children: "Press Escape to cancel" })] }));
+        return (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: "yellow", borderBottom: false, borderLeft: false, borderRight: false, children: [_jsxs(Text, { color: "yellow", children: ["No history found ", searchQuery && `for "${searchQuery}"`] }), _jsx(Text, { dimColor: true, children: "Press Escape to cancel" })] }));
     }
     const formatTimestamp = (timestamp) => {
         const date = new Date(timestamp);
@@ -63,5 +64,12 @@ export const HistorySearch = ({ searchQuery, onSelect, onCancel, }) => {
             return diffMinutes > 0 ? `${diffMinutes}m ago` : "just now";
         }
     };
-    return (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: "blue", borderBottom: false, borderLeft: false, borderRight: false, paddingTop: 1, gap: 1, children: [_jsx(Box, { children: _jsxs(Text, { color: "blue", bold: true, children: ["Prompt History ", searchQuery && `(filtering: "${searchQuery}")`] }) }), entries.map((entry, index) => (_jsxs(Box, { flexDirection: "column", children: [_jsx(Text, { color: index === selectedIndex ? "black" : "white", backgroundColor: index === selectedIndex ? "blue" : undefined, wrap: "truncate-end", children: entry.prompt.replace(/\n/g, " ") }), index === selectedIndex && (_jsx(Box, { marginLeft: 4, children: _jsx(Text, { color: "gray", dimColor: true, children: formatTimestamp(entry.timestamp) }) }))] }, index))), _jsx(Box, { children: _jsx(Text, { dimColor: true, children: "Use \u2191\u2193 to navigate, Enter to select, Escape to cancel" }) })] }));
+    // Calculate visible window
+    const startIndex = Math.max(0, Math.min(selectedIndex - Math.floor(MAX_VISIBLE_ITEMS / 2), Math.max(0, entries.length - MAX_VISIBLE_ITEMS)));
+    const visibleEntries = entries.slice(startIndex, startIndex + MAX_VISIBLE_ITEMS);
+    return (_jsxs(Box, { flexDirection: "column", borderStyle: "single", borderColor: "blue", borderBottom: false, borderLeft: false, borderRight: false, gap: 1, children: [_jsx(Box, { children: _jsxs(Text, { color: "blue", bold: true, children: ["Prompt History ", searchQuery && `(filtering: "${searchQuery}")`] }) }), _jsx(Box, { flexDirection: "column", children: visibleEntries.map((entry, index) => {
+                    const actualIndex = startIndex + index;
+                    const isSelected = actualIndex === selectedIndex;
+                    return (_jsxs(Box, { flexDirection: "row", justifyContent: "space-between", children: [_jsx(Box, { flexShrink: 1, children: _jsx(Text, { color: isSelected ? "black" : "white", backgroundColor: isSelected ? "blue" : undefined, wrap: "truncate-end", children: entry.prompt.replace(/\n/g, " ") }) }), isSelected && (_jsx(Box, { marginLeft: 2, flexShrink: 0, children: _jsx(Text, { color: "gray", dimColor: true, children: formatTimestamp(entry.timestamp) }) }))] }, actualIndex));
+                }) }), _jsx(Box, { children: _jsx(Text, { dimColor: true, children: "Use \u2191\u2193 to navigate, Enter to select, Escape to cancel" }) })] }));
 };