koishi-plugin-cocoyyy-console 1.2.2-alpha.0 → 1.2.2-alpha.1

package/lib/index.js

@@ -1781,6 +1781,68 @@ function extractImageAndText(content) {
   return { imageUrls, text };
 }
 __name(extractImageAndText, "extractImageAndText");
+function extractFromForwardChildren(children) {
+  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("")}`);
+    }
+  }
+  return { imageUrls: allImageUrls, text: allTexts.join("\n") };
+}
+__name(extractFromForwardChildren, "extractFromForwardChildren");
+async function fetchForwardMessages(session, forwardId) {
+  try {
+    const result = await session.bot.internal.getForwardMsg({ id: forwardId });
+    const messages = result?.messages ?? [];
+    const allTexts = [];
+    const allImageUrls = [];
+    for (const msg of messages) {
+      const sender = msg?.sender?.nickname ?? String(msg?.sender?.user_id ?? "未知");
+      const segments = msg?.message ?? msg?.content ?? [];
+      const msgTexts = [];
+      for (const seg of segments) {
+        if (seg.type === "text") {
+          const t = seg.data?.text ?? "";
+          if (t) msgTexts.push(t);
+        } else if (seg.type === "image") {
+          const url = seg.data?.url ?? seg.data?.file ?? "";
+          if (url) allImageUrls.push(url);
+        }
+      }
+      if (msgTexts.length > 0) {
+        allTexts.push(`[${sender}]: ${msgTexts.join("")}`);
+      }
+    }
+    return { imageUrls: allImageUrls, text: allTexts.join("\n") };
+  } catch (e) {
+    logger.error("[fetchForwardMessages Error]: 获取转发消息失败: " + e);
+    return { imageUrls: [], text: "" };
+  }
+}
+__name(fetchForwardMessages, "fetchForwardMessages");
+function extractForwardId(elements, content) {
+  const forwardEl = elements.find((el) => el.type === "forward");
+  if (forwardEl?.attrs?.id) return String(forwardEl.attrs.id);
+  const match = content.match(/<forward\b[^>]*\bid=["']?([^"'\s>/]+)["']?/i);
+  return match?.[1] ?? null;
+}
+__name(extractForwardId, "extractForwardId");
 async function checkShitOrNot(session, parttern_msg, config) {
   chatHistory = [];
   chatHistory.push({ role: "system", content: [{ type: "text", text: parttern_msg }] });
@@ -1805,6 +1867,59 @@ async function checkShitOrNot(session, parttern_msg, config) {
   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);
+  let imageUrls = [];
+  let text = "";
+  let isForward = false;
+  const forwardId = extractForwardId(quoteElements, quoteContent);
+  if (forwardId) {
+    isForward = true;
+    logger.info("[checkShitOrNot_v2 Info]: 检测到转发聊天记录，ID: " + forwardId);
+    const forwardEl = quoteElements.find((el) => el.type === "forward");
+    const forwardChildren = forwardEl?.children ?? [];
+    if (forwardChildren.length > 0) {
+      const extracted = extractFromForwardChildren(forwardChildren);
+      imageUrls = extracted.imageUrls;
+      text = extracted.text;
+      logger.info("[checkShitOrNot_v2 Info]: 从 children 提取转发内容，共 " + forwardChildren.length + " 条");
+    }
+    if (!text && imageUrls.length === 0) {
+      logger.info("[checkShitOrNot_v2 Info]: children 为空，尝试通过 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]: 引用消息中未发现文本或图片");
+    return isForward ? "转发聊天记录中未发现可分析的文本或图片" : "引用消息中未发现文本或图片";
+  }
+  logger.info("[checkShitOrNot_v2 Info]: 提取图片 URL: " + imageUrls.join(", "));
+  const displayText = isForward ? `以下是转发聊天记录的内容:
+
+${text}` : text;
+  const targetContent = [
+    ...displayText ? [{ type: "text", text: displayText }] : [],
+    ...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_v2 Error]: " + err2);
+    return null;
+  }
+  return resp2;
+}
+__name(checkShitOrNot_v2, "checkShitOrNot_v2");
 
 // src/services/son_func/shit_or_not_command.ts
 var local_config = null;
@@ -1851,6 +1966,26 @@ 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,4 +2,5 @@ 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>;
-export { checkShitOrNot, ocrImage };
+declare function checkShitOrNot_v2(session: Session, parttern_msg: string, config: AiAPIConfig): Promise<string>;
+export { checkShitOrNot, checkShitOrNot_v2, ocrImage };

package/package.json

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