index.js

#!/usr/bin/env node

const fs = require('fs-extra');
const path = require('path');
const puppeteer = require('puppeteer');
const configPath = process.argv.pop();

if (!configPath || !fs.existsSync(path.resolve(configPath))) {
  console.error('Error: config not provided');
  process.exit(1);
}

const tasks = require(path.resolve(configPath));

const DEFAULT_WIDTH = 1280;
const DEFAULT_HEIGHT = 800;

(async () => {
  try {
    await execute();
  } catch (e) {
    console.error(e);
    process.exit(1);
  }

  async function execute() {
    const browser = await puppeteer.launch();
    const page = await browser.newPage();

    for (const task of tasks) {
      const { height, width, fullPage, emulate, auth, waitFor, element, routes } = task;

      if (width || height) {
        await page.setViewport({
          width: width || DEFAULT_WIDTH,
          height: height || DEFAULT_HEIGHT,
        });
      }

      if (emulate) {
        await page.emulate(puppeteer.KnownDevices[emulate]);
      }

      for (const route of routes) {
        const { url, output } = route;
        const outputPath = eval('`' + output + '`');

        try {
          if (!/\.pdf|\.jpg|\.jpeg|\.png$/.test(outputPath)) {
            console.warn(`Warning: output format for ${output} is not supported`);
            continue;
          }

          if (auth) {
            const [username, password] = auth.split(';');
            await page.authenticate({ username, password });
          }

          await page.goto(url);

          if (waitFor) {
            if (typeof waitFor === 'number') {
              await page.waitFor(waitFor);
            } else if (typeof waitFor === 'string') {
              await page.waitForSelector(waitFor);
            } else {
              console.warn(`Warning: invalid waitFor value ${waitFor}`);
            }
          }

          await fs.ensureDir(path.dirname(path.resolve(outputPath)));

          if (element) {
            const el = await page.$(element);
            if (/\.pdf$/.test(outputPath)) {
              console.warn('Warning: taking element screenshot in PDF not supported');
            } else {
              await el.screenshot({ path: path.resolve(outputPath) });
            }
          } else {
            if (/\.pdf$/.test(outputPath)) {
              await page.pdf({ path: path.resolve(outputPath) });
            } else {
              await page.screenshot({ path: path.resolve(outputPath), fullPage });
            }
          }

          console.log(`Saved ${url} screenshot to ${outputPath}`);
        } catch (e) {
          console.error(`Failed to save ${url} screenshot to ${outputPath}`, e);
        }
      }
    }

    await browser.close();
  }
})();