@browserless/pdf 10.8.0-beta.0 → 10.8.0-beta.2

package/package.json

@@ -2,7 +2,7 @@
   "name": "@browserless/pdf",
   "description": "Sensible good defaults for exporting a website as PDF",
   "homepage": "https://browserless.js.org/#/?id=pdfurl-options",
-  "version": "10.8.0-beta.0",
+  "version": "10.8.0-beta.2",
   "main": "src",
   "repository": {
     "directory": "packages/pdf",
@@ -38,5 +38,5 @@
     "test": "exit 0"
   },
   "license": "MIT",
-  "gitHead": "e7c845edfd2552f48b8f6f3b70d887ec497955b0"
+  "gitHead": "7f293ddfa791439d087ce5ee3d87eca4d1abb08c"
 }

package/src/index.js

@@ -4,6 +4,7 @@ const { takeScreenshot } = require('@browserless/screenshot')
 const debug = require('debug-logfmt')('browserless:pdf')
 const createGoto = require('@browserless/goto')
 const timeSpan = require('@kikobeats/time-span')({ format: require('pretty-ms') })
+const { setTimeout } = require('node:timers/promises')
 
 const getMargin = unit => {
   if (!unit) return unit
@@ -42,9 +43,24 @@ module.exports = ({ goto, ...gotoOpts } = {}) => {
       } else {
         await goto(page, { ...opts, url, waitUntil, waitUntilAuto })
         async function waitUntilAuto (page) {
-          const { isWhite } = await takeScreenshot({ page, goto, opts })
+          let isWhite = false
+          let retryCount = 0
+          const maxRetries = 3
+
+          do {
+            const screenshotResult = await takeScreenshot({ page, goto, opts })
+            isWhite = screenshotResult.isWhite
+
+            if (isWhite && retryCount < maxRetries) {
+              retryCount++
+              debug('screenshot:retry', { waitUntil, isWhite, retryCount, maxRetries })
+              await setTimeout(500)
+              await goto.waitUntilAuto(page, { timeout: opts.timeout })
+            }
+          } while (isWhite && retryCount < maxRetries)
+
           pdfBuffer = await generatePdf(page)
-          debug('screenshot', { waitUntil, isWhite, duration: timePdf() })
+          debug('screenshot', { waitUntil, isWhite, retryCount, duration: timePdf() })
         }
       }