AWS Lambda OG Image Generator

index.js

const chromium = require('/opt/scraper-deps/node_modules/@sparticuz/chromium');
const puppeteer = require('/opt/scraper-deps/node_modules/puppeteer-core');

const handler = async (event) => {
  
  const title = event.title || event.queryStringParameters.title || null;
  if (!title) {
    return {
      statusCode: 400,
      body: JSON.stringify({
        message: 'Bad request!'
      })
    }
  }
  
  const url = `{YOUR_HTML_ENDPOINT}?title=${title}`

  let browser;
  try {
    browser = await puppeteer.launch({
      args: [...chromium.args],
      defaultViewport: chromium.defaultViewport,
      executablePath: await chromium.executablePath(),
      headless: chromium.headless,
      ignoreHTTPSErrors: true,
    });
    
    const page = await browser.newPage();
    await page.setViewport({ width: 1920, height: 1080 });
    await page.goto(url);

    const ogContainer = '.og-container';
    await page.waitForSelector(ogContainer);
    const element = await page.$(ogContainer);

    // Capture screenshot and save it
    const image = await element.screenshot();
    const base64Image = image.toString('base64');
    
    const response = {
      statusCode: 200,
      body: JSON.stringify({
        content: base64Image
      }),
    };
    return response;
  } catch (e) {
    console.log('e', e);
  } finally {
    browser.close();
  }
};

module.exports.handler = handler;