screenshot.js

exports.getScreenShot = async (req, res) => {
    const { address } = req.query;
    // web address you want to screenshot , Default is medium.com
    const webAddress = `https://${( address || 'medium')}.com`;
    try {
        const browser = await puppeteer.launch();
        const page = await browser.newPage();
        await page.goto(webAddress);
        await page.screenshot({ path: `./assets/snapshot.png` });
        await browser.close();
        res.send('screenshot saved');
    }
    catch (err) {
        res.status(500).send(err.message);
    }
};