zulip/tools/screenshots/thread-screenshot.js

"use strict";

/* global $, CSS */

const fs = require("node:fs");
const path = require("node:path");

const {program} = require("commander");
require("css.escape");
const puppeteer = require("puppeteer");

const options = {};

program
    .arguments("<narrow_uri> <narrow> <unread_msg_id> <image_path> <realm_url>")
    .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=500,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: 530, 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);

        // remove unread marker and don't select message
        const marker = `.message-list[data-message-list-id="${CSS.escape(
            message_list_id,
        )}"] .unread_marker`;
        await page.evaluate((sel) => {
            $(sel).remove();
        }, marker);

        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);

        // This is done so as to get white background while capturing screenshots.
        const background_selectors = [".app-main", ".message-feed", ".message_header"];
        await page.evaluate((selectors) => {
            for (const selector of selectors) {
                $(selector).css("background-color", "white");
            }
        }, background_selectors);

        // Compute screenshot area, with some padding around the message group
        const clip = {...(await messageListBox.boundingBox())};
        clip.width -= 64;
        clip.y += 10;
        clip.height -= 8;
        const imagePath = options.imagePath;
        const imageDir = path.dirname(imagePath);
        await fs.promises.mkdir(imageDir, {recursive: true});
        await page.screenshot({path: imagePath, clip});
    } catch (error) {
        console.log(error);
        process.exit(1);
    } finally {
        await browser.close();
    }
}

run();