From 3f4b0b44cf84e27fe2d1b61854f14bdee7f76a5b Mon Sep 17 00:00:00 2001 From: idranme Date: Wed, 14 Aug 2024 23:04:15 +0800 Subject: [PATCH] feat: cache recalled message content --- src/common/config.ts | 1 + src/common/data.ts | 20 ++++++++++++++++++++ src/common/types.ts | 1 + src/main/main.ts | 4 +++- src/onebot11/action/msg/GetMsg.ts | 10 ++++------ src/onebot11/action/msg/SendMsg.ts | 7 +++---- src/onebot11/constructor.ts | 10 ++++------ src/renderer/index.ts | 5 +++++ 8 files changed, 41 insertions(+), 17 deletions(-) diff --git a/src/common/config.ts b/src/common/config.ts index 3ef929c..5fb5ee5 100644 --- a/src/common/config.ts +++ b/src/common/config.ts @@ -52,6 +52,7 @@ export class ConfigUtil { autoDeleteFile: false, autoDeleteFileSecond: 60, musicSignUrl: '', + msgCacheExpire: 8000 } if (!fs.existsSync(this.configPath)) { diff --git a/src/common/data.ts b/src/common/data.ts index ccd36cc..3f9770c 100644 --- a/src/common/data.ts +++ b/src/common/data.ts @@ -9,6 +9,8 @@ import { NTQQGroupApi } from '../ntqqapi/api/group' import { log } from './utils/log' import { isNumeric } from './utils/helper' import { NTQQFriendApi, NTQQUserApi } from '../ntqqapi/api' +import { RawMessage } from '../ntqqapi/types' +import { getConfigUtil } from './config' export let groups: Group[] = [] export let friends: Friend[] = [] @@ -128,3 +130,21 @@ export function getSelfUid() { export function getSelfUin() { return selfInfo['uin'] } + +const messages: Map = new Map() +let expire: number + +/** 缓存已撤回消息内容 */ +export function addMsgCache(msg: RawMessage) { + expire ??= getConfigUtil().getConfig().msgCacheExpire! + const id = msg.msgId + messages.set(id, msg) + setTimeout(() => { + messages.delete(id) + }, expire) +} + +/** 获取已撤回消息内容 */ +export function getMsgCache(msgId: string) { + return messages.get(msgId) +} \ No newline at end of file diff --git a/src/common/types.ts b/src/common/types.ts index d4c6986..4705dd0 100644 --- a/src/common/types.ts +++ b/src/common/types.ts @@ -30,6 +30,7 @@ export interface Config { ffmpeg?: string // ffmpeg路径 musicSignUrl?: string ignoreBeforeLoginMsg?: boolean + msgCacheExpire?: number } export interface LLOneBotError { diff --git a/src/main/main.ts b/src/main/main.ts index 5ba38f5..49868cd 100644 --- a/src/main/main.ts +++ b/src/main/main.ts @@ -21,7 +21,8 @@ import { setSelfInfo, getSelfInfo, getSelfUid, - getSelfUin + getSelfUin, + addMsgCache } from '../common/data' import { hookNTQQApiCall, hookNTQQApiReceive, ReceiveCmdS, registerReceiveHook, startHook } from '../ntqqapi/hook' import { OB11Constructor } from '../onebot11/constructor' @@ -222,6 +223,7 @@ function onLoad() { if (!oriMessageId) { continue } + addMsgCache(message) OB11Constructor.RecallEvent(message, oriMessageId).then((recallEvent) => { if (recallEvent) { //log('post recall event', recallEvent) diff --git a/src/onebot11/action/msg/GetMsg.ts b/src/onebot11/action/msg/GetMsg.ts index 5f53294..e3a0e2b 100644 --- a/src/onebot11/action/msg/GetMsg.ts +++ b/src/onebot11/action/msg/GetMsg.ts @@ -4,6 +4,7 @@ import BaseAction from '../BaseAction' import { ActionName } from '../types' import { NTQQMsgApi } from '@/ntqqapi/api' import { MessageUnique } from '@/common/utils/MessageUnique' +import { getMsgCache } from '@/common/data' export interface PayloadType { message_id: number | string @@ -29,12 +30,9 @@ class GetMsg extends BaseAction { peerUid: msgIdWithPeer.Peer.peerUid, chatType: msgIdWithPeer.Peer.chatType } - const msg = await NTQQMsgApi.getMsgsByMsgId( - peer, - [msgIdWithPeer?.MsgId || payload.message_id.toString()] - ) - const retMsg = await OB11Constructor.message(msg.msgList[0]) - retMsg.message_id = MessageUnique.createMsg(peer, msg.msgList[0].msgId)! + const msg = getMsgCache(msgIdWithPeer.MsgId) ?? (await NTQQMsgApi.getMsgsByMsgId(peer, [msgIdWithPeer.MsgId])).msgList[0] + const retMsg = await OB11Constructor.message(msg) + retMsg.message_id = MessageUnique.createMsg(peer, msg.msgId)! retMsg.message_seq = retMsg.message_id retMsg.real_id = retMsg.message_id return retMsg diff --git a/src/onebot11/action/msg/SendMsg.ts b/src/onebot11/action/msg/SendMsg.ts index cae7707..a08c3d5 100644 --- a/src/onebot11/action/msg/SendMsg.ts +++ b/src/onebot11/action/msg/SendMsg.ts @@ -119,9 +119,8 @@ export async function createSendElements( if (!peer) { continue } - let atQQ = sendMsg.data?.qq - if (atQQ) { - atQQ = atQQ.toString() + if (sendMsg.data?.qq) { + const atQQ = String(sendMsg.data.qq) if (atQQ === 'all') { // todo：查询剩余的at全体次数 const groupCode = peer.peerUid @@ -161,7 +160,7 @@ export async function createSendElements( } break case OB11MessageDataType.reply: { - if (sendMsg.data.id) { + if (sendMsg.data?.id) { const replyMsgId = await MessageUnique.getMsgIdAndPeerByShortId(+sendMsg.data.id) if (!replyMsgId) { log('回复消息不存在', replyMsgId) diff --git a/src/onebot11/constructor.ts b/src/onebot11/constructor.ts index 2bc85a8..c94335f 100644 --- a/src/onebot11/constructor.ts +++ b/src/onebot11/constructor.ts @@ -590,21 +590,19 @@ export class OB11Constructor { msg: RawMessage, shortId: number ): Promise { - let msgElement = msg.elements.find( + const msgElement = msg.elements.find( (element) => element.grayTipElement?.subElementType === GrayTipElementSubType.RECALL, ) if (!msgElement) { return } - const isGroup = msg.chatType === ChatType.group const revokeElement = msgElement.grayTipElement.revokeElement - if (isGroup) { + if (msg.chatType === ChatType.group) { const operator = await getGroupMember(msg.peerUid, revokeElement.operatorUid) - const sender = await getGroupMember(msg.peerUid, revokeElement.origMsgSenderUid!) return new OB11GroupRecallNoticeEvent( parseInt(msg.peerUid), - parseInt(sender?.uin!), - parseInt(operator?.uin!), + parseInt(msg.senderUin!), + parseInt(operator?.uin || msg.senderUin!), shortId, ) } diff --git a/src/renderer/index.ts b/src/renderer/index.ts index 70d9083..5015d46 100644 --- a/src/renderer/index.ts +++ b/src/renderer/index.ts @@ -219,6 +219,11 @@ async function onSettingWindowCreated(view: Element) { `${window.LiteLoader.plugins['LLOneBot'].path.data}/logs`, SettingButton('打开', 'config-open-log-path'), ), + SettingItem( + '已撤回消息内容缓存时长', + '单位为毫秒', + `
`, + ), ]), SettingList([ SettingItem('GitHub 仓库', `https://github.com/LLOneBot/LLOneBot`, SettingButton('点个星星', 'open-github')),