const puppeteer = require("puppeteer"); const commander = require("commander"); const path = require("path"); const mkdirp = require("mkdirp"); 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 devlogin admin button and click on it await page.waitForSelector('.btn-admin'); await page.click('.btn-admin'); // 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: clip }); } catch (e) { console.log(e); process.exit(1); } finally { await browser.close(); } } run();