"use strict"; const path = require("path"); const commander = require("commander"); const mkdirp = require("mkdirp"); const puppeteer = require("puppeteer"); const host = "localhost:9991"; const options = {}; commander .arguments(" ") .action((messageId, imagePath) => { options.messageId = messageId; options.imagePath = imagePath; console.log(`Capturing screenshot for message ${messageId} to ${imagePath}`); }) .parse(process.argv); if (options.messageId === undefined) { console.error("no messageId specified!"); process.exit(1); } // TODO: Refactor to share code with frontend_tests/puppeteer_tests/00-realm-creation.js async function run() { const browser = await puppeteer.launch({ args: [ "--window-size=1400,1024", "--no-sandbox", "--disable-setuid-sandbox", // Helps render fonts correctly on Ubuntu: https://github.com/puppeteer/puppeteer/issues/661 "--font-render-hinting=none", ], defaultViewport: null, headless: true, }); try { const page = await browser.newPage(); // deviceScaleFactor:2 gives better quality screenshots (higher pixel density) await page.setViewport({width: 1280, height: 1024, deviceScaleFactor: 2}); await page.goto("http://" + host); // wait for Iago devlogin button and click on it. await page.waitForSelector('[value="iago@zulip.com"]'); // By waiting till DOMContentLoaded we're confirming that Iago is logged in. await Promise.all([ page.waitForNavigation({waitUntil: "domcontentloaded"}), page.click('[value="iago@zulip.com"]'), ]); // Navigate to message and capture screenshot await page.goto(`http://${host}/#narrow/near/${options.messageId}`); const messageSelector = `#zfilt${options.messageId}`; await page.waitForSelector(messageSelector); // remove unread marker and don't select message const marker = `#zfilt${options.messageId} .unread_marker`; await page.evaluate((sel) => $(sel).remove(), marker); // eslint-disable-line no-undef await page.evaluate(() => navigate.up()); // eslint-disable-line no-undef const messageBox = await page.$(messageSelector); const messageGroup = (await messageBox.$x(".."))[0]; // Compute screenshot area, with some padding around the message group const clip = Object.assign({}, await messageGroup.boundingBox()); clip.y -= 5; clip.x -= 5; clip.width += 10; clip.height += 10; const imagePath = options.imagePath; const imageDir = path.dirname(imagePath); mkdirp.sync(imageDir); await page.screenshot({path: imagePath, clip}); } catch (error) { console.log(error); process.exit(1); } finally { await browser.close(); } } run();