2022-06-16 04:13:44 -05:00
|
|
|
const puppeteer = require("puppeteer");
|
|
|
|
const { PuppeteerBlocker } = require("@cliqz/adblocker-puppeteer");
|
|
|
|
const fetch = require("cross-fetch");
|
|
|
|
const config = require("../../src/config.js");
|
|
|
|
const fs = require("fs");
|
2022-05-26 11:15:07 -05:00
|
|
|
|
2022-06-16 04:13:44 -05:00
|
|
|
const screenshotDirectory =
|
|
|
|
config.API.STORAGE_LOCATION + "/LinkWarden/screenshot's/";
|
|
|
|
const pdfDirectory = config.API.STORAGE_LOCATION + "/LinkWarden/pdf's/";
|
2022-05-26 11:15:07 -05:00
|
|
|
|
2022-06-16 04:13:44 -05:00
|
|
|
if (!fs.existsSync(screenshotDirectory)) {
|
2022-05-26 11:15:07 -05:00
|
|
|
fs.mkdirSync(screenshotDirectory, { recursive: true });
|
|
|
|
}
|
|
|
|
|
2022-06-16 04:13:44 -05:00
|
|
|
if (!fs.existsSync(pdfDirectory)) {
|
2022-05-26 11:15:07 -05:00
|
|
|
fs.mkdirSync(pdfDirectory, { recursive: true });
|
|
|
|
}
|
|
|
|
|
|
|
|
module.exports = async (link, id) => {
|
|
|
|
const browser = await puppeteer.launch();
|
|
|
|
const page = await browser.newPage();
|
|
|
|
|
|
|
|
await PuppeteerBlocker.fromPrebuiltAdsAndTracking(fetch).then((blocker) => {
|
|
|
|
blocker.enableBlockingInPage(page);
|
|
|
|
});
|
|
|
|
|
2022-06-16 04:13:44 -05:00
|
|
|
await page.goto(link, { waitUntil: "load", timeout: 0 });
|
|
|
|
|
|
|
|
await page.screenshot({
|
|
|
|
path: screenshotDirectory + id + ".png",
|
|
|
|
fullPage: true,
|
|
|
|
});
|
|
|
|
await page.pdf({ path: pdfDirectory + id + ".pdf", format: "a4" });
|
2022-05-26 11:15:07 -05:00
|
|
|
|
|
|
|
await browser.close();
|
2022-06-16 04:13:44 -05:00
|
|
|
};
|