koishi-plugin-aka-ai-generator 0.8.5 → 0.8.7

package/lib/index.js

@@ -69,14 +69,35 @@ function sanitizeString(str) {
 __name(sanitizeString, "sanitizeString");
 async function downloadImageAsBase64(ctx, url, timeout, logger, maxSize = 10 * 1024 * 1024) {
   try {
-    const response = await ctx.http.get(url, {
-      responseType: "arraybuffer",
-      timeout: timeout * 1e3,
-      headers: {
-        "Accept": "image/*"
+    let buffer;
+    if (url.startsWith("internal:")) {
+      logger.debug("检测到 Koishi 内部协议 URL，使用 ctx.http.file() 获取", { url: url.substring(0, 80) });
+      try {
+        const fileResult = await ctx.http.file(url, { timeout: timeout * 1e3 });
+        buffer = Buffer.from(fileResult.data);
+        logger.info("Koishi 内部协议资源获取成功", {
+          url: url.substring(0, 80),
+          size: buffer.length,
+          type: fileResult.type
+        });
+      } catch (internalError) {
+        logger.error("Koishi 内部协议资源获取失败", {
+          url: url.substring(0, 80),
+          error: sanitizeError(internalError),
+          message: internalError?.message
+        });
+        throw new Error(`无法获取飞书/Lark 图片资源: ${internalError?.message || "未知错误"}`);
       }
-    });
-    const buffer = Buffer.from(response);
+    } else {
+      const response = await ctx.http.get(url, {
+        responseType: "arraybuffer",
+        timeout: timeout * 1e3,
+        headers: {
+          "Accept": "image/*"
+        }
+      });
+      buffer = Buffer.from(response);
+    }
     if (buffer.length > maxSize) {
       throw new Error(`图片大小超过限制 (${(maxSize / 1024 / 1024).toFixed(1)}MB)`);
     }
@@ -2205,8 +2226,26 @@ function apply(ctx, config) {
     while (true) {
       const msg = await session.prompt(mode === "multiple" ? 6e4 : 3e4);
       if (!msg) return { error: "等待超时" };
+      logger.info("getInputData 收到消息", {
+        platform: session.platform,
+        msgType: typeof msg,
+        msgLength: msg?.length,
+        msgPreview: typeof msg === "string" ? msg.substring(0, 200) : "non-string",
+        rawMsg: msg
+      });
       const elements = import_koishi2.h.parse(msg);
+      logger.info("getInputData 解析元素", {
+        platform: session.platform,
+        elementsCount: elements?.length,
+        elementTypes: elements?.map((e) => e.type),
+        elementsDetail: JSON.stringify(elements?.slice(0, 5))
+      });
       const images = import_koishi2.h.select(elements, "img");
+      logger.info("getInputData 图片元素", {
+        platform: session.platform,
+        imagesCount: images?.length,
+        imagesAttrs: images?.map((img) => ({ src: img.attrs?.src?.substring(0, 100), allAttrs: Object.keys(img.attrs || {}) }))
+      });
       const textElements = import_koishi2.h.select(elements, "text");
       const text = textElements.map((el) => el.attrs.content).join(" ").trim();
       if (images.length > 0) {

package/lib/providers/utils.d.ts

@@ -10,6 +10,7 @@ export declare function sanitizeString(str: string): string;
 /**
  * 下载图片并转换为 Base64
  * 包含 MIME 类型检测和大小限制
+ * 支持 Koishi 内部协议 URL (如 internal:lark/... internal:onebot/... 等)
  */
 export declare function downloadImageAsBase64(ctx: Context, url: string, timeout: number, logger: any, maxSize?: number): Promise<{
     data: string;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-aka-ai-generator",
   "description": "自用AI生成插件(GPTGod & Yunwu)",
-  "version": "0.8.5",
+  "version": "0.8.7",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [