GuanM/NapCatQQ
1
0
Fork 0
mirror of https://github.com/NapNeko/NapCatQQ.git synced 2025-07-19 12:03:37 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: 简化代码 #415

Browse Source
This commit is contained in:
手瓜一十雪
2024-09-29 12:19:04 +08:00
parent 377e3c253f
commit 5745f388a9
1 changed files with 49 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
src/onebot/action/go-cqhttp/GetForwardMsg.ts
View File

@@ -1,9 +1,16 @@
import BaseAction from '../BaseAction'; import BaseAction from '../BaseAction';
import { OB11ForwardMessage, OB11Message, OB11MessageData, OB11MessageDataType, OB11MessageNode } from '@/onebot'; import { OB11ForwardMessage, OB11Message, OB11MessageData, OB11MessageDataType, OB11MessageForward, OB11MessageNode as OriginalOB11MessageNode } from '@/onebot';
import { ActionName } from '../types'; import { ActionName } from '../types';
import { FromSchema, JSONSchema } from 'json-schema-to-ts'; import { FromSchema, JSONSchema } from 'json-schema-to-ts';
import { MessageUnique } from '@/common/message-unique'; import { MessageUnique } from '@/common/message-unique';
type OB11MessageNode = OriginalOB11MessageNode & {
data: {
content?: Array<OB11MessageData>;
message: Array<OB11MessageData>;
};
};
const SchemaData = { const SchemaData = {
type: 'object', type: 'object',
properties: { properties: {
@@ -17,59 +24,61 @@ type Payload = FromSchema<typeof SchemaData>;
export class GoCQHTTPGetForwardMsgAction extends BaseAction<Payload, any> { export class GoCQHTTPGetForwardMsgAction extends BaseAction<Payload, any> {
actionName = ActionName.GoCQHTTP_GetForwardMsg; actionName = ActionName.GoCQHTTP_GetForwardMsg;
payloadSchema = SchemaData; payloadSchema = SchemaData;
async parseForward(msg: OB11Message[]) {
let retMsg: Array<OB11MessageNode & {
data: { message: Array<OB11MessageData> }
}> = [];
for (let message of msg) { private createTemplateNode(message: OB11Message): OB11MessageNode {
let templateNode: OB11MessageNode & { return {
data: { message: Array<OB11MessageData> } type: OB11MessageDataType.node,
} = { data: {
type: OB11MessageDataType.node, user_id: message.user_id,
data: { nickname: message.sender.nickname,
user_id: 10001, message: [],
nickname: "QQ用户", content: []
message: [], }
content: [] };
} }
};
templateNode.data.nickname = message.sender.nickname; async parseForward(messages: OB11Message[]): Promise<OB11MessageNode[]> {
templateNode.data.user_id = message.user_id; const retMsg: OB11MessageNode[] = [];
for (let msgdata of message.message) { for (const message of messages) {
if (typeof msgdata !== 'string' && msgdata.type === OB11MessageDataType.forward) { const templateNode = this.createTemplateNode(message);
templateNode.data.message.push(...(await this.parseForward(msgdata.data.content)));
} for (const msgdata of message.message) {
if (typeof msgdata !== 'string') { if ((msgdata as OB11MessageData).type === OB11MessageDataType.forward) {
templateNode.data.message.push(msgdata); const newNode = this.createTemplateNode(message);
newNode.data.content = await this.parseForward((msgdata as OB11MessageForward).data.content);
templateNode.data.message.push(newNode);
} else {
templateNode.data.message.push(msgdata as OB11MessageData);
} }
} }
retMsg.push(templateNode); retMsg.push(templateNode);
} }
return retMsg;
return retMsg;
} }
async _handle(payload: Payload): Promise<any> { async _handle(payload: Payload): Promise<any> {
const msgId = payload.message_id || payload.id; const msgId = payload.message_id || payload.id;
if (!msgId) { if (!msgId) {
throw Error('message_id is required'); throw new Error('message_id is required');
} }
const rootMsgId = MessageUnique.getShortIdByMsgId(msgId); const rootMsgId = MessageUnique.getShortIdByMsgId(msgId);
const rootMsg = MessageUnique.getMsgIdAndPeerByShortId(rootMsgId || parseInt(msgId)); const rootMsg = MessageUnique.getMsgIdAndPeerByShortId(rootMsgId || parseInt(msgId));
if (!rootMsg) { if (!rootMsg) {
throw Error('msg not found'); throw new Error('msg not found');
} }
const data = await this.core.apis.MsgApi.getMultiMsg(rootMsg.Peer, rootMsg.MsgId, rootMsg.MsgId); const data = await this.core.apis.MsgApi.getMultiMsg(rootMsg.Peer, rootMsg.MsgId, rootMsg.MsgId);
if (!data || data.result !== 0) { if (!data || data.result !== 0) {
throw Error('找不到相关的聊天记录' + data?.errMsg); throw new Error('找不到相关的聊天记录' + data?.errMsg);
} }
const msgList = data.msgList; const msgList = data.msgList;
const messages = (await Promise.all(msgList.map(async msg => { const messages = (await Promise.all(msgList.map(async msg => {
const resMsg = await this.obContext.apis.MsgApi const resMsg = await this.obContext.apis.MsgApi.parseMessage(msg);
.parseMessage(msg); if (!resMsg) return null;
if (!resMsg) return;
resMsg.message_id = MessageUnique.createUniqueMsgId({ resMsg.message_id = MessageUnique.createUniqueMsgId({
guildId: '', guildId: '',
chatType: msg.chatType, chatType: msg.chatType,
@@ -77,15 +86,16 @@ export class GoCQHTTPGetForwardMsgAction extends BaseAction<Payload, any> {
}, msg.msgId)!; }, msg.msgId)!;
return resMsg; return resMsg;
}))).filter(msg => !!msg); }))).filter(msg => !!msg);
messages.map(msg => {
(<OB11ForwardMessage>msg).content = msg.message; if (this.obContext.configLoader.configData.messagePostFormat === 'array') {
delete (<any>msg).message;
});
//遍历合并的所有消息
if (this.obContext.configLoader.configData.messagePostFormat == 'array') {
return await this.parseForward(messages); return await this.parseForward(messages);
} }
messages.forEach(msg => {
(msg as OB11ForwardMessage).content = msg.message;
delete (msg as any).message;
});
return { messages }; return { messages };
} }
} }