gocqhttp-twitter-bot/dist/command.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.query = exports.view = exports.unsubAll = exports.unsub = exports.list = exports.sub = exports.parseCmd = void 0;
const fs = require("fs");
const path = require("path");
const datetime_1 = require("./datetime");
const loggers_1 = require("./loggers");
const twitter_1 = require("./twitter");
const logger = (0, loggers_1.getLogger)('command');
function parseCmd(message) {
    message = message.trim();
    message = message.replace('\\\\', '\\0x5c');
    message = message.replace('\\\"', '\\0x22');
    message = message.replace('\\\'', '\\0x27');
    const strs = message.match(/'[\s\S]*?'|(?:\S+=)?"[\s\S]*?"|\S+/mg);
    const cmd = (strs === null || strs === void 0 ? void 0 : strs.length) ? strs[0].length ? strs[0].substring(0, 1) === '/' ? strs[0].substring(1) : '' : '' : '';
    const args = (strs !== null && strs !== void 0 ? strs : []).slice(1).map(arg => {
        arg = arg.replace(/^(\S+=)?["']+(?!.*=)|["']+$/g, '$1');
        arg = arg.replace('\\0x27', '\\\'');
        arg = arg.replace('\\0x22', '\\\"');
        arg = arg.replace('\\0x5c', '\\\\');
        return arg;
    });
    return {
        cmd,
        args,
    };
}
exports.parseCmd = parseCmd;
function linkFinder(userName, chat, lock) {
    const normalizedLink = (0, twitter_1.linkBuilder)({ userName });
    const link = Object.keys(lock.threads).find(realLink => normalizedLink === realLink.replace(/\/@/, '/').toLowerCase());
    if (!link)
        return [null, -1];
    const index = lock.threads[link].subscribers.findIndex(({ chatID, chatType }) => chat.chatID === chatID && chat.chatType === chatType);
    return [link, index];
}
function sub(chat, args, reply, lock, lockfile) {
    if (chat.chatType === "temp") {
        return reply('请先添加机器人为好友。');
    }
    if (args.length === 0) {
        return reply('找不到要订阅 Instagram 限时动态的链接。');
    }
    const matched = (0, twitter_1.parseLink)(args[0]);
    if (!matched) {
        return reply(`订阅链接格式错误：
示例：
https://www.instagram.com/tomoyo_kurosawa_/`);
    }
    const subscribeTo = (link, config = {}) => {
        const { id, msg = `已为此聊天订阅 ${link} 的 Instagram 限时动态` } = config;
        if (id) {
            lock.feed.push(link);
            lock.threads[link] = {
                id,
                offset: '0',
                subscribers: [],
                updatedAt: '',
            };
        }
        lock.threads[link].subscribers.push(chat);
        logger.warn(`chat ${JSON.stringify(chat)} has subscribed ${link}`);
        fs.writeFileSync(path.resolve(lockfile), JSON.stringify(lock));
        reply(msg);
    };
    const tryFindSub = (userName) => {
        const [realLink, index] = linkFinder(userName, chat, lock);
        if (index > -1) {
            reply('此聊天已订阅此链接的 Instagram 限时动态。');
            return true;
        }
        if (realLink) {
            subscribeTo(realLink);
            return true;
        }
        return false;
    };
    const newSub = (userName) => {
        const link = (0, twitter_1.linkBuilder)(matched);
        subscribeTo(link, { id: Number(userName.split(':')[1]) });
    };
    if (!tryFindSub(matched.userName)) {
        twitter_1.ScreenNameNormalizer.normalizeLive(matched.userName).then(userName => {
            if (!userName)
                return reply(`找不到用户 ${matched.userName.replace(/^@?(.*)$/, '@$1')}。`);
            else
                newSub(userName);
        });
    }
}
exports.sub = sub;
function unsubAll(chat, args, reply, lock, lockfile) {
    if (chat.chatType === "temp") {
        return reply('请先添加机器人为好友。');
    }
    Object.entries(lock.threads).forEach(([link, { subscribers }]) => {
        const index = subscribers.indexOf(chat);
        if (index === -1)
            return;
        subscribers.splice(index, 1);
        fs.writeFileSync(path.resolve(lockfile), JSON.stringify(lock));
        logger.warn(`chat ${JSON.stringify(chat)} has unsubscribed ${link}`);
    });
    return reply(`已为此聊天退订所有 Instagram 限时动态链接。`);
}
exports.unsubAll = unsubAll;
function unsub(chat, args, reply, lock, lockfile) {
    var _a;
    if (chat.chatType === "temp") {
        return reply('请先添加机器人为好友。');
    }
    if (args.length === 0) {
        return reply('找不到要退订 Instagram 限时动态的链接。');
    }
    const match = (_a = (0, twitter_1.parseLink)(args[0])) === null || _a === void 0 ? void 0 : _a.userName;
    if (!match) {
        return reply('链接格式有误。');
    }
    const [link, index] = linkFinder(match, chat, lock);
    if (index === -1)
        return list(chat, args, msg => reply('您没有订阅此链接的 Instagram 限时动态。\n' + msg), lock);
    else {
        lock.threads[link].subscribers.splice(index, 1);
        fs.writeFileSync(path.resolve(lockfile), JSON.stringify(lock));
        logger.warn(`chat ${JSON.stringify(chat)} has unsubscribed ${link}`);
        return reply(`已为此聊天退订 ${link} 的 Instagram 限时动态`);
    }
}
exports.unsub = unsub;
function list(chat, _, reply, lock) {
    if (chat.chatType === "temp") {
        return reply('请先添加机器人为好友。');
    }
    const links = [];
    Object.keys(lock.threads).forEach(key => {
        if (lock.threads[key].subscribers.find(({ chatID, chatType }) => chat.chatID === chatID && chat.chatType === chatType))
            links.push(`${key} ${(0, datetime_1.relativeDate)(lock.threads[key].updatedAt)}`);
    });
    return reply('此聊天中订阅的 Instagram 限时动态动态链接：\n' + links.join('\n'));
}
exports.list = list;
function view(chat, args, reply) {
    const promptOnError = (func) => (...args) => {
        try {
            func(...args);
        }
        catch (e) {
            reply('机器人尚未加载完毕，请稍后重试。');
        }
    };
    if (args.length === 0) {
        return reply('找不到要查看 Instagram 限时动态的链接。');
    }
    const match = (0, twitter_1.parseLink)(args[0]);
    if (!(match === null || match === void 0 ? void 0 : match.userName)) {
        return reply('链接格式有误。');
    }
    if (match === null || match === void 0 ? void 0 : match.storyId) {
        return promptOnError(twitter_1.sendStory)(match.userName, match.storyId, chat);
    }
    const conf = {};
    const confZH = {
        count: '最大查看数量',
        skip: '跳过数量',
    };
    for (const arg of args.slice(1)) {
        const optMatch = /^(count|skip)=(.*)/.exec(arg);
        if (!optMatch)
            return reply(`未定义的查看参数：${arg}。`);
        const optKey = optMatch[1];
        if (optMatch.length === 1 || !/^-?\d*$/.test(optMatch[2]))
            return reply(`${confZH[optKey]}参数应为数值。`);
        if (optMatch[2] === '')
            return reply(`${confZH[optKey]}参数值不可为空。`);
        conf[optKey] = Number(optMatch[2]);
    }
    promptOnError(twitter_1.sendAllStories)(match.userName, chat, conf.skip, conf.count);
}
exports.view = view;
function query(chat, args, reply) {
    var _a;
    if (args.length === 0) {
        return reply('找不到要查询 Instagram 限时动态的用户。');
    }
    const match = (_a = (0, twitter_1.parseLink)(args[0])) === null || _a === void 0 ? void 0 : _a.userName;
    if (!match) {
        return reply('链接格式有误。');
    }
    try {
        (0, twitter_1.sendTimeline)(match, chat);
    }
    catch (e) {
        logger.error(`error querying timeline, error: ${e}`);
        reply('机器人尚未加载完毕，请稍后重试。');
    }
}
exports.query = query;