refactor: move checkSendMessage and handleForwardNode to separate files

src/onebot11/action/msg/SendMsg/handle-forward-node.ts

@@ -0,0 +1,163 @@
+import { ChatType, ElementType, Group, NTQQMsgApi, Peer, RawMessage, SendMessageElement } from '@/core';
+import { OB11MessageNode } from '@/onebot11/types';
+import { selfInfo } from '@/core/data';
+import { dbUtil } from '@/core/utils/db';
+import createSendElements from '@/onebot11/action/msg/SendMsg/create-send-elements';
+import { logDebug, logError } from '@/common/utils/log';
+import { sleep } from '@/common/utils/helper';
+import fs from 'node:fs';
+import { normalize, sendMsg } from '@/onebot11/action/msg/SendMsg/index';
+
+async function cloneMsg(msg: RawMessage): Promise<RawMessage | undefined> {
+  const selfPeer = {
+    chatType: ChatType.friend,
+    peerUid: selfInfo.uid
+  };
+
+  logDebug('克隆的目标消息', msg);
+
+  const sendElements: SendMessageElement[] = [];
+
+  for (const element of msg.elements) {
+    sendElements.push(element as SendMessageElement);
+  }
+
+  if (sendElements.length === 0) {
+    logDebug('需要clone的消息无法解析，将会忽略掉', msg);
+  }
+
+  logDebug('克隆消息', sendElements);
+
+  try {
+    const nodeMsg = await NTQQMsgApi.sendMsg(selfPeer, sendElements, true);
+    await sleep(500); // 防止风控
+    return nodeMsg;
+  } catch (e) {
+    logError(e, '克隆转发消息失败,将忽略本条消息', msg);
+  }
+}
+
+export async function handleForwardNode(destPeer: Peer, messageNodes: OB11MessageNode[], group: Group | undefined): Promise<RawMessage | null> {
+  const selfPeer = {
+    chatType: ChatType.friend,
+    peerUid: selfInfo.uid
+  };
+  let nodeMsgIds: string[] = [];
+
+  // 先判断一遍是不是id和自定义混用
+  const needClone =
+    messageNodes.filter(node => node.data.id).length &&
+    messageNodes.filter(node => !node.data.id).length;
+
+  for (const messageNode of messageNodes) {
+    // 一个node表示一个人的消息
+    const nodeId = messageNode.data.id;
+    // 有nodeId表示一个子转发消息卡片
+    if (nodeId) {
+      const nodeMsg = await dbUtil.getMsgByShortId(parseInt(nodeId));
+      if (!needClone) {
+        nodeMsgIds.push(nodeMsg!.msgId);
+      } else {
+        if (nodeMsg!.peerUid !== selfInfo.uid) {
+          // need cloning
+          const clonedMsg = await cloneMsg(nodeMsg!);
+          if (clonedMsg) {
+            nodeMsgIds.push(clonedMsg.msgId);
+          }
+        }
+      }
+    } else {
+      // 自定义的消息
+      // 提取消息段，发给自己生成消息id
+      try {
+        const { sendElements, deleteAfterSentFiles } = await createSendElements(normalize(messageNode.data.content), group);
+        logDebug('开始生成转发节点', sendElements);
+        const sendElementsSplit: SendMessageElement[][] = [];
+        let splitIndex = 0;
+        for (const sendElement of sendElements) {
+          if (!sendElementsSplit[splitIndex]) {
+            sendElementsSplit[splitIndex] = [];
+          }
+
+          if (sendElement.elementType === ElementType.FILE || sendElement.elementType === ElementType.VIDEO) {
+            if (sendElementsSplit[splitIndex].length > 0) {
+              splitIndex++;
+            }
+            sendElementsSplit[splitIndex] = [sendElement];
+            splitIndex++;
+          } else {
+            sendElementsSplit[splitIndex].push(sendElement);
+          }
+          logDebug(sendElementsSplit);
+        }
+        // log("分割后的转发节点", sendElementsSplit)
+        const MsgNodeList: Promise<RawMessage>[] = [];
+        for (const sendElementsSplitElement of sendElementsSplit) {
+          MsgNodeList.push(sendMsg(selfPeer, sendElementsSplitElement, [], true));
+          await sleep(Math.trunc(sendElementsSplit.length / 10) * 100);
+          //await sleep(10);
+        }
+        for (const msgNode of MsgNodeList) {
+          const result = await msgNode;
+          nodeMsgIds.push(result.msgId);
+          logDebug('转发节点生成成功', result.msgId);
+        }
+        deleteAfterSentFiles.map(f => fs.unlink(f, () => {
+        }));
+
+      } catch (e) {
+        logDebug('生成转发消息节点失败', e);
+      }
+    }
+  }
+
+  // 检查srcPeer是否一致，不一致则需要克隆成自己的消息, 让所有srcPeer都变成自己的，使其保持一致才能够转发
+  const nodeMsgArray: Array<RawMessage> = [];
+  let srcPeer: Peer | undefined = undefined;
+  let needSendSelf = false;
+  for (const msgId of nodeMsgIds) {
+    const nodeMsg = await dbUtil.getMsgByLongId(msgId);
+    if (nodeMsg) {
+      nodeMsgArray.push(nodeMsg);
+      if (!srcPeer) {
+        srcPeer = { chatType: nodeMsg.chatType, peerUid: nodeMsg.peerUid };
+      } else if (srcPeer.peerUid !== nodeMsg.peerUid) {
+        needSendSelf = true;
+        srcPeer = selfPeer;
+      }
+    }
+  }
+  logDebug('nodeMsgArray', nodeMsgArray);
+  nodeMsgIds = nodeMsgArray.map(msg => msg.msgId);
+  if (needSendSelf) {
+    logDebug('需要克隆转发消息');
+    for (const [index, msg] of nodeMsgArray.entries()) {
+      if (msg.peerUid !== selfInfo.uid) {
+        const clonedMsg = await cloneMsg(msg);
+        if (clonedMsg) {
+          nodeMsgIds[index] = clonedMsg.msgId;
+        }
+      }
+    }
+  }
+  // elements之间用换行符分隔
+  // let _sendForwardElements: SendMessageElement[] = []
+  // for(let i = 0; i < sendForwardElements.length; i++){
+  //     _sendForwardElements.push(sendForwardElements[i])
+  //     _sendForwardElements.push(SendMsgElementConstructor.text("\n\n"))
+  // }
+  // const nodeMsg = await NTQQApi.sendMsg(selfPeer, _sendForwardElements, true);
+  // nodeIds.push(nodeMsg.msgId)
+  // await sleep(500);
+  // 开发转发
+  if (nodeMsgIds.length === 0) {
+    throw Error('转发消息失败，生成节点为空');
+  }
+  try {
+    logDebug('开发转发', srcPeer, destPeer, nodeMsgIds);
+    return await NTQQMsgApi.multiForwardMsg(srcPeer!, destPeer, nodeMsgIds);
+  } catch (e) {
+    logError('forward failed', e);
+    return null;
+  }
+}