@skrillex1224/playwright-toolkit 2.1.151 → 2.1.152

package/README.md

@@ -209,6 +209,12 @@ await page.context().addCookies(cookies);
 
 // 全页面滚动截图 (自动检测所有滚动元素，强制展开后截图)
 const base64Image = await Share.captureScreen(page);
+
+// 移动端宽度截图（moblie 拼写保持兼容）
+const mobileImage = await Share.captureScreen(page, { moblie: true });
+
+// 仅恢复宽度（restore 支持 true / false / 'width-only' / 'height-only'）
+const image2 = await Share.captureScreen(page, { restore: 'width-only' });
 // 返回 base64 编码的 PNG 图片
 ```
 

package/dist/index.cjs

@@ -3059,6 +3059,8 @@ var logger11 = createInternalLogger("Share");
 var DEFAULT_TIMEOUT_MS = 50 * 1e3;
 var DEFAULT_PAYLOAD_SNAPSHOT_MAX_LEN = 500;
 var DEFAULT_POLL_INTERVAL_MS = 120;
+var DEFAULT_MOBILE_SCREENSHOT_WIDTH = 414;
+var MOBILE_VIEWPORT_MAX_WIDTH = 480;
 var normalizePrefix = (value) => value == null ? void 0 : String(value).trim();
 var toSnapshot = (value, maxLen) => {
   const text = String(value || "");
@@ -3447,16 +3449,25 @@ var Share = {
    * @param {import('playwright').Page} page
    * @param {Object} [options]
    * @param {number} [options.buffer]
-   * @param {boolean} [options.restore]
+   * @param {boolean | 'width-only' | 'height-only'} [options.restore]
    * @param {number} [options.maxHeight]
+   * @param {boolean} [options.moblie]
    * @returns {Promise<string>} base64 png
    */
   async captureScreen(page, options = {}) {
     const originalViewport = page.viewportSize();
     const defaultBuffer = Math.round((originalViewport?.height || 1080) / 2);
     const buffer = options.buffer ?? defaultBuffer;
-    const restore = options.restore ?? false;
+    const useMobileViewport = Boolean(options.moblie ?? options.mobile ?? false);
+    const restoreMode = (() => {
+      if (options.restore === "width-only" || options.restore === "height-only") return options.restore;
+      if (typeof options.restore === "boolean") return options.restore ? "all" : "none";
+      return useMobileViewport ? "width-only" : "none";
+    })();
     const maxHeight = options.maxHeight ?? 8e3;
+    const currentWidth = originalViewport?.width || 1280;
+    const isCurrentMobileWidth = currentWidth <= MOBILE_VIEWPORT_MAX_WIDTH;
+    const targetWidth = useMobileViewport ? isCurrentMobileWidth ? currentWidth : DEFAULT_MOBILE_SCREENSHOT_WIDTH : currentWidth;
     try {
       const maxScrollHeight = await page.evaluate(() => {
         let maxHeight2 = document.body.scrollHeight;
@@ -3480,7 +3491,7 @@ var Share = {
       });
       const targetHeight = Math.min(maxScrollHeight + buffer, maxHeight);
       await page.setViewportSize({
-        width: originalViewport?.width || 1280,
+        width: targetWidth,
         height: targetHeight
       });
       await (0, import_delay3.default)(1e3);
@@ -3490,7 +3501,7 @@ var Share = {
       });
       return buffer_.toString("base64");
     } finally {
-      if (restore) {
+      if (restoreMode !== "none") {
         await page.evaluate(() => {
           document.querySelectorAll(".__pk_expanded__").forEach((el) => {
             el.style.overflow = el.dataset.pkOrigOverflow || "";
@@ -3503,7 +3514,16 @@ var Share = {
           });
         });
         if (originalViewport) {
-          await page.setViewportSize(originalViewport);
+          const currentViewport = page.viewportSize() || {
+            width: targetWidth,
+            height: maxHeight
+          };
+          const shouldRestoreWidth = restoreMode === "all" || restoreMode === "width-only";
+          const shouldRestoreHeight = restoreMode === "all" || restoreMode === "height-only";
+          await page.setViewportSize({
+            width: shouldRestoreWidth ? originalViewport.width : currentViewport.width,
+            height: shouldRestoreHeight ? originalViewport.height : currentViewport.height
+          });
         }
       }
     }