Puppeteer on LambdaでWebページのキャプチャを撮る

'use strict';

const AWS = require('aws-sdk');
const launchChrome = require('@serverless-chrome/lambda');
const CDP = require('chrome-remote-interface');
const puppeteer = require('puppeteer');

const SAVE_BUCKET_NAME = '<キャプチャを保存するバケット名>';

/**
 * Lambdaハンドラ
 * @param {Object} event Lambdaイベントデータ
 * @param {Object} context Contextオブジェクト
 * @param {function} callback コールバックオプション
 */  
exports.handler = async (event, context, callback) => {
    let slsChrome = null;
    let browser = null;
    let page = null;

    try {
        // 前処理
        // serverless-chromeを起動し、PuppeteerからWebSocketで接続する
        slsChrome = await launchChrome();
        browser = await puppeteer.connect({ 
            browserWSEndpoint: (await CDP.Version()).webSocketDebuggerUrl 
        });
        page = await browser.newPage();

        // ブラウザ操作
        await page.goto('https://www.pressmantech.com', { waitUntil: 'domcontentloaded' });
        // page.click('a[rel="bookmark"]');
        // ↑本来はこうしたい。けどなぜか動作しないのでJavascriptでリンクをクリック
        page.evaluate(() => document.querySelector('a[rel="bookmark"]').click());
        await page.waitForNavigation({ timeout: 30000, waitUntil: 'domcontentloaded' });
        // PDFで日本語フォントを表示するためにWebフォントを強制的に適用させる
        // 注：クロスドメインなiframe内コンテンツは操作できないため豆腐のまま
        await page.evaluate(() => {
            var style = document.createElement('style');
            style.textContent = `
                @import url('//fonts.googleapis.com/css?family=Source+Code+Pro');
                @import url('//fonts.googleapis.com/earlyaccess/notosansjp.css');`;
            document.head.appendChild(style);
            document.body.style.fontFamily = "'Noto Sans JP', sans-serif";
            
            document.querySelectorAll('pre > code').forEach((el, idx) => {
                el.style.fontFamily = "'Source Code Pro', 'Noto Sans JP', monospace";
            });
        });
        await page.waitFor(1000);
        const jpgBuf = await page.screenshot({ fullPage: true, type: 'jpeg' });
        const pdfBuf = await page.pdf({ printBackground: true });

        // S3に保存
        const s3 = new AWS.S3();
        const now = new Date();
        now.setHours(now.getHours() + 9);
        const nowStr = '' +
            now.getFullYear() + '-' +
            (now.getMonth() + 1 + '').padStart(2, '0') + '-' +
            (now.getDate() + '').padStart(2, '0') + ' ' + 
            (now.getHours() + '').padStart(2, '0') + ':' +
            (now.getMinutes() + '').padStart(2, '0') + ':' + 
            (now.getSeconds() + '').padStart(2, '0');
        const fileName = nowStr.replace(/[\-:]/g, '_').replace(/\s/g, '__');
        let s3Param = {
            Bucket: SAVE_BUCKET_NAME,
            Key: null,
            Body: null
        };

        s3Param.Key = fileName + '.jpg';
        s3Param.Body =  jpgBuf;
        await s3.putObject(s3Param).promise();

        s3Param.Key = fileName + '.pdf';
        s3Param.Body = pdfBuf;
        await s3.putObject(s3Param).promise();

        return callback(null, JSON.stringify({ result: 'OK', createDate: nowStr }));
    } catch (err) {
        console.error(err);
        return callback(null, JSON.stringify({ result: 'NG' }));
    } finally {
        if (page) {
            await page.close();
        }

        if (browser) {
            await browser.disconnect();
        }

        if (slsChrome) { 
            await slsChrome.kill();
        }
    }
};