@mariozechner/pi-coding-agent 0.51.1 → 0.51.2

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

@@ -9,7 +9,7 @@ import * as path from "node:path";
 import { getOAuthProviders, } from "@mariozechner/pi-ai";
 import { CombinedAutocompleteProvider, Container, fuzzyFilter, Loader, Markdown, matchesKey, ProcessTerminal, Spacer, Text, TruncatedText, TUI, visibleWidth, } from "@mariozechner/pi-tui";
 import { spawn, spawnSync } from "child_process";
-import { APP_NAME, getAuthPath, getDebugLogPath, getShareViewerUrl, isBunBinary, isBunRuntime, VERSION, } from "../../config.js";
+import { APP_NAME, getAuthPath, getDebugLogPath, getShareViewerUrl, getUpdateInstruction, VERSION, } from "../../config.js";
 import { parseSkillBlock } from "../../core/agent-session.js";
 import { FooterDataProvider } from "../../core/footer-data-provider.js";
 import { KeybindingsManager } from "../../core/keybindings.js";
@@ -218,7 +218,8 @@ export class InteractiveMode {
             description: cmd.description,
         }));
         // Convert extension commands to SlashCommand format
-        const extensionCommands = (this.session.extensionRunner?.getRegisteredCommands() ?? []).map((cmd) => ({
+        const builtinCommandNames = new Set(slashCommands.map((c) => c.name));
+        const extensionCommands = (this.session.extensionRunner?.getRegisteredCommands(builtinCommandNames) ?? []).map((cmd) => ({
             name: cmd.name,
             description: cmd.description ?? "(extension command)",
             getArgumentCompletions: cmd.getArgumentCompletions,
@@ -724,6 +725,8 @@ export class InteractiveMode {
                 extensionDiagnostics.push({ type: "error", message: error.error, path: error.path });
             }
         }
+        const commandDiagnostics = this.session.extensionRunner?.getCommandDiagnostics() ?? [];
+        extensionDiagnostics.push(...commandDiagnostics);
         const shortcutDiagnostics = this.session.extensionRunner?.getShortcutDiagnostics() ?? [];
         extensionDiagnostics.push(...shortcutDiagnostics);
         if (extensionDiagnostics.length > 0) {
@@ -1112,6 +1115,8 @@ export class InteractiveMode {
                 }
                 return result;
             },
+            getToolsExpanded: () => this.toolOutputExpanded,
+            setToolsExpanded: (expanded) => this.setToolsExpanded(expanded),
         };
     }
     /**
@@ -2110,6 +2115,9 @@ export class InteractiveMode {
         // Wait for any pending renders to complete
         // requestRender() uses process.nextTick(), so we wait one tick
         await new Promise((resolve) => process.nextTick(resolve));
+        // Drain any in-flight Kitty key release events before stopping.
+        // This prevents escape sequences from leaking to the parent shell over slow SSH.
+        await this.ui.terminal.drainInput(1000);
         this.stop();
         process.exit(0);
     }
@@ -2211,10 +2219,13 @@ export class InteractiveMode {
         }
     }
     toggleToolOutputExpansion() {
-        this.toolOutputExpanded = !this.toolOutputExpanded;
+        this.setToolsExpanded(!this.toolOutputExpanded);
+    }
+    setToolsExpanded(expanded) {
+        this.toolOutputExpanded = expanded;
         for (const child of this.chatContainer.children) {
             if (isExpandable(child)) {
-                child.setExpanded(this.toolOutputExpanded);
+                child.setExpanded(expanded);
             }
         }
         this.ui.requestRender();
@@ -2292,9 +2303,7 @@ export class InteractiveMode {
         this.ui.requestRender();
     }
     showNewVersionNotification(newVersion) {
-        const action = isBunBinary
-            ? `Download from: ${theme.fg("accent", "https://github.com/badlogic/pi-mono/releases/latest")}`
-            : `Run: ${theme.fg("accent", `${isBunRuntime ? "bun" : "npm"} install -g @mariozechner/pi-coding-agent`)}`;
+        const action = theme.fg("accent", getUpdateInstruction("@mariozechner/pi-coding-agent"));
         const updateInstruction = theme.fg("muted", `New version ${newVersion} is available. `) + action;
         const changelogUrl = theme.fg("accent", "https://github.com/badlogic/pi-mono/blob/main/packages/coding-agent/CHANGELOG.md");
         const changelogLine = theme.fg("muted", "Changelog: ") + changelogUrl;