@testdriverai/runner 7.9.103-test → 7.9.104-test

package/lib/automation.js

@@ -52,6 +52,26 @@ const PY_IMPORT = IS_LINUX
   ? "import os; os.environ.setdefault('DISPLAY', ':0'); import pyautogui, sys; pyautogui.FAILSAFE = False; "
   : 'import pyautogui, sys; pyautogui.FAILSAFE = False; ';
 
+/**
+ * Read width/height from a PNG buffer's IHDR chunk without decoding the image.
+ * PNG layout: 8-byte signature, then the IHDR chunk whose 4-byte width and
+ * height live at byte offsets 16 and 20 (big-endian). Returns {} if the buffer
+ * doesn't look like a PNG so callers degrade to the resize fallback.
+ * @param {Buffer} buffer
+ * @returns {{ width?: number, height?: number }}
+ */
+function readPngDimensions(buffer) {
+  const PNG_SIGNATURE = '89504e470d0a1a0a';
+  if (!buffer || buffer.length < 24 ||
+      buffer.slice(0, 8).toString('hex') !== PNG_SIGNATURE) {
+    return {};
+  }
+  return {
+    width: buffer.readUInt32BE(16),
+    height: buffer.readUInt32BE(20),
+  };
+}
+
 /**
  * Run a pyautogui Python script via subprocess.
  * @param {string} script — Python code (pyautogui + sys are already imported via PY_IMPORT prefix)
@@ -600,12 +620,17 @@ class Automation extends EventEmitter {
         const buffer = await this._captureScreenshotBuffer();
         console.log(`[automation] Step 2: Screenshot captured, size: ${buffer.length} bytes`);
 
+        // Read the PNG dimensions from the IHDR header (bytes 16–24) without
+        // decoding. The SDK uses these to decide whether it can pass the s3Key
+        // straight to the API (no resize needed) or must download + resize.
+        const dimensions = readPngDimensions(buffer);
+
         // Upload screenshot to S3 and return key
         console.log('[automation] Step 3: Uploading to S3...');
         const s3Key = await this._uploadToS3(buffer, this._sandboxId, 'image/png');
         console.log(`[automation] Step 4: Upload complete, s3Key: ${s3Key}`);
 
-        return { s3Key };
+        return { s3Key, ...dimensions };
       }
 
       case 'ping':

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@testdriverai/runner",
-  "version": "7.9.103-test",
+  "version": "7.9.104-test",
   "description": "TestDriver Runner - Ably-based remote automation agent with Node.js automation",
   "main": "index.js",
   "bin": {