@skrillex1224/playwright-toolkit 2.1.275 → 2.1.276

package/dist/index.cjs

@@ -8169,7 +8169,10 @@ var buildWatermarkifyRenderHtml = ({ imageSrc, overlaySvg, width, height, imageH
         </html>
     `;
 };
-var composeScreenshotBufferWithBrowser = async (page, buffer, overlaySvg, imageInfo = {}) => {
+var normalizeWatermarkifyRenderMode = (value) => {
+  return String(value || "default").trim().toLowerCase() === "cloakbrowser" ? "cloakbrowser" : "default";
+};
+var composeScreenshotBufferWithBrowser = async (page, buffer, overlaySvg, imageInfo = {}, options = {}) => {
   if (!page || typeof page.context !== "function") {
     logger14.warning("watermarkify \u6D4F\u89C8\u5668\u5408\u6210\u8DF3\u8FC7: \u7F3A\u5C11\u53EF\u7528 page");
     return buffer;
@@ -8193,15 +8196,35 @@ var composeScreenshotBufferWithBrowser = async (page, buffer, overlaySvg, imageI
       height: viewportHeight
     }).catch(() => {
     });
-    await renderPage.setContent(buildWatermarkifyRenderHtml({
-      imageSrc: `data:${imageInfo.mimeType || "image/png"};base64,${buffer.toString("base64")}`,
-      overlaySvg,
-      width: safeWidth,
-      height: safeHeight,
-      imageHeight: safeImageHeight
-    }), {
-      waitUntil: "load"
-    });
+    const renderMode = normalizeWatermarkifyRenderMode(options.mode);
+    if (renderMode === "cloakbrowser") {
+      const renderHtml = buildWatermarkifyRenderHtml({
+        imageSrc: `data:${imageInfo.mimeType || "image/png"};base64,${buffer.toString("base64")}`,
+        overlaySvg,
+        width: safeWidth,
+        height: safeHeight,
+        imageHeight: safeImageHeight
+      });
+      await renderPage.goto("about:blank", {
+        waitUntil: "commit"
+      }).catch(() => {
+      });
+      await renderPage.evaluate((html) => {
+        document.open();
+        document.write(html);
+        document.close();
+      }, renderHtml);
+    } else {
+      await renderPage.setContent(buildWatermarkifyRenderHtml({
+        imageSrc: `data:${imageInfo.mimeType || "image/png"};base64,${buffer.toString("base64")}`,
+        overlaySvg,
+        width: safeWidth,
+        height: safeHeight,
+        imageHeight: safeImageHeight
+      }), {
+        waitUntil: "load"
+      });
+    }
     await renderPage.waitForFunction(() => {
       const image = document.getElementById("pk-base-image");
       return image instanceof HTMLImageElement && image.complete && image.naturalWidth > 0 && image.naturalHeight > 0;
@@ -9136,7 +9159,7 @@ var buildWatermarkifySvg = (meta, imageWidth, imageHeight) => {
         </svg>
     `;
 };
-var watermarkifyScreenshotBuffer = async (buffer, meta, page = null) => {
+var watermarkifyScreenshotBuffer = async (buffer, meta, page = null, options = {}) => {
   const hasWatermark = meta?.watermark?.enabled !== false && normalizeText(meta?.watermarkText);
   const hasStrip = meta?.strip?.enabled !== false && Array.isArray(meta?.stripSegments) && meta.stripSegments.length > 0;
   if (!Buffer.isBuffer(buffer) || !meta || !hasWatermark && !hasStrip) {
@@ -9157,7 +9180,7 @@ var watermarkifyScreenshotBuffer = async (buffer, meta, page = null) => {
   if (!overlaySvg) {
     return buffer;
   }
-  return await composeScreenshotBufferWithBrowser(page, buffer, overlaySvg, outputImageInfo);
+  return await composeScreenshotBufferWithBrowser(page, buffer, overlaySvg, outputImageInfo, options);
 };
 
 // src/internals/compression.js
@@ -9755,6 +9778,7 @@ var Share = {
    * @param {number} [options.maxBytes] 默认 5MiB，返回 base64 超过后会压缩
    * @param {'jpeg'|'jpg'} [options.type] 压缩输出格式，默认 jpeg
    * @param {boolean|Object} [options.compression] 传 false 可关闭压缩
+   * @param {'default'|'cloakbrowser'} [options.mode] 截图水印合成模式，默认 default
    * @returns {Promise<string>} base64 image
    */
   async captureScreen(page, options = {}) {
@@ -9775,7 +9799,9 @@ var Share = {
         ...screenshotWatermarkify,
         capturedAt
       });
-      outputBuffer = await watermarkifyScreenshotBuffer(rawBuffer, watermarkifyMeta, page);
+      outputBuffer = await watermarkifyScreenshotBuffer(rawBuffer, watermarkifyMeta, page, {
+        mode: options.mode === "cloakbrowser" ? "cloakbrowser" : "default"
+      });
     }
     return await compressImageBufferToBase64(outputBuffer, compression);
   }