@astralyn/pi 1.0.5 → 1.2.0

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

@@ -1,5 +1,5 @@
+import { confirmStopBranchTasks, hotkeysTaskSection } from "./tasks/host-hooks.js";
 import { bindTasksUI } from "./tasks/index.js";
-import { cleanTaskText, firstCommandLine } from "./tasks/text.js";
 /**
  * Interactive mode for the coding agent.
  * Handles TUI rendering and user interaction, delegating business logic to AgentSession.
@@ -22,7 +22,7 @@ import { findExtensionStackMatches, recordCrash, takeUnnotifiedCrash } from "../
 import { DEFAULT_THINKING_LEVEL, THINKING_LEVEL_OPTIONS } from "../../core/defaults.js";
 import { FooterDataProvider } from "../../core/footer-data-provider.js";
 import { configureHttpDispatcher, formatHttpIdleTimeoutMs } from "../../core/http-dispatcher.js";
-import { KEYBINDINGS, KeybindingsManager } from "../../core/keybindings.js";
+import { KeybindingsManager } from "../../core/keybindings.js";
 import { createCompactionSummaryMessage, createCustomMessage } from "../../core/messages.js";
 import { defaultModelPerProvider, findExactModelReferenceMatch, resolveModelScopeFromModels, } from "../../core/model-resolver.js";
 import { CredentialSynchronizationError } from "../../core/model-runtime.js";
@@ -86,6 +86,7 @@ import { InteractiveEditorHost } from "./editor-host.js";
 import { editInExternalEditor } from "./external-editor.js";
 import { refreshModelCatalogs } from "./model-catalog-refresh.js";
 import { getModelSearchText } from "./model-search.js";
+import { ProgramStatusReporter } from "./program-status-reporter.js";
 import { shareSession } from "./session-share.js";
 import { getAvailableThemes, getAvailableThemesWithPaths, getEditorTheme, getMarkdownTheme, getThemeByName, onThemeChange, SYSTEM_THEME_NAME, setRegisteredThemes, stopThemeWatcher, Theme, theme, } from "./theme/theme.js";
 import { InteractiveThemeController } from "./theme/theme-controller.js";
@@ -93,7 +94,7 @@ import { disposeChatRows, PendingToolMap, ToolChatContainer } from "./tool-view/
 import { ToolExecutionComponent } from "./tool-view/tool-execution.js";
 import { createInteractiveTui, createInteractiveTuiReference } from "./tui-renderer.js";
 export { createInteractiveTui, createInteractiveTuiReference } from "./tui-renderer.js";
-function isWorkingStatusEditor(editor) {
+function isEmbeddedStatus(editor) {
     return ("embedWorkingStatus" in editor &&
         editor.embedWorkingStatus === true &&
         "setWorkingStatusIndicator" in editor &&
@@ -131,22 +132,6 @@ function isCompactionCostNotice(item) {
 function isUsageSessionEntry(item) {
     return "type" in item && item.type === "usage";
 }
-/**
- * Task rows for `/hotkeys`. Task keybindings act outside the editor, so they never showed up in
- * the editor-driven sections; labels come from KEYBINDINGS so the table cannot drift from the
- * binding registry.
- */
-function hotkeysTaskSection(display) {
-    const detach = KEYBINDINGS["app.tasks.detach"];
-    const detachSelected = KEYBINDINGS["app.tasks.detachSelected"];
-    return `
-**Tasks**
-| Key | Action |
-|-----|--------|
-| \`${display("app.tasks.detach")}\` | ${detach.description} |
-| \`${display("app.tasks.detachSelected")}\` | ${detachSelected.description} (while /tasks is open) |
-`;
-}
 // EIO: tty reads/ioctls from an orphaned background process group, or writes after hangup.
 // ENOTTY: the tty was revoked (macOS) and stdin is no longer a terminal.
 const DEAD_TERMINAL_ERROR_CODES = new Set(["EIO", "EPIPE", "ENOTCONN", "ENOTTY"]);
@@ -326,6 +311,8 @@ export class InteractiveMode {
         this.compactionQueuedMessages = [];
         // Shutdown state
         this.shutdownRequested = false;
+        /** Reports working, blocked, done, and error states to terminals that support OSC 7501. */
+        this.programStatus = new ProgramStatusReporter(() => this.ui.terminal, () => this.sessionManager.getSessionName());
         /** The `/bug` hint is shown at most once per session so error output stays readable. */
         this.bugReportHintShown = false;
         this.installChangeWarningShown = false;
@@ -700,6 +687,7 @@ export class InteractiveMode {
         // Start the UI before initializing extensions so session_start handlers can use interactive dialogs
         this.ui.start();
         this.isInitialized = true;
+        this.programStatus.report();
         this.ensurePngTranscoder();
         this.themeController.applyFromSettings();
         // The header and startup notices bake theme colors into their text, so build them once the terminal
@@ -1588,6 +1576,7 @@ export class InteractiveMode {
         const session = this.session;
         this.unsubscribe?.();
         this.unsubscribe = undefined;
+        this.programStatus.reset();
         this.applyRuntimeSettings();
         if (options.renderBeforeBind) {
             this.renderCurrentSessionState();
@@ -1737,7 +1726,7 @@ export class InteractiveMode {
     }
     setEditorWorkingStatusIndicator(indicator) {
         this.defaultEditor.setWorkingStatusIndicator(undefined);
-        if (!isWorkingStatusEditor(this.editor))
+        if (!isEmbeddedStatus(this.editor))
             return false;
         this.editor.setWorkingStatusIndicator(indicator);
         return true;
@@ -1773,7 +1762,7 @@ export class InteractiveMode {
         }
     }
     showWorkingStatusIndicator() {
-        const colorFn = isWorkingStatusEditor(this.editor)
+        const colorFn = isEmbeddedStatus(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));
@@ -2070,7 +2059,7 @@ export class InteractiveMode {
     /**
      * Show a selector for extensions.
      */
-    showExtensionSelector(title, options, opts) {
+    showExtensionSelector(title, options, opts, blocked = { kind: "question", message: title }) {
         return new Promise((resolve) => {
             if (opts?.signal?.aborted) {
                 resolve(undefined);
@@ -2099,6 +2088,8 @@ export class InteractiveMode {
             this.editorContainer.clear();
             this.editorContainer.addChild(this.extensionSelector);
             this.ui.setFocus(this.extensionSelector);
+            // Extension dialogs share the editor slot: opening one replaces the status of a displaced one.
+            this.programStatus.setBlocked("extension-dialog", blocked);
             this.ui.requestRender();
         });
     }
@@ -2110,6 +2101,7 @@ export class InteractiveMode {
         this.editorContainer.clear();
         this.editorContainer.addChild(this.editor);
         this.extensionSelector = undefined;
+        this.programStatus.setBlocked("extension-dialog", undefined);
         this.ui.setFocus(this.editor);
         this.ui.requestRender();
     }
@@ -2119,7 +2111,7 @@ export class InteractiveMode {
     async showExtensionConfirm(title, message, opts) {
         // The message is context, not a heading: render it as the selector's
         // description instead of folding it into the accent-bold title.
-        const result = await this.showExtensionSelector(title, ["Yes", "No"], { ...opts, description: message });
+        const result = await this.showExtensionSelector(title, ["Yes", "No"], { ...opts, description: message }, { kind: "permission", message: title });
         return result === "Yes";
     }
     async promptForMissingSessionCwd(error) {
@@ -2153,6 +2145,7 @@ export class InteractiveMode {
             this.editorContainer.clear();
             this.editorContainer.addChild(this.extensionInput);
             this.ui.setFocus(this.extensionInput);
+            this.programStatus.setBlocked("extension-dialog", { kind: "question", message: title });
             this.ui.requestRender();
         });
     }
@@ -2164,6 +2157,7 @@ export class InteractiveMode {
         this.editorContainer.clear();
         this.editorContainer.addChild(this.editor);
         this.extensionInput = undefined;
+        this.programStatus.setBlocked("extension-dialog", undefined);
         this.ui.setFocus(this.editor);
         this.ui.requestRender();
     }
@@ -2183,6 +2177,7 @@ export class InteractiveMode {
             this.editorContainer.clear();
             this.editorContainer.addChild(this.extensionEditor);
             this.ui.setFocus(this.extensionEditor);
+            this.programStatus.setBlocked("extension-dialog", { kind: "question", message: title });
             this.ui.requestRender();
         });
     }
@@ -2193,6 +2188,7 @@ export class InteractiveMode {
         this.editorContainer.clear();
         this.editorContainer.addChild(this.editor);
         this.extensionEditor = undefined;
+        this.programStatus.setBlocked("extension-dialog", undefined);
         this.ui.setFocus(this.editor);
         this.ui.requestRender();
     }
@@ -2712,6 +2708,7 @@ export class InteractiveMode {
             await this.init();
         }
         this.footer.invalidate();
+        this.programStatus.handleEvent(event);
         switch (event.type) {
             case "agent_start":
                 this.pendingTools.clear();
@@ -2817,6 +2814,7 @@ export class InteractiveMode {
                                 const component = new ToolExecutionComponent(content.name, content.id, content.arguments, {
                                     showImages: this.settingsManager.getShowImages(),
                                     imageWidthCells: this.settingsManager.getImageWidthCells(),
+                                    outputPad: this.outputPad,
                                 }, this.getRegisteredToolDefinition(content.name), this.ui, this.sessionManager.getCwd());
                                 component.setExpanded(this.toolOutputExpanded);
                                 this.chatContainer.addChild(component);
@@ -2887,6 +2885,7 @@ export class InteractiveMode {
                     component = new ToolExecutionComponent(event.toolName, event.toolCallId, event.args, {
                         showImages: this.settingsManager.getShowImages(),
                         imageWidthCells: this.settingsManager.getImageWidthCells(),
+                        outputPad: this.outputPad,
                     }, this.getRegisteredToolDefinition(event.toolName), this.ui, this.sessionManager.getCwd());
                     component.setExpanded(this.toolOutputExpanded);
                     this.chatContainer.addChild(component);
@@ -2909,7 +2908,7 @@ export class InteractiveMode {
                     this.maybeShowInstallChangeWarning();
                 const component = this.pendingTools.get(event.toolCallId);
                 if (component) {
-                    component.updateResult({ ...event.result, isError: event.isError });
+                    component.updateResult({ ...event.result, isError: event.isError, durationMs: event.durationMs });
                     this.pendingTools.delete(event.toolCallId);
                     this.ui.requestRender();
                 }
@@ -2940,7 +2939,10 @@ export class InteractiveMode {
                 this.defaultEditor.onEscape = () => {
                     this.session.abortCompaction();
                 };
-                this.showStatusIndicator(new CompactionStatusIndicator(this.ui, event.reason));
+                const colorFn = isEmbeddedStatus(this.editor)
+                    ? (text) => (this.editor.borderColor ?? theme.getThinkingBorderColor(this.session.thinkingLevel || "off"))(text)
+                    : undefined;
+                this.showStatusIndicator(new CompactionStatusIndicator(this.ui, event.reason, colorFn));
                 this.ui.requestRender();
                 break;
             }
@@ -3029,7 +3031,10 @@ export class InteractiveMode {
                     this.showStatusIndicator(new BranchSummaryStatusIndicator(this.ui));
                 }
                 else {
-                    this.showStatusIndicator(new CompactionStatusIndicator(this.ui, event.reason));
+                    const colorFn = isEmbeddedStatus(this.editor)
+                        ? (text) => (this.editor.borderColor ?? theme.getThinkingBorderColor(this.session.thinkingLevel || "off"))(text)
+                        : undefined;
+                    this.showStatusIndicator(new CompactionStatusIndicator(this.ui, event.reason, colorFn));
                 }
                 this.ui.requestRender();
                 break;
@@ -3093,7 +3098,7 @@ export class InteractiveMode {
         if (!renderer) {
             return;
         }
-        const component = new CustomEntryComponent(entry, renderer);
+        const component = new CustomEntryComponent(entry, renderer, this.outputPad);
         component.setExpanded(this.toolOutputExpanded);
         if (!component.hasContent()) {
             return;
@@ -3110,7 +3115,7 @@ export class InteractiveMode {
     addMessageToChat(message, options) {
         switch (message.role) {
             case "bashExecution": {
-                const component = new BashExecutionComponent(message.command, this.ui, message.excludeFromContext);
+                const component = new BashExecutionComponent(message.command, this.ui, message.excludeFromContext, this.outputPad);
                 if (message.output) {
                     component.appendOutput(message.output);
                 }
@@ -3129,14 +3134,14 @@ export class InteractiveMode {
             }
             case "compactionSummary": {
                 this.chatContainer.addChild(new Spacer(1));
-                const component = new CompactionSummaryMessageComponent(message, this.getMarkdownThemeWithSettings());
+                const component = new CompactionSummaryMessageComponent(message, this.getMarkdownThemeWithSettings(), this.outputPad);
                 component.setExpanded(this.toolOutputExpanded);
                 this.chatContainer.addChild(component);
                 break;
             }
             case "branchSummary": {
                 this.chatContainer.addChild(new Spacer(1));
-                const component = new BranchSummaryMessageComponent(message, this.getMarkdownThemeWithSettings());
+                const component = new BranchSummaryMessageComponent(message, this.getMarkdownThemeWithSettings(), this.outputPad);
                 component.setExpanded(this.toolOutputExpanded);
                 this.chatContainer.addChild(component);
                 break;
@@ -3152,7 +3157,7 @@ export class InteractiveMode {
                     const skillBlock = parseSkillBlock(textContent);
                     if (skillBlock) {
                         // Render skill block (collapsible)
-                        const component = new SkillInvocationMessageComponent(skillBlock, this.getMarkdownThemeWithSettings());
+                        const component = new SkillInvocationMessageComponent(skillBlock, this.getMarkdownThemeWithSettings(), this.outputPad);
                         component.setExpanded(this.toolOutputExpanded);
                         this.chatContainer.addChild(component);
                         // Render user message separately if present
@@ -3221,6 +3226,7 @@ export class InteractiveMode {
                         const component = new ToolExecutionComponent(content.name, content.id, content.arguments, {
                             showImages: this.settingsManager.getShowImages(),
                             imageWidthCells: this.settingsManager.getImageWidthCells(),
+                            outputPad: this.outputPad,
                         }, this.getRegisteredToolDefinition(content.name), this.ui, this.sessionManager.getCwd());
                         component.setExpanded(this.toolOutputExpanded);
                         this.chatContainer.addChild(component);
@@ -3274,6 +3280,9 @@ export class InteractiveMode {
      * @param options.populateHistory Add user messages to editor history
      */
     renderSessionEntries(entries, options = {}) {
+        // Selection coordinates point into the transcript being replaced (#9311).
+        if (this.renderer instanceof TuiAltScreen)
+            this.renderer.resetTextSelection();
         const items = entries.flatMap((entry) => {
             if (entry.type === "custom" || (entry.type === "usage" && entry.kind === "cache_warm")) {
                 return [entry];
@@ -4170,21 +4179,14 @@ export class InteractiveMode {
                 onOutputPadChange: (padding) => {
                     this.settingsManager.setOutputPad(padding);
                     this.outputPad = padding;
-                    if (this.streamingComponent || this.session.isStreaming) {
-                        for (const child of this.chatContainer.children) {
-                            if (child instanceof AssistantMessageComponent ||
-                                child instanceof CustomMessageComponent ||
-                                child instanceof UserMessageComponent) {
+                    for (const container of [this.chatContainer, this.pendingMessagesContainer]) {
+                        for (const child of container.children) {
+                            if ("setOutputPad" in child && typeof child.setOutputPad === "function") {
                                 child.setOutputPad(padding);
                             }
                         }
-                        if (this.streamingComponent) {
-                            this.streamingComponent.setOutputPad(padding);
-                        }
-                        this.ui.requestRender();
-                        return;
                     }
-                    this.rebuildChatFromMessages();
+                    this.ui.requestRender();
                 },
                 onAutocompleteMaxVisibleChange: (maxVisible) => {
                     this.settingsManager.setAutocompleteMaxVisible(maxVisible);
@@ -4619,19 +4621,10 @@ export class InteractiveMode {
                 }
                 // Ask about summarization
                 done(); // Close selector first
-                // Results are saved on their launch branch, so leaving it stops its running tasks.
                 const stopped = this.session.tasksStoppedByTreeNavigation(entryId);
-                if (stopped.length > 0) {
-                    const shown = stopped
-                        .slice(0, 5)
-                        .map((task) => `  ${cleanTaskText(firstCommandLine(task.command ?? task.title))}`);
-                    if (stopped.length > shown.length)
-                        shown.push(`  …and ${stopped.length - shown.length} more`);
-                    const confirmed = await this.showExtensionConfirm(`Stop ${stopped.length} running task${stopped.length === 1 ? "" : "s"}?`, `They were started on the branch you are leaving and will be cancelled:\n${shown.join("\n")}`);
-                    if (!confirmed) {
-                        this.showTreeSelector(entryId);
-                        return;
-                    }
+                if (!(await confirmStopBranchTasks(stopped, (title, message) => this.showExtensionConfirm(title, message)))) {
+                    this.showTreeSelector(entryId);
+                    return;
                 }
                 // Loop until user makes a complete choice or cancels to tree
                 let wantsSummary = false;
@@ -5161,7 +5154,7 @@ export class InteractiveMode {
             this.ui.requestRender();
         };
         try {
-            await this.loginProvider(dialog, providerId, "api_key");
+            await this.loginProvider(dialog, providerId, providerName, "api_key");
             restoreEditor();
             await this.completeProviderAuthentication(providerId, providerName, "api_key", previousModel);
         }
@@ -5249,12 +5242,18 @@ export class InteractiveMode {
             dialog.showProgress(event.message);
         }
     }
-    async loginProvider(dialog, providerId, method) {
-        await this.session.modelRuntime.login(providerId, method, {
-            signal: dialog.signal,
-            prompt: (prompt) => this.showAuthPrompt(dialog, prompt, providerId),
-            notify: (event) => this.notifyAuthDialog(dialog, event),
-        }, { getDeviceId: () => this.settingsManager.getOrCreateDeviceId() });
+    async loginProvider(dialog, providerId, providerName, method) {
+        this.programStatus.setBlocked("login", { kind: "auth", message: `Log in to ${providerName}` });
+        try {
+            await this.session.modelRuntime.login(providerId, method, {
+                signal: dialog.signal,
+                prompt: (prompt) => this.showAuthPrompt(dialog, prompt, providerId),
+                notify: (event) => this.notifyAuthDialog(dialog, event),
+            }, { getDeviceId: () => this.settingsManager.getOrCreateDeviceId() });
+        }
+        finally {
+            this.programStatus.setBlocked("login", undefined);
+        }
     }
     async showLoginDialog(providerId, providerName, onBack) {
         const previousModel = this.session.model;
@@ -5270,7 +5269,7 @@ export class InteractiveMode {
             this.ui.requestRender();
         };
         try {
-            await this.loginProvider(dialog, providerId, "oauth");
+            await this.loginProvider(dialog, providerId, providerName, "oauth");
             restoreEditor();
             await this.completeProviderAuthentication(providerId, providerName, "oauth", previousModel);
             if (providerId === RADIUS_PROVIDER_ID)
@@ -5861,7 +5860,7 @@ export class InteractiveMode {
         if (eventResult?.result) {
             const result = eventResult.result;
             // Create UI component for display
-            this.bashComponent = new BashExecutionComponent(command, this.ui, excludeFromContext);
+            this.bashComponent = new BashExecutionComponent(command, this.ui, excludeFromContext, this.outputPad);
             if (this.session.isStreaming) {
                 this.pendingMessagesContainer.addChild(this.bashComponent);
                 this.pendingBashComponents.push(this.bashComponent);
@@ -5882,7 +5881,7 @@ export class InteractiveMode {
         }
         // Normal execution path (possibly with custom operations)
         const isDeferred = this.session.isStreaming;
-        this.bashComponent = new BashExecutionComponent(command, this.ui, excludeFromContext);
+        this.bashComponent = new BashExecutionComponent(command, this.ui, excludeFromContext, this.outputPad);
         if (isDeferred) {
             // Show in pending area when agent is streaming
             this.pendingMessagesContainer.addChild(this.bashComponent);