از همه صفحات خود عکس بگیرید – انجمن DEV

export const snapshotPages = async (
  pages: string[],
  snapshotDir: string,
  toRelative: (url: string) => string,
) => {
  const browser = await puppeteer.launch();

  if (!fs.existsSync(snapshotDir)) {
    fs.mkdirSync(snapshotDir);
  }

  // process 10 pages at a time
  await processPromisesBatch(pages, 10, async (url) => {
    const page = await browser.newPage();

    // eslint-disable-next-line
    console.log(`Opening ${url}`);
    await page.goto(url, {
      waitUntil: 'networkidle2',
    });

    const relative = toRelative(url);

    const snapshotPath = path.join(snapshotDir, relative);

    const dir = path.dirname(snapshotPath);
    try {
      if (!fs.existsSync(dir)) {
        fs.mkdirSync(dir);
      }

      await page.screenshot({ path: `${snapshotPath}.png`, fullPage: true });
    } catch (err) {
      // eslint-disable-next-line
      console.log('err: ', err);
    }
  })

  await browser.close();
};