@dreb/coding-agent 2.22.1 → 2.24.0

package/dist/modes/interactive/interactive-mode.js

@@ -28,6 +28,7 @@ import { getChangelogPath, getNewEntries, parseChangelog } from "../../utils/cha
 import { copyToClipboard } from "../../utils/clipboard.js";
 import { extensionForImageMimeType, readClipboardImage } from "../../utils/clipboard-image.js";
 import { parseGitUrl } from "../../utils/git.js";
+import { extractCopyableText, getMessagePreview, getMessageRoleLabel } from "../../utils/message-text.js";
 import { ensureTool } from "../../utils/tools-manager.js";
 import { ArminComponent } from "./components/armin.js";
 import { AssistantMessageComponent } from "./components/assistant-message.js";
@@ -36,6 +37,7 @@ import { BorderedLoader } from "./components/bordered-loader.js";
 import { BranchSummaryMessageComponent } from "./components/branch-summary-message.js";
 import { BuddyComponent } from "./components/buddy-component.js";
 import { CompactionSummaryMessageComponent } from "./components/compaction-summary-message.js";
+import { CopySelectorComponent } from "./components/copy-selector.js";
 import { CustomEditor } from "./components/custom-editor.js";
 import { CustomMessageComponent } from "./components/custom-message.js";
 import { DaxnutsComponent } from "./components/daxnuts.js";
@@ -1711,6 +1713,7 @@ export class InteractiveMode {
         this.defaultEditor.onAction("app.session.tree", () => this.showTreeSelector());
         this.defaultEditor.onAction("app.session.fork", () => this.showUserMessageSelector());
         this.defaultEditor.onAction("app.session.resume", () => this.showSessionSelector());
+        this.defaultEditor.onAction("app.clipboard.copyMessages", () => this.showCopySelector());
         this.defaultEditor.onChange = (text) => {
             const wasBashMode = this.isBashMode;
             this.isBashMode = text.trimStart().startsWith("!");
@@ -2226,6 +2229,33 @@ export class InteractiveMode {
                 this.ui.requestRender();
                 break;
             }
+            case "length_retry": {
+                // Remove the ghost streaming component left from the truncated stream
+                if (this.streamingComponent) {
+                    this.chatContainer.removeChild(this.streamingComponent);
+                    this.streamingComponent = undefined;
+                    this.streamingMessage = undefined;
+                }
+                // Clear any pending tool components from the truncated partial
+                for (const [, component] of this.pendingTools.entries()) {
+                    this.chatContainer.removeChild(component);
+                }
+                this.pendingTools.clear();
+                // Show retry status
+                this.statusContainer.clear();
+                if (this.loadingAnimation) {
+                    this.loadingAnimation.stop();
+                    this.loadingAnimation = undefined;
+                }
+                if (this.retryLoader) {
+                    this.retryLoader.stop();
+                    this.retryLoader = undefined;
+                }
+                this.retryLoader = new Loader(this.ui, (spinner) => theme.fg("warning", spinner), (text) => theme.fg("muted", text), `Response truncated, retrying with larger token budget (${event.attempt}/${event.maxAttempts})... (${keyText("app.interrupt")} to cancel)`);
+                this.statusContainer.addChild(this.retryLoader);
+                this.ui.requestRender();
+                break;
+            }
             case "background_agent_start":
             case "background_agent_end": {
                 this.updateBackgroundAgentStatus();
@@ -3655,18 +3685,68 @@ export class InteractiveMode {
         }
     }
     async handleCopyCommand() {
-        const text = this.session.getLastAssistantText();
-        if (!text) {
-            this.showError("No agent messages to copy yet.");
+        this.showCopySelector();
+    }
+    showCopySelector() {
+        const messages = this.session.messages;
+        if (messages.length === 0) {
+            this.showStatus("No messages to copy");
             return;
         }
-        try {
-            await copyToClipboard(text);
-            this.showStatus("Copied last agent message to clipboard");
-        }
-        catch (error) {
-            this.showError(error instanceof Error ? error.message : String(error));
+        // Build items from session messages
+        const items = messages.map((msg, index) => ({
+            index,
+            roleLabel: getMessageRoleLabel(msg),
+            preview: getMessagePreview(msg),
+        }));
+        // Hide buddy while selector is open to free vertical space
+        const hadBuddy = this.buddyComponent !== null;
+        if (hadBuddy) {
+            this.widgetContainerBelow.clear();
+            this.ui.requestRender();
         }
+        // Calculate max visible items based on terminal height
+        // Reserve lines for: header(4) + borders(2) + spacers(2) + scroll indicator(1) + footer(1) + padding(2) = ~12 lines overhead
+        const terminalRows = this.ui.terminal.rows;
+        const overhead = 12;
+        const maxVisible = Math.max(3, Math.min(15, terminalRows - overhead));
+        this.showSelector((done) => {
+            const selector = new CopySelectorComponent(items, async (selectedIndices) => {
+                done();
+                // Restore buddy
+                if (hadBuddy)
+                    this.renderWidgets();
+                this.ui.requestRender();
+                if (selectedIndices.length === 0) {
+                    this.showWarning("No messages selected");
+                    return;
+                }
+                // Extract text from selected messages in chronological order
+                const selectedTexts = selectedIndices
+                    .map((i) => extractCopyableText(messages[i]))
+                    .filter((text) => text.length > 0);
+                if (selectedTexts.length === 0) {
+                    this.showWarning("Selected messages have no copyable text");
+                    return;
+                }
+                const combined = selectedTexts.join("\n\n---\n\n");
+                const result = await copyToClipboard(combined);
+                const count = selectedTexts.length;
+                if (result.method === "osc52") {
+                    this.showStatus(`Sent ${count} message${count === 1 ? "" : "s"} to terminal clipboard (OSC 52)`);
+                }
+                else {
+                    this.showStatus(`Copied ${count} message${count === 1 ? "" : "s"} to clipboard`);
+                }
+            }, () => {
+                done();
+                // Restore buddy
+                if (hadBuddy)
+                    this.renderWidgets();
+                this.ui.requestRender();
+            }, maxVisible);
+            return { component: selector, focus: selector.getMessageList() };
+        });
     }
     handleNameCommand(text) {
         const name = text.replace(/^\/name\s*/, "").trim();