gocqhttp-twitter-bot/dist/mirai.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.Message = void 0;
const axios_1 = require("axios");
const fs_1 = require("fs");
const mirai_ts_1 = require("mirai-ts");
const message_1 = require("mirai-ts/dist/message");
const temp = require("temp");
const command_1 = require("./command");
const loggers_1 = require("./loggers");
const logger = loggers_1.getLogger('qqbot');
exports.Message = message_1.default;
class default_1 {
    constructor(opt) {
        this.getChat = (msg) => __awaiter(this, void 0, void 0, function* () {
            switch (msg.type) {
                case 'FriendMessage':
                    return {
                        chatID: msg.sender.id,
                        chatType: "private" /* Private */,
                    };
                case 'GroupMessage':
                    return {
                        chatID: msg.sender.group.id,
                        chatType: "group" /* Group */,
                    };
                case 'TempMessage':
                    const friendList = yield this.bot.api.friendList();
                    // already befriended
                    if (friendList.some(friendItem => friendItem.id === msg.sender.id)) {
                        return {
                            chatID: msg.sender.id,
                            chatType: "private" /* Private */,
                        };
                    }
                    return {
                        chatID: {
                            qq: msg.sender.id,
                            group: msg.sender.group.id,
                        },
                        chatType: "temp" /* Temp */,
                    };
            }
        });
        this.sendTo = (subscriber, msg) => (() => {
            switch (subscriber.chatType) {
                case 'group':
                    return this.bot.api.sendGroupMessage(msg, subscriber.chatID);
                case 'private':
                    return this.bot.api.sendFriendMessage(msg, subscriber.chatID);
                // currently disabled
                case 'temp':
                    return this.bot.api.sendTempMessage(msg, subscriber.chatID.qq, subscriber.chatID.group);
            }
        })()
            .then(response => {
            logger.info(`pushing data to ${JSON.stringify(subscriber.chatID)} was successful, response:`);
            logger.info(response);
        })
            .catch(reason => {
            logger.error(`error pushing data to ${JSON.stringify(subscriber.chatID)}, reason: ${reason}`);
            throw Error(reason);
        });
        this.uploadPic = (img, timeout = -1) => {
            if (timeout)
                timeout = Math.floor(timeout);
            if (timeout === 0 || timeout < -1) {
                return Promise.reject('Error: timeout must be greater than 0ms');
            }
            let imgFile;
            if (img.imageId !== '')
                return Promise.resolve();
            if (img.url !== '') {
                if (img.url.split(':')[0] !== 'data') {
                    return Promise.reject('Error: URL must be of protocol "data"');
                }
                if (img.url.split(',')[0].split(';')[1] !== 'base64') {
                    return Promise.reject('Error: data URL must be of encoding "base64"');
                }
                temp.track();
                try {
                    const tempFile = temp.openSync();
                    fs_1.writeSync(tempFile.fd, Buffer.from(img.url.split(',')[1], 'base64'));
                    fs_1.closeSync(tempFile.fd);
                    imgFile = tempFile.path;
                }
                catch (error) {
                    logger.error(error);
                }
            }
            try {
                this.bot.axios.defaults.timeout = timeout === -1 ? 0 : timeout;
                logger.info(`uploading ${JSON.stringify(exports.Message.Image(img.imageId, `${img.url.split(',')[0]},[...]`, img.path))}...`);
                return this.bot.api.uploadImage('group', imgFile || img.path)
                    .then(response => {
                    logger.info(`uploading ${img.path} as group image was successful, response:`);
                    logger.info(JSON.stringify(response));
                    img.url = '';
                    img.path = response.path.split(/[/\\]/).slice(-1)[0];
                })
                    .catch(reason => {
                    logger.error(`error uploading ${img.path}, reason: ${reason}`);
                    throw Error(reason);
                });
            }
            finally {
                temp.cleanup();
                this.bot.axios.defaults.timeout = 0;
            }
        };
        this.initBot = () => {
            this.bot = new mirai_ts_1.default({
                authKey: this.botInfo.access_token,
                enableWebsocket: false,
                host: this.botInfo.host,
                port: this.botInfo.port,
            });
            this.bot.axios.defaults.maxContentLength = this.bot.axios.defaults.maxBodyLength = Infinity;
            this.bot.on('NewFriendRequestEvent', evt => {
                logger.debug(`detected new friend request event: ${JSON.stringify(evt)}`);
                this.bot.api.groupList()
                    .then((groupList) => {
                    if (groupList.some(groupItem => groupItem.id === evt.groupId)) {
                        evt.respond(0); // allow
                        return logger.info(`accepted friend request from ${evt.fromId} (from group ${evt.groupId})`);
                    }
                    logger.warn(`received friend request from ${evt.fromId} (from group ${evt.groupId})`);
                    logger.warn('please manually accept this friend request');
                });
            });
            this.bot.on('BotInvitedJoinGroupRequestEvent', evt => {
                logger.debug(`detected group invitation event: ${JSON.stringify(evt)}`);
                this.bot.api.friendList()
                    .then((friendList) => {
                    if (friendList.some(friendItem => friendItem.id = evt.fromId)) {
                        evt.respond(0); // allow
                        return logger.info(`accepted group invitation from ${evt.fromId} (friend)`);
                    }
                    logger.warn(`received group invitation from ${evt.fromId} (unknown)`);
                    logger.warn('please manually accept this group invitation');
                });
            });
            this.bot.on('message', (msg) => __awaiter(this, void 0, void 0, function* () {
                const chat = yield this.getChat(msg);
                const cmdObj = command_1.parseCmd(msg.plain);
                switch (cmdObj.cmd) {
                    case 'twitterfleets_sub':
                    case 'twitterfleets_subscribe':
                        this.botInfo.sub(chat, cmdObj.args, msg.reply);
                        break;
                    case 'twitterfleets_unsub':
                    case 'twitterfleets_unsubscribe':
                        this.botInfo.unsub(chat, cmdObj.args, msg.reply);
                        break;
                    case 'ping':
                    case 'twitterfleets':
                        this.botInfo.list(chat, cmdObj.args, msg.reply);
                        break;
                    case 'help':
                        msg.reply(`推特故事搬运机器人：
/twitterfleets - 查询当前聊天中的推特故事订阅
/twitterfleets_subscribe [链接] - 订阅 Twitter Fleets 搬运
/twitterfleets_unsubscribe [链接] - 退订 Twitter Fleets 搬运`);
                }
            }));
        };
        // TODO doesn't work if connection is dropped after connection
        this.listen = (logMsg) => {
            if (logMsg !== '') {
                logger.warn(logMsg !== null && logMsg !== void 0 ? logMsg : 'Listening...');
            }
            axios_1.default.get(`http://${this.botInfo.host}:${this.botInfo.port}/about`)
                .then(() => __awaiter(this, void 0, void 0, function* () {
                if (logMsg !== '') {
                    this.bot.listen();
                    yield this.login();
                }
                setTimeout(() => this.listen(''), 5000);
            }))
                .catch(() => {
                logger.error(`Error connecting to bot provider at ${this.botInfo.host}:${this.botInfo.port}`);
                setTimeout(() => this.listen('Retry listening...'), 2500);
            });
        };
        this.login = (logMsg) => __awaiter(this, void 0, void 0, function* () {
            logger.warn(logMsg !== null && logMsg !== void 0 ? logMsg : 'Logging in...');
            yield this.bot.link(this.botInfo.bot_id)
                .then(() => logger.warn(`Logged in as ${this.botInfo.bot_id}`))
                .catch(() => {
                logger.error(`Cannot log in. Do you have a bot logged in as ${this.botInfo.bot_id}?`);
                setTimeout(() => this.login('Retry logging in...'), 2500);
            });
        });
        this.connect = () => {
            this.initBot();
            this.listen();
        };
        logger.warn(`Initialized mirai-ts for ${opt.host}:${opt.port} with access_token ${opt.access_token}`);
        this.botInfo = opt;
    }
}
exports.default = default_1;