"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(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)}`); const hash = sha1(JSON.stringify(subscriber) + text); if (this.redisClient && this.redisClient.exists(hash)) { logger.info('key hash exists, skip this subscriber'); return; } this.bot.bot('send_msg', { message_type: subscriber.chatType, user_id: subscriber.chatID, group_id: subscriber.chatID, discuss_id: subscriber.chatID, message: msg, }); if (this.redisClient) { this.redisClient.set(hash, 'true'); this.redisClient.expire(hash, this.redisConfig.redisExpireTime); } }); }, 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; } } exports.default = default_1;