@astralyn/pi 0.84.10 → 0.84.12

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

@@ -7,7 +7,7 @@ import * as fs from "node:fs";
 import * as os from "node:os";
 import * as path from "node:path";
 import * as TuiLayouts from "@earendil-works/pi-tui";
-import { CombinedAutocompleteProvider, Container, fuzzyFilter, Markdown, matchesKey, ProcessTerminal, Spacer, setKeybindings, Text, TruncatedText, TuiAltScreen, TuiMainScreen, visibleWidth, } from "@earendil-works/pi-tui";
+import { CombinedAutocompleteProvider, Container, fuzzyFilter, Markdown, matchesKey, ProcessTerminal, Spacer, setCapabilityOverrides, setKeybindings, Text, TruncatedText, TuiAltScreen, TuiMainScreen, visibleWidth, } from "@earendil-works/pi-tui";
 import chalk from "chalk";
 import { spawn } from "child_process";
 import { APP_NAME, APP_TITLE, CONFIG_DIR_NAME, getAgentDir, getAuthPath, getDebugLogPath, getDocsPath, INSTALL_TELEMETRY_URL, UPDATE_CHANGELOG_HINT, VERSION, } from "../../config.js";
@@ -185,6 +185,7 @@ export function createInteractiveTui(options) {
             searchCurrentMatchStyle: (text) => theme.bold(theme.inverse(styleSearchMatch(text))),
             openUrl: openBrowser,
             onRightClickPaste: options.onRightClickPaste,
+            copyOnSelect: options.fullscreenCopyOnSelect,
             copySelection: async (text) => {
                 try {
                     await copyToClipboard(text);
@@ -319,6 +320,7 @@ export class InteractiveMode {
          */
         this.isShuttingDown = false;
         this.runtimeHost = runtimeHost;
+        setCapabilityOverrides(this.settingsManager.getTerminalCapabilityOverrides());
         const tuiMode = options.tuiMode ?? this.settingsManager.getTuiMode();
         this.options = { ...options, tuiMode };
         this.autoTrustOnReloadCwd = options.autoTrustOnReloadCwd;
@@ -335,6 +337,7 @@ export class InteractiveMode {
             showHardwareCursor: this.settingsManager.getShowHardwareCursor(),
             logDirectory: getAgentDir(),
             onRightClickPaste: this.onRightClickPaste,
+            fullscreenCopyOnSelect: this.settingsManager.getFullscreenCopyOnSelect(),
         });
         this.ui = createInteractiveTuiReference(() => this.renderer);
         this.ui.setClearOnShrink(this.settingsManager.getClearOnShrink());
@@ -582,6 +585,7 @@ export class InteractiveMode {
             logDirectory: getAgentDir(),
             terminal,
             onRightClickPaste: this.onRightClickPaste,
+            fullscreenCopyOnSelect: this.settingsManager.getFullscreenCopyOnSelect(),
         });
         nextUi.setClearOnShrink(clearOnShrink);
         nextUi.onDebug = onDebug;
@@ -1485,8 +1489,12 @@ export class InteractiveMode {
         this.transcriptScrollView?.setScrollbar(this.settingsManager.getFullscreenScrollbar());
     }
     applyRuntimeSettings() {
+        setCapabilityOverrides(this.settingsManager.getTerminalCapabilityOverrides());
         configureHttpDispatcher(this.settingsManager.getHttpIdleTimeoutMs());
         this.applyFullscreenScrollbarSetting();
+        if (this.renderer instanceof TuiAltScreen) {
+            this.renderer.setCopyOnSelect(this.settingsManager.getFullscreenCopyOnSelect());
+        }
         this.footer.setSession(this.session);
         this.footer.setAutoCompactEnabled(this.session.autoCompactionEnabled);
         this.footerDataProvider.setCwd(this.sessionManager.getCwd());
@@ -1683,6 +1691,9 @@ export class InteractiveMode {
             this.statusContainer.addChild(this.idleStatus);
         }
     }
+    showWorkingStatusIndicator() {
+        this.showStatusIndicator(new WorkingStatusIndicator(this.ui, this.workingMessage ?? this.defaultWorkingMessage, this.workingIndicatorOptions));
+    }
     setWorkingVisible(visible) {
         this.workingVisible = visible;
         if (!visible) {
@@ -1691,7 +1702,7 @@ export class InteractiveMode {
             return;
         }
         if (this.session.isStreaming && this.activeStatusIndicator?.kind !== "working") {
-            this.showStatusIndicator(new WorkingStatusIndicator(this.ui, this.workingMessage ?? this.defaultWorkingMessage, this.workingIndicatorOptions));
+            this.showWorkingStatusIndicator();
         }
         this.ui.requestRender();
     }
@@ -2318,7 +2329,7 @@ export class InteractiveMode {
         this.defaultEditor.onAction("app.tools.expand", () => this.toggleToolOutputExpansion());
         this.defaultEditor.onAction("app.thinking.toggle", () => this.toggleThinkingBlockVisibility());
         this.defaultEditor.onAction("app.editor.external", () => void this.handleOpenExternalEditor());
-        this.defaultEditor.onAction("app.message.copy", () => void this.handleCopyCommand({ flashConfirmation: true }));
+        this.defaultEditor.onAction("app.message.copy", () => void this.handleCopyCommand({ flashConfirmation: true, preferSelection: true }));
         this.defaultEditor.onAction("app.message.followUp", () => this.handleFollowUp());
         this.defaultEditor.onAction("app.message.dequeue", () => this.handleDequeue());
         this.defaultEditor.onAction("app.session.new", () => this.handleClearCommand());
@@ -2585,31 +2596,24 @@ export class InteractiveMode {
         switch (event.type) {
             case "agent_start":
                 this.clearPendingTools();
-                if (this.settingsManager.getShowTerminalProgress()) {
-                    this.ui.terminal.setProgress(true);
-                }
                 // Restore main escape handler if retry handler is still active
                 // (retry success event fires later, but we need main handler now)
                 if (this.retryEscapeHandler) {
                     this.defaultEditor.onEscape = this.retryEscapeHandler;
                     this.retryEscapeHandler = undefined;
                 }
-                if (this.workingVisible) {
-                    this.showStatusIndicator(new WorkingStatusIndicator(this.ui, this.workingMessage ?? this.defaultWorkingMessage, this.workingIndicatorOptions));
-                }
-                else {
-                    this.clearStatusIndicator();
-                }
-                this.ui.requestRender();
                 break;
             case "turn_start":
-                // A mid-run compaction replaces the working indicator. Restore it before the
-                // next provider request without waiting for another agent_start event.
                 if (this.settingsManager.getShowTerminalProgress()) {
                     this.ui.terminal.setProgress(true);
                 }
-                if (this.workingVisible && this.activeStatusIndicator?.kind !== "working") {
-                    this.showStatusIndicator(new WorkingStatusIndicator(this.ui, this.workingMessage ?? this.defaultWorkingMessage, this.workingIndicatorOptions));
+                if (this.workingVisible) {
+                    if (this.activeStatusIndicator?.kind !== "working") {
+                        this.showWorkingStatusIndicator();
+                    }
+                }
+                else {
+                    this.clearStatusIndicator();
                 }
                 this.ui.requestRender();
                 break;
@@ -2789,33 +2793,31 @@ export class InteractiveMode {
                         this.showStatus("Auto-compaction cancelled");
                     }
                 }
-                else {
-                    if (event.result) {
-                        const entries = this.sessionManager.buildContextEntries();
-                        if (entries[0]?.type !== "compaction") {
-                            throw new Error("Completed compaction is missing from the session context");
-                        }
-                        this.clearChat();
-                        // The latest compaction is prepended for model context; append it below at its chronological position.
-                        this.renderSessionEntries(entries.slice(1));
-                        this.addMessageToChat(createCompactionSummaryMessage(event.result.summary, event.result.tokensBefore, new Date().toISOString()));
-                        if (event.result.usage) {
-                            this.addCompactionCostNotice({
-                                type: "compaction_cost",
-                                kind: "compaction",
-                                usage: event.result.usage,
-                            });
-                        }
-                        this.footer.invalidate();
+                else if (event.result) {
+                    const entries = this.sessionManager.buildContextEntries();
+                    if (entries[0]?.type !== "compaction") {
+                        throw new Error("Completed compaction is missing from the session context");
                     }
-                    if (event.errorMessage) {
-                        if (event.reason === "manual") {
-                            this.showError(event.errorMessage);
-                        }
-                        else {
-                            this.chatContainer.addChild(new Spacer(1));
-                            this.chatContainer.addChild(new Text(theme.fg("error", event.errorMessage), 1, 0));
-                        }
+                    this.clearChat();
+                    // The latest compaction is prepended for model context; append it below at its chronological position.
+                    this.renderSessionEntries(entries.slice(1));
+                    this.addMessageToChat(createCompactionSummaryMessage(event.result.summary, event.result.tokensBefore, new Date().toISOString()));
+                    if (event.result.usage) {
+                        this.addCompactionCostNotice({
+                            type: "compaction_cost",
+                            kind: "compaction",
+                            usage: event.result.usage,
+                        });
+                    }
+                    this.footer.invalidate();
+                }
+                else if (event.errorMessage) {
+                    if (event.reason === "manual") {
+                        this.showError(event.errorMessage);
+                    }
+                    else {
+                        this.chatContainer.addChild(new Spacer(1));
+                        this.chatContainer.addChild(new Text(theme.fg("error", event.errorMessage), 1, 0));
                     }
                 }
                 void this.flushCompactionQueue({ willRetry: event.willRetry });
@@ -3460,17 +3462,19 @@ export class InteractiveMode {
         }
         this.showStatus(`Tool output: ${expanded ? "expanded" : "collapsed"}`);
     }
+    /** Update rendered assistant messages without rebuilding live tool components. */
+    updateThinkingBlockVisibility() {
+        for (const child of this.chatContainer.children) {
+            if (child instanceof AssistantMessageComponent) {
+                child.setHideThinkingBlock(this.hideThinkingBlock);
+            }
+        }
+        this.ui.requestRender();
+    }
     toggleThinkingBlockVisibility() {
         this.hideThinkingBlock = !this.hideThinkingBlock;
         this.settingsManager.setHideThinkingBlock(this.hideThinkingBlock);
-        // Rebuild chat from session messages
-        this.rebuildChatFromMessages();
-        // If streaming, re-add the streaming component with updated visibility and re-render
-        if (this.streamingComponent && this.streamingMessage) {
-            this.streamingComponent.setHideThinkingBlock(this.hideThinkingBlock);
-            this.streamingComponent.updateContent(this.streamingMessage);
-            this.chatContainer.addChild(this.streamingComponent);
-        }
+        this.updateThinkingBlockVisibility();
         this.showStatus(`Thinking blocks: ${this.hideThinkingBlock ? "hidden" : "visible"}`);
     }
     async handleOpenExternalEditor() {
@@ -3790,6 +3794,7 @@ export class InteractiveMode {
                 tuiMode: this.ui.mode,
                 fullscreenExitOutput: this.settingsManager.getFullscreenExitOutput(),
                 fullscreenScrollbar: this.settingsManager.getFullscreenScrollbar(),
+                fullscreenCopyOnSelect: this.settingsManager.getFullscreenCopyOnSelect(),
                 warnings: this.settingsManager.getWarnings(),
             }, {
                 onAutoCompactChange: (enabled) => {
@@ -3866,12 +3871,7 @@ export class InteractiveMode {
                 onHideThinkingBlockChange: (hidden) => {
                     this.hideThinkingBlock = hidden;
                     this.settingsManager.setHideThinkingBlock(hidden);
-                    for (const child of this.chatContainer.children) {
-                        if (child instanceof AssistantMessageComponent) {
-                            child.setHideThinkingBlock(hidden);
-                        }
-                    }
-                    this.rebuildChatFromMessages();
+                    this.updateThinkingBlockVisibility();
                 },
                 onMermaidRenderingModeChange: (mode) => {
                     this.settingsManager.setMermaidRenderingMode(mode);
@@ -3965,6 +3965,11 @@ export class InteractiveMode {
                     this.settingsManager.setFullscreenScrollbar(mode);
                     this.applyFullscreenScrollbarSetting();
                 },
+                onFullscreenCopyOnSelectChange: (enabled) => {
+                    this.settingsManager.setFullscreenCopyOnSelect(enabled);
+                    if (this.renderer instanceof TuiAltScreen)
+                        this.renderer.setCopyOnSelect(enabled);
+                },
                 onWarningsChange: (warnings) => {
                     this.settingsManager.setWarnings(warnings);
                 },
@@ -4154,6 +4159,7 @@ export class InteractiveMode {
             const selectModel = async (model, persist) => {
                 try {
                     await this.session.setModel(model, { persist });
+                    this.updateAvailableProviderCount();
                     this.footer.invalidate();
                     this.updateEditorBorderColor();
                     done();
@@ -5002,8 +5008,8 @@ export class InteractiveMode {
                 activeHeader.setExpanded(this.toolOutputExpanded);
             }
             setRegisteredThemes(this.session.resourceLoader.getThemes().themes);
-            await this.themeController.applyFromSettings();
             this.applyRuntimeSettings();
+            await this.themeController.applyFromSettings();
             this.setupAutocompleteProvider();
             const runner = this.session.extensionRunner;
             this.setupExtensionShortcuts(runner);
@@ -5125,6 +5131,13 @@ export class InteractiveMode {
         });
     }
     async handleCopyCommand(options = {}) {
+        if (options.preferSelection &&
+            this.ui instanceof TuiAltScreen &&
+            !this.ui.getCopyOnSelect() &&
+            this.ui.hasActiveSelection()) {
+            await this.ui.copyActiveSelectionToClipboard();
+            return;
+        }
         const text = this.session.getLastAssistantText();
         if (!text) {
             this.showError("No agent messages to copy yet.");