gocqhttp-twitter-bot/dist/twitter.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const fs = require("fs");
const log4js = require("log4js");
const path = require("path");
const redis = require("redis");
const sha1 = require("sha1");
const Twitter = require("twitter");
const webshot_1 = require("./webshot");
const logger = log4js.getLogger('twitter');
logger.level = global.loglevel;
class default_1 {
    constructor(opt) {
        this.launch = () => {
            if (this.redisConfig) {
                this.redisClient = redis.createClient({
                    host: this.redisConfig.redisHost,
                    port: this.redisConfig.redisPort,
                });
            }
            this.webshot = new webshot_1.default(() => setTimeout(this.work, this.workInterval * 1000));
        };
        this.work = () => {
            const lock = this.lock;
            if (this.workInterval < 1)
                this.workInterval = 1;
            if (lock.feed.length === 0) {
                setTimeout(() => {
                    this.work();
                }, this.workInterval * 1000);
                return;
            }
            if (lock.workon >= lock.feed.length)
                lock.workon = 0;
            if (!lock.threads[lock.feed[lock.workon]] ||
                !lock.threads[lock.feed[lock.workon]].subscribers ||
                lock.threads[lock.feed[lock.workon]].subscribers.length === 0) {
                logger.warn(`nobody subscribes thread ${lock.feed[lock.workon]}, removing from feed`);
                delete lock.threads[lock.feed[lock.workon]];
                lock.feed.splice(lock.workon, 1);
                fs.writeFileSync(path.resolve(this.lockfile), JSON.stringify(lock));
                this.work();
                return;
            }
            logger.debug(`pulling feed ${lock.feed[lock.workon]}`);
            const promise = new Promise(resolve => {
                let match = lock.feed[lock.workon].match(/https:\/\/twitter.com\/([^\/]+)\/lists\/([^\/]+)/);
                let config;
                let endpoint;
                if (match) {
                    config = {
                        owner_screen_name: match[1],
                        slug: match[2],
                        tweet_mode: 'extended',
                    };
                    endpoint = 'lists/statuses';
                }
                else {
                    match = lock.feed[lock.workon].match(/https:\/\/twitter.com\/([^\/]+)/);
                    if (match) {
                        config = {
                            screen_name: match[1],
                            exclude_replies: false,
                            tweet_mode: 'extended',
                        };
                        endpoint = 'statuses/user_timeline';
                    }
                }
                if (endpoint) {
                    const offset = lock.threads[lock.feed[lock.workon]].offset;
                    if (offset > 0)
                        config.since_id = offset;
                    this.client.get(endpoint, config, (error, tweets, response) => {
                        if (error) {
                            if (error instanceof Array && error.length > 0 && error[0].code === 34) {
                                logger.warn(`error on fetching tweets for ${lock.feed[lock.workon]}: ${JSON.stringify(error)}`);
                                lock.threads[lock.feed[lock.workon]].subscribers.forEach(subscriber => {
                                    logger.info(`sending notfound message of ${lock.feed[lock.workon]} to ${JSON.stringify(subscriber)}`);
                                    this.bot.bot('send_msg', {
                                        message_type: subscriber.chatType,
                                        user_id: subscriber.chatID,
                                        group_id: subscriber.chatID,
                                        discuss_id: subscriber.chatID,
                                        message: `链接 ${lock.feed[lock.workon]} 指向的用户或列表不存在，请退订。`,
                                    });
                                });
                            }
                            else {
                                logger.error(`unhandled error on fetching tweets for ${lock.feed[lock.workon]}: ${JSON.stringify(error)}`);
                            }
                            resolve();
                        }
                        else
                            resolve(tweets);
                    });
                }
            });
            promise.then((tweets) => {
                logger.debug(`api returned ${JSON.stringify(tweets)} for feed ${lock.feed[lock.workon]}`);
                if (!tweets || tweets.length === 0) {
                    lock.threads[lock.feed[lock.workon]].updatedAt = new Date().toString();
                    return;
                }
                if (lock.threads[lock.feed[lock.workon]].offset === -1) {
                    lock.threads[lock.feed[lock.workon]].offset = tweets[0].id_str;
                    return;
                }
                if (lock.threads[lock.feed[lock.workon]].offset === 0)
                    tweets.splice(1);
                return this.webshot(this.mode, tweets, (msg, text) => {
                    lock.threads[lock.feed[lock.workon]].subscribers.forEach(subscriber => {
                        logger.info(`pushing data of thread ${lock.feed[lock.workon]} to ${JSON.stringify(subscriber)}`);
                        let hash = JSON.stringify(subscriber) + text;
                        logger.debug(hash);
                        hash = sha1(JSON.stringify(subscriber) + text);
                        logger.debug(hash);
                        const send = () => {
                            this.bot.bot('send_msg', {
                                message_type: subscriber.chatType,
                                user_id: subscriber.chatID,
                                group_id: subscriber.chatID,
                                discuss_id: subscriber.chatID,
                                message: this.mode === 0 ? msg : text,
                            });
                        };
                        if (this.redisClient) {
                            this.redisClient.exists(hash, (err, res) => {
                                logger.debug('redis: ', res);
                                if (err) {
                                    logger.error('redis error: ', err);
                                }
                                else if (res) {
                                    logger.info('key hash exists, skip this subscriber');
                                    return;
                                }
                                send();
                                this.redisClient.set(hash, 'true', 'EX', this.redisConfig.redisExpireTime, (setErr, setRes) => {
                                    logger.debug('redis: ', setRes);
                                    if (setErr) {
                                        logger.error('redis error: ', setErr);
                                    }
                                });
                            });
                        }
                        else
                            send();
                    });
                }, this.webshotDelay)
                    .then(() => {
                    lock.threads[lock.feed[lock.workon]].offset = tweets[0].id_str;
                    lock.threads[lock.feed[lock.workon]].updatedAt = new Date().toString();
                });
            })
                .then(() => {
                lock.workon++;
                let timeout = this.workInterval * 1000 / lock.feed.length;
                if (timeout < 1000)
                    timeout = 1000;
                fs.writeFileSync(path.resolve(this.lockfile), JSON.stringify(lock));
                setTimeout(() => {
                    this.work();
                }, timeout);
            });
        };
        this.client = new Twitter({
            consumer_key: opt.consumer_key,
            consumer_secret: opt.consumer_secret,
            access_token_key: opt.access_token_key,
            access_token_secret: opt.access_token_secret,
        });
        this.lockfile = opt.lockfile;
        this.lock = opt.lock;
        this.workInterval = opt.workInterval;
        this.bot = opt.bot;
        this.webshotDelay = opt.webshotDelay;
        this.redisConfig = opt.redis;
        this.mode = opt.mode;
    }
}
exports.default = default_1;