53 lines
1.6 KiB
JavaScript
53 lines
1.6 KiB
JavaScript
import puppeteer from "puppeteer";
|
|
import { scrollPageToBottom } from "puppeteer-autoscroll-down";
|
|
import express from "express";
|
|
import "dotenv/config";
|
|
|
|
const folderToServe = "/Users/tarasis/Programming/websites/rmcg.dev/www/";
|
|
|
|
const port = process.env.PORT || 4000;
|
|
|
|
const app = express();
|
|
|
|
app.use(express.static(folderToServe));
|
|
app.get("/screenshot", async (req, res) => {
|
|
const dynamicPage = express();
|
|
dynamicPage.use(express.static(folderToServe));
|
|
|
|
const dynamicServer = dynamicPage.listen(0, async () => {
|
|
const dynamicPort = dynamicServer.address().port;
|
|
console.log(
|
|
`Dynamic server is running at http://localhost:${dynamicPort}`
|
|
);
|
|
|
|
const browser = await puppeteer.launch();
|
|
const page = await browser.newPage();
|
|
await page.goto(`http://localhost:${dynamicPort}`);
|
|
|
|
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:${dynamicPort}`);
|
|
|
|
await dynamicServer.close();
|
|
});
|
|
});
|
|
|
|
app.listen(port, () => {
|
|
console.log(`Server is running at http://localhost:${port}`);
|
|
});
|