"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 koishi_1 = require("koishi"); require("koishi-adapter-onebot"); const command_1 = require("./command"); const loggers_1 = require("./loggers"); const utils_1 = require("./utils"); const logger = loggers_1.getLogger('qqbot'); const cqUrlFix = (factory) => (...args) => factory(...args).replace(/(?<=\[CQ:.*)url=(?=(base64|file|https?):\/\/)/, 'file='); exports.Message = { Image: cqUrlFix(koishi_1.segment.image), Video: cqUrlFix(koishi_1.segment.video), Voice: cqUrlFix(koishi_1.segment.audio), ellipseBase64: (msg) => msg.replace(/(?<=\[CQ:.*base64:\/\/).*?(,|\])/g, '...$1'), separateAttachment: (msg) => { const attachments = []; const message = msg.replace(/\[CQ:(video|record),.*?\]/g, code => { attachments.push(code); return ''; }); return { message, attachments }; }, }; class default_1 { constructor(opt) { this.messageQueues = {}; this.tempSenders = {}; this.next = (type, id) => { const queue = this.messageQueues[`${type}:${id}`]; if (queue && queue.length) { queue[0]().then(() => { queue.shift(); if (!queue.length) delete this.messageQueues[`${type}:${id}`]; else this.next(type, id); }); } }; this.enqueue = (type, id, resolver) => { var _a, _b; let wasEmpty = false; const queue = (_a = this.messageQueues)[_b = `${type}:${id}`] || (_a[_b] = (() => { wasEmpty = true; return []; })()); queue.push(() => koishi_1.sleep(200).then(resolver)); logger.debug(`no. of message currently queued for ${type}:${id}: ${queue.length}`); if (wasEmpty) this.next(type, id); }; this.getChat = (session) => __awaiter(this, void 0, void 0, function* () { switch (session.subtype) { case 'private': if (session.sender.groupId) { const friendList = yield session.bot.getFriendList(); if (!friendList.some(friendItem => friendItem.userId === session.userId)) { this.tempSenders[session.userId] = session.sender.groupId; return { chatID: { qq: Number(session.userId), group: Number(session.sender.groupId), }, chatType: "temp", }; } } return { chatID: Number(session.userId), chatType: "private", }; case 'group': return { chatID: Number(session.groupId), chatType: "group", }; } }); this.sendToGroup = (groupID, message) => new Promise((resolve, reject) => { this.enqueue('group', groupID, () => this.bot.sendMessage(groupID, message).then(resolve).catch(reject)); }); this.sendToUser = (userID, message) => new Promise((resolve, reject) => { this.enqueue('private', userID, () => this.bot.sendPrivateMessage(userID, message).then(resolve).catch(reject)); }); this.sendTo = (subscriber, messageChain) => Promise.all((splitted => [splitted.message, ...splitted.attachments])(exports.Message.separateAttachment(messageChain)).map(msg => { switch (subscriber.chatType) { case 'group': return this.sendToGroup(subscriber.chatID.toString(), msg); case 'private': return this.sendToUser(subscriber.chatID.toString(), msg); case 'temp': return this.sendToUser(subscriber.chatID.qq.toString(), msg); } })) .then(response => { if (response === undefined) return; logger.info(`pushing data to ${JSON.stringify(subscriber.chatID)} was successful, response: ${response}`); }) .catch(reason => { reason = exports.Message.ellipseBase64(reason); logger.error(`error pushing data to ${JSON.stringify(subscriber.chatID)}, reason: ${reason}`); throw Error(reason); }); this.initBot = () => { this.app = new koishi_1.App({ type: 'onebot', server: `ws://${this.botInfo.host}:${this.botInfo.port}`, selfId: this.botInfo.bot_id.toString(), token: this.botInfo.access_token, axiosConfig: { maxContentLength: Infinity, }, processMessage: msg => msg.trim(), }); this.app.on('friend-request', (session) => __awaiter(this, void 0, void 0, function* () { const userString = `${session.username}(${session.userId})`; let groupId; let groupString; if (session.username in this.tempSenders) groupId = this.tempSenders[session.userId].toString(); logger.debug(`detected new friend request event: ${userString}`); return session.bot.getGroupList().then(groupList => { if (groupList.some(groupItem => { const test = groupItem.groupId === groupId; if (test) groupString = `${groupItem.groupName}(${groupId})`; return test; })) { session.bot.handleFriendRequest(session.messageId, true); return logger.info(`accepted friend request from ${userString} (from group ${groupString})`); } utils_1.chainPromises(groupList.map(groupItem => (done) => Promise.resolve(done || this.bot.getGroupMember(groupItem.groupId, session.userId).then(() => { groupString = `${groupItem.groupName}(${groupItem.groupId})`; session.bot.handleFriendRequest(session.messageId, true); logger.info(`accepted friend request from ${userString} (found in group ${groupString})`); return true; }).catch(() => false)))).then(done => { if (done) return; logger.warn(`received friend request from ${userString} (stranger)`); logger.warn('please manually accept this friend request'); }); }); })); this.app.on('group-request', (session) => __awaiter(this, void 0, void 0, function* () { const userString = `${session.username}(${session.userId})`; const groupString = `${session.groupName}(${session.groupId})`; logger.debug(`detected group invitation event: ${groupString}}`); return session.bot.getFriendList().then(friendList => { if (friendList.some(friendItem => friendItem.userId = session.userId)) { session.bot.handleGroupRequest(session.messageId, true); return logger.info(`accepted group invitation from ${userString} (friend)`); } logger.warn(`received group invitation from ${userString} (stranger)`); logger.warn('please manually accept this group invitation'); }); })); this.app.middleware((session) => __awaiter(this, void 0, void 0, function* () { const chat = yield this.getChat(session); const cmdObj = command_1.parseCmd(session.content); const reply = (msg) => __awaiter(this, void 0, void 0, function* () { return session.sendQueued(msg); }); switch (cmdObj.cmd) { case 'twitterfleets_view': case 'twitterfleets_get': command_1.view(chat, cmdObj.args, reply); break; case 'twitterfleets_sub': case 'twitterfleets_subscribe': this.botInfo.sub(chat, cmdObj.args, reply); break; case 'twitterfleets_unsub': case 'twitterfleets_unsubscribe': this.botInfo.unsub(chat, cmdObj.args, reply); break; case 'ping': case 'twitterfleets': this.botInfo.list(chat, cmdObj.args, reply); break; case 'help': if (cmdObj.args.length === 0) { reply(`推特故事搬运机器人： /twitterfleets - 查询当前聊天中的推特故事订阅 /twitterfleets_view〈链接〉- 查看该用户当前可见的所有 Fleets /twitterfleets_subscribe [链接] - 订阅 Twitter Fleets 搬运 /twitterfleets_unsubscribe [链接] - 退订 Twitter Fleets 搬运`); } } }), true); }; this.listen = (logMsg = 'connecting to bot provider...') => __awaiter(this, void 0, void 0, function* () { logger.warn(logMsg); try { yield this.app.start(); } catch (err) { logger.error(`error connecting to bot provider at ${this.app.options.server}, will retry in 2.5s...`); yield koishi_1.sleep(2500); yield this.listen('retry connecting...'); } }); this.connect = () => __awaiter(this, void 0, void 0, function* () { this.initBot(); yield this.listen(); this.bot = this.app.getBot('onebot'); }); logger.warn(`Initialized koishi on ${opt.host}:${opt.port} with access_token ${opt.access_token}`); this.botInfo = opt; } } exports.default = default_1;