koishi-plugin-chat-patch 2.1.1 → 2.2.0

package/lib/api-handlers.d.ts

@@ -8,6 +8,7 @@ export declare class ApiHandlers {
     private fileManager;
     private messageHandler;
     private logger;
+    private currentTempVideo;
     constructor(ctx: Context, config: Config, fileManager: FileManager, messageHandler: MessageHandler);
     registerApiHandlers(): void;
     private isFileUrl;

package/lib/index.js

@@ -128,26 +128,40 @@ var Utils = class {
     }
   }
   // 清理对象中的base64内容，改为持久化存储
-  cleanBase64Content(obj) {
+  // isBotMessage: 是否为机器人发送的消息
+  cleanBase64Content(obj, isBotMessage = false) {
     if (obj === null || obj === void 0) {
       return obj;
     }
     if (typeof obj === "string") {
       if (this.isBase64(obj)) {
+        if (isBotMessage && !obj.startsWith("data:image/")) {
+          return "[富媒体内容已省略]";
+        }
         return this.persistBase64Image(obj);
       }
       return obj;
     }
     if (Array.isArray(obj)) {
-      return obj.map((item) => this.cleanBase64Content(item));
+      return obj.map((item) => this.cleanBase64Content(item, isBotMessage));
     }
     if (typeof obj === "object") {
       const cleaned = {};
       for (const [key, value] of Object.entries(obj)) {
+        if (isBotMessage && obj.type && !["text", "image", "img"].includes(obj.type)) {
+          if (typeof value === "string" && (key === "src" || key === "url" || key === "file") && this.isBase64(value)) {
+            cleaned[key] = "[富媒体内容已省略]";
+            continue;
+          }
+        }
         if (typeof value === "string" && (key === "src" || key === "url" || key === "file" || key === "data" || key === "content") && this.isBase64(value)) {
-          cleaned[key] = this.persistBase64Image(value);
+          if (isBotMessage && !value.startsWith("data:image/")) {
+            cleaned[key] = "[富媒体内容已省略]";
+          } else {
+            cleaned[key] = this.persistBase64Image(value);
+          }
         } else {
-          cleaned[key] = this.cleanBase64Content(value);
+          cleaned[key] = this.cleanBase64Content(value, isBotMessage);
         }
       }
       return cleaned;
@@ -285,17 +299,22 @@ var MessageHandler = class {
     }
   }
   // 处理消息中的媒体元素并缓存
-  async processMediaElements(elements) {
+  // 只缓存用户的图片和语音，不缓存视频和文件
+  async processMediaElements(elements, isUserMessage = true) {
     if (!elements) return elements;
     for (const el of elements) {
       if (["image", "img", "mface"].includes(el.type)) {
         const src = el.attrs.src || el.attrs.url || el.attrs.file;
-        if (src) el.attrs.src = await this.downloadAndCacheMedia(src, "image");
-      } else if (["audio", "video"].includes(el.type)) {
+        if (src && isUserMessage) {
+          el.attrs.src = await this.downloadAndCacheMedia(src, "image");
+        }
+      } else if (el.type === "audio") {
         const src = el.attrs.src || el.attrs.url || el.attrs.file;
-        if (src) el.attrs.src = await this.downloadAndCacheMedia(src, "media");
+        if (src && isUserMessage) {
+          el.attrs.src = await this.downloadAndCacheMedia(src, "media");
+        }
       }
-      if (el.children) await this.processMediaElements(el.children);
+      if (el.children) await this.processMediaElements(el.children, isUserMessage);
     }
     return elements;
   }
@@ -306,10 +325,10 @@ var MessageHandler = class {
       const isDirect = session.isDirect || session.channelId?.includes("private");
       const timestamp = Date.now();
       if (session.elements) {
-        await this.processMediaElements(session.elements);
+        await this.processMediaElements(session.elements, true);
       }
       if (session.quote?.elements) {
-        await this.processMediaElements(session.quote.elements);
+        await this.processMediaElements(session.quote.elements, true);
       }
       let quoteInfo = void 0;
       if (session.quote) {
@@ -350,11 +369,11 @@ var MessageHandler = class {
         timestamp,
         channelId: session.channelId,
         selfId: session.selfId,
-        elements: this.utils.cleanBase64Content(elements),
+        elements: this.utils.cleanBase64Content(elements, false),
         type: "user",
         guildName,
         platform: session.platform || "unknown",
-        quote: quoteInfo ? this.utils.cleanBase64Content(quoteInfo) : void 0,
+        quote: quoteInfo ? this.utils.cleanBase64Content(quoteInfo, false) : void 0,
         isDirect: !!isDirect
       };
       await this.fileManager.addMessageToFile(messageInfo);
@@ -371,8 +390,8 @@ var MessageHandler = class {
         timestamp,
         guildName,
         channelType: session.type || 0,
-        elements: this.utils.cleanBase64Content(elements),
-        quote: quoteInfo ? this.utils.cleanBase64Content(quoteInfo) : void 0,
+        elements: this.utils.cleanBase64Content(elements, false),
+        quote: quoteInfo ? this.utils.cleanBase64Content(quoteInfo, false) : void 0,
         isDirect: session.isDirect,
         bot: {
           avatar: session.bot.user?.avatar,
@@ -393,9 +412,6 @@ var MessageHandler = class {
       const guildName = await this.updateChannelInfoToFile(session);
       const isDirect = session.isDirect || finalChannelId?.includes("private");
       const timestamp = Date.now();
-      if (session.event?.message?.elements) {
-        await this.processMediaElements(session.event.message.elements);
-      }
       let content = session.content || "";
       if (!content && session.event?.message?.elements) {
         content = this.utils.extractTextContent(session.event.message.elements).trim();
@@ -438,7 +454,7 @@ var MessageHandler = class {
         timestamp,
         channelId: finalChannelId,
         selfId: session.selfId,
-        elements: this.utils.cleanBase64Content(session.event?.message?.elements),
+        elements: this.utils.cleanBase64Content(session.event?.message?.elements, true),
         type: "bot",
         guildName,
         platform: session.platform || "unknown",
@@ -461,7 +477,7 @@ var MessageHandler = class {
         timestamp,
         guildName,
         channelType: session.event?.channel?.type || session.type || 0,
-        elements: this.utils.cleanBase64Content(session.event?.message?.elements),
+        elements: this.utils.cleanBase64Content(session.event?.message?.elements, true),
         quote: quoteInfo,
         isDirect: !!isDirect,
         sending: true,
@@ -642,6 +658,8 @@ var ApiHandlers = class {
     __name(this, "ApiHandlers");
   }
   logger;
+  // 当前临时缓存的视频文件路径
+  currentTempVideo = null;
   registerApiHandlers() {
     this.ctx.console.addListener("clear-all-indexeddb-data", async () => {
       try {
@@ -1067,6 +1085,38 @@ var ApiHandlers = class {
         return { success: false, error: error?.message || String(error) };
       }
     });
+    this.ctx.console.addListener("fetch-video-temp", async (data) => {
+      try {
+        this.logInfo("收到视频临时加载请求:", data.url);
+        if (this.isFileUrl(data.url)) {
+          const result = await this.handleLocalFileRequest(data.url);
+          return result;
+        }
+        const response = await fetch(data.url, {
+          headers: {
+            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36",
+            "Referer": ""
+          }
+        });
+        if (!response.ok) {
+          throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+        }
+        const buffer = await response.arrayBuffer();
+        const base64 = Buffer.from(buffer).toString("base64");
+        const contentType = response.headers.get("content-type") || "video/mp4";
+        this.logInfo("视频下载成功:", { size: buffer.byteLength, contentType });
+        return {
+          success: true,
+          base64,
+          contentType,
+          dataUrl: `data:${contentType};base64,${base64}`,
+          size: buffer.byteLength
+        };
+      } catch (error) {
+        this.logger.error("视频临时加载失败:", error);
+        return { success: false, error: error?.message || String(error) };
+      }
+    });
   }
   // 检查是否为文件 URL
   isFileUrl(url) {
@@ -1197,6 +1247,27 @@ var usage = `
 `;
 async function apply(ctx, config) {
   const logger = ctx.logger("chat-patch");
+  const mediaDir = import_node_path3.default.join(ctx.baseDir, "data", "chat-patch", "persist-media", "media");
+  if (require("node:fs").existsSync(mediaDir)) {
+    try {
+      const files = require("node:fs").readdirSync(mediaDir);
+      let deletedCount = 0;
+      for (const file of files) {
+        const filePath = import_node_path3.default.join(mediaDir, file);
+        try {
+          require("node:fs").unlinkSync(filePath);
+          deletedCount++;
+        } catch (e) {
+          logger.warn("删除媒体文件失败:", filePath, e);
+        }
+      }
+      if (deletedCount > 0) {
+        logger.info(`启动时清理了 ${deletedCount} 个旧版本的媒体缓存文件`);
+      }
+    } catch (e) {
+      logger.warn("清理媒体文件夹失败:", e);
+    }
+  }
   const fileManager = new FileManager(ctx, config);
   const messageHandler = new MessageHandler(ctx, config, fileManager);
   const apiHandlers = new ApiHandlers(ctx, config, fileManager, messageHandler);

package/lib/utils.d.ts

@@ -0,0 +1,13 @@
+import { Config } from './config';
+import { Context } from 'koishi';
+export declare class Utils {
+    private config;
+    private ctx?;
+    constructor(config: Config, ctx?: Context);
+    isPlatformBlocked(platform: string): boolean;
+    extractTextContent(elements: any[]): string;
+    private isBase64;
+    persistBase64Image(base64Data: string): string;
+    private cleanupPersistImages;
+    cleanBase64Content(obj: any, isBotMessage?: boolean): any;
+}

package/package.json

@@ -1,49 +1,52 @@
-{
-  "name": "koishi-plugin-chat-patch",
-  "description": "[<ruby>chat-patch<rp>（</rp><rt>点我预览效果</rt><rp>）</rp></ruby>](https://i0.hdslb.com/bfs/openplatform/71074dfc9e5256fc3333d8bd8478bec1af874046.png) 视奸小插件((bushi( (低性能警告)。手机端适配。灵感来自 chat 插件。",
-  "version": "2.1.1",
-  "main": "lib/index.js",
-  "typings": "lib/index.d.ts",
-  "files": [
-    "client",
-    "dist",
-    "lib",
-    "src"
-  ],
-  "license": "MIT",
-  "homepage": "https://github.com/koishi-shangxue-plugins/koishi-shangxue-apps/",
-  "bugs": {
-    "url": "https://github.com/koishi-shangxue-plugins/koishi-shangxue-apps/issues"
-  },
-  "keywords": [
-    "koishi",
-    "plugin",
-    "chat",
-    "聊天室",
-    "console"
-  ],
-  "koishi": {
-    "service": {
-      "required": [
-        "console"
-      ]
-    }
-  },
-  "peerDependencies": {
-    "@koishijs/plugin-console": "^5.11.0",
-    "koishi": "^4.16.0"
-  },
-  "devDependencies": {
-    "@koishijs/client": "^5.11.0",
-    "@types/mime-types": "^3.0.1",
-    "autoprefixer": "^10.4.23",
-    "koishi": "^4.16.0",
-    "postcss": "^8.5.6",
-    "tailwindcss": "^3.4.17"
-  },
-  "dependencies": {
-    "@element-plus/icons-vue": "^2.3.2",
-    "element-plus": "^2.13.0",
-    "mime-types": "^3.0.2"
-  }
-}
+{
+  "name": "koishi-plugin-chat-patch",
+  "description": "[<ruby>chat-patch<rp>（</rp><rt>点我预览效果</rt><rp>）</rp></ruby>](https://i0.hdslb.com/bfs/openplatform/71074dfc9e5256fc3333d8bd8478bec1af874046.png) 视奸小插件((bushi( (低性能警告)。手机端适配。灵感来自 chat 插件。",
+  "version": "2.2.0",
+  "main": "lib/index.js",
+  "typings": "lib/index.d.ts",
+  "files": [
+    "client",
+    "dist",
+    "lib",
+    "src"
+  ],
+  "license": "MIT",
+  "homepage": "https://github.com/koishi-shangxue-plugins/koishi-shangxue-apps/",
+  "bugs": {
+    "url": "https://github.com/koishi-shangxue-plugins/koishi-shangxue-apps/issues"
+  },
+  "contributors": [
+    "shangxueink <1919892171@qq.com>"
+  ],
+  "keywords": [
+    "koishi",
+    "plugin",
+    "chat",
+    "聊天室",
+    "console"
+  ],
+  "koishi": {
+    "service": {
+      "required": [
+        "console"
+      ]
+    }
+  },
+  "peerDependencies": {
+    "@koishijs/plugin-console": "^5.11.0",
+    "koishi": "^4.16.0"
+  },
+  "devDependencies": {
+    "@koishijs/client": "^5.11.0",
+    "@types/mime-types": "^3.0.1",
+    "autoprefixer": "^10.4.23",
+    "koishi": "^4.16.0",
+    "postcss": "^8.5.6",
+    "tailwindcss": "^3.4.17"
+  },
+  "dependencies": {
+    "@element-plus/icons-vue": "^2.3.2",
+    "element-plus": "^2.13.0",
+    "mime-types": "^3.0.2"
+  }
+}