single-file-cli 2.0.34 → 2.0.35

package/build.sh

@@ -1,6 +1,6 @@
 #!/bin/sh
 
-deno vendor "npm:single-file-core@1.3.30"
+deno vendor "npm:single-file-core@1.4.0"
 
 echo "
 import { build } from 'npm:esbuild';

package/deno.json

@@ -1,6 +1,6 @@
 {
   "name": "@single-file/single-file-cli",
-  "version": "2.0.34",
+  "version": "2.0.35",
   "description": "SingleFile CLI",
   "nodeModulesDir": true,
   "exports": {

package/lib/cdp-client.js

@@ -32,6 +32,8 @@ const NETWORK_STATES = ["InteractiveTime", "networkIdle", "networkAlmostIdle", "
 const MINIMIZED_WINDOW_STATE = "minimized";
 const SINGLE_FILE_WORLD_NAME = "singlefile";
 const EMPTY_PAGE_URL = "about:blank";
+const CAPTURE_SCREENSHOT_FUNCTION_NAME = "captureScreenshot";
+const SET_SCREENSHOT_FUNCTION_NAME = "setScreenshot";
 
 export { initialize, getPageData, closeBrowser };
 
@@ -106,12 +108,15 @@ async function getPageData(options) {
 			setTimeout(() => resolve, options.browserWaitDelay);
 		}
 		if (options.embedScreenshot && options.compressContent) {
-			const { data } = await Page.captureScreenshot({ format: "png", captureBeyondViewport: true });
-			const dataURI = "data:image/png;base64," + data;
-			options.embeddedImage = Array.from(new Uint8Array(await (await fetch(dataURI)).arrayBuffer()));
+			await Runtime.addBinding({ name: CAPTURE_SCREENSHOT_FUNCTION_NAME, executionContextId: contextId });
+			Runtime.addEventListener("bindingCalled", ({ params }) => {
+				if (params.name === CAPTURE_SCREENSHOT_FUNCTION_NAME) {
+					onCaptureScreenshot({ Page, Runtime }, contextId);
+				}
+			});
 		}
 		const { result } = await Runtime.evaluate({
-			expression: `singlefile.getPageData(${JSON.stringify(options)})`,
+			expression: `(${getPageDataScriptSource.toString()})(${JSON.stringify(options)},${JSON.stringify(SET_SCREENSHOT_FUNCTION_NAME)})`,
 			awaitPromise: true,
 			returnByValue: true,
 			contextId
@@ -140,6 +145,31 @@ async function getPageData(options) {
 	}
 }
 
+async function onCaptureScreenshot({ Page, Runtime }, contextId) {
+	const { data } = await Page.captureScreenshot({ format: "png", captureBeyondViewport: true });
+	const dataURI = "data:image/png;base64," + data;
+	const screenshotData = Array.from(new Uint8Array(await (await fetch(dataURI)).arrayBuffer()));
+	await Runtime.evaluate({
+		expression: `globalThis.${SET_SCREENSHOT_FUNCTION_NAME}(${JSON.stringify(screenshotData)})`,
+		contextId
+	});
+}
+
+function getPageDataScriptSource(options, SET_SCREENSHOT_FUNCTION_NAME) {
+	if (options.embedScreenshot && options.compressContent) {
+		let screenshot = new Promise(resolve => globalThis[SET_SCREENSHOT_FUNCTION_NAME] = resolve);
+		let pendingCapture;
+		options.onprogress = async event => {
+			if (event.type == event.RESOURCES_INITIALIZING && globalThis.captureScreenshot && window == top && !pendingCapture) {
+				pendingCapture = true;
+				globalThis.captureScreenshot("");
+				options.embeddedImage = await screenshot;
+			}
+		};
+	}
+	return singlefile.getPageData(options);
+}
+
 async function loadPage({ Page, Runtime }, options) {
 	await Runtime.enable();
 	await Page.enable();