@mariozechner/pi-coding-agent 0.37.2 → 0.37.4

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

@@ -6,7 +6,6 @@ import * as crypto from "node:crypto";
 import * as fs from "node:fs";
 import * as os from "node:os";
 import * as path from "node:path";
-import Clipboard from "@crosscopy/clipboard";
 import { getOAuthProviders } from "@mariozechner/pi-ai";
 import { CombinedAutocompleteProvider, Container, getEditorKeybindings, Loader, Markdown, matchesKey, ProcessTerminal, Spacer, Text, TruncatedText, TUI, visibleWidth, } from "@mariozechner/pi-tui";
 import { spawn, spawnSync } from "child_process";
@@ -18,6 +17,7 @@ import { loadSkills } from "../../core/skills.js";
 import { loadProjectContextFiles } from "../../core/system-prompt.js";
 import { getChangelogPath, parseChangelog } from "../../utils/changelog.js";
 import { copyToClipboard } from "../../utils/clipboard.js";
+import { extensionForImageMimeType, readClipboardImage } from "../../utils/clipboard-image.js";
 import { ArminComponent } from "./components/armin.js";
 import { AssistantMessageComponent } from "./components/assistant-message.js";
 import { BashExecutionComponent } from "./components/bash-execution.js";
@@ -97,6 +97,12 @@ export class InteractiveMode {
     // Extension widgets (components rendered above the editor)
     extensionWidgets = new Map();
     widgetContainer;
+    // Custom footer from extension (undefined = use built-in footer)
+    customFooter = undefined;
+    // Built-in header (logo + keybinding hints + changelog)
+    builtInHeader = undefined;
+    // Custom header from extension (undefined = use built-in header)
+    customHeader = undefined;
     // Convenience accessors
     get agent() {
         return this.session.agent;
@@ -197,7 +203,7 @@ export class InteractiveMode {
             theme.fg("muted", " to suspend") +
             "\n" +
             theme.fg("dim", deleteToLineEnd) +
-            theme.fg("muted", " to delete line") +
+            theme.fg("muted", " to delete to end") +
             "\n" +
             theme.fg("dim", cycleThinkingLevel) +
             theme.fg("muted", " to cycle thinking") +
@@ -223,6 +229,9 @@ export class InteractiveMode {
             theme.fg("dim", "!") +
             theme.fg("muted", " to run bash") +
             "\n" +
+            theme.fg("dim", "!!") +
+            theme.fg("muted", " to run bash (no context)") +
+            "\n" +
             theme.fg("dim", followUp) +
             theme.fg("muted", " to queue follow-up") +
             "\n" +
@@ -231,10 +240,10 @@ export class InteractiveMode {
             "\n" +
             theme.fg("dim", "drop files") +
             theme.fg("muted", " to attach");
-        const header = new Text(`${logo}\n${instructions}`, 1, 0);
+        this.builtInHeader = new Text(`${logo}\n${instructions}`, 1, 0);
         // Setup UI layout
         this.ui.addChild(new Spacer(1));
-        this.ui.addChild(header);
+        this.ui.addChild(this.builtInHeader);
         this.ui.addChild(new Spacer(1));
         // Add changelog if provided
         if (this.changelogMarkdown) {
@@ -340,6 +349,11 @@ export class InteractiveMode {
                     this.showError(`Extension sendMessage failed: ${err instanceof Error ? err.message : String(err)}`);
                 });
             },
+            sendUserMessageHandler: (content, options) => {
+                this.session.sendUserMessage(content, options).catch((err) => {
+                    this.showError(`Extension sendUserMessage failed: ${err instanceof Error ? err.message : String(err)}`);
+                });
+            },
             appendEntryHandler: (customType, data) => {
                 this.sessionManager.appendCustomEntry(customType, data);
             },
@@ -523,6 +537,64 @@ export class InteractiveMode {
         }
         this.ui.requestRender();
     }
+    /**
+     * Set a custom footer component, or restore the built-in footer.
+     */
+    setExtensionFooter(factory) {
+        // Dispose existing custom footer
+        if (this.customFooter?.dispose) {
+            this.customFooter.dispose();
+        }
+        // Remove current footer from UI
+        if (this.customFooter) {
+            this.ui.removeChild(this.customFooter);
+        }
+        else {
+            this.ui.removeChild(this.footer);
+        }
+        if (factory) {
+            // Create and add custom footer
+            this.customFooter = factory(this.ui, theme);
+            this.ui.addChild(this.customFooter);
+        }
+        else {
+            // Restore built-in footer
+            this.customFooter = undefined;
+            this.ui.addChild(this.footer);
+        }
+        this.ui.requestRender();
+    }
+    /**
+     * Set a custom header component, or restore the built-in header.
+     */
+    setExtensionHeader(factory) {
+        // Header may not be initialized yet if called during early initialization
+        if (!this.builtInHeader) {
+            return;
+        }
+        // Dispose existing custom header
+        if (this.customHeader?.dispose) {
+            this.customHeader.dispose();
+        }
+        // Remove current header from UI
+        if (this.customHeader) {
+            this.ui.removeChild(this.customHeader);
+        }
+        else {
+            this.ui.removeChild(this.builtInHeader);
+        }
+        if (factory) {
+            // Create and add custom header at position 1 (after initial spacer)
+            this.customHeader = factory(this.ui, theme);
+            this.ui.children.splice(1, 0, this.customHeader);
+        }
+        else {
+            // Restore built-in header at position 1
+            this.customHeader = undefined;
+            this.ui.children.splice(1, 0, this.builtInHeader);
+        }
+        this.ui.requestRender();
+    }
     /**
      * Create the ExtensionUIContext for extensions.
      */
@@ -534,6 +606,8 @@ export class InteractiveMode {
             notify: (message, type) => this.showExtensionNotify(message, type),
             setStatus: (key, text) => this.setExtensionStatus(key, text),
             setWidget: (key, content) => this.setExtensionWidget(key, content),
+            setFooter: (factory) => this.setExtensionFooter(factory),
+            setHeader: (factory) => this.setExtensionHeader(factory),
             setTitle: (title) => this.ui.terminal.setTitle(title),
             custom: (factory) => this.showExtensionCustom(factory),
             setEditorText: (text) => this.editor.setText(text),
@@ -763,18 +837,16 @@ export class InteractiveMode {
     }
     async handleClipboardImagePaste() {
         try {
-            if (!Clipboard.hasImage()) {
-                return;
-            }
-            const imageData = await Clipboard.getImageBinary();
-            if (!imageData || imageData.length === 0) {
+            const image = await readClipboardImage();
+            if (!image) {
                 return;
             }
             // Write to temp file
             const tmpDir = os.tmpdir();
-            const fileName = `pi-clipboard-${crypto.randomUUID()}.png`;
+            const ext = extensionForImageMimeType(image.mimeType) ?? "png";
+            const fileName = `pi-clipboard-${crypto.randomUUID()}.${ext}`;
             const filePath = path.join(tmpDir, fileName);
-            fs.writeFileSync(filePath, Buffer.from(imageData));
+            fs.writeFileSync(filePath, Buffer.from(image.bytes));
             // Insert file path directly
             this.editor.insertTextAtCursor(filePath);
             this.ui.requestRender();
@@ -955,7 +1027,6 @@ export class InteractiveMode {
                 }
                 else if (event.message.role === "user") {
                     this.addMessageToChat(event.message);
-                    this.editor.setText("");
                     this.updatePendingMessagesDisplay();
                     this.ui.requestRender();
                 }
@@ -1657,6 +1728,7 @@ export class InteractiveMode {
                 autoCompact: this.session.autoCompactionEnabled,
                 showImages: this.settingsManager.getShowImages(),
                 autoResizeImages: this.settingsManager.getImageAutoResize(),
+                blockImages: this.settingsManager.getBlockImages(),
                 steeringMode: this.session.steeringMode,
                 followUpMode: this.session.followUpMode,
                 thinkingLevel: this.session.thinkingLevel,
@@ -1682,6 +1754,9 @@ export class InteractiveMode {
                 onAutoResizeImagesChange: (enabled) => {
                     this.settingsManager.setImageAutoResize(enabled);
                 },
+                onBlockImagesChange: (blocked) => {
+                    this.settingsManager.setBlockImages(blocked);
+                },
                 onSteeringModeChange: (mode) => {
                     this.session.setSteeringMode(mode);
                 },
@@ -2254,6 +2329,7 @@ export class InteractiveMode {
 | \`Ctrl+V\` | Paste image from clipboard |
 | \`/\` | Slash commands |
 | \`!\` | Run bash command |
+| \`!!\` | Run bash command (excluded from context) |
 `;
         // Add extension-registered shortcuts
         const extensionRunner = this.session.extensionRunner;