koishi-plugin-chat-patch 1.0.7 → 1.0.9

package/src/file-manager.ts

@@ -3,11 +3,13 @@ import fs from 'node:fs'
 import { Context, Logger } from 'koishi'
 import { ChatData, MessageInfo } from './types'
 import { Config } from './config'
+import { Utils } from './utils'
 
 export class FileManager {
     private dataFilePath: string
     private fileOperationLock = Promise.resolve()
     private logger: Logger
+    private utils: Utils
 
     constructor(
         private ctx: Context,
@@ -15,6 +17,7 @@ export class FileManager {
     ) {
         this.dataFilePath = path.resolve(ctx.baseDir, 'data', 'chat-patch', 'chat-data.json')
         this.logger = ctx.logger('chat-patch')
+        this.utils = new Utils(config)
     }
 
     // 确保目录存在
@@ -119,8 +122,11 @@ export class FileManager {
                 messageInfo.timestamp = Date.now()
             }
 
+            // 清理消息中的base64内容
+            const cleanedMessageInfo = this.utils.cleanBase64Content(messageInfo) as MessageInfo
+
             const beforeCount = data.messages[channelKey].length
-            data.messages[channelKey].push(messageInfo)
+            data.messages[channelKey].push(cleanedMessageInfo)
             const afterCount = data.messages[channelKey].length
 
             // 限制消息数量 - 保留最新的消息

package/src/message-handler.ts

@@ -127,12 +127,12 @@ export class MessageHandler {
                 timestamp: timestamp,
                 channelId: session.channelId,
                 selfId: session.selfId,
-                elements: elements,
+                elements: this.utils.cleanBase64Content(elements),
                 type: 'user',
                 guildId: session.guildId,
                 guildName: guildName,
                 platform: session.platform || 'unknown',
-                quote: quoteInfo
+                quote: quoteInfo ? this.utils.cleanBase64Content(quoteInfo) : undefined
             }
 
             await this.fileManager.addMessageToFile(messageInfo)
@@ -151,8 +151,8 @@ export class MessageHandler {
                 guildId: session.guildId,
                 guildName: guildName,
                 channelType: session.type || 0,
-                elements: elements,
-                quote: quoteInfo,
+                elements: this.utils.cleanBase64Content(elements),
+                quote: quoteInfo ? this.utils.cleanBase64Content(quoteInfo) : undefined,
                 bot: {
                     avatar: session.bot.user?.avatar,
                     name: session.bot.user?.name,
@@ -189,7 +189,7 @@ export class MessageHandler {
                 timestamp: timestamp,
                 channelId: session.event?.channel?.id || session.channelId,
                 selfId: session.selfId,
-                elements: session.event?.message?.elements,
+                elements: this.utils.cleanBase64Content(session.event?.message?.elements),
                 type: 'bot',
                 guildId: session.event?.guild?.id || session.guildId,
                 guildName: guildName,
@@ -212,7 +212,7 @@ export class MessageHandler {
                 guildId: session.event?.guild?.id || session.guildId,
                 guildName: guildName,
                 channelType: session.event?.channel?.type || session.type || 0,
-                elements: session.event?.message?.elements,
+                elements: this.utils.cleanBase64Content(session.event?.message?.elements),
                 bot: {
                     avatar: session.bot.user?.avatar,
                     name: session.bot.user?.name,

package/src/utils.ts

@@ -40,4 +40,61 @@ export class Utils {
         }
         return text
     }
+
+    // 检查字符串是否为base64格式
+    private isBase64(str: string): boolean {
+        if (!str || typeof str !== 'string') return false
+
+        // 检查是否以data:开头的base64格式
+        if (str.startsWith('data:')) {
+            return str.includes('base64,')
+        }
+
+        // 检查纯base64字符串（长度大于100且符合base64格式）
+        if (str.length > 100) {
+            const base64Regex = /^[A-Za-z0-9+/]*={0,2}$/
+            return base64Regex.test(str)
+        }
+
+        return false
+    }
+
+    // 清理对象中的base64内容
+    cleanBase64Content(obj: any): any {
+        if (obj === null || obj === undefined) {
+            return obj
+        }
+
+        if (typeof obj === 'string') {
+            if (this.isBase64(obj)) {
+                return '暂不支持记录base64内容'
+            }
+            return obj
+        }
+
+        if (Array.isArray(obj)) {
+            return obj.map(item => this.cleanBase64Content(item))
+        }
+
+        if (typeof obj === 'object') {
+            const cleaned: any = {}
+            for (const [key, value] of Object.entries(obj)) {
+                // 特别处理常见的base64字段
+                if (typeof value === 'string' && (
+                    key === 'src' ||
+                    key === 'url' ||
+                    key === 'file' ||
+                    key === 'data' ||
+                    key === 'content'
+                ) && this.isBase64(value)) {
+                    cleaned[key] = '暂不支持记录base64内容'
+                } else {
+                    cleaned[key] = this.cleanBase64Content(value)
+                }
+            }
+            return cleaned
+        }
+
+        return obj
+    }
 }