249 lines
8.1 KiB
TypeScript
249 lines
8.1 KiB
TypeScript
import MiraiTs, { MiraiApiHttpConfig } from 'mirai-ts'
|
||
import { Mutex } from 'async-mutex'
|
||
import yargs from 'yargs'
|
||
import { readFileSync } from 'fs'
|
||
import { Member } from 'mirai-ts/dist/types/contact'
|
||
import { Plain } from 'mirai-ts/dist/types/message-type'
|
||
|
||
const args = yargs.option('config', {
|
||
alias: 'c',
|
||
description: 'path to the config json',
|
||
type: 'string'
|
||
}).help().alias('help', 'h').argv
|
||
|
||
type ExpectedType<T extends (arg: unknown) => unknown> =
|
||
T extends (arg: unknown) => arg is infer R ? R : never
|
||
type Definition = Record<string, (arg: unknown) => unknown>
|
||
export type FromDefinition<D extends Definition> = {
|
||
[K in keyof D]: ExpectedType<D[K]>
|
||
}
|
||
export const getTypeChecker = <D extends Definition>(d: D) => {
|
||
const keys = Object.keys(d) as (keyof Definition)[]
|
||
return (e: unknown): e is FromDefinition<D> => {
|
||
return typeof e === 'object' && keys.every(k => {
|
||
return d[k]((e as any)[k])
|
||
})
|
||
}
|
||
}
|
||
const configDefinition = {
|
||
qq: (x: unknown): x is number => typeof x === 'number',
|
||
mahConfig: (x: unknown): x is MiraiApiHttpConfig => typeof x === 'object',
|
||
groups: (x: unknown): x is number[] => Array.isArray(x) && x.every(e => typeof e === 'number'),
|
||
excludeQQ: (x: unknown): x is number[] => Array.isArray(x) && x.every(e => typeof e === 'number'),
|
||
}
|
||
const parseConfig = () => {
|
||
if (typeof args.config !== 'string') {
|
||
throw new Error('missing config path')
|
||
}
|
||
const parsed = JSON.parse(readFileSync(args.config).toString('utf-8'))
|
||
const configChecker = getTypeChecker(configDefinition)
|
||
if (!configChecker(parsed)) {
|
||
throw new Error('Invalid config')
|
||
}
|
||
return parsed
|
||
}
|
||
|
||
const { qq: botQQ, mahConfig, groups: groupNumbers, excludeQQ } = parseConfig();
|
||
const mirai = new MiraiTs(mahConfig)
|
||
|
||
type StoredMessage = {
|
||
author: number
|
||
id: number
|
||
isOriginal: boolean
|
||
}
|
||
|
||
class StoredMessages {
|
||
private threshold = 200
|
||
private data = [new Map<number, StoredMessage>(), new Map<number, StoredMessage>()]
|
||
private currentIndex = 0
|
||
private otherIndex = 1
|
||
private get current() { return this.data[this.currentIndex] }
|
||
private get other() { return this.data[this.otherIndex] }
|
||
|
||
public add(original: number, translated: StoredMessage) {
|
||
if (this.current.size > this.threshold) {
|
||
// 假如保存的消息过多，那么把另一个 buffer （老数据）给清空后拿来使用
|
||
[this.currentIndex, this.otherIndex] = [this.otherIndex, this.currentIndex]
|
||
this.current.clear()
|
||
}
|
||
const current = this.current
|
||
current.set(original, translated)
|
||
}
|
||
|
||
public translate(messageId: number) {
|
||
return this.current.get(messageId) || this.other.get(messageId)
|
||
}
|
||
}
|
||
|
||
async function app() {
|
||
// 登录 QQ
|
||
await mirai.link(botQQ);
|
||
|
||
const groups = groupNumbers.map(group => ({
|
||
group,
|
||
members: new Map<number, string>(),
|
||
stored: new StoredMessages(),
|
||
async updateMemberList() {
|
||
const list: Member[] = await mirai.api.memberList(this.group)
|
||
this.members = new Map(list.map(m => [m.id, m.memberName]))
|
||
}
|
||
}));
|
||
// 保存成员列表
|
||
await Promise.all(groups.map(g => g.updateMemberList()))
|
||
|
||
const mutex = new Mutex()
|
||
// 对收到的消息进行处理
|
||
// message 本质相当于同时绑定了 FriendMessage GroupMessage TempMessage
|
||
// 你也可以单独对某一类消息进行监听
|
||
mirai.on('GroupMessage', async msg => {
|
||
const fromGroup = msg.sender.group.id
|
||
if (!groups.find(({ group }) => group === fromGroup)) {
|
||
return
|
||
}
|
||
const messageId = msg.messageChain[0].id
|
||
const messageAuthor = msg.sender.id
|
||
const originalGroup = groups.find(({ group }) => group === fromGroup)
|
||
|
||
if (excludeQQ.includes(messageAuthor)) {
|
||
return
|
||
}
|
||
|
||
const releaseMutex = await mutex.acquire()
|
||
try {
|
||
const promises = groups
|
||
.filter(({ group }) => group !== fromGroup)
|
||
.map(async ({ group, stored, members }) => {
|
||
// 把 qq 转换成纯文本的时候，优先使用哪个群里的群名片
|
||
const qqToName = (qq: number) => {
|
||
const name = members.get(qq) || originalGroup?.members?.get(qq)
|
||
if (name !== undefined) {
|
||
return name
|
||
}
|
||
for (const { members } of groups) {
|
||
const found = members.get(qq)
|
||
if (found !== undefined) {
|
||
return found
|
||
}
|
||
}
|
||
}
|
||
|
||
let quote: StoredMessage | undefined
|
||
let atMeCounter = 0
|
||
const processed = msg.messageChain.filter(x => {
|
||
// 处理 @
|
||
if (x.type === 'At') {
|
||
// 避免转发回复时的 @
|
||
if (quote && x.target === botQQ) {
|
||
return atMeCounter++ === 0
|
||
}
|
||
return true
|
||
}
|
||
// 处理回复
|
||
if (x.type !== 'Quote') {
|
||
return true
|
||
}
|
||
quote = stored.translate(x.id) || quote
|
||
return false
|
||
}).map(x => {
|
||
// 处理 @
|
||
if (x.type !== 'At') {
|
||
return x
|
||
}
|
||
|
||
if (x.target === botQQ) {
|
||
x = { ...x, target: quote?.author || botQQ }
|
||
}
|
||
|
||
// 转发的消息不应该继续 @ 人，因为人可能并不在被转发的群
|
||
// 就算在，被好几个群同时 @，也很奇怪（
|
||
// 因此，除非是为了回复的 @，否则一律转成纯文本
|
||
// 就算确实是为了回复的 @，假如转发的群里这个人不在，那也转成纯文本
|
||
// 此外，假如是回复的 @，但是这并不是回复对象的那个群，那么也转成纯文本
|
||
if (x.target !== quote?.author ||
|
||
!members.has(x.target) ||
|
||
!quote.isOriginal) {
|
||
// 把 @ 转换成纯文本的时候，优先使用哪个群里的群名片
|
||
const name = qqToName(x.target) || x.display
|
||
return { type: 'Plain' as const, text: `@${name}` }
|
||
}
|
||
|
||
return x
|
||
})
|
||
|
||
const authorName = qqToName(messageAuthor)
|
||
if (authorName !== undefined) {
|
||
if (processed[1]?.type !== 'Plain') {
|
||
processed.splice(1, 0, { type: 'Plain', text: '' })
|
||
}
|
||
const firstPlain = processed[1] as Plain
|
||
firstPlain.text = `${authorName}：` + firstPlain.text
|
||
}
|
||
|
||
try {
|
||
const sent = await mirai.api.sendGroupMessage(processed, group, quote?.id)
|
||
return { sentId: sent.messageId, targetStorage: stored }
|
||
}
|
||
catch (e) {
|
||
console.warn(e)
|
||
}
|
||
})
|
||
|
||
const results = (await Promise.all(promises))
|
||
.filter(<T>(x?: T): x is T => x !== undefined)
|
||
for (const { sentId, targetStorage } of results) {
|
||
const others = results.filter(other => other.targetStorage !== targetStorage)
|
||
const forwarded: StoredMessage = {
|
||
author: messageAuthor,
|
||
id: sentId,
|
||
isOriginal: false
|
||
}
|
||
for (const other of others) {
|
||
targetStorage.add(other.sentId, forwarded)
|
||
}
|
||
targetStorage.add(messageId, forwarded)
|
||
originalGroup?.stored?.add(sentId, {
|
||
author: messageAuthor,
|
||
id: messageId,
|
||
isOriginal: true
|
||
})
|
||
}
|
||
}
|
||
finally {
|
||
releaseMutex()
|
||
}
|
||
})
|
||
|
||
// 处理各种事件类型
|
||
// 事件订阅说明（名称均与 mirai-api-http 中事件名一致）
|
||
// https://github.com/RedBeanN/node-mirai/blob/master/event.md
|
||
// console.log("on other event");
|
||
// https://github.com/project-mirai/mirai-api-http/blob/master/EventType.md#群消息撤回
|
||
mirai.on('GroupRecallEvent', async event => {
|
||
const releaseMutex = await mutex.acquire()
|
||
try {
|
||
for (const group of groups) {
|
||
if (group.group === event.group.id) {
|
||
continue
|
||
}
|
||
const matched = group.stored.translate(event.messageId)
|
||
if (!matched) {
|
||
continue
|
||
}
|
||
try {
|
||
await mirai.api.recall(matched.id)
|
||
}
|
||
catch (e) {
|
||
console.warn(e)
|
||
}
|
||
}
|
||
}
|
||
finally {
|
||
releaseMutex()
|
||
}
|
||
});
|
||
|
||
// 开始监听
|
||
mirai.listen();
|
||
}
|
||
|
||
app(); |