Add delay option in tools

Allow user to specify a delay in milliseconds before taking the
screenshot. The delay is an optional argument and defaults to 0.

Signed-off-by: Vinicius Tinti <viniciustinti@gmail.com>

README.md

@@ -112,6 +112,13 @@ docker run --shm-size 1G --rm -v /tmp/screenshots:/screenshots \
  <screenshot,full_screenshot,screenshot_series,full_screenshot_series> 'https://www.google.com' 1366x768
 ```
 
+### screenshot tools syntax
+
+`<tool> <url> <width>x<height> [<delay_in_ms>]`
+
+* `delay_in_ms`: is optional (defaults to `0`)
+  * Waits for `delay_in_ms` milliseconds before taking the screenshot
+
 ### `screenshot`
 
 ```bash

tools/fullScreenshot.js

@@ -1,5 +1,9 @@
 #!/usr/bin/env node
 
+function sleep(ms) {
+    return new Promise(resolve => {if (ms > 0) {setTimeout(resolve,ms)}})
+}
+
 process.on('uncaughtException', (error) => {
     console.error(error);
     process.exit(1);
@@ -36,6 +40,12 @@ if (typeof process.argv[3] === 'string') {
     var [width, height] = process.argv[3].split('x').map(v => parseInt(v, 10));
 }
 
+var delay = 0;
+
+if (typeof process.argv[4] === 'string') {
+    delay = parseInt(process.argv[4], 10);
+}
+
 var isMobile = false;
 
 let filename = `full_screenshot_${width}_${height}.png`;
@@ -59,6 +69,8 @@ let filename = `full_screenshot_${width}_${height}.png`;
 
     await page.goto(url, {waitUntil: 'networkidle2'});
 
+    await sleep(delay);
+
     await page.screenshot({path: `/screenshots/${filename}`, fullPage: true});
 
     browser.close();

tools/fullScreenshotSeries.js

@@ -1,5 +1,9 @@
 #!/usr/bin/env node
 
+function sleep(ms) {
+    return new Promise(resolve => {if (ms > 0) {setTimeout(resolve,ms)}})
+}
+
 process.on('uncaughtException', (error) => {
     console.error(error);
     process.exit(1);
@@ -36,6 +40,12 @@ if (typeof process.argv[3] === 'string') {
     var [width, height] = process.argv[3].split('x').map(v => parseInt(v, 10));
 }
 
+var delay = 0;
+
+if (typeof process.argv[4] === 'string') {
+    delay = parseInt(process.argv[4], 10);
+}
+
 var isMobile = false;
 
 let filename = `${dateStr}_full_screenshot_${width}_${height}.png`;
@@ -59,6 +69,8 @@ let filename = `${dateStr}_full_screenshot_${width}_${height}.png`;
 
     await page.goto(url, {waitUntil: 'networkidle2'});
 
+    await sleep(delay);
+
     await page.screenshot({path: `/screenshots/${filename}`, fullPage: true});
 
     browser.close();

tools/screenshot.js

@@ -1,5 +1,9 @@
 #!/usr/bin/env node
 
+function sleep(ms) {
+    return new Promise(resolve => {if (ms > 0) {setTimeout(resolve,ms)}})
+}
+
 process.on('uncaughtException', (error) => {
     console.error(error);
     process.exit(1);
@@ -36,6 +40,12 @@ if (typeof process.argv[3] === 'string') {
     var [width, height] = process.argv[3].split('x').map(v => parseInt(v, 10));
 }
 
+var delay = 0;
+
+if (typeof process.argv[4] === 'string') {
+    delay = parseInt(process.argv[4], 10);
+}
+
 var isMobile = false;
 
 let filename = `screenshot_${width}_${height}.png`;
@@ -59,6 +69,8 @@ let filename = `screenshot_${width}_${height}.png`;
 
     await page.goto(url, {waitUntil: 'networkidle2'});
 
+    await sleep(delay);
+
     await page.screenshot({path: `/screenshots/${filename}`, fullPage: false});
 
     browser.close();

tools/screenshotSeries.js

@@ -1,5 +1,9 @@
 #!/usr/bin/env node
 
+function sleep(ms) {
+    return new Promise(resolve => {if (ms > 0) {setTimeout(resolve,ms)}})
+}
+
 process.on('uncaughtException', (error) => {
     console.error(error);
     process.exit(1);
@@ -36,6 +40,12 @@ if (typeof process.argv[3] === 'string') {
     var [width, height] = process.argv[3].split('x').map(v => parseInt(v, 10));
 }
 
+var delay = 0;
+
+if (typeof process.argv[4] === 'string') {
+    delay = parseInt(process.argv[4], 10);
+}
+
 var isMobile = false;
 
 let filename = `${dateStr}_screenshot_${width}_${height}.png`;
@@ -59,6 +69,8 @@ let filename = `${dateStr}_screenshot_${width}_${height}.png`;
 
     await page.goto(url, {waitUntil: 'networkidle2'});
 
+    await sleep(delay);
+
     await page.screenshot({path: `/screenshots/${filename}`, fullPage: false});
 
     browser.close();