"use strict"; /* global $, CSS */ const path = require("path"); const {program} = require("commander"); require("css.escape"); const mkdirp = require("mkdirp"); const puppeteer = require("puppeteer"); const options = {}; program .arguments(" ") .action((narrow_uri, narrow, unread_msg_id, imagePath, realmUrl) => { options.narrowUri = narrow_uri; options.narrow = narrow; options.messageId = unread_msg_id; options.imagePath = imagePath; options.realmUrl = realmUrl; console.log(`Capturing screenshot for ${narrow} to ${imagePath}`); }) .parse(process.argv); if (options.imagePath === undefined) { console.error("no image path specified!"); process.exit(1); } // TODO: Refactor to share code with web/e2e-tests/realm-creation.test.ts 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: "new", }); 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(`${options.realmUrl}/devlogin`); // 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(`${options.narrowUri}`, { waitUntil: "networkidle2", }); // eslint-disable-next-line no-undef const message_list_id = await page.evaluate(() => zulip_test.current_msg_list.id); const messageListSelector = "#message-lists-container"; await page.waitForSelector(messageListSelector); const messageSelector = `#message-row-${message_list_id}-${CSS.escape(options.messageId)}`; await page.waitForSelector(messageSelector); const messageListBox = await page.$(messageListSelector); await page.evaluate((msg) => $(msg).removeClass("selected_message"), messageSelector); // Compute screenshot area, with some padding around the message group const clip = {...(await messageListBox.boundingBox())}; clip.x -= 5; clip.width += 10; clip.y += 5; 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();