added waitFor

Author: metalwarrior665

INPUT_SCHEMA.json

@@ -86,6 +86,12 @@
             "description": "Only works for puppeteer type. Be careful that Chrome is not guaranteed to work with Puppeteer.",
             "default": false
         },
+        "waitFor": {
+            "title": "Wait for",
+            "type": "string",
+            "description": "Only works for puppeteer type. Will wait on each page. You can provide number in ms or a selector.",
+            "editor": "textfield"
+        },
         "useGoogleBotHeaders": {
             "title": "Use Google Bot headers",
             "type": "boolean",

src/main.js

@@ -23,6 +23,7 @@ Apify.main(async () => {
         retireInstanceAfterRequestCount = 10,
         headfull = false,
         useChrome = false,
+        waitFor,
         useGoogleBotHeaders = false,
     } = input;
 
@@ -70,6 +71,12 @@ Apify.main(async () => {
     }
 
     const handlePageFunction = async ({ request, $, html, page, response }) => {
+        if (page) {
+            // We wait for number in ms or a selector
+            const maybeNumber = Number(waitFor);
+            await page.waitFor(maybeNumber || maybeNumber === 0 ? maybeNumber : waitFor);
+        }
+
         let screenshotUrl;
         let htmlUrl;
         if (saveSnapshots) {