koishi-plugin-chat-patch 2.2.0 → 2.4.0

package/src/index.ts

@@ -38,7 +38,6 @@ export { Config } from './config'
 export async function apply(ctx: Context, config: Config) {
   const logger = ctx.logger('chat-patch')
 
-  // 启动时清空 media 文件夹（为旧版本做清理工作）
   const mediaDir = path.join(ctx.baseDir, 'data', 'chat-patch', 'persist-media', 'media')
   if (require('node:fs').existsSync(mediaDir)) {
     try {
@@ -61,17 +60,14 @@ export async function apply(ctx: Context, config: Config) {
     }
   }
 
-  // 初始化各个模块
   const fileManager = new FileManager(ctx, config)
   const messageHandler = new MessageHandler(ctx, config, fileManager)
   const apiHandlers = new ApiHandlers(ctx, config, fileManager, messageHandler)
   const utils = new Utils(config, ctx)
 
-  // 初始化数据
   const initialData = fileManager.readChatDataFromFile()
   const cleanedData = fileManager.cleanExcessMessages(initialData)
 
-  // 如果清理了数据，立即写回文件
   const originalCount = Object.values(initialData.messages).reduce((total, msgs) => total + msgs.length, 0)
   const cleanedCount = Object.values(cleanedData.messages).reduce((total, msgs) => total + msgs.length, 0)
 
@@ -79,7 +75,6 @@ export async function apply(ctx: Context, config: Config) {
     fileManager.writeChatDataToFile(cleanedData)
   }
 
-  // 日志调试函数
   function logInfo(...args: any[]) {
     if (config.loggerinfo) {
       (logger.info as (...args: any[]) => void)(...args)
@@ -94,39 +89,67 @@ export async function apply(ctx: Context, config: Config) {
     总消息数: Object.values(cleanedData.messages).reduce((total, msgs) => total + msgs.length, 0)
   })
 
-  // 监听用户消息
-  ctx.on('message', async (session) => {
-    // 检查平台是否被屏蔽
-    if (utils.isPlatformBlocked(session.platform || 'unknown')) {
-      logInfo(`忽略来自被屏蔽平台的消息: ${session.platform}`)
-      return
-    }
-    // logInfo(session)
-    // 广播消息事件给前端处理
-    await messageHandler.broadcastMessageEvent(session)
+  ctx.on('message', (session) => {
+    if (utils.isPlatformBlocked(session.platform || 'unknown')) return
+
+    const timestamp = Date.now()
+
+    ctx.console.broadcast('chat-message-event', {
+      type: 'message',
+      selfId: session.selfId,
+      platform: session.platform || 'unknown',
+      channelId: session.channelId,
+      messageId: session.event?.message?.id || `msg-${timestamp}`,
+      content: session.content || '',
+      userId: session.userId || 'unknown',
+      username: session.username || session.userId || 'unknown',
+      avatar: session.event?.user?.avatar,
+      timestamp: timestamp,
+      isDirect: session.isDirect,
+      elements: utils.cleanBase64Content(session.elements, false),
+      bot: {
+        avatar: session.bot.user?.avatar,
+        name: session.bot.user?.name,
+      }
+    })
+
+    messageHandler.recordUserMessage(session, timestamp)
   })
 
-  // 监听机器人发送的消息
-  ctx.on('before-send', async (session) => {
-    // 检查平台是否被屏蔽
-    if (utils.isPlatformBlocked(session.platform || 'unknown')) {
-      logInfo(`忽略来自被屏蔽平台的机器人消息: ${session.platform}`)
-      return
-    }
-    // 广播机器人消息事件给前端处理
-    await messageHandler.broadcastBotMessageEvent(session)
+  ctx.on('before-send', (session) => {
+    if (utils.isPlatformBlocked(session.platform || 'unknown')) return
+
+    const timestamp = Date.now()
+
+    ctx.console.broadcast('chat-bot-message-event', {
+      type: 'bot-message',
+      selfId: session.selfId,
+      platform: session.platform || 'unknown',
+      channelId: session.channelId,
+      messageId: `bot-msg-${timestamp}`,
+      content: session.content || '',
+      userId: session.selfId,
+      username: session.bot.user?.name || `Bot-${session.selfId}`,
+      avatar: session.bot.user?.avatar,
+      timestamp: timestamp,
+      sending: true,
+      elements: utils.cleanBase64Content(session.event?.message?.elements, true),
+      bot: {
+        avatar: session.bot.user?.avatar,
+        name: session.bot.user?.name,
+      }
+    })
+
+    messageHandler.recordBotMessage(session, timestamp)
   })
 
-  // 插件启动时设置定期清理过期消息
   ctx.on('ready', async () => {
     logInfo('插件启动完成，开始监听消息')
 
-    // 定期清理超量消息
-    setInterval(() => {
+    ctx.setInterval(() => {
       const data = fileManager.readChatDataFromFile()
       const cleanedData = fileManager.cleanExcessMessages(data)
 
-      // 如果有消息被清理，写回文件
       const originalCount = Object.values(data.messages).reduce((total, msgs) => total + msgs.length, 0)
       const cleanedCount = Object.values(cleanedData.messages).reduce((total, msgs) => total + msgs.length, 0)
 
@@ -134,15 +157,19 @@ export async function apply(ctx: Context, config: Config) {
         fileManager.writeChatDataToFile(cleanedData)
         logInfo('定期清理完成，清理了', originalCount - cleanedCount, '条超量消息')
       }
-    }, 300000) // 每5分钟清理一次
+    }, 300000)
   })
 
-  // 注册所有 API 处理器
   apiHandlers.registerApiHandlers()
 
-  // 注册控制台页面
   ctx.console.addEntry({
     dev: path.resolve(__dirname, '../client/index.ts'),
     prod: path.resolve(__dirname, '../dist'),
   })
+
+  ctx.on('dispose', () => {
+
+    fileManager.dispose()
+    logInfo('插件已卸载，所有待处理的消息已写入')
+  })
 }

package/src/message-handler.ts

@@ -8,7 +8,7 @@ import { } from '@koishijs/plugin-console'
 export class MessageHandler {
   private logger: Logger
   private utils: Utils
-  // 存储正确的 channelId 映射，key 是 selfId，value 是正确的 channelId
+
   private correctChannelIds: Map<string, string> = new Map()
 
   constructor(
@@ -20,118 +20,152 @@ export class MessageHandler {
     this.utils = new Utils(config)
   }
 
-  // 设置正确的 channelId
+  recordUserMessage(session: Session, timestamp: number) {
+
+    setImmediate(() => {
+      this.processUserMessage(session, timestamp).catch(error => {
+        this.logger.error('记录用户消息失败:', error)
+      })
+    })
+  }
+
+  recordBotMessage(session: Session, timestamp: number) {
+
+    setImmediate(() => {
+      this.processBotMessage(session, timestamp).catch(error => {
+        this.logger.error('记录机器人消息失败:', error)
+      })
+    })
+  }
+
   setCorrectChannelId(selfId: string, channelId: string) {
     this.correctChannelIds.set(selfId, channelId)
     this.logInfo('设置正确的 channelId:', { selfId, channelId })
   }
 
-  // 获取正确的 channelId
   getCorrectChannelId(selfId: string): string | undefined {
     return this.correctChannelIds.get(selfId)
   }
 
-  // 更新机器人信息到JSON文件
-  async updateBotInfoToFile(session: Session) {
-    const data = this.fileManager.readChatDataFromFile()
-
-    const botInfo: BotInfo = {
-      selfId: session.selfId,
-      platform: session.platform || 'unknown',
-      username: session.bot.user?.name || `Bot-${session.selfId}`,
-      avatar: session.bot.user?.avatar,
-      status: 'online'
-    }
-
-    data.bots[session.selfId] = botInfo
-    this.fileManager.writeChatDataToFile(data)
-    this.logInfo('更新机器人信息到文件:', botInfo.username)
-  }
-
-  // 更新频道信息到JSON文件
-  async updateChannelInfoToFile(session: Session): Promise<string> {
-    const isDirect = session.isDirect || session.channelId?.includes('private')
-    let guildName = session.channelId
-    // 直接使用 session.username 作为私聊用户名
-    const directUserName = session.username || session.event?.user?.name || session.userId
+  updateBotInfoToFile(session: Session) {
 
-    const data = this.fileManager.readChatDataFromFile()
-
-    if (!isDirect) {
+    setImmediate(() => {
       try {
-        // 获取群组名称
-        if (session.guildId && session.bot.getGuild && typeof session.bot.getGuild === 'function') {
-          const guild = await session.bot.getGuild(session.guildId)
-          guildName = guild?.name || session.channelId
-        }
+        const data = this.fileManager.readChatDataFromFile()
 
-        // 如果没有getGuild方法，尝试使用getChannel方法
-        if (session.guildId && !session.bot.getGuild && session.bot.getChannel && typeof session.bot.getChannel === 'function') {
-          try {
-            const channel = await session.bot.getChannel(session.guildId)
-            guildName = channel?.name || session.channelId
-          } catch (channelError) {
-            this.logInfo('获取频道信息失败，使用频道ID作为备用:', channelError)
-          }
+        const botInfo: BotInfo = {
+          selfId: session.selfId,
+          platform: session.platform || 'unknown',
+          username: session.bot.user?.name || `Bot-${session.selfId}`,
+          avatar: session.bot.user?.avatar,
+          status: 'online'
         }
+
+        data.bots[session.selfId] = botInfo
+        this.fileManager.writeChatDataToFile(data)
+        this.logInfo('更新机器人信息到文件:', botInfo.username)
       } catch (error) {
-        this.logInfo('获取频道信息失败，使用频道ID作为备用:', error)
-        guildName = session.channelId
+        this.logger.error('更新机器人信息失败:', error)
       }
-    }
+    })
+  }
 
-    if (!data.channels[session.selfId]) {
-      data.channels[session.selfId] = {}
-    }
+  updateChannelInfoToFile(session: Session): string {
+    const isDirect = session.isDirect || session.channelId?.includes('private')
+    const directUserName = session.username || session.event?.user?.name || session.userId
 
-    // 获取现有频道信息
-    const existingChannel = data.channels[session.selfId][session.channelId]
+    const data = this.fileManager.readChatDataFromFile()
+    const existingChannel = data.channels[session.selfId]?.[session.channelId]
 
-    // 构造频道名称
-    let finalName: string
+    let immediateName = session.channelId
     if (isDirect) {
-      // 私聊频道：优先使用真实用户名
       if (directUserName && directUserName !== session.userId) {
-        finalName = `私聊（${directUserName}）`
+        immediateName = `私聊（${directUserName}）`
       } else if (existingChannel?.name && !existingChannel.name.includes('未知')) {
-        // 如果已有名称且不是"未知用户"，保持原名称
-        finalName = existingChannel.name
+        immediateName = existingChannel.name
       } else if (session.platform && session.platform.toLowerCase().includes('sandbox')) {
-        // sandbox 平台直接使用 userId 作为频道名称
-        finalName = `私聊（${session.userId}）`
+        immediateName = `私聊（${session.userId}）`
       } else {
-        finalName = '私聊（未知用户）'
+        immediateName = '私聊（未知用户）'
       }
-    } else {
-      finalName = guildName || session.channelId
+    } else if (existingChannel?.guildName) {
+      immediateName = existingChannel.guildName
     }
 
-    const channelInfo: ChannelInfo = {
-      id: session.channelId,
-      name: finalName,
-      type: session.type || 0,
-      channelId: session.channelId,
-      guildName: guildName,
-      isDirect: !!isDirect
-    }
+    setImmediate(async () => {
+      try {
+        let guildName = session.channelId
 
-    // 记录名称变化
-    if (existingChannel && existingChannel.name !== finalName) {
-      this.logInfo('更新频道名称:', {
-        channelId: session.channelId,
-        oldName: existingChannel.name,
-        newName: finalName
-      })
-    }
+        if (!isDirect) {
+          try {
 
-    data.channels[session.selfId][session.channelId] = channelInfo
-    this.fileManager.writeChatDataToFile(data)
-    this.logInfo('更新频道信息到文件:', channelInfo.name)
+            if (session.guildId && session.bot.getGuild && typeof session.bot.getGuild === 'function') {
+              const guild = await session.bot.getGuild(session.guildId)
+              guildName = guild?.name || session.channelId
+            } else if (session.guildId && !session.bot.getGuild && session.bot.getChannel && typeof session.bot.getChannel === 'function') {
+              try {
+                const channel = await session.bot.getChannel(session.guildId)
+                guildName = channel?.name || session.channelId
+              } catch (channelError) {
+                this.logInfo('获取频道信息失败，使用频道ID作为备用:', channelError)
+              }
+            }
+          } catch (error) {
+            this.logInfo('获取频道信息失败，使用频道ID作为备用:', error)
+            guildName = session.channelId
+          }
+        }
+
+        const freshData = this.fileManager.readChatDataFromFile()
+        if (!freshData.channels[session.selfId]) {
+          freshData.channels[session.selfId] = {}
+        }
 
-    return guildName
+        const existingChannel = freshData.channels[session.selfId][session.channelId]
+
+        let finalName: string
+        if (isDirect) {
+          if (directUserName && directUserName !== session.userId) {
+            finalName = `私聊（${directUserName}）`
+          } else if (existingChannel?.name && !existingChannel.name.includes('未知')) {
+            finalName = existingChannel.name
+          } else if (session.platform && session.platform.toLowerCase().includes('sandbox')) {
+            finalName = `私聊（${session.userId}）`
+          } else {
+            finalName = '私聊（未知用户）'
+          }
+        } else {
+          finalName = guildName || session.channelId
+        }
+
+        const channelInfo: ChannelInfo = {
+          id: session.channelId,
+          name: finalName,
+          type: session.type || 0,
+          channelId: session.channelId,
+          guildName: guildName,
+          isDirect: !!isDirect
+        }
+
+        if (existingChannel && existingChannel.name !== finalName) {
+          this.logInfo('更新频道名称:', {
+            channelId: session.channelId,
+            oldName: existingChannel.name,
+            newName: finalName
+          })
+        }
+
+        freshData.channels[session.selfId][session.channelId] = channelInfo
+        this.fileManager.writeChatDataToFile(freshData)
+        this.logInfo('更新频道信息到文件:', channelInfo.name)
+      } catch (error) {
+        this.logger.error('异步更新频道信息失败:', error)
+      }
+    })
+
+    return immediateName
   }
 
-  // 下载并缓存媒体文件
   public async downloadAndCacheMedia(url: string, type: 'image' | 'media' | 'avatar') {
     try {
       if (!url || url.startsWith('data:') || url.startsWith('file:')) return url
@@ -145,7 +179,6 @@ export class MessageHandler {
         require('node:fs').mkdirSync(dir, { recursive: true })
       }
 
-      // 使用 URL 的 hash 作为文件名，避免重复下载
       const crypto = require('node:crypto')
       const hash = crypto.createHash('md5').update(url).digest('hex')
       const ext = require('node:path').extname(new URL(url).pathname) || (type === 'image' ? '.jpg' : '.mp4')
@@ -165,47 +198,54 @@ export class MessageHandler {
     }
   }
 
-  // 处理消息中的媒体元素并缓存
-  // 只缓存用户的图片和语音，不缓存视频和文件
-  private async processMediaElements(elements: h[], isUserMessage: boolean = 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 && 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 && isUserMessage) {
-          // 只缓存用户消息的语音
-          el.attrs.src = await this.downloadAndCacheMedia(src, 'media')
+  private processMediaElementsAsync(elements: h[], isUserMessage: boolean = true) {
+    if (!elements) return
+
+    setImmediate(async () => {
+      try {
+        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 && isUserMessage) {
+
+              this.downloadAndCacheMedia(src, 'image').catch(e => {
+                this.logger.warn('缓存图片失败:', e)
+              })
+            }
+          } else if (el.type === 'audio') {
+            const src = el.attrs.src || el.attrs.url || el.attrs.file
+            if (src && isUserMessage) {
+
+              this.downloadAndCacheMedia(src, 'media').catch(e => {
+                this.logger.warn('缓存语音失败:', e)
+              })
+            }
+          }
+
+          if (el.children) this.processMediaElementsAsync(el.children, isUserMessage)
         }
+      } catch (error) {
+        this.logger.error('处理媒体元素失败:', error)
       }
-      // 视频和文件不再缓存，保持原始URL
-      if (el.children) await this.processMediaElements(el.children, isUserMessage)
-    }
-    return elements
+    })
   }
 
-  async broadcastMessageEvent(session: Session) {
+  private async processUserMessage(session: Session, timestamp?: number) {
     try {
-      await this.updateBotInfoToFile(session)
-      const guildName = await this.updateChannelInfoToFile(session)
-      const isDirect = session.isDirect || session.channelId?.includes('private')
+      if (!timestamp) timestamp = Date.now()
+
+      this.updateBotInfoToFile(session)
 
-      const timestamp = Date.now()
+      const guildName = this.updateChannelInfoToFile(session)
+      const isDirect = session.isDirect || session.channelId?.includes('private')
 
-      // 处理媒体缓存 - 用户消息只缓存图片和语音
       if (session.elements) {
-        await this.processMediaElements(session.elements, true)
+        this.processMediaElementsAsync(session.elements, true)
       }
       if (session.quote?.elements) {
-        await this.processMediaElements(session.quote.elements, true)
+        this.processMediaElementsAsync(session.quote.elements, true)
       }
 
-      // 处理 quote 信息
       let quoteInfo: QuoteInfo | undefined = undefined
       if (session.quote) {
         quoteInfo = {
@@ -224,7 +264,6 @@ export class MessageHandler {
         }
       }
 
-      // 从 session 中提取用户消息内容
       let content = ''
       let elements: h[] = []
 
@@ -244,7 +283,6 @@ export class MessageHandler {
         }
       }
 
-      // 创建消息信息对象
       const messageInfo: MessageInfo = {
         id: session.event?.message?.id || `msg-${timestamp}`,
         content: content || session.content || '',
@@ -288,47 +326,37 @@ export class MessageHandler {
 
       this.ctx.console.broadcast('chat-message-event', messageEvent)
     } catch (error) {
-      this.logger.error('广播消息事件失败:', error)
+      this.logger.error('处理用户消息失败:', error)
     }
   }
 
-  // 处理机器人发送的消息
-  async broadcastBotMessageEvent(session: Session) {
+  private async processBotMessage(session: Session, timestamp?: number) {
     try {
-      // 获取正确的 channelId
-      const correctChannelId = this.getCorrectChannelId(session.selfId)
+      if (!timestamp) timestamp = Date.now()
 
-      // 使用正确的 channelId，如果没有则使用 session.channelId
+      const correctChannelId = this.getCorrectChannelId(session.selfId)
       const finalChannelId = correctChannelId || session.channelId
 
-      await this.updateBotInfoToFile(session)
-      const guildName = await this.updateChannelInfoToFile(session)
-      const isDirect = session.isDirect || finalChannelId?.includes('private')
-
-      const timestamp = Date.now()
+      this.updateBotInfoToFile(session)
 
-      // 机器人消息不缓存任何媒体
-      // 保持原始URL，不进行缓存处理
+      const guildName = this.updateChannelInfoToFile(session)
+      const isDirect = session.isDirect || finalChannelId?.includes('private')
 
-      // 优先使用 session.content，它包含了完整的消息内容（含标签）
       let content = session.content || ''
 
       if (!content && session.event?.message?.elements) {
         content = this.utils.extractTextContent(session.event.message.elements).trim()
       }
 
-      // 尝试从 content 中提取 quote id 并构建 quote 对象
       let quoteInfo: QuoteInfo | undefined = undefined
       const quoteMatch = content.match(/<quote id="([^"]+)"\/>/)
       if (quoteMatch) {
         const quoteId = quoteMatch[1]
         const data = this.fileManager.readChatDataFromFile()
         const channelKey = `${session.selfId}:${finalChannelId}`
-        // 在当前频道的消息历史中查找被引用的消息
         const quotedMsg = data.messages[channelKey]?.find(m => m.id === quoteId)
 
         if (quotedMsg) {
-          // 如果是虚拟 ID，尝试获取其真实的 messageId
           const realId = quotedMsg.id.startsWith('bot-msg-') ? (quotedMsg as any).realId : quotedMsg.id
 
           quoteInfo = {
@@ -345,10 +373,8 @@ export class MessageHandler {
             },
             timestamp: quotedMsg.timestamp
           }
-          // 移除 content 中的 quote 标签，避免重复显示
           content = content.replace(/<quote id="[^"]+"\/>\s*/, '')
 
-          // 修正 content 中的 quote 标签为真实 ID，以便 bot.sendMessage 能够正确识别
           if (realId) {
             session.content = session.content.replace(/id="[^"]+"/, `id="${realId}"`)
           }
@@ -374,6 +400,7 @@ export class MessageHandler {
         sending: true // 标记为正在发送
       }
 
+      // 异步保存消息（不阻塞）
       await this.fileManager.addMessageToFile(messageInfo)
 
       const messageEvent = {
@@ -401,7 +428,7 @@ export class MessageHandler {
 
       this.ctx.console.broadcast('chat-bot-message-event', messageEvent)
     } catch (error) {
-      this.logger.error('广播机器人消息事件失败:', error)
+      this.logger.error('处理机器人消息失败:', error)
     }
   }