@earendil-works/pi-coding-agent 0.84.3 → 0.85.0

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, getCapabilities, hyperlink, Markdown, matchesKey, ProcessTerminal, Spacer, setKeybindings, Text, TruncatedText, TuiAltScreen, TuiMainScreen, visibleWidth, } from "@earendil-works/pi-tui";
+import { CombinedAutocompleteProvider, Container, fuzzyFilter, getCapabilities, hyperlink, Markdown, matchesKey, 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, VERSION, } from "../../config.js";
@@ -26,19 +26,20 @@ import { formatMissingSessionCwdPrompt, MissingSessionCwdError } from "../../cor
 import { SessionManager, sessionEntryToContextMessages } from "../../core/session-manager.js";
 import { BUILTIN_SLASH_COMMANDS } from "../../core/slash-commands.js";
 import { isInstallTelemetryEnabled } from "../../core/telemetry.js";
+import { withBuiltInRenderers } from "../../core/tools/renderers/index.js";
 import { hasTrustRequiringProjectResources, ProjectTrustStore } from "../../core/trust-manager.js";
 import { getUsageCostBreakdown } from "../../core/usage-totals.js";
 import { getChangelogPath, getNewEntries, normalizeChangelogLinks, parseChangelog } from "../../utils/changelog.js";
 import { copyToClipboard, readClipboardText } from "../../utils/clipboard.js";
 import { extensionForImageMimeType, readClipboardImage } from "../../utils/clipboard-image.js";
 import { parseGitUrl } from "../../utils/git.js";
-import { openBrowser } from "../../utils/open-browser.js";
 import { getCwdRelativePath } from "../../utils/paths.js";
 import { getPiUserAgent } from "../../utils/pi-user-agent.js";
 import { killTrackedDetachedChildren } from "../../utils/shell.js";
 import { loadAllHighlightLanguages } from "../../utils/syntax-highlight.js";
 import { ensureTool } from "../../utils/tools-manager.js";
 import { checkForNewPiVersion } from "../../utils/version-check.js";
+import { createChatViewport } from "./chat-viewport.js";
 import { ArminComponent } from "./components/armin.js";
 import { AssistantMessageComponent } from "./components/assistant-message.js";
 import { BashExecutionComponent } from "./components/bash-execution.js";
@@ -76,6 +77,14 @@ import { getModelSearchText } from "./model-search.js";
 import { shareSession } from "./session-share.js";
 import { getAvailableThemes, getAvailableThemesWithPaths, getEditorTheme, getMarkdownTheme, getThemeByName, onThemeChange, setRegisteredThemes, stopThemeWatcher, Theme, theme, } from "./theme/theme.js";
 import { InteractiveThemeController } from "./theme/theme-controller.js";
+import { createInteractiveTui, createInteractiveTuiReference } from "./tui-renderer.js";
+export { createInteractiveTui, createInteractiveTuiReference } from "./tui-renderer.js";
+function isWorkingStatusEditor(editor) {
+    return ("embedWorkingStatus" in editor &&
+        editor.embedWorkingStatus === true &&
+        "setWorkingStatusIndicator" in editor &&
+        typeof editor.setWorkingStatusIndicator === "function");
+}
 function isExpandable(obj) {
     return typeof obj === "object" && obj !== null && "setExpanded" in obj && typeof obj.setExpanded === "function";
 }
@@ -177,60 +186,6 @@ function formatLoginProviderCompletionDescription(provider) {
     const authTypes = provider.authTypes.map(formatAuthSelectorProviderType).join("/");
     return provider.name === provider.id ? authTypes : `${provider.name} · ${authTypes}`;
 }
-/** Composition root for selecting the interactive terminal renderer. */
-export function createInteractiveTui(options) {
-    const terminal = options.terminal ?? new ProcessTerminal();
-    if (options.tuiMode === "fullscreen") {
-        const styleSearchMatch = (text) => theme.bg("searchMatchBg", theme.fg("searchMatchText", text));
-        return new TuiAltScreen(terminal, options.showHardwareCursor, options.logDirectory, {
-            searchMatchStyle: (text) => theme.underline(styleSearchMatch(text)),
-            searchCurrentMatchStyle: (text) => theme.bold(theme.inverse(styleSearchMatch(text))),
-            openUrl: openBrowser,
-            onRightClickPaste: options.onRightClickPaste,
-            copySelection: async (text) => {
-                try {
-                    await copyToClipboard(text);
-                    return true;
-                }
-                catch {
-                    return false;
-                }
-            },
-        });
-    }
-    return new TuiMainScreen(terminal, options.showHardwareCursor, options.logDirectory);
-}
-/** Stable reference for components while InteractiveMode replaces the active renderer. */
-export function createInteractiveTuiReference(getTui) {
-    return new Proxy({}, {
-        get: (_target, property) => {
-            const tui = getTui();
-            const value = Reflect.get(tui, property, tui);
-            if (typeof value !== "function")
-                return value;
-            let methodTui = tui;
-            let method = value;
-            return (...args) => {
-                const currentTui = getTui();
-                if (currentTui !== methodTui) {
-                    const currentMethod = Reflect.get(currentTui, property, currentTui);
-                    if (typeof currentMethod !== "function") {
-                        throw new TypeError(`TUI property ${String(property)} is not callable`);
-                    }
-                    methodTui = currentTui;
-                    method = currentMethod;
-                }
-                return Reflect.apply(method, methodTui, args);
-            };
-        },
-        set: (_target, property, value) => {
-            const tui = getTui();
-            return Reflect.set(tui, property, value, tui);
-        },
-        has: (_target, property) => Reflect.has(getTui(), property),
-        getPrototypeOf: () => Reflect.getPrototypeOf(getTui()),
-    });
-}
 export class InteractiveMode {
     runtimeHost;
     renderer;
@@ -262,11 +217,12 @@ export class InteractiveMode {
     onInputCallback;
     pendingUserInputs = [];
     activeStatusIndicator = undefined;
+    activeWorkingIndicatorEmbedded = false;
     idleStatus = new IdleStatus();
     workingMessage = undefined;
     workingVisible = true;
     workingIndicatorOptions = undefined;
-    defaultWorkingMessage = "Working...";
+    defaultWorkingMessage = "Working";
     defaultHiddenThinkingLabel = "Thinking...";
     hiddenThinkingLabel = this.defaultHiddenThinkingLabel;
     lastSigintTime = 0;
@@ -350,6 +306,7 @@ export class InteractiveMode {
     }
     constructor(runtimeHost, options = {}) {
         this.runtimeHost = runtimeHost;
+        setCapabilityOverrides(this.settingsManager.getTerminalCapabilityOverrides());
         const tuiMode = options.tuiMode ?? this.settingsManager.getTuiMode();
         this.options = { ...options, tuiMode };
         this.autoTrustOnReloadCwd = options.autoTrustOnReloadCwd;
@@ -366,6 +323,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());
@@ -387,6 +345,7 @@ export class InteractiveMode {
         this.defaultEditor = new CustomEditor(this.ui, getEditorTheme(), this.keybindings, {
             paddingX: editorPaddingX,
             autocompleteMaxVisible,
+            embedWorkingStatus: true,
         });
         this.editor = this.defaultEditor;
         this.editorContainer = new Container();
@@ -613,6 +572,7 @@ export class InteractiveMode {
             logDirectory: getAgentDir(),
             terminal,
             onRightClickPaste: this.onRightClickPaste,
+            fullscreenCopyOnSelect: this.settingsManager.getFullscreenCopyOnSelect(),
         });
         nextUi.setClearOnShrink(clearOnShrink);
         nextUi.onDebug = onDebug;
@@ -657,25 +617,20 @@ export class InteractiveMode {
         }
         // Keep one component tree and remount it when changing renderers.
         this.renderWidgets(); // Initialize with default spacer
-        this.transcriptScrollView = new TuiLayouts.ScrollView(this.documentContainer, {
-            follow: "end",
-            primary: true,
-            overscroll: "chain",
+        const viewport = createChatViewport({
+            document: this.documentContainer,
+            pendingMessages: this.pendingMessagesContainer,
+            status: this.statusContainer,
+            widgetsAbove: this.widgetContainerAbove,
+            editor: this.editorContainer,
+            widgetsBelow: this.widgetContainerBelow,
+            footer: this.footerContainer,
             scrollbar: this.settingsManager.getFullscreenScrollbar(),
-            scrollbarStyle: (text) => theme.bg("scrollbarThumb", text),
+            scrollbarTrackStyle: (text) => theme.fg("scrollbarTrack", text),
+            scrollbarThumbStyle: (text) => theme.fg("scrollbarThumb", text),
         });
-        const dock = new TuiLayouts.VStack([
-            { component: this.pendingMessagesContainer, shrink: 1, minSize: 0 },
-            { component: this.statusContainer, shrink: 1, minSize: 0 },
-            { component: this.widgetContainerAbove, shrink: 1, minSize: 0 },
-            { component: this.editorContainer, shrink: 1, minSize: 3 },
-            { component: this.widgetContainerBelow, shrink: 1, minSize: 0 },
-            { component: this.footerContainer, shrink: 1, minSize: 1 },
-        ]);
-        this.fullscreenLayoutRoot = new TuiLayouts.VStack([
-            { component: this.transcriptScrollView, basis: 0, grow: 1, shrink: 1, minSize: 1 },
-            { component: dock, basis: "auto", grow: 0, shrink: 1, minSize: 1 },
-        ]);
+        this.transcriptScrollView = viewport.transcript;
+        this.fullscreenLayoutRoot = viewport.root;
         this.mountInteractiveTui(this.renderer, [
             this.documentContainer,
             this.pendingMessagesContainer,
@@ -1521,8 +1476,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());
@@ -1583,8 +1542,12 @@ export class InteractiveMode {
     /**
      * Get a registered tool definition by name (for custom rendering).
      */
+    /**
+     * Extension-registered definition, falling back to the built-in one. The renderer components take
+     * whatever this returns, so they never reach into the tool registry themselves.
+     */
     getRegisteredToolDefinition(toolName) {
-        return this.session.getToolDefinition(toolName);
+        return withBuiltInRenderers(toolName, this.session.getToolDefinition(toolName));
     }
     getMarkdownTransformers() {
         return [this.mermaidMarkdownTransformer, ...this.session.extensionRunner.getMarkdownTransformers()];
@@ -1654,24 +1617,49 @@ export class InteractiveMode {
         this.footerDataProvider.setExtensionStatus(key, text);
         this.ui.requestRender();
     }
+    setEditorWorkingStatusIndicator(indicator) {
+        this.defaultEditor.setWorkingStatusIndicator(undefined);
+        if (!isWorkingStatusEditor(this.editor))
+            return false;
+        this.editor.setWorkingStatusIndicator(indicator);
+        return true;
+    }
     showStatusIndicator(indicator) {
         this.activeStatusIndicator?.dispose();
         this.activeStatusIndicator = indicator;
+        this.activeWorkingIndicatorEmbedded = false;
         this.statusContainer.clear();
+        this.setEditorWorkingStatusIndicator(undefined);
+        if (indicator instanceof WorkingStatusIndicator && this.setEditorWorkingStatusIndicator(indicator)) {
+            this.activeWorkingIndicatorEmbedded = true;
+            return;
+        }
         this.statusContainer.addChild(indicator);
     }
     clearStatusIndicator(kind) {
         if (kind && this.activeStatusIndicator?.kind !== kind) {
             return;
         }
-        const hadActiveStatusIndicator = this.activeStatusIndicator !== undefined;
-        this.activeStatusIndicator?.dispose();
+        const clearedIndicator = this.activeStatusIndicator;
+        const clearedIndicatorWasEmbedded = clearedIndicator?.kind === "working" && this.activeWorkingIndicatorEmbedded;
+        clearedIndicator?.dispose();
         this.activeStatusIndicator = undefined;
+        this.activeWorkingIndicatorEmbedded = false;
         this.statusContainer.clear();
-        if (hadActiveStatusIndicator && this.options.tuiMode === "regular" && this.ui.getClearOnShrink()) {
+        this.setEditorWorkingStatusIndicator(undefined);
+        if (clearedIndicator &&
+            !clearedIndicatorWasEmbedded &&
+            this.options.tuiMode === "regular" &&
+            this.ui.getClearOnShrink()) {
             this.statusContainer.addChild(this.idleStatus);
         }
     }
+    showWorkingStatusIndicator() {
+        const colorFn = isWorkingStatusEditor(this.editor)
+            ? (text) => (this.editor.borderColor ?? theme.getThinkingBorderColor(this.session.thinkingLevel || "off"))(text)
+            : undefined;
+        this.showStatusIndicator(new WorkingStatusIndicator(this.ui, this.workingMessage ?? this.defaultWorkingMessage, this.workingIndicatorOptions, colorFn));
+    }
     setWorkingVisible(visible) {
         this.workingVisible = visible;
         if (!visible) {
@@ -1680,7 +1668,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();
     }
@@ -2142,6 +2130,13 @@ export class InteractiveMode {
             this.editor = this.defaultEditor;
         }
         this.editorContainer.addChild(this.editor);
+        if (this.activeStatusIndicator instanceof WorkingStatusIndicator) {
+            this.statusContainer.clear();
+            this.activeWorkingIndicatorEmbedded = this.setEditorWorkingStatusIndicator(this.activeStatusIndicator);
+            if (!this.activeWorkingIndicatorEmbedded) {
+                this.statusContainer.addChild(this.activeStatusIndicator);
+            }
+        }
         this.ui.setFocus(this.editor);
         this.ui.requestRender();
     }
@@ -2300,7 +2295,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());
@@ -2567,17 +2562,21 @@ export class InteractiveMode {
         switch (event.type) {
             case "agent_start":
                 this.pendingTools.clear();
-                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;
                 }
+                break;
+            case "turn_start":
+                if (this.settingsManager.getShowTerminalProgress()) {
+                    this.ui.terminal.setProgress(true);
+                }
                 if (this.workingVisible) {
-                    this.showStatusIndicator(new WorkingStatusIndicator(this.ui, this.workingMessage ?? this.defaultWorkingMessage, this.workingIndicatorOptions));
+                    if (this.activeStatusIndicator?.kind !== "working") {
+                        this.showWorkingStatusIndicator();
+                    }
                 }
                 else {
                     this.clearStatusIndicator();
@@ -2679,6 +2678,7 @@ export class InteractiveMode {
                         for (const [, component] of this.pendingTools.entries()) {
                             component.setArgsComplete();
                         }
+                        this.maybeShowAssistantDiagnostics(this.streamingMessage);
                         this.maybeShowCacheMissNotice(this.streamingMessage);
                     }
                     this.streamingComponent = undefined;
@@ -3042,6 +3042,7 @@ export class InteractiveMode {
                     }
                 }
                 if (message.stopReason !== "aborted" && message.stopReason !== "error") {
+                    this.maybeShowAssistantDiagnostics(message);
                     const miss = cacheMisses.get(message);
                     if (miss)
                         this.addCacheMissNotice(miss);
@@ -3098,6 +3099,32 @@ export class InteractiveMode {
         this.chatContainer.addChild(new Spacer(1));
         this.chatContainer.addChild(new Text(theme.fg("warning", `${label}: ${formatTokens(tokens)} tokens billed${cost}`), 1, 0));
     }
+    maybeShowAssistantDiagnostics(message) {
+        if (!this.settingsManager.getShowCacheMissNotices())
+            return;
+        for (const diagnostic of message.diagnostics ?? []) {
+            if (diagnostic.type !== "anthropic_input_transformations")
+                continue;
+            const transformations = diagnostic.details?.transformations;
+            if (!Array.isArray(transformations))
+                continue;
+            const dropped = transformations.flatMap((transformation) => {
+                if (typeof transformation !== "object" || transformation === null)
+                    return [];
+                const details = transformation;
+                if (details.type !== "thinking_dropped")
+                    return [];
+                const reason = typeof details.reason === "string" ? details.reason : "unknown reason";
+                const location = typeof details.path === "string" ? ` at ${details.path}` : "";
+                return [`${reason}${location}`];
+            });
+            if (dropped.length === 0)
+                continue;
+            const noun = dropped.length === 1 ? "thinking block" : `${dropped.length} thinking blocks`;
+            this.chatContainer.addChild(new Spacer(1));
+            this.chatContainer.addChild(new Text(theme.fg("warning", `Anthropic dropped ${noun}: ${dropped.join("; ")}`), 1, 0));
+        }
+    }
     /**
      * Show a transcript notice when a completed assistant message paid for a
      * significant cache miss. Only states observable facts: the miss itself,
@@ -3395,6 +3422,9 @@ export class InteractiveMode {
             const level = this.session.thinkingLevel || "off";
             this.editor.borderColor = theme.getThinkingBorderColor(level);
         }
+        if (this.activeStatusIndicator?.kind === "working") {
+            this.activeStatusIndicator.invalidate();
+        }
         this.ui.requestRender();
     }
     cycleThinkingLevel() {
@@ -3447,18 +3477,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.chatContainer.clear();
-        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() {
@@ -3775,6 +3806,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) => {
@@ -3851,13 +3883,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.chatContainer.clear();
-                    this.rebuildChatFromMessages();
+                    this.updateThinkingBlockVisibility();
                 },
                 onMermaidRenderingModeChange: (mode) => {
                     this.settingsManager.setMermaidRenderingMode(mode);
@@ -3951,6 +3977,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);
                 },
@@ -4140,6 +4171,7 @@ export class InteractiveMode {
             const selectModel = async (model, persist) => {
                 try {
                     await this.session.setModel(model, { persist });
+                    this.updateAvailableProviderCount();
                     this.footer.invalidate();
                     this.updateEditorBorderColor();
                     done();
@@ -4988,8 +5020,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);
@@ -5111,6 +5143,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.");