@skrillex1224/playwright-toolkit 2.1.222 → 2.1.224

package/dist/index.js

@@ -334,18 +334,18 @@ var fallbackLog = {
   error: (...args) => console.error(...args),
   debug: (...args) => console.debug ? console.debug(...args) : console.log(...args)
 };
-var resolveLogMethod = (logger14, name) => {
-  if (logger14 && typeof logger14[name] === "function") {
-    return logger14[name].bind(logger14);
+var resolveLogMethod = (logger15, name) => {
+  if (logger15 && typeof logger15[name] === "function") {
+    return logger15[name].bind(logger15);
   }
-  if (name === "warning" && logger14 && typeof logger14.warn === "function") {
-    return logger14.warn.bind(logger14);
+  if (name === "warning" && logger15 && typeof logger15.warn === "function") {
+    return logger15.warn.bind(logger15);
   }
   return fallbackLog[name];
 };
 var defaultLogger = null;
-var setDefaultLogger = (logger14) => {
-  defaultLogger = logger14;
+var setDefaultLogger = (logger15) => {
+  defaultLogger = logger15;
 };
 var resolveLogger = (explicitLogger) => {
   if (explicitLogger && typeof explicitLogger.info === "function") {
@@ -372,8 +372,8 @@ var colorize = (text, color) => {
 var createBaseLogger = (prefix = "", explicitLogger) => {
   const name = prefix ? String(prefix) : "";
   const dispatch = (methodName, icon, message, color) => {
-    const logger14 = resolveLogger(explicitLogger);
-    const logFn = resolveLogMethod(logger14, methodName);
+    const logger15 = resolveLogger(explicitLogger);
+    const logFn = resolveLogMethod(logger15, methodName);
     const timestamp = colorize(`[${formatTimestamp()}]`, ANSI.gray);
     const line = formatLine(name, icon, message);
     const coloredLine = colorize(line, color);
@@ -4906,7 +4906,7 @@ var createTemplateLogger = (baseLogger = createBaseLogger()) => {
 };
 var getDefaultBaseLogger = () => createBaseLogger("");
 var Logger = {
-  setLogger: (logger14) => setDefaultLogger(logger14),
+  setLogger: (logger15) => setDefaultLogger(logger15),
   info: (message) => getDefaultBaseLogger().info(message),
   success: (message) => getDefaultBaseLogger().success(message),
   warning: (message) => getDefaultBaseLogger().warning(message),
@@ -4914,8 +4914,8 @@ var Logger = {
   error: (message) => getDefaultBaseLogger().error(message),
   debug: (message) => getDefaultBaseLogger().debug(message),
   start: (message) => getDefaultBaseLogger().start(message),
-  useTemplate: (logger14) => {
-    if (logger14) return createTemplateLogger(createBaseLogger("", logger14));
+  useTemplate: (logger15) => {
+    if (logger15) return createTemplateLogger(createBaseLogger("", logger15));
     return createTemplateLogger();
   }
 };
@@ -6244,8 +6244,147 @@ var watermarkifyScreenshotBuffer = async (buffer, meta, page = null) => {
   return await composeScreenshotBufferWithBrowser(page, buffer, overlaySvg, imageInfo);
 };
 
+// src/internals/screenshot-compression.js
+import { Jimp, JimpMime, ResizeStrategy } from "jimp";
+var logger13 = createInternalLogger("ScreenshotCompression");
+var DEFAULT_SCREENSHOT_MAX_BYTES = 5 * 1024 * 1024;
+var DEFAULT_SCREENSHOT_OUTPUT_TYPE = "jpeg";
+var DEFAULT_SCREENSHOT_QUALITY = 0.72;
+var DEFAULT_SCREENSHOT_MIN_QUALITY = 0.38;
+var DEFAULT_SCREENSHOT_MIN_SCALE = 0.25;
+var SUPPORTED_SCREENSHOT_OUTPUT_TYPES = /* @__PURE__ */ new Set(["jpeg"]);
+var toPositiveInteger = (value, fallback = 0) => {
+  const number = Math.floor(Number(value) || 0);
+  return number > 0 ? number : fallback;
+};
+var normalizeQuality2 = (value, fallback) => {
+  const number = Number(value);
+  if (!Number.isFinite(number) || number <= 0) return fallback;
+  const normalized = number > 1 ? number / 100 : number;
+  return Math.min(1, Math.max(0.01, normalized));
+};
+var normalizeScale = (value, fallback) => {
+  const number = Number(value);
+  if (!Number.isFinite(number) || number <= 0) return fallback;
+  return Math.min(1, Math.max(0.05, number));
+};
+var normalizeScreenshotOutputType = (value) => {
+  const raw = String(value || DEFAULT_SCREENSHOT_OUTPUT_TYPE).trim().toLowerCase();
+  const normalized = raw === "jpg" ? "jpeg" : raw;
+  return SUPPORTED_SCREENSHOT_OUTPUT_TYPES.has(normalized) ? normalized : DEFAULT_SCREENSHOT_OUTPUT_TYPE;
+};
+var getBase64BytesFromBuffer = (buffer) => Math.ceil(buffer.length / 3) * 4;
+var toJpegQuality = (value) => Math.round(normalizeQuality2(value, DEFAULT_SCREENSHOT_QUALITY) * 100);
+var resolveCaptureScreenCompression = (options = {}) => {
+  const explicit = options.compression;
+  const source = explicit && typeof explicit === "object" && !Array.isArray(explicit) ? explicit : {};
+  const enabled = explicit !== false && source.enabled !== false && options.compress !== false;
+  const quality = normalizeQuality2(
+    source.quality ?? options.quality,
+    DEFAULT_SCREENSHOT_QUALITY
+  );
+  const minQuality = Math.min(
+    quality,
+    normalizeQuality2(source.minQuality ?? options.minQuality, DEFAULT_SCREENSHOT_MIN_QUALITY)
+  );
+  return {
+    enabled,
+    maxBytes: toPositiveInteger(
+      source.maxBytes ?? source.maxBase64Bytes ?? options.maxBytes ?? options.maxBase64Bytes,
+      DEFAULT_SCREENSHOT_MAX_BYTES
+    ),
+    outputType: normalizeScreenshotOutputType(
+      source.type ?? source.outputType ?? options.type ?? options.outputType
+    ),
+    quality,
+    minQuality,
+    minScale: normalizeScale(
+      source.minScale ?? options.minScale,
+      DEFAULT_SCREENSHOT_MIN_SCALE
+    )
+  };
+};
+var encodeJpeg = async (sourceImage, compression, scale, quality) => {
+  const width = Math.max(1, Math.round(sourceImage.bitmap.width * scale));
+  const height = Math.max(1, Math.round(sourceImage.bitmap.height * scale));
+  const image = sourceImage.clone();
+  if (scale < 0.999) {
+    image.resize({
+      w: width,
+      h: height,
+      mode: ResizeStrategy.BILINEAR
+    });
+  }
+  const buffer = await image.getBuffer(JimpMime.jpeg, { quality });
+  return {
+    buffer,
+    bytes: getBase64BytesFromBuffer(buffer),
+    width,
+    height,
+    quality,
+    scale: Number(scale.toFixed(3)),
+    format: compression.outputType
+  };
+};
+var compressScreenshotBuffer = async (buffer, compression) => {
+  const sourceImage = await Jimp.read(buffer);
+  const maxQuality = toJpegQuality(compression.quality);
+  const minQuality = Math.min(maxQuality, toJpegQuality(compression.minQuality));
+  let quality = maxQuality;
+  let scale = 1;
+  let smallest = null;
+  for (let attempt = 0; attempt < 12; attempt += 1) {
+    const candidate = await encodeJpeg(sourceImage, compression, scale, quality);
+    if (!smallest || candidate.bytes < smallest.bytes) {
+      smallest = candidate;
+    }
+    if (candidate.bytes <= compression.maxBytes) {
+      return { ...candidate, withinLimit: true };
+    }
+    if (quality > minQuality) {
+      quality = Math.max(minQuality, Math.floor(quality * 0.75));
+      continue;
+    }
+    const ratio = Math.sqrt(compression.maxBytes / Math.max(1, candidate.bytes));
+    const nextScale = Math.max(
+      compression.minScale,
+      Math.min(scale * 0.85, scale * ratio * 0.94)
+    );
+    if (nextScale >= scale * 0.99 || scale <= compression.minScale) {
+      break;
+    }
+    scale = nextScale;
+  }
+  const finalCandidate = await encodeJpeg(sourceImage, compression, compression.minScale, minQuality);
+  const fallback = !smallest || finalCandidate.bytes < smallest.bytes ? finalCandidate : smallest;
+  return { ...fallback, withinLimit: fallback.bytes <= compression.maxBytes };
+};
+var compressScreenshotBufferToBase64 = async (buffer, compression) => {
+  const originalBytes = getBase64BytesFromBuffer(buffer);
+  if (!compression.enabled || originalBytes <= compression.maxBytes) {
+    return buffer.toString("base64");
+  }
+  const result = await compressScreenshotBuffer(buffer, compression).catch((error) => {
+    logger13.warning(`captureScreen \u538B\u7F29\u5931\u8D25\uFF0C\u8FD4\u56DE\u539F\u56FE: ${error instanceof Error ? error.message : String(error)}`);
+    return null;
+  });
+  if (!result?.buffer) {
+    return buffer.toString("base64");
+  }
+  if (result.withinLimit) {
+    logger13.info(
+      `captureScreen \u5DF2\u538B\u7F29: ${originalBytes} -> ${result.bytes} bytes, format=${result.format}, quality=${result.quality}, scale=${result.scale}, size=${result.width}x${result.height}`
+    );
+  } else {
+    logger13.warning(
+      `captureScreen \u538B\u7F29\u540E\u4ECD\u8D85\u8FC7\u76EE\u6807: ${originalBytes} -> ${result.bytes} bytes, maxBytes=${compression.maxBytes}, format=${result.format}, quality=${result.quality}, scale=${result.scale}`
+    );
+  }
+  return result.buffer.toString("base64");
+};
+
 // src/share.js
-var logger13 = createInternalLogger("Share");
+var logger14 = createInternalLogger("Share");
 var DEFAULT_TIMEOUT_MS2 = 50 * 1e3;
 var DEFAULT_PAYLOAD_SNAPSHOT_MAX_LEN = 500;
 var DEFAULT_POLL_INTERVAL_MS = 120;
@@ -6382,7 +6521,7 @@ var createDomShareMonitor = async (page, options = {}) => {
   const onMatch = typeof options.onMatch === "function" ? options.onMatch : null;
   const onTelemetry = typeof options.onTelemetry === "function" ? options.onTelemetry : null;
   let matched = false;
-  logger13.info(`DOM \u76D1\u542C\u51C6\u5907\u6302\u8F7D: selectors=${toJsonInline(selectors, 120)}, mode=${mode}`);
+  logger14.info(`DOM \u76D1\u542C\u51C6\u5907\u6302\u8F7D: selectors=${toJsonInline(selectors, 120)}, mode=${mode}`);
   const monitor = await Mutation.useMonitor(page, selectors, {
     mode,
     onMutation: (context = {}) => {
@@ -6400,12 +6539,12 @@ ${text}`;
         });
       }
       if (mutationCount <= 5 || mutationCount % 50 === 0) {
-        logger13.info(`DOM \u53D8\u5316\u5DF2\u6355\u83B7: mutationCount=${mutationCount}, mutationNodes=${mutationNodes.length}`);
+        logger14.info(`DOM \u53D8\u5316\u5DF2\u6355\u83B7: mutationCount=${mutationCount}, mutationNodes=${mutationNodes.length}`);
       }
       const [candidate] = Utils.parseLinks(rawDom, { prefix }) || [];
       if (!candidate) return;
       matched = true;
-      logger13.success("captureLink.domHit", `DOM \u547D\u4E2D\u5206\u4EAB\u94FE\u63A5: mutationCount=${mutationCount}, link=${candidate}`);
+      logger14.success("captureLink.domHit", `DOM \u547D\u4E2D\u5206\u4EAB\u94FE\u63A5: mutationCount=${mutationCount}, link=${candidate}`);
       if (onMatch) {
         onMatch({
           link: candidate,
@@ -6421,7 +6560,7 @@ ${text}`;
   return {
     stop: async () => {
       const result = await monitor.stop();
-      logger13.info(`DOM \u76D1\u542C\u5DF2\u505C\u6B62: totalMutations=${result?.totalMutations || 0}`);
+      logger14.info(`DOM \u76D1\u542C\u5DF2\u505C\u6B62: totalMutations=${result?.totalMutations || 0}`);
       return result;
     }
   };
@@ -6461,8 +6600,8 @@ var Share = {
     if (share.mode === "response" && apiMatchers.length === 0) {
       throw new Error("Share.captureLink requires share.xurl[0] api matcher when mode=response");
     }
-    logger13.start("captureLink", `mode=${share.mode}, timeoutMs=${timeoutMs}, prefix=${share.prefix}`);
-    logger13.info(`captureLink \u914D\u7F6E: xurl=${toJsonInline(share.xurl)}, domMode=${domMode}, domSelectors=${toJsonInline(domSelectors, 120)}`);
+    logger14.start("captureLink", `mode=${share.mode}, timeoutMs=${timeoutMs}, prefix=${share.prefix}`);
+    logger14.info(`captureLink \u914D\u7F6E: xurl=${toJsonInline(share.xurl)}, domMode=${domMode}, domSelectors=${toJsonInline(domSelectors, 120)}`);
     const stats = {
       actionTimedOut: false,
       domMutationCount: 0,
@@ -6487,7 +6626,7 @@ var Share = {
         link: validated,
         payloadText: String(payloadText || "")
       };
-      logger13.info(`\u5019\u9009\u94FE\u63A5\u5DF2\u786E\u8BA4: source=${source}, link=${validated}`);
+      logger14.info(`\u5019\u9009\u94FE\u63A5\u5DF2\u786E\u8BA4: source=${source}, link=${validated}`);
       return true;
     };
     const resolveResponseCandidate = (responseText) => {
@@ -6522,7 +6661,7 @@ var Share = {
       try {
         await monitor.stop();
       } catch (error) {
-        logger13.warning(`\u505C\u6B62 DOM \u76D1\u542C\u5931\u8D25: ${error instanceof Error ? error.message : String(error)}`);
+        logger14.warning(`\u505C\u6B62 DOM \u76D1\u542C\u5931\u8D25: ${error instanceof Error ? error.message : String(error)}`);
       }
     };
     const onResponse = async (response) => {
@@ -6535,29 +6674,29 @@ var Share = {
           stats.responseSampleUrls.push(url);
         }
         if (stats.responseObserved <= 5) {
-          logger13.info(`\u63A5\u53E3\u54CD\u5E94\u91C7\u6837(${stats.responseObserved}): ${url}`);
+          logger14.info(`\u63A5\u53E3\u54CD\u5E94\u91C7\u6837(${stats.responseObserved}): ${url}`);
         }
         if (!apiMatchers.some((matcher) => url.includes(matcher))) return;
         stats.responseMatched += 1;
         stats.lastMatchedUrl = url;
-        logger13.info(`\u63A5\u53E3\u547D\u4E2D\u5339\u914D(${stats.responseMatched}): ${url}`);
+        logger14.info(`\u63A5\u53E3\u547D\u4E2D\u5339\u914D(${stats.responseMatched}): ${url}`);
         const text = await response.text();
         const hit = resolveResponseCandidate(text);
         if (!hit?.link) {
           if (stats.responseMatched <= 3) {
-            logger13.info(`\u63A5\u53E3\u89E3\u6790\u5B8C\u6210\u4F46\u672A\u63D0\u53D6\u5230\u5206\u4EAB\u94FE\u63A5: payloadSize=${text.length}`);
+            logger14.info(`\u63A5\u53E3\u89E3\u6790\u5B8C\u6210\u4F46\u672A\u63D0\u53D6\u5230\u5206\u4EAB\u94FE\u63A5: payloadSize=${text.length}`);
           }
           return;
         }
         stats.responseResolved += 1;
-        logger13.success("captureLink.responseHit", `\u63A5\u53E3\u547D\u4E2D\u5206\u4EAB\u94FE\u63A5: url=${url}, link=${hit.link}`);
+        logger14.success("captureLink.responseHit", `\u63A5\u53E3\u547D\u4E2D\u5206\u4EAB\u94FE\u63A5: url=${url}, link=${hit.link}`);
         setCandidate("response", hit.link, hit.payloadText);
       } catch (error) {
-        logger13.warning(`\u63A5\u53E3\u54CD\u5E94\u5904\u7406\u5F02\u5E38: ${error instanceof Error ? error.message : String(error)}`);
+        logger14.warning(`\u63A5\u53E3\u54CD\u5E94\u5904\u7406\u5F02\u5E38: ${error instanceof Error ? error.message : String(error)}`);
       }
     };
     if (share.mode === "dom") {
-      logger13.info("\u5F53\u524D\u4E3A DOM \u6A21\u5F0F\uFF0C\u4EC5\u542F\u7528 DOM \u76D1\u542C");
+      logger14.info("\u5F53\u524D\u4E3A DOM \u6A21\u5F0F\uFF0C\u4EC5\u542F\u7528 DOM \u76D1\u542C");
       domMonitor = await createDomShareMonitor(page, {
         prefix: share.prefix,
         selectors: domSelectors,
@@ -6572,14 +6711,14 @@ var Share = {
       });
     }
     if (share.mode === "response") {
-      logger13.info(`\u5F53\u524D\u4E3A\u63A5\u53E3\u6A21\u5F0F\uFF0C\u6302\u8F7D response \u76D1\u542C: apiMatchers=${toJsonInline(apiMatchers, 160)}`);
+      logger14.info(`\u5F53\u524D\u4E3A\u63A5\u53E3\u6A21\u5F0F\uFF0C\u6302\u8F7D response \u76D1\u542C: apiMatchers=${toJsonInline(apiMatchers, 160)}`);
       page.on("response", onResponse);
     }
     const deadline = Date.now() + timeoutMs;
     const getRemainingMs = () => Math.max(0, deadline - Date.now());
     try {
       const actionTimeout = getRemainingMs();
-      logger13.start("captureLink.performActions", `\u6267\u884C\u52A8\u4F5C\u9884\u7B97=${actionTimeout}ms`);
+      logger14.start("captureLink.performActions", `\u6267\u884C\u52A8\u4F5C\u9884\u7B97=${actionTimeout}ms`);
       if (actionTimeout > 0) {
         let timer = null;
         let actionError = null;
@@ -6593,21 +6732,21 @@ var Share = {
         const actionResult = await Promise.race([actionPromise, timeoutPromise]);
         if (timer) clearTimeout(timer);
         if (actionResult === "__ACTION_ERROR__") {
-          logger13.fail("captureLink.performActions", actionError);
+          logger14.fail("captureLink.performActions", actionError);
           throw actionError;
         }
         if (actionResult === "__ACTION_TIMEOUT__") {
           stats.actionTimedOut = true;
-          logger13.warning(`performActions \u5DF2\u8D85\u65F6 (${actionTimeout}ms)\uFF0C\u52A8\u4F5C\u53EF\u80FD\u4ECD\u5728\u5F02\u6B65\u6267\u884C`);
+          logger14.warning(`performActions \u5DF2\u8D85\u65F6 (${actionTimeout}ms)\uFF0C\u52A8\u4F5C\u53EF\u80FD\u4ECD\u5728\u5F02\u6B65\u6267\u884C`);
         } else {
-          logger13.success("captureLink.performActions", "\u6267\u884C\u52A8\u4F5C\u5B8C\u6210");
+          logger14.success("captureLink.performActions", "\u6267\u884C\u52A8\u4F5C\u5B8C\u6210");
         }
       }
       let nextProgressLogTs = Date.now() + 3e3;
       while (true) {
         const selected = share.mode === "dom" ? candidates.dom : candidates.response;
         if (selected?.link) {
-          logger13.success("captureLink", `\u6355\u83B7\u6210\u529F: source=${share.mode}, link=${selected.link}`);
+          logger14.success("captureLink", `\u6355\u83B7\u6210\u529F: source=${share.mode}, link=${selected.link}`);
           return {
             link: selected.link,
             payloadText: selected.payloadText,
@@ -6619,7 +6758,7 @@ var Share = {
         if (remaining <= 0) break;
         const now = Date.now();
         if (now >= nextProgressLogTs) {
-          logger13.info(
+          logger14.info(
             `captureLink \u7B49\u5F85\u4E2D: remaining=${remaining}ms, domMutationCount=${stats.domMutationCount}, responseMatched=${stats.responseMatched}`
           );
           nextProgressLogTs = now + 5e3;
@@ -6627,11 +6766,11 @@ var Share = {
         await delay2(Math.max(0, Math.min(DEFAULT_POLL_INTERVAL_MS, remaining)));
       }
       if (share.mode === "response" && stats.responseMatched === 0) {
-        logger13.warning(
+        logger14.warning(
           `\u63A5\u53E3\u76D1\u542C\u672A\u547D\u4E2D: apiMatchers=${toJsonInline(apiMatchers, 220)}, \u54CD\u5E94\u6837\u672CURLs=${toJsonInline(stats.responseSampleUrls, 420)}`
         );
       }
-      logger13.warning(
+      logger14.warning(
         `captureLink \u8D85\u65F6\u672A\u62FF\u5230\u94FE\u63A5: mode=${share.mode}, actionTimedOut=${stats.actionTimedOut}, domMutationCount=${stats.domMutationCount}, responseObserved=${stats.responseObserved}, responseMatched=${stats.responseMatched}, lastMatchedUrl=${stats.lastMatchedUrl || "none"}`
       );
       return {
@@ -6643,7 +6782,7 @@ var Share = {
     } finally {
       if (share.mode === "response") {
         page.off("response", onResponse);
-        logger13.info("response \u76D1\u542C\u5DF2\u5378\u8F7D");
+        logger14.info("response \u76D1\u542C\u5DF2\u5378\u8F7D");
       }
       await stopDomMonitor();
     }
@@ -6656,7 +6795,10 @@ var Share = {
    * @param {number} [options.buffer]
    * @param {boolean} [options.restore]
    * @param {number} [options.maxHeight]
-   * @returns {Promise<string>} base64 png
+   * @param {number} [options.maxBytes] 默认 5MiB，返回 base64 超过后会压缩
+   * @param {'jpeg'|'jpg'} [options.type] 压缩输出格式，默认 jpeg
+   * @param {boolean|Object} [options.compression] 传 false 可关闭压缩
+   * @returns {Promise<string>} base64 image
    */
   async captureScreen(page, options = {}) {
     const originalViewport = await resolveCurrentViewportSize(page);
@@ -6665,6 +6807,7 @@ var Share = {
     const restore = options.restore ?? false;
     const maxHeight = options.maxHeight ?? 8e3;
     const screenshotWatermarkify = resolveCaptureScreenWatermarkify(page, options.watermarkify);
+    const compression = resolveCaptureScreenCompression(options);
     try {
       const maxScrollHeight = await page.evaluate(() => {
         let maxHeight2 = document.body.scrollHeight;
@@ -6698,15 +6841,15 @@ var Share = {
         type: "png",
         maxClipHeight: targetHeight
       });
-      if (!screenshotWatermarkify.enabled) {
-        return rawBuffer.toString("base64");
+      let outputBuffer = rawBuffer;
+      if (screenshotWatermarkify.enabled) {
+        const watermarkifyMeta = await resolveScreenshotWatermarkifyMeta(page, {
+          ...screenshotWatermarkify,
+          capturedAt
+        });
+        outputBuffer = await watermarkifyScreenshotBuffer(rawBuffer, watermarkifyMeta, page);
       }
-      const watermarkifyMeta = await resolveScreenshotWatermarkifyMeta(page, {
-        ...screenshotWatermarkify,
-        capturedAt
-      });
-      const buffer_ = await watermarkifyScreenshotBuffer(rawBuffer, watermarkifyMeta, page);
-      return buffer_.toString("base64");
+      return await compressScreenshotBufferToBase64(outputBuffer, compression);
     } finally {
       if (restore) {
         await page.evaluate(() => {