58 lines
2.1 KiB
JavaScript
58 lines
2.1 KiB
JavaScript
|
const puppeteer = require("puppeteer");
|
||
|
|
const { scrollPageToBottom } = require("puppeteer-autoscroll-down");
|
||
|
|
const express = require("express");
|
||
|
|
|
||
|
|
const app = express();
|
||
|
|
|
||
|
|
const folderToServe = "/Users/tarasis/Programming/websites/rmcg.dev/www/";
|
||
|
|
|
||
|
|
let dynamicPort = 6100;
|
||
|
|
|
||
|
|
app.get("/screenshot", async (req, res) => {
|
||
|
|
//pretend generated html
|
||
|
|
const dynamicPage = express();
|
||
|
|
// console.log("🚀 ~ app.get ~ dynamicPage:", dynamicPage);
|
||
|
|
const currentDynamicPort = dynamicPort++;
|
||
|
|
console.log("🚀 ~ app.get ~ currentDynamicPort:", currentDynamicPort);
|
||
|
|
dynamicPage.use(
|
||
|
|
// express.static("/Users/tarasis/Programming/websites/rmcg.dev/www")
|
||
|
|
express.static(folderToServe)
|
||
|
|
);
|
||
|
|
|
||
|
|
const dynamicServer = dynamicPage.listen(currentDynamicPort, async () => {
|
||
|
|
console.log(
|
||
|
|
`Dynamic server is running at http://localhost:${currentDynamicPort}`
|
||
|
|
);
|
||
|
|
|
||
|
|
const browser = await puppeteer.launch();
|
||
|
|
// console.log("🚀 ~ dynamicServer ~ browser:", browser)
|
||
|
|
const page = await browser.newPage();
|
||
|
|
// await page.goto(req.query.url); // URL is given by the "user" (your client-side application)
|
||
|
|
await page.goto(`http://localhost:${currentDynamicPort}`); // URL is given by the "user" (your client-side application)
|
||
|
|
|
||
|
|
await page.waitForResponse((response) => response.status() === 200);
|
||
|
|
await scrollPageToBottom(page);
|
||
|
|
const screenshotBuffer = await page.screenshot({ fullPage: true });
|
||
|
|
|
||
|
|
// Respond with the image
|
||
|
|
res.writeHead(200, {
|
||
|
|
"Content-Type": "image/png",
|
||
|
|
"Content-Length": screenshotBuffer.length,
|
||
|
|
});
|
||
|
|
res.end(screenshotBuffer);
|
||
|
|
// other option is just to save the file to disk
|
||
|
|
// await page.screenshot({ path: 'example.png' });
|
||
|
|
// Close the browser
|
||
|
|
|
||
|
|
await browser.close();
|
||
|
|
console.log(
|
||
|
|
`Closing dynamic server http://localhost:${currentDynamicPort}`
|
||
|
|
);
|
||
|
|
await dynamicServer.close();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
app.listen(4000, () => {
|
||
|
|
console.log("opened server on 4000");
|
||
|
|
});
|