koishi-plugin-cocoyyy-console 1.2.2-alpha.2 → 1.2.2

package/lib/index.js

@@ -1781,77 +1781,65 @@ function extractImageAndText(content) {
   return { imageUrls, text };
 }
 __name(extractImageAndText, "extractImageAndText");
-function parseOneBotForwardNodes(nodes) {
-  const allTexts = [];
-  const allImageUrls = [];
-  for (const node of nodes) {
-    const sender = node?.sender?.nickname ?? String(node?.sender?.user_id ?? "未知");
-    const segments = node?.message ?? node?.content ?? [];
-    const msgTexts = [];
-    for (const seg of segments) {
-      if (seg?.type === "text") {
+function parseSegments(segments, depth) {
+  const texts = [];
+  const imageUrls = [];
+  for (const seg of segments) {
+    switch (seg?.type) {
+      case "text": {
         const t = seg.data?.text ?? "";
-        if (t) msgTexts.push(t);
-      } else if (seg?.type === "image") {
+        if (t) texts.push(t);
+        break;
+      }
+      case "image": {
         const url = seg.data?.url ?? seg.data?.file ?? "";
-        if (url) allImageUrls.push(url);
+        if (url) imageUrls.push(url);
+        break;
       }
-    }
-    if (msgTexts.length > 0) {
-      allTexts.push(`[${sender}]: ${msgTexts.join("")}`);
+      case "forward": {
+        if (depth >= 1) {
+          texts.push("[嵌套转发消息]");
+        } else {
+          const nestedNodes = seg.data?.content ?? [];
+          if (nestedNodes.length > 0) {
+            const nested = parseOneBotForwardNodes(nestedNodes, depth + 1);
+            if (nested.text) texts.push(`[嵌套转发记录]
+${nested.text}
+[嵌套转发记录结束]`);
+            imageUrls.push(...nested.imageUrls);
+          } else {
+            texts.push("[嵌套转发消息]");
+          }
+        }
+        break;
+      }
+      case "json":
+      case "xml":
+        texts.push("[卡片消息]");
+        break;
+      default:
+        break;
     }
   }
-  return { imageUrls: allImageUrls, text: allTexts.join("\n") };
+  return { texts, imageUrls };
 }
-__name(parseOneBotForwardNodes, "parseOneBotForwardNodes");
-function extractFromForwardChildren(children) {
+__name(parseSegments, "parseSegments");
+function parseOneBotForwardNodes(nodes, depth = 0) {
   const allTexts = [];
   const allImageUrls = [];
-  for (const msgEl of children) {
-    if (msgEl.type !== "message") continue;
-    const msgChildren = msgEl.children ?? [];
-    const authorEl = msgChildren.find((c) => c.type === "author");
-    const senderName = authorEl?.attrs?.name ?? String(authorEl?.attrs?.id ?? "未知");
-    const msgTexts = [];
-    for (const child of msgChildren) {
-      if (child.type === "text") {
-        const t = child.attrs?.content ?? "";
-        if (t) msgTexts.push(t);
-      } else if (child.type === "img" || child.type === "image") {
-        const url = child.attrs?.src ?? child.attrs?.url ?? "";
-        if (url) allImageUrls.push(url);
-      }
-    }
-    if (msgTexts.length > 0) {
-      allTexts.push(`[${senderName}]: ${msgTexts.join("")}`);
+  logger.info("[parseOneBotForwardNodes Info]:Depth [" + depth + "], parse node count : " + nodes.length);
+  for (const node of nodes) {
+    const sender = node?.sender?.nickname ?? String(node?.sender?.user_id ?? "未知");
+    const segments = node?.message ?? node?.content ?? [];
+    const { texts, imageUrls } = parseSegments(segments, depth);
+    allImageUrls.push(...imageUrls);
+    if (texts.length > 0) {
+      allTexts.push(`[${sender}]: ${texts.join("")}`);
     }
   }
   return { imageUrls: allImageUrls, text: allTexts.join("\n") };
 }
-__name(extractFromForwardChildren, "extractFromForwardChildren");
-async function fetchForwardMessages(session, forwardId) {
-  const internal = session.bot.internal;
-  let result;
-  try {
-    result = await internal.getForwardMsg({ id: forwardId });
-  } catch (e1) {
-    logger.warn("[fetchForwardMessages Warn]: getForwardMsg 调用失败，尝试 get_forward_msg: " + e1);
-    try {
-      result = await internal.get_forward_msg({ id: forwardId });
-    } catch (e2) {
-      logger.error("[fetchForwardMessages Error]: 两种 API 调用均失败: " + e2);
-      return { imageUrls: [], text: "" };
-    }
-  }
-  logger.info("[fetchForwardMessages Info]: API 返回结构预览: " + JSON.stringify(result)?.slice(0, 300));
-  const messages = result?.messages ?? [];
-  if (messages.length === 0) {
-    logger.warn("[fetchForwardMessages Warn]: API 返回 messages 为空，result keys: " + Object.keys(result ?? {}).join(", "));
-    return { imageUrls: [], text: "" };
-  }
-  return parseOneBotForwardNodes(messages);
-}
-__name(fetchForwardMessages, "fetchForwardMessages");
+__name(parseOneBotForwardNodes, "parseOneBotForwardNodes");
 function extractForwardId(elements, content) {
   const forwardEl = elements.find((el) => el.type === "forward");
   if (forwardEl?.attrs?.id) return String(forwardEl.attrs.id);
@@ -1860,76 +1848,37 @@ function extractForwardId(elements, content) {
 }
 __name(extractForwardId, "extractForwardId");
 async function checkShitOrNot(session, parttern_msg, config) {
-  chatHistory = [];
-  chatHistory.push({ role: "system", content: [{ type: "text", text: parttern_msg }] });
-  const quoteContent = session.quote.content;
-  logger.info("[checkShitOrNot Info]: 引用消息内容: " + quoteContent);
-  const { imageUrls, text } = extractImageAndText(quoteContent);
-  if (!text && imageUrls.length === 0) {
-    logger.warn("[checkShitOrNot Warn]: 引用消息中未发现文本或图片");
-    return "引用消息中未发现文本或图片";
-  }
-  logger.info("[checkShitOrNot Info]: 提取图片 URL: " + imageUrls.join(", "));
-  const targetContent = [
-    ...text ? [{ type: "text", text }] : [],
-    ...imageUrls.map((url) => ({ type: "image_url", image_url: { url } }))
-  ];
-  chatHistory.push({ role: "user", content: targetContent });
-  const { err: err2, resp: resp2 } = await callOpenRouter(config.api_url, config.api_key, config.api_model, chatHistory);
-  if (err2 != null) {
-    logger.error("[checkShitOrNot Error]: " + err2);
-    return null;
-  }
-  return resp2;
-}
-__name(checkShitOrNot, "checkShitOrNot");
-async function checkShitOrNot_v2(session, parttern_msg, config) {
   chatHistory = [];
   chatHistory.push({ role: "system", content: [{ type: "text", text: parttern_msg }] });
   const quoteContent = session.quote.content;
   const quoteElements = session.quote.elements ?? [];
-  logger.info("[checkShitOrNot_v2 Info]: 引用消息内容: " + quoteContent);
-  logger.info("[checkShitOrNot_v2 Info]: quoteElements 长度: " + quoteElements.length);
+  logger.info("[checkShitOrNot Info]: 引用消息内容: " + quoteContent);
+  logger.info("[checkShitOrNot Info]: quoteElements 长度: " + quoteElements.length);
   let imageUrls = [];
   let text = "";
   let isForward = false;
   const forwardId = extractForwardId(quoteElements, quoteContent);
   if (forwardId) {
     isForward = true;
-    logger.info("[checkShitOrNot_v2 Info]: 检测到转发聊天记录，ID: " + forwardId);
+    logger.info("[checkShitOrNot Info]: 检测到转发聊天记录，ID: " + forwardId);
     const forwardEl = quoteElements.find((el) => el.type === "forward");
     const attrContent = Array.isArray(forwardEl?.attrs?.content) ? forwardEl.attrs.content : [];
     if (attrContent.length > 0) {
-      logger.info("[checkShitOrNot_v2 Info]: 从 attrs.content 提取转发内容，共 " + attrContent.length + " 条");
+      logger.info("[checkShitOrNot Info]: 从 attrs.content 提取转发内容，共 " + attrContent.length + " 条");
       const extracted = parseOneBotForwardNodes(attrContent);
       imageUrls = extracted.imageUrls;
       text = extracted.text;
     }
-    if (!text && imageUrls.length === 0) {
-      const forwardChildren = forwardEl?.children ?? [];
-      if (forwardChildren.length > 0) {
-        logger.info("[checkShitOrNot_v2 Info]: 从 children 提取转发内容，共 " + forwardChildren.length + " 条");
-        const extracted = extractFromForwardChildren(forwardChildren);
-        imageUrls = extracted.imageUrls;
-        text = extracted.text;
-      }
-    }
-    if (!text && imageUrls.length === 0) {
-      logger.info("[checkShitOrNot_v2 Info]: 本地提取失败，尝试通过 NapCat API 拉取转发消息");
-      const extracted = await fetchForwardMessages(session, forwardId);
-      imageUrls = extracted.imageUrls;
-      text = extracted.text;
-    }
   } else {
     const extracted = extractImageAndText(quoteContent);
     imageUrls = extracted.imageUrls;
     text = extracted.text;
   }
   if (!text && imageUrls.length === 0) {
-    logger.warn("[checkShitOrNot_v2 Warn]: 引用消息中未发现文本或图片");
+    logger.warn("[checkShitOrNot Warn]: 引用消息中未发现文本或图片");
     return isForward ? "转发聊天记录中未发现可分析的文本或图片" : "引用消息中未发现文本或图片";
   }
-  logger.info("[checkShitOrNot_v2 Info]: 提取文本长度: " + text.length + "，图片数: " + imageUrls.length);
+  logger.info("[checkShitOrNot Info]: 提取文本长度: " + text.length + "，图片数: " + imageUrls.length);
   const displayText = isForward ? `以下是转发聊天记录的内容:
 
 ${text}` : text;
@@ -1940,12 +1889,12 @@ ${text}` : text;
   chatHistory.push({ role: "user", content: targetContent });
   const { err: err2, resp: resp2 } = await callOpenRouter(config.api_url, config.api_key, config.api_model, chatHistory);
   if (err2 != null) {
-    logger.error("[checkShitOrNot_v2 Error]: " + err2);
+    logger.error("[checkShitOrNot Error]: " + err2);
     return null;
   }
   return resp2;
 }
-__name(checkShitOrNot_v2, "checkShitOrNot_v2");
+__name(checkShitOrNot, "checkShitOrNot");
 
 // src/services/son_func/shit_or_not_command.ts
 var local_config = null;
@@ -1992,26 +1941,6 @@ son判断模式列表:
       think_flag = false;
     }
   });
-  ctx.command("sonv2 <参数>", "判定是否为史").action(async ({ session }, ...args) => {
-    if (!dev_mode) {
-      if (!is_at_bot_quote(session)) return "请提供正确格式，如： [引用消息] @bot son [判断模式]";
-    }
-    if (think_flag) return "请等待思考结果";
-    const parttern = args?.[0];
-    if (!parttern) return "请提供正确判断模式，如： [引用消息] @bot son [判断模式]";
-    if (!local_config) return "未加载配置文件，请查看日志";
-    const parttern_msg = local_config.shit_or_not?.prompts[parttern];
-    if (!parttern_msg) return "未找到判断模式，请查看日志";
-    await session.send("已收到，正在品鉴中...");
-    think_flag = true;
-    try {
-      let exec = await checkShitOrNot_v2(session, parttern_msg, ai_config);
-      if (exec == null) return "品鉴失败，请查看日志";
-      return exec;
-    } finally {
-      think_flag = false;
-    }
-  });
 }
 __name(registerShitOrNotCommands, "registerShitOrNotCommands");
 function extractAtId(content) {

package/lib/services/son_func/shit_or_not_service.d.ts

@@ -2,5 +2,4 @@ import { Session } from 'koishi';
 import { AiAPIConfig } from '../../config/config';
 declare function ocrImage(buffer: Buffer): Promise<string>;
 declare function checkShitOrNot(session: Session, parttern_msg: string, config: AiAPIConfig): Promise<string>;
-declare function checkShitOrNot_v2(session: Session, parttern_msg: string, config: AiAPIConfig): Promise<string>;
-export { checkShitOrNot, checkShitOrNot_v2, ocrImage };
+export { checkShitOrNot, ocrImage };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-cocoyyy-console",
   "description": "自用koishi插件，功能包含复读，记录黑历史，*人等",
-  "version": "1.2.2-alpha.2",
+  "version": "1.2.2",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "contributors": [