mirror of
https://github.com/zulip/zulip.git
synced 2025-10-24 08:33:43 +00:00
7229088b6c
To take screenshots, we log in as Desdemonia who doesn't seem to have permissions to the streams where we capture screenshots. This commit fixes it by logging in as Iago. Also added a waitForNavigation until DOMContentLoaded to avoid flakes that could occur because of us going to another URL before Iago gets logged in.
77 lines
2.9 KiB
JavaScript
77 lines
2.9 KiB
JavaScript
const puppeteer = require("puppeteer");
|
|
const commander = require("commander");
|
|
const path = require("path");
|
|
const mkdirp = require("mkdirp");
|
|
const host = "localhost:9991";
|
|
const options = {};
|
|
|
|
commander
|
|
.arguments('<message_id> <image_path>')
|
|
.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: clip });
|
|
} catch (e) {
|
|
console.log(e);
|
|
process.exit(1);
|
|
} finally {
|
|
await browser.close();
|
|
}
|
|
}
|
|
|
|
run();
|