el.xwx.moe/api/modules/getData.js

37 lines
1.1 KiB
JavaScript
Raw Normal View History

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-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-06-16 04:13:44 -05:00
if (!fs.existsSync(screenshotDirectory)) {
fs.mkdirSync(screenshotDirectory, { recursive: true });
}
2022-06-16 04:13:44 -05:00
if (!fs.existsSync(pdfDirectory)) {
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" });
await browser.close();
2022-06-16 04:13:44 -05:00
};