@pi-archimedes/image-paste 2.3.0 → 2.5.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-archimedes/image-paste",
-  "version": "2.3.0",
+  "version": "2.5.0",
   "type": "module",
   "keywords": [
     "pi-package"
@@ -15,8 +15,8 @@
     "@earendil-works/pi-tui": ">=0.1.0"
   },
   "devDependencies": {
-    "@earendil-works/pi-coding-agent": "^0.84.2",
-    "@earendil-works/pi-tui": "^0.84.2",
+    "@earendil-works/pi-coding-agent": "^0.84.4",
+    "@earendil-works/pi-tui": "^0.84.4",
     "typescript": "^6.0.0"
   },
   "pi": {

package/src/default-export.test.ts

@@ -0,0 +1,41 @@
+import { describe, expect, it, vi } from "vitest";
+import defaultExport from "./index.js";
+
+describe("image-paste default export (standalone factory)", () => {
+	it("is a function that registers paste + per-session lifecycle on the pi extension", () => {
+		expect(typeof defaultExport).toBe("function");
+
+		const on = vi.fn();
+		const registerShortcut = vi.fn();
+		const registerMessageRenderer = vi.fn();
+
+		const pi = { on, registerShortcut, registerMessageRenderer } as any;
+
+		defaultExport(pi);
+
+		// registerImagePaste wires the preview renderer + shortcuts
+		expect(registerMessageRenderer).toHaveBeenCalled();
+		expect(registerShortcut).toHaveBeenCalled();
+
+		// input handler (from registerImagePaste) + session lifecycle handlers
+		const events = on.mock.calls.map((c: Array<unknown>) => c[0]);
+		expect(events).toEqual(
+			expect.arrayContaining(["input", "session_start", "session_shutdown"]),
+		);
+	});
+
+	it("resets the per-session queue on session_start and clears it on shutdown", () => {
+		const on = vi.fn();
+		const pi = { on, registerShortcut: vi.fn(), registerMessageRenderer: vi.fn() } as any;
+		defaultExport(pi);
+
+		const startHandler = on.mock.calls.find((c: Array<unknown>) => c[0] === "session_start")?.[1];
+		const shutdownHandler = on.mock.calls.find((c: Array<unknown>) => c[0] === "session_shutdown")?.[1];
+		expect(typeof startHandler).toBe("function");
+		expect(typeof shutdownHandler).toBe("function");
+
+		// Must not throw when wired to a session context
+		expect(() => (startHandler as (...a: unknown[]) => unknown)({}, { hasUI: true })).not.toThrow();
+		expect(() => (shutdownHandler as (...a: unknown[]) => unknown)({}, {})).not.toThrow();
+	});
+});

package/src/index.ts

@@ -188,3 +188,19 @@ export function shutdownImagePaste(): void {
   _queue = null;
   _pasting = false;
 }
+
+// ── Default export (for standalone pi.extensions loading) ─────────────────
+
+// Pi's standalone loader requires a default export functioning as the
+// extension factory. Meta imports the named exports directly (register
+// + session lifecycle), so this default export is standalone-only.
+export default function (pi: ExtensionAPI): void {
+  registerImagePaste(pi);
+  // Session lifecycle: pi requires per-session ctx for image pasting.
+  pi.on("session_start", (_event, ctx) => {
+    initImagePasteSession(ctx);
+  });
+  pi.on("session_shutdown", (_event, _ctx) => {
+    shutdownImagePaste();
+  });
+}

package/src/preview.ts

@@ -12,10 +12,6 @@ interface PreviewDetails {
 export function registerImagePreview(pi: ExtensionAPI): void {
   pi.registerMessageRenderer<PreviewDetails>(CUSTOM_TYPE, (message, _options, theme) => {
     try {
-      // Theme.fg is runtime-available but not exposed in the Theme type definition
-      const fg = (theme as any).fg?.bind(theme) as ((color: string, text: string) => string) | undefined;
-      if (!fg) return undefined;
-
       // Extract image data from content array
       const content = message.content;
       if (!Array.isArray(content) || content.length === 0) return undefined;
@@ -28,7 +24,7 @@ export function registerImagePreview(pi: ExtensionAPI): void {
           container.addChild(new Spacer(1));
           container.addChild(
             new Image(item.data, item.mimeType, {
-              fallbackColor: (text: string) => fg("toolOutput", text),
+              fallbackColor: (text: string) => theme.fg("toolOutput", text),
             }, {
               maxWidthCells: 60,
             }),