gocqhttp-twitter-bot
Version:
GoCQHTTP Twitter Bot
255 lines (254 loc) • 14 kB
JavaScript
;
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),
toString: () => session.userId,
},
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, noErrors = false) => 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 => {
logger.error(exports.Message.ellipseBase64(`error pushing data to ${JSON.stringify(subscriber.chatID)}, reason: ${reason}`));
if (!noErrors)
throw reason instanceof Error ? reason : 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;
})) {
return session.bot.handleFriendRequest(session.messageId, true)
.then(() => { logger.info(`accepted friend request from ${userString} (from group ${groupString})`); })
.catch(error => { logger.error(`error accepting friend request from ${userString}, error: ${error}`); });
}
utils_1.chainPromises(groupList.map(groupItem => (done) => Promise.resolve(done ||
this.bot.getGroupMember(groupItem.groupId, session.userId).then(() => {
groupString = `${groupItem.groupName}(${groupItem.groupId})`;
return session.bot.handleFriendRequest(session.messageId, true)
.then(() => { logger.info(`accepted friend request from ${userString} (found in group ${groupString})`); })
.catch(error => { logger.error(`error accepting friend request from ${userString}, error: ${error}`); })
.then(() => 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)) {
return session.bot.handleGroupRequest(session.messageId, true)
.then(() => { logger.info(`accepted group invitation from ${userString} (friend)`); })
.catch(error => { logger.error(`error accepting group invitation from ${userString}, error: ${error}`); });
}
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* () {
const userString = `${session.username}(${session.userId})`;
return (chat.chatType === "group" ? this.sendToGroup : this.sendToUser)(chat.chatID.toString(), msg)
.catch(error => { logger.error(`error replying to message from ${userString}, error: ${error}`); });
});
switch (cmdObj.cmd) {
case 'twitter_view':
case 'twitter_get':
command_1.view(chat, cmdObj.args, reply);
break;
case 'twitter_query':
case 'twitter_gettimeline':
command_1.query(chat, cmdObj.args, reply);
break;
case 'twitter_sub':
case 'twitter_subscribe':
this.botInfo.sub(chat, cmdObj.args, reply);
break;
case 'twitter_unsub':
case 'twitter_unsubscribe':
this.botInfo.unsub(chat, cmdObj.args, reply);
break;
case 'ping':
case 'twitter':
this.botInfo.list(chat, cmdObj.args, reply);
break;
case 'help':
if (cmdObj.args.length === 0) {
reply(`推特搬运机器人：
/twitter - 查询当前聊天中的推文订阅
/twitter_sub[scribe]〈链接|用户名〉- 订阅 Twitter 推文搬运
/twitter_unsub[scribe]〈链接|用户名〉- 退订 Twitter 推文搬运
/twitter_view〈链接〉- 查看推文
/twitter_query〈链接|用户名〉[参数列表...] - 查询时间线（详见 /help twitter_query）\
${chat.chatType === "temp" ?
'\n（当前游客模式下无法使用订阅功能，请先添加本账号为好友。）' : ''}`);
}
else if (cmdObj.args[0] === 'twitter_query') {
reply(`查询时间线中的推文：
/twitter_query〈链接|用户名〉[〈参数 1〉=〈值 1〉〈参数 2〉=〈值 2〉...]
参数列表（方框内全部为可选，留空则为默认）：
count：查询数量上限（类型：非零整数，最大值正负 50）[默认值：10]
since：查询起始点（类型：正整数或日期）[默认值：（空，无限过去）]
until：查询结束点（类型：正整数或日期）[默认值：（空，当前时刻）]
noreps 忽略回复推文（类型：on/off）[默认值：on（是）]
norts：忽略原生转推（类型：on/off）[默认值：off（否）]`)
.then(() => reply(`\
起始点和结束点为正整数时取推特推文编号作为比较基准，否则会尝试作为日期读取。
推荐的日期格式：2012-12-22 12:22 UTC+2 （日期和时间均为可选，可分别添加）
count 为正时，从新向旧查询；为负时，从旧向新查询
count 与 since/until 并用时，取二者中实际查询结果较少者
例子：/twitter_query RiccaTachibana count=5 since="2019-12-30\
UTC+9" until="2020-01-06 UTC+8" norts=on
从起始时间点（含）到结束时间点（不含）从新到旧获取最多 5 条推文，\
其中不包含原生转推（实际上用户只发了 1 条）`));
}
}
}), 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;
//# sourceMappingURL=koishi.js.map