@opengsd/gsd-pi 1.8.1 → 1.9.0

package/dist/resources/extensions/mac-tools/index.js

@@ -15,6 +15,7 @@ import { StringEnum, Type } from "@gsd/pi-ai";
 import { execFileSync } from "node:child_process";
 import { statSync, readdirSync } from "node:fs";
 import path from "node:path";
+import { constrainScreenshot, getScreenshotBufferDimensions, } from "../browser-tools/screenshot-constraints.js";
 // ---------------------------------------------------------------------------
 // Paths
 // ---------------------------------------------------------------------------
@@ -79,6 +80,27 @@ function ensureCompiled() {
         throw new Error(`Swift compilation failed (${action.toLowerCase()}):\n${stderr || stdout || err.message}`);
     }
 }
+function macScreenshotQualityToPercent(quality) {
+    if (quality === undefined || !Number.isFinite(quality))
+        return 80;
+    return Math.min(100, Math.max(1, Math.round(quality * 100)));
+}
+export async function prepareMacScreenshotPayload(data, quality) {
+    const format = data.format;
+    const mimeType = format === "png" ? "image/png" : "image/jpeg";
+    const rawBuffer = Buffer.from(data.imageData, "base64");
+    const constrainedBuffer = await constrainScreenshot(null, rawBuffer, mimeType, macScreenshotQualityToPercent(quality));
+    const dimensions = constrainedBuffer === rawBuffer
+        ? null
+        : await getScreenshotBufferDimensions(constrainedBuffer);
+    return {
+        imageData: constrainedBuffer.toString("base64"),
+        width: dimensions?.width ?? data.width,
+        height: dimensions?.height ?? data.height,
+        format,
+        mimeType,
+    };
+}
 /**
  * Invoke the mac-agent CLI with a command and optional params.
  * Handles compilation caching, stdin/stdout JSON protocol, and error surfacing.
@@ -646,18 +668,18 @@ export default function (pi) {
             if (!result.success) {
                 throw new Error("mac_screenshot: " + result.error);
             }
-            const data = result.data;
-            const imageData = data.imageData;
-            const format = data.format;
-            const width = data.width;
-            const height = data.height;
-            const mimeType = format === "png" ? "image/png" : "image/jpeg";
+            const screenshot = await prepareMacScreenshotPayload(result.data, args.quality);
             return {
                 content: [
-                    { type: "text", text: `Screenshot: ${width}x${height} ${format}` },
-                    { type: "image", data: imageData, mimeType },
+                    { type: "text", text: `Screenshot: ${screenshot.width}x${screenshot.height} ${screenshot.format}` },
+                    { type: "image", data: screenshot.imageData, mimeType: screenshot.mimeType },
                 ],
-                details: { width, height, format, mimeType },
+                details: {
+                    width: screenshot.width,
+                    height: screenshot.height,
+                    format: screenshot.format,
+                    mimeType: screenshot.mimeType,
+                },
             };
         },
     });