gocqhttp-twitter-bot/dist/webshot.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const CallableInstance = require("callable-instance");
const https = require("https");
const log4js = require("log4js");
const pngjs_1 = require("pngjs");
const puppeteer = require("puppeteer");
const read = require("read-all-stream");
const logger = log4js.getLogger('webshot');
logger.level = global.loglevel;
class Webshot extends CallableInstance {
    constructor(onready) {
        super('webshot');
        this.renderWebshot = (url, height, webshotDelay) => {
            const promise = new Promise(resolve => {
                const width = 1080;
                logger.info(`shooting ${width}*${height} webshot for ${url}`);
                this.browser.newPage()
                    .then(page => {
                    page.setUserAgent('Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/67.0.3396.99 Safari/537.36')
                        .then(() => page.setViewport({
                        width,
                        height,
                        isMobile: true,
                    }))
                        .then(() => page.goto(url))
                        .then(() => page.addStyleTag({
                        content: 'html{zoom:2}header{display:none!important}',
                    }))
                        .then(() => page.waitFor(webshotDelay))
                        .then(() => page.screenshot())
                        .then(screenshot => {
                        new pngjs_1.PNG({
                            filterType: 4,
                        }).on('parsed', function () {
                            let boundary = null;
                            let x = 0;
                            for (let y = 0; y < this.height; y++) {
                                const idx = (this.width * y + x) << 2;
                                if (this.data[idx] !== 255) {
                                    boundary = y;
                                    break;
                                }
                            }
                            if (boundary !== null) {
                                logger.info(`found boundary at ${boundary}, cropping image`);
                                this.data = this.data.slice(0, (this.width * boundary) << 2);
                                this.height = boundary;
                                boundary = null;
                                x = Math.floor(this.width / 2);
                                for (let y = this.height - 1; y >= 0; y--) {
                                    const idx = (this.width * y + x) << 2;
                                    if (this.data[idx] !== 255) {
                                        boundary = y;
                                        break;
                                    }
                                }
                                if (boundary != null) {
                                    logger.info(`found boundary at ${boundary}, trimming image`);
                                    this.data = this.data.slice(0, (this.width * boundary) << 2);
                                    this.height = boundary;
                                }
                                read(this.pack(), 'base64').then(data => {
                                    logger.info(`finished webshot for ${url}`);
                                    resolve({ data, boundary });
                                });
                            }
                            else if (height >= 8 * 1920) {
                                logger.warn('too large, consider as a bug, returning');
                                read(this.pack(), 'base64').then(data => {
                                    logger.info(`finished webshot for ${url}`);
                                    resolve({ data, boundary: 0 });
                                });
                            }
                            else {
                                logger.info('unable to found boundary, try shooting a larger image');
                                resolve({ data: '', boundary });
                            }
                        }).parse(screenshot);
                    })
                        .then(() => page.close());
                });
            });
            return promise.then(data => {
                if (data.boundary === null)
                    return this.renderWebshot(url, height + 1920, webshotDelay);
                else
                    return data.data;
            });
        };
        this.fetchImage = (url) => new Promise(resolve => {
            logger.info(`fetching ${url}`);
            https.get(url, res => {
                if (res.statusCode === 200) {
                    read(res, 'base64').then(data => {
                        logger.info(`successfully fetched ${url}`);
                        resolve(data);
                    });
                }
                else {
                    logger.error(`failed to fetch ${url}: ${res.statusCode}`);
                    resolve();
                }
            }).on('error', (err) => {
                logger.error(`failed to fetch ${url}: ${err.message}`);
                resolve();
            });
        });
        puppeteer.launch({ args: ['--no-sandbox', '--disable-setuid-sandbox', '--lang=zh-CN,zh'] })
            .then(browser => this.browser = browser)
            .then(() => {
            logger.info('launched puppeteer browser');
            if (onready)
                onready();
        });
    }
    webshot(tweets, callback, webshotDelay) {
        let promise = new Promise(resolve => {
            resolve();
        });
        tweets.forEach(twi => {
            let cqstr = '';
            const url = `https://mobile.twitter.com/${twi.user.screen_name}/status/${twi.id_str}`;
            promise = promise.then(() => this.renderWebshot(url, 1920, webshotDelay))
                .then(base64Webshot => {
                if (base64Webshot)
                    cqstr += `[CQ:image,file=base64://${base64Webshot}]`;
            });
            if (twi.extended_entities) {
                twi.extended_entities.media.forEach(media => {
                    promise = promise.then(() => this.fetchImage(media.media_url_https))
                        .then(base64Image => {
                        if (base64Image)
                            cqstr += `[CQ:image,file=base64://${base64Image}]`;
                    });
                });
            }
            if (twi.entities && twi.entities.urls && twi.entities.urls.length) {
                promise = promise.then(() => {
                    const urls = twi.entities.urls
                        .filter(urlObj => urlObj.indices[0] < twi.display_text_range[1])
                        .map(urlObj => urlObj.expanded_url);
                    if (urls.length) {
                        cqstr += '\n';
                        cqstr += urls.join('\n');
                    }
                });
            }
            promise.then(() => callback(cqstr));
        });
        return promise;
    }
}
exports.default = Webshot;