@mariozechner/pi-coding-agent 0.37.2 → 0.37.3

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,8 @@ 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;
     // Convenience accessors
     get agent() {
         return this.session.agent;
@@ -340,6 +342,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 +530,33 @@ 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();
+    }
     /**
      * Create the ExtensionUIContext for extensions.
      */
@@ -534,6 +568,7 @@ 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),
             setTitle: (title) => this.ui.terminal.setTitle(title),
             custom: (factory) => this.showExtensionCustom(factory),
             setEditorText: (text) => this.editor.setText(text),
@@ -763,18 +798,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();
@@ -1657,6 +1690,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 +1716,9 @@ export class InteractiveMode {
                 onAutoResizeImagesChange: (enabled) => {
                     this.settingsManager.setImageAutoResize(enabled);
                 },
+                onBlockImagesChange: (blocked) => {
+                    this.settingsManager.setBlockImages(blocked);
+                },
                 onSteeringModeChange: (mode) => {
                     this.session.setSteeringMode(mode);
                 },