gocqhttp-twitter-bot/dist/twitter.js

"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.sendAllStories = exports.sendStory = exports.sendTimeline = exports.ScreenNameNormalizer = exports.SessionManager = exports.parseLink = exports.linkBuilder = void 0;
const crypto = require("crypto");
const fs = require("fs");
const http = require("http");
const path = require("path");
const url_1 = require("url");
const util_1 = require("util");
const instagram_private_api_1 = require("instagram-private-api");
const socks_proxy_agent_1 = require("socks-proxy-agent");
const datetime_1 = require("./datetime");
const loggers_1 = require("./loggers");
const utils_1 = require("./utils");
const webshot_1 = require("./webshot");
const parseLink = (link) => {
    let match = /instagram\.com\/stories\/([^\/?#]+)\/(\d+)/.exec(link);
    if (match)
        return { userName: ScreenNameNormalizer.normalize(match[1]).split(':')[0], storyId: match[2] };
    match =
        /instagram\.com\/([^\/?#]+)/.exec(link) ||
            /^([^\/?#]+)$/.exec(link);
    if (match)
        return { userName: ScreenNameNormalizer.normalize(match[1]).split(':')[0] };
    return;
};
exports.parseLink = parseLink;
const linkBuilder = (config) => {
    if (!config.userName)
        return;
    if (!config.storyId)
        return `https://www.instagram.com/${config.userName}/`;
    return `https://www.instagram.com/stories/${config.userName}/${config.storyId}/`;
};
exports.linkBuilder = linkBuilder;
const igErrorIsAuthError = (error) => / 401/.test(error.message) || error instanceof instagram_private_api_1.IgLoginRequiredError || error instanceof instagram_private_api_1.IgCookieNotFoundError;
class SessionManager {
    constructor(client, file, credentials, codeServicePort) {
        this.init = () => {
            this.ig.state.generateDevice(this.username);
            this.ig.request.end$.subscribe(() => { this.save(); });
            const filePath = path.resolve(this.lockfile);
            if (fs.existsSync(filePath)) {
                try {
                    const serialized = JSON.parse(fs.readFileSync(filePath, 'utf8'));
                    return this.ig.state.deserialize(serialized).then(() => {
                        logger.info(`successfully loaded client session cookies for user ${this.username}`);
                    });
                }
                catch (err) {
                    logger.error(`failed to load client session cookies from file ${this.lockfile}: `, err);
                    return Promise.resolve();
                }
            }
            else {
                return this.login().catch((err) => {
                    logger.error(`error while trying to log in as user ${this.username}, error: ${err}`);
                    logger.warn('attempting to retry after 1 minute...');
                    if (fs.existsSync(filePath))
                        fs.unlinkSync(filePath);
                    (0, util_1.promisify)(setTimeout)(60000).then(this.init);
                });
            }
        };
        this.handle2FA = (submitter) => new Promise((resolve, reject) => {
            const token = crypto.randomBytes(20).toString('hex');
            logger.info('please submit the code with a one-time token from your browser with this path:');
            logger.info(`/confirm-2fa?code=<the code you received>&token=${token}`);
            let working;
            const server = http.createServer((req, res) => {
                const { pathname, query } = (0, url_1.parse)(req.url, true);
                if (!working && pathname === '/confirm-2fa' && query.token === token &&
                    typeof (query.code) === 'string' && /^\d{6}$/.test(query.code)) {
                    const code = query.code;
                    logger.debug(`received code: ${code}`);
                    working = true;
                    submitter(code)
                        .then(response => { res.write('OK'); res.end(); server.close(() => resolve(response)); })
                        .catch(err => { res.write('Error'); res.end(); reject(err); })
                        .finally(() => { working = false; });
                }
            });
            server.listen(this.codeServicePort);
        });
        this.login = () => this.ig.simulate.preLoginFlow()
            .then(() => this.ig.account.login(this.username, this.password))
            .catch((err) => {
            if (err instanceof instagram_private_api_1.IgLoginTwoFactorRequiredError) {
                const { two_factor_identifier, totp_two_factor_on } = err.response.body.two_factor_info;
                logger.debug(`2FA info: ${JSON.stringify(err.response.body.two_factor_info)}`);
                logger.info(`login is requesting two-factor authentication via ${totp_two_factor_on ? 'TOTP' : 'SMS'}`);
                return this.handle2FA(code => this.ig.account.twoFactorLogin({
                    username: this.username,
                    verificationCode: code,
                    twoFactorIdentifier: two_factor_identifier,
                    verificationMethod: totp_two_factor_on ? '0' : '1',
                }));
            }
            throw err;
        })
            .then(user => {
            logger.info(`successfully logged in as ${this.username}`);
            return user;
        });
        this.save = () => this.ig.state.serialize()
            .then((serialized) => {
            delete serialized.constants;
            return fs.writeFileSync(path.resolve(this.lockfile), JSON.stringify(serialized, null, 2), 'utf-8');
        });
        this.ig = client;
        this.lockfile = file;
        [this.username, this.password] = credentials;
        this.codeServicePort = codeServicePort;
    }
}
exports.SessionManager = SessionManager;
class ScreenNameNormalizer {
    static normalizeLive(username) {
        return __awaiter(this, void 0, void 0, function* () {
            if (this._queryUser) {
                return yield this._queryUser(username)
                    .catch((err) => {
                    if (!(err instanceof instagram_private_api_1.IgExactUserNotFoundError)) {
                        logger.warn(`error looking up user: ${err.message}`);
                        return `${username}:`;
                    }
                    return null;
                });
            }
            return this.normalize(username);
        });
    }
}
exports.ScreenNameNormalizer = ScreenNameNormalizer;
ScreenNameNormalizer.normalize = (username) => `${username.toLowerCase().replace(/^@/, '')}:`;
let sendTimeline = (username, receiver) => {
    throw Error();
};
exports.sendTimeline = sendTimeline;
let sendStory = (username, storyId, receiver) => {
    throw Error();
};
exports.sendStory = sendStory;
let sendAllStories = (username, receiver, startIndex, count) => {
    throw Error();
};
exports.sendAllStories = sendAllStories;
const logger = (0, loggers_1.getLogger)('instagram');
const maxTrials = 3;
const retryInterval = 1500;
const ordinal = (n) => {
    switch ((Math.trunc(n / 10) % 10 === 1) ? 0 : n % 10) {
        case 1:
            return `${n}st`;
        case 2:
            return `${n}nd`;
        case 3:
            return `${n}rd`;
        default:
            return `${n}th`;
    }
};
const retryOnError = (doWork, onRetry) => new Promise(resolve => {
    const retry = (reason, count) => {
        setTimeout(() => {
            let terminate = false;
            onRetry(reason, count, defaultValue => { terminate = true; resolve(defaultValue); });
            if (!terminate)
                doWork().then(resolve).catch(error => retry(error, count + 1));
        }, retryInterval);
    };
    doWork().then(resolve).catch(error => retry(error, 1));
});
class default_1 {
    constructor(opt) {
        this.launch = () => {
            this.webshot = new webshot_1.default(this.wsUrl, this.mode, () => {
                const subscribedIds = this.lock.feed.map(feed => this.lock.threads[feed].id.toString());
                for (const id in this.cache) {
                    if (this.cache[id].pullOrder !== 0 && !subscribedIds.includes(id)) {
                        logger.warn(`disabling pull job of unsubscribed user @${this.cache[id].user.username}`);
                        this.cache[id].pullOrder = 0;
                    }
                }
                const userIdCache = this.pullOrders;
                if (Object.values(userIdCache).length !== userIdCache.length) {
                    this.pullOrders = utils_1.Arr.shuffle(userIdCache);
                    fs.writeFileSync(path.resolve(this.cachefile), JSON.stringify(this.cache));
                }
                const timeout = Math.max(1000, this.workInterval * 1000 / this.lock.feed.length);
                setInterval(() => { this.pullOrders = utils_1.Arr.shuffle(this.pullOrders); }, 21600000);
                setTimeout(this.workForAll, timeout);
                this.work();
            });
        };
        this.queryUserObject = (userName) => this.client.user.searchExact(userName)
            .catch((error) => {
            if (igErrorIsAuthError(error)) {
                logger.warn('login required, logging in again...');
                return this.session.login().then(() => this.client.user.searchExact(userName));
            }
            else
                throw error;
        });
        this.queryUser = (rawUserName) => {
            const username = ScreenNameNormalizer.normalize(rawUserName).split(':')[0];
            for (const { user } of Object.values(this.cache)) {
                if (user.username === username)
                    return Promise.resolve(`${username}:${user.pk}`);
            }
            return this.queryUserObject(username)
                .then(({ pk, username, full_name }) => {
                this.cache[pk] = { user: { pk, username, full_name }, stories: {}, pullOrder: 0 };
                fs.writeFileSync(path.resolve(this.cachefile), JSON.stringify(this.cache));
                logger.info(`initialized cache item for user ${full_name} (@${username})`);
                return `${username}:${pk}`;
            });
        };
        this.workOnMedia = (mediaItems, sendMedia) => (0, utils_1.chainPromises)(mediaItems.map(({ msgs, text, author, original }) => {
            const findFilePath = (mediaMsg) => /=file:\/\/(.*?)]/.exec(mediaMsg)[1];
            const filePath = findFilePath(msgs);
            return () => (fs.existsSync(filePath) ?
                Promise.resolve() :
                this.webshot.fetchBestCandidate(original).then(mediaMsg => {
                    logger.warn(`media file missing, refetched media for ${author}/${original.code}`);
                    return (0, util_1.promisify)(fs.rename)(findFilePath(mediaMsg), filePath);
                })).then(() => sendMedia(msgs, text, author));
        }));
        this.sendStories = (source, ...to) => (msg, text, author) => {
            to.forEach(subscriber => {
                logger.info(`pushing data${source ? ` of ${source}` : ''} to ${JSON.stringify(subscriber)}`);
                retryOnError(() => this.bot.sendTo(subscriber, msg), (_, count, terminate) => {
                    if (count <= maxTrials) {
                        logger.warn(`retry sending to ${subscriber.chatID} for the ${ordinal(count)} time...`);
                    }
                    else {
                        logger.warn(`${count - 1} consecutive failures while sending message chain, trying plain text instead...`);
                        terminate(this.bot.sendTo(subscriber, author + text, true));
                    }
                });
            });
        };
        this.workForAll = () => {
            const timeout = Math.max(1000, this.workInterval * 1000 / this.lock.feed.length);
            if (this.isInactiveTime) {
                setTimeout(this.workForAll, timeout);
                return;
            }
            logger.debug(`current cache: ${JSON.stringify(this.cache)}`);
            (0, utils_1.chainPromises)(Object.entries(this.lock.threads).map(([feed, thread]) => () => {
                const id = thread.id;
                const userName = parseLink(feed).userName;
                logger.debug(`preparing to add user @${userName} to next pull task...`);
                if (id in this.cache) {
                    const item = this.cache[id];
                    if (item.pullOrder === 0)
                        item.pullOrder = -1;
                    return Promise.resolve();
                }
                return (0, util_1.promisify)(setTimeout)((Math.random() * 2 + 1) * 5000).then(() => this.client.user.info(id).then(({ pk, username, full_name }) => {
                    this.cache[id] = { user: { pk, username, full_name }, stories: {}, pullOrder: -1 };
                    fs.writeFileSync(path.resolve(this.cachefile), JSON.stringify(this.cache));
                    logger.info(`initialized cache item for user ${full_name} (@${username})`);
                }));
            }))
                .then(() => {
                const userIdCache = Object.values(this.cache).some(item => item.pullOrder < 0) ?
                    this.pullOrders = utils_1.Arr.shuffle(Object.keys(this.cache)).map(Number) :
                    this.pullOrders;
                return (0, utils_1.chainPromises)(utils_1.Arr.chunk(userIdCache, 20).map(userIds => () => {
                    logger.info(`pulling stories from users:${userIds.map(id => ` @${this.cache[id].user.username}`)}`);
                    return this.client.feed.reelsMedia({ userIds }).request()
                        .then(({ reels }) => (0, utils_1.chainPromises)(Object.keys(reels).map(userId => this.cache[userId]).map(({ user, stories }) => () => this.queryUserObject(user.username)
                        .catch((error) => {
                        if (error instanceof instagram_private_api_1.IgExactUserNotFoundError) {
                            return this.client.user.info(user.pk)
                                .then(({ username, full_name }) => {
                                user.username = username;
                                user.full_name = full_name;
                            });
                        }
                    }).finally(() => Promise.all(reels[user.pk].items
                        .filter(item => !(item.pk in stories))
                        .map(item => this.webshot([Object.assign(Object.assign({}, item), { user })], (msgs, text, author) => stories[item.pk] = { pk: item.pk, msgs, text, author, original: item }, this.webshotDelay)))))))
                        .finally(() => {
                        fs.writeFileSync(path.resolve(this.cachefile), JSON.stringify(this.cache));
                        Object.values(this.lock.threads).forEach(thread => {
                            if (userIds.includes(thread.id)) {
                                thread.updatedAt = this.cache[thread.id].updated = Date();
                            }
                        });
                    });
                }), (lp1, lp2) => () => lp1().then(() => (0, util_1.promisify)(setTimeout)(timeout).then(lp2)));
            })
                .catch((error) => {
                if (error instanceof instagram_private_api_1.IgNetworkError) {
                    if (error.cause.message === "Unexpected '<'") {
                        logger.warn('login required, logging in again...');
                        return this.session.login().then(this.workForAll);
                    }
                    logger.warn(`error while fetching stories for all: ${JSON.stringify(error.cause)}`);
                }
                else if (igErrorIsAuthError(error)) {
                    logger.warn('login required, logging in again...');
                    this.session.login().then(this.workForAll);
                }
                else {
                    logger.error(`unhandled error on fetching media for all: ${error}`);
                }
            })
                .then(() => {
                setTimeout(this.workForAll, this.workInterval * 1000);
            });
        };
        this.work = () => {
            const lock = this.lock;
            const timeout = Math.max(1000, this.workInterval * 1000 / lock.feed.length);
            const nextTurn = () => { setTimeout(this.work, timeout); return; };
            if (lock.feed.length === 0)
                return nextTurn();
            if (lock.workon >= lock.feed.length)
                lock.workon = 0;
            const currentFeed = lock.feed[lock.workon];
            if (!lock.threads[currentFeed] ||
                !lock.threads[currentFeed].subscribers ||
                lock.threads[currentFeed].subscribers.length === 0) {
                logger.warn(`nobody subscribes thread ${currentFeed}, removing from feed`);
                delete lock.threads[currentFeed];
                (this.cache[parseLink(currentFeed).userName] || {}).pullOrder = 0;
                fs.writeFileSync(path.resolve(this.cachefile), JSON.stringify(this.cache));
                lock.feed.splice(lock.workon, 1);
                fs.writeFileSync(path.resolve(this.lockfile), JSON.stringify(lock));
                return this.work();
            }
            logger.debug(`searching for new items from ${currentFeed} in cache`);
            const match = /https:\/\/www\.instagram\.com\/([^\/]+)/.exec(currentFeed);
            if (!match) {
                logger.error(`current feed "${currentFeed}" is invalid, please remove this feed manually`);
                lock.workon++;
                return nextTurn();
            }
            const cachedFeed = this.cache[lock.threads[currentFeed].id];
            if (!cachedFeed)
                return nextTurn();
            const newer = (item) => utils_1.BigNumOps.compare(item.pk, lock.threads[currentFeed].offset) > 0;
            const promise = Promise.resolve(Object.values(cachedFeed.stories)
                .filter(newer)
                .sort((i1, i2) => utils_1.BigNumOps.compare(i2.pk, i1.pk))
                .slice(-5));
            promise.then((mediaItems) => {
                const currentThread = lock.threads[currentFeed];
                if (!mediaItems || mediaItems.length === 0)
                    return;
                const updateOffset = () => currentThread.offset = mediaItems[0].pk;
                if (currentThread.offset === '-1') {
                    updateOffset();
                    return;
                }
                const questionIndex = mediaItems.findIndex(story => story.original.story_questions);
                if (questionIndex > 0)
                    mediaItems.splice(0, questionIndex);
                if (currentThread.offset === '0')
                    mediaItems.splice(1);
                return this.workOnMedia(mediaItems.slice(0).reverse(), this.sendStories(`thread ${currentFeed}`, ...currentThread.subscribers))
                    .then(updateOffset)
                    .then(() => {
                    if (questionIndex > -1) {
                        currentThread.subscribers.forEach(subscriber => {
                            const username = cachedFeed.user.username;
                            const author = `${cachedFeed.user.full_name} (@${username}) `;
                            this.bot.sendTo(subscriber, `请注意，用户${author}已开启问答互动。需退订请回复：/igstory_unsub ${username}${(questionIndex > 0) ? `\n本次推送已截止于此条动态，下次推送在 ${this.workInterval} 秒后。` : ''}`);
                        });
                    }
                });
            })
                .then(() => {
                lock.workon++;
                fs.writeFileSync(path.resolve(this.lockfile), JSON.stringify(lock));
                nextTurn();
            });
        };
        this.client = new instagram_private_api_1.IgApiClient();
        if (opt.proxyUrl) {
            try {
                const url = new URL(opt.proxyUrl);
                if (!/^socks(?:4a?|5h?)?:$/.test(url.protocol))
                    throw Error();
                if (!url.port)
                    url.port = '1080';
                this.client.request.defaults.agent = new socks_proxy_agent_1.SocksProxyAgent({
                    hostname: url.hostname,
                    port: url.port,
                    userId: url.username,
                    password: url.password,
                });
            }
            catch (e) {
                logger.warn(`invalid socks proxy url: ${opt.proxyUrl}, ignoring`);
            }
        }
        this.session = new SessionManager(this.client, opt.sessionLockfile, opt.credentials, opt.codeServicePort);
        this.lockfile = opt.lockfile;
        this.lock = opt.lock;
        this.cachefile = opt.cachefile;
        this.cache = opt.cache;
        this.inactiveHours = opt.inactiveHours;
        this.workInterval = opt.workInterval;
        this.bot = opt.bot;
        this.webshotDelay = opt.webshotDelay;
        this.mode = opt.mode;
        this.wsUrl = opt.wsUrl;
        const workNow = (config) => {
            const { action, retryAction, reply, rawUserName } = config;
            return this.queryUser(rawUserName)
                .then(userNameId => {
                const userId = userNameId.split(':')[1];
                const lastUpdated = new Date((this.cache[userId] || {}).updated || 0).getTime();
                const storyCount = lastUpdated && Object.keys(this.cache[userId].stories || {}).length;
                if (new Date().getTime() - lastUpdated > this.workInterval * 1000 && storyCount)
                    return userId;
                return this.client.feed.reelsMedia({ userIds: [userId] }).items()
                    .then(storyItems => Promise.all(storyItems
                    .filter(item => !(item.pk in this.cache[userId].stories))
                    .map(item => this.webshot([Object.assign(Object.assign({}, item), { user: this.cache[userId].user })], (msgs, text, author) => this.cache[userId].stories[item.pk] = { pk: item.pk, msgs, text, author, original: item }, this.webshotDelay))).then(() => userId).finally(() => this.cache[userId].updated = Date()));
            })
                .then(action)
                .catch((error) => {
                if (error instanceof instagram_private_api_1.IgExactUserNotFoundError) {
                    reply(`找不到用户 ${rawUserName.replace(/^@?(.*)$/, '@$1')}。`);
                }
                else if (error instanceof instagram_private_api_1.IgNetworkError) {
                    if (error.cause.message === "Unexpected '<'") {
                        logger.warn('login required, logging in again...');
                        return this.session.login().then(retryAction);
                    }
                    logger.warn(`error while fetching stories for ${rawUserName}: ${JSON.stringify(error.cause)}`);
                    reply(`获取 Stories 时出现错误：原因： ${error.cause}`);
                }
                else if (igErrorIsAuthError(error)) {
                    logger.warn('login required, logging in again...');
                    reply('等待登录中，稍后会处理请求，请稍候……');
                    this.session.login().then(retryAction);
                }
                else {
                    logger.error(`unhandled error while fetching stories for ${rawUserName}: ${error}`);
                    reply(`获取 Stories 时发生未知错误： ${error}`);
                }
            });
        };
        ScreenNameNormalizer._queryUser = this.queryUser;
        exports.sendTimeline = (rawUserName, receiver) => {
            const reply = msg => this.bot.sendTo(receiver, msg);
            workNow({
                rawUserName,
                action: userId => {
                    const userName = this.cache[userId].user.username;
                    const storyItems = Object.values(this.cache[userId].stories)
                        .sort((i1, i2) => -utils_1.BigNumOps.compare(i2.pk, i1.pk));
                    if (storyItems.length === 0)
                        return reply(`当前用户 (@${userName}) 没有可用的 Instagram 限时动态。`);
                    return reply('#. 编号：发送时间\n' + storyItems.map(({ original }, index) => `\n${index + 1}. ${original.pk}: ${(0, datetime_1.relativeDate)(original.taken_at * 1000)}`).join(''))
                        .then(() => reply(`请使用 /igstory_view ${userName} skip=<#-1> count=1
或 /igstory_view https://www.instagram.com/stories/${userName}/<编号>/
查看指定的限时动态。`));
                },
                reply,
                retryAction: () => (0, exports.sendTimeline)(rawUserName, receiver),
            });
        };
        exports.sendStory = (rawUserName, storyId, receiver) => {
            const reply = msg => this.bot.sendTo(receiver, msg);
            const sender = this.sendStories(`instagram stories for ${rawUserName}`, receiver);
            workNow({
                rawUserName,
                action: userId => {
                    if (!(storyId in this.cache[userId].stories))
                        return reply('此动态不存在或已过期。');
                    return this.workOnMedia([this.cache[userId].stories[storyId]], sender);
                },
                reply,
                retryAction: () => (0, exports.sendStory)(rawUserName, storyId, receiver),
            });
        };
        exports.sendAllStories = (rawUserName, receiver, startIndex = 0, count = 10) => {
            const reply = msg => this.bot.sendTo(receiver, msg);
            if (startIndex < 0)
                return reply('跳过数量参数值应为非负整数。');
            if (count < 1)
                return reply('最大查看数量参数值应为正整数。');
            const sender = this.sendStories(`instagram stories for ${rawUserName}`, receiver);
            workNow({
                rawUserName,
                action: userId => {
                    const userName = this.cache[userId].user.username;
                    const storyItems = Object.values(this.cache[userId].stories)
                        .sort((i1, i2) => -utils_1.BigNumOps.compare(i2.pk, i1.pk));
                    if (storyItems.length === 0)
                        return reply(`当前用户 (@${userName}) 没有可用的 Instagram 限时动态。`);
                    if (startIndex + 1 > storyItems.length)
                        return reply('跳过数量到达或超过当前用户可用的限时动态数量。');
                    const endIndex = Math.min(storyItems.length, startIndex + count);
                    const sendRangeText = `${startIndex + 1}${endIndex - startIndex > 1 ? `-${endIndex}` : ''}`;
                    return this.workOnMedia(storyItems.slice(startIndex, endIndex), sender)
                        .then(() => reply(`已显示当前用户 ${storyItems.length} 条可用限时动态中的第 ${sendRangeText} 条。`));
                },
                reply,
                retryAction: () => (0, exports.sendAllStories)(rawUserName, receiver, startIndex, count)
            });
        };
    }
    get pullOrders() {
        const arr = [];
        Object.values(this.cache).forEach(item => {
            if (item.pullOrder > 0)
                arr[item.pullOrder - 1] = item.user.pk;
        });
        return arr;
    }
    ;
    set pullOrders(arr) {
        Object.values(this.cache).forEach(item => {
            item.pullOrder = arr.indexOf(item.user.pk) + 1;
        });
    }
    get isInactiveTime() {
        const timeToEpoch = (h = 0, m = 0) => new Date().setHours(h, m, 0, 0);
        return this.inactiveHours
            .map(rangeStr => ((start, end) => ({ start, end }))(...rangeStr
            .split('-', 2)
            .map(timeStr => timeToEpoch(...timeStr.split(':', 2)
            .map(Number)))))
            .some(range => {
            const now = new Date().getTime();
            return now >= range.start && now < range.end;
        });
    }
}
exports.default = default_1;