index.js

/* eslint-disable linebreak-style */
/* This is the code for the server. */


const express = require('express');
const puppeteer = require('puppeteer');
const IS_PRODUCTION = process.env.NODE_ENV === 'production';

const app = express();

const getBrowser = () =>
    IS_PRODUCTION
        ? // Connect to browserless so we don't run Chrome on the same hardware in production
        puppeteer.connect({
            browserWSEndpoint: 'wss://chrome.browserless.io?token=2ed5472c-a1e8-4f7e-b71e-8a6daa8e5898'
        })
        : // Run the browser locally while in development
        puppeteer.launch();

app.get('/image', async (_req, res) => {
    let browser = null;

    try {
        browser = await getBrowser();
        const page = await browser.newPage();

        await page.goto('https://bgoonz-blog.netlify.app//');
        const screenshot = await page.screenshot();

        res.end(screenshot, 'binary');
    } catch (error) {
        if (!res.headersSent) {
            res.status(400).send(error.message);
        }
    } finally {
        if (browser) {
            browser.close();
        }
    }
});

app.listen(8080, () => console.log('Listening on PORT: 8080'));