@skrillex1224/playwright-toolkit 2.1.233 → 2.1.234

package/dist/index.cjs

@@ -312,6 +312,7 @@ var ActorInfo = {
     name: "\u901A\u4E49\u5343\u95EE",
     domain: "www.qianwen.com",
     path: "/chat",
+    device: Device.Mobile,
     share: {
       mode: "response",
       prefix: "https://www.qianwen.com/share/chat/",
@@ -469,6 +470,9 @@ function createInternalLogger(moduleName, explicitLogger) {
 // src/internals/screenshot.js
 var import_delay = __toESM(require("delay"), 1);
 
+// src/internals/constants.js
+var PageRuntimeStateKey = "__playwright_toolkit_runtime_state__";
+
 // src/internals/viewport.js
 var toPositiveInt = (value) => {
   const number = Math.round(Number(value) || 0);
@@ -505,6 +509,7 @@ var SUPPORTED_TYPES = /* @__PURE__ */ new Set(["png", "jpeg", "webp"]);
 var EXPANDED_SCROLLABLE_CLASS = "__pk_expanded__";
 var DEFAULT_MAX_HEIGHT = 8e3;
 var DEFAULT_SETTLE_MS = 1e3;
+var DEFAULT_MOBILE_SETTLE_MS = 50;
 var toPositiveNumber = (value, fallback = 0) => {
   const n = Number(value);
   if (!Number.isFinite(n) || n <= 0) return fallback;
@@ -527,6 +532,7 @@ var normalizeQuality = (value, type) => {
   if (rounded < 0 || rounded > 100) return void 0;
   return rounded;
 };
+var resolvePageDevice = (page) => normalizeDevice(page?.[PageRuntimeStateKey]?.device);
 var buildFullPageClip = (metrics, viewport, maxClipHeight) => {
   const contentSize = metrics && typeof metrics === "object" ? metrics.contentSize || null : null;
   const width = Math.max(1, Math.ceil(viewport.width || contentSize?.width || 1));
@@ -543,7 +549,7 @@ var buildFullPageClip = (metrics, viewport, maxClipHeight) => {
   };
 };
 var expandScrollableContent = async (page, options = {}) => {
-  return await page.evaluate(({ className, forceScrollableHeight, visibleOnly }) => {
+  return await page.evaluate(({ className, forceScrollableHeight, visibleOnly, expandDocumentElements }) => {
     const body = document.body;
     const root = document.documentElement;
     const scrollingElement = document.scrollingElement;
@@ -568,8 +574,8 @@ var expandScrollableContent = async (page, options = {}) => {
       if (Number(style.opacity) === 0) return false;
       return rect.width > 0 && rect.height > 0;
     };
-    const rememberOriginalBoxStyles = (el) => {
-      if (!el || isDocumentElement(el) || el.classList.contains(className)) {
+    const rememberOriginalBoxStyles = (el, { includeDocumentElement = false } = {}) => {
+      if (!el || !includeDocumentElement && isDocumentElement(el) || el.classList.contains(className)) {
         return;
       }
       el.dataset.pkOrigOverflow = el.style.overflow;
@@ -582,6 +588,19 @@ var expandScrollableContent = async (page, options = {}) => {
       el.dataset.pkOrigMaxHeightPriority = el.style.getPropertyPriority("max-height") || "";
       el.classList.add(className);
     };
+    const expandDocumentElementsToHeight = (height) => {
+      if (!expandDocumentElements) return;
+      const targetHeight = Math.ceil(Number(height) || 0);
+      if (targetHeight <= 0) return;
+      [root, body, scrollingElement].forEach((el) => {
+        if (!el) return;
+        rememberOriginalBoxStyles(el, { includeDocumentElement: true });
+        el.style.setProperty("overflow", "visible", "important");
+        el.style.setProperty("height", `${targetHeight}px`, "important");
+        el.style.setProperty("min-height", `${targetHeight}px`, "important");
+        el.style.setProperty("max-height", "none", "important");
+      });
+    };
     const isScrollableY = (el, style, rect) => {
       if (!el || el.scrollHeight <= el.clientHeight + 1) {
         return false;
@@ -717,38 +736,61 @@ var expandScrollableContent = async (page, options = {}) => {
       }
     });
     maxHeight = Math.max(maxHeight, measureDocumentScrollHeight(), measureExpandedRenderedBottom());
+    if (expandDocumentElements) {
+      for (let pass = 0; pass < 2; pass += 1) {
+        expandDocumentElementsToHeight(maxHeight);
+        maxHeight = Math.max(maxHeight, measureDocumentScrollHeight(), measureExpandedRenderedBottom());
+      }
+    }
     return Math.ceil(maxHeight);
   }, {
     className: EXPANDED_SCROLLABLE_CLASS,
     forceScrollableHeight: options.forceScrollableHeight !== false,
-    visibleOnly: options.visibleOnly !== false
+    visibleOnly: options.visibleOnly !== false,
+    expandDocumentElements: options.expandDocumentElements === true
   });
 };
 var prepareExpandedFullPageScreenshot = async (page, options = {}) => {
   const originalViewport = await resolveCurrentViewportSize(page);
   const maxHeight = toPositiveInteger(options.maxHeight, DEFAULT_MAX_HEIGHT);
-  const settleMs = Math.max(0, Number(options.settleMs ?? DEFAULT_SETTLE_MS) || 0);
+  const preserveViewport = options.preserveViewport ?? resolvePageDevice(page) === Device.Mobile;
+  const defaultSettleMs = preserveViewport ? DEFAULT_MOBILE_SETTLE_MS : DEFAULT_SETTLE_MS;
+  const requestedSettleMs = Math.max(0, Number(options.settleMs ?? defaultSettleMs) || 0);
+  const settleMs = preserveViewport ? Math.min(requestedSettleMs, DEFAULT_MOBILE_SETTLE_MS) : requestedSettleMs;
   const maxScrollHeight = await expandScrollableContent(page, {
     forceScrollableHeight: options.forceScrollableHeight,
-    visibleOnly: options.visibleOnly !== false
+    visibleOnly: options.visibleOnly !== false,
+    expandDocumentElements: preserveViewport
   });
   const targetHeight = Math.min(maxScrollHeight, maxHeight);
-  await page.setViewportSize({
-    width: originalViewport.width,
-    height: targetHeight
-  });
+  let viewportResized = false;
+  if (!preserveViewport) {
+    await page.setViewportSize({
+      width: originalViewport.width,
+      height: targetHeight
+    });
+    viewportResized = true;
+  }
   if (settleMs > 0) {
     await (0, import_delay.default)(settleMs);
   }
   return {
     originalViewport,
     maxScrollHeight,
-    targetHeight
+    targetHeight,
+    preserveViewport,
+    viewportResized
   };
 };
 var restoreExpandedFullPageScreenshot = async (page, state = {}) => {
   await page.evaluate((className) => {
-    document.querySelectorAll(`.${className}`).forEach((el) => {
+    const targets = new Set([
+      ...document.querySelectorAll(`.${className}`),
+      document.documentElement,
+      document.body,
+      document.scrollingElement
+    ].filter((el) => el?.classList?.contains(className)));
+    targets.forEach((el) => {
       el.style.setProperty("overflow", el.dataset.pkOrigOverflow || "", el.dataset.pkOrigOverflowPriority || "");
       el.style.setProperty("height", el.dataset.pkOrigHeight || "", el.dataset.pkOrigHeightPriority || "");
       el.style.setProperty("min-height", el.dataset.pkOrigMinHeight || "", el.dataset.pkOrigMinHeightPriority || "");
@@ -764,7 +806,7 @@ var restoreExpandedFullPageScreenshot = async (page, state = {}) => {
       el.classList.remove(className);
     });
   }, EXPANDED_SCROLLABLE_CLASS);
-  if (state?.originalViewport) {
+  if (state?.originalViewport && state?.viewportResized) {
     await page.setViewportSize(state.originalViewport);
   }
 };
@@ -1262,9 +1304,6 @@ var ProxyMeterRuntime = {
   getProxyMeterSnapshot
 };
 
-// src/internals/constants.js
-var PageRuntimeStateKey = "__playwright_toolkit_runtime_state__";
-
 // src/runtime-env.js
 var BROWSER_PROFILE_SCHEMA_VERSION = 1;
 var rememberedRuntimeState = null;
@@ -7379,11 +7418,11 @@ var resolveScreenshotWatermarkifyMeta = async (page, options = {}) => {
     watermark: options.watermark,
     strip: options.strip,
     stripLogoSrc,
-    device: resolvePageDevice(page)
+    device: resolvePageDevice2(page)
   };
 };
 var buildFontFamily = () => 'MiSans, "SF Pro Display", "PingFang SC", "Helvetica Neue", Arial, sans-serif';
-var resolvePageDevice = (page) => normalizeDevice(page?.[PageRuntimeStateKey]?.device);
+var resolvePageDevice2 = (page) => normalizeDevice(page?.[PageRuntimeStateKey]?.device);
 var findStripSegment = (segments, kind) => {
   const found = Array.isArray(segments) ? segments.find((segment) => segment?.kind === kind) : null;
   return found && typeof found === "object" ? found : {};