koishi-plugin-chat-patch 2.0.1 → 2.1.0

package/src/api-handlers.ts

@@ -34,23 +34,20 @@ export class ApiHandlers {
     // 获取所有聊天数据的 API
     this.ctx.console.addListener('get-chat-data' as any, async () => {
       try {
+        // 优化性能：只读取基础信息，不读取庞大的消息体
         const data = this.fileManager.readChatDataFromFile()
-        const cleanedData = this.fileManager.cleanExcessMessages(data)
-
-        this.logInfo('获取聊天数据:', {
-          机器人数量: Object.keys(cleanedData.bots).length,
-          频道数量: Object.keys(cleanedData.channels).reduce((total, botId) =>
-            total + Object.keys(cleanedData.channels[botId] || {}).length, 0),
-          消息频道数: Object.keys(cleanedData.messages).length,
-          总消息数: Object.values(cleanedData.messages).reduce((total, msgs) => total + msgs.length, 0)
-        })
+
+        this.logInfo('获取基础聊天数据')
 
         return {
           success: true,
           data: {
-            ...cleanedData,
-            pinnedBots: cleanedData.pinnedBots,
-            pinnedChannels: cleanedData.pinnedChannels
+            bots: data.bots || {},
+            channels: data.channels || {},
+            pinnedBots: data.pinnedBots || [],
+            pinnedChannels: data.pinnedChannels || [],
+            // 不返回 messages，由前端按需拉取
+            messages: {}
           }
         }
       } catch (error: any) {
@@ -257,32 +254,31 @@ export class ApiHandlers {
         const result = await bot.sendMessage(data.channelId, parsedContent)
         this.logInfo('消息发送成功:', result)
 
-        // 广播机器人消息发送成功事件，包含真实的 messageId
         const messageId = Array.isArray(result) ? result[0] : result;
+
+        // 发送成功后，更新本地存储中的虚拟消息为真实 ID
         if (messageId) {
-          const bot = this.ctx.bots.find((b: any) => b.selfId === data.selfId);
-          const botMessageEvent = {
-            type: 'bot-message-sent',
-            selfId: data.selfId,
-            platform: bot?.platform || 'unknown',
-            channelId: data.channelId,
-            messageId: messageId,
-            content: messageContent,
-            userId: data.selfId,
-            username: bot?.user?.name || `Bot-${data.selfId}`,
-            avatar: bot?.user?.avatar,
-            timestamp: Date.now(),
-            guildName: '', // 这个信息在前端会补充
-            channelType: 0, // 这个信息在前端会补充
-            elements: [], // 这个信息在前端会补充
-            isDirect: false // 这个信息在前端会补充
-          };
-
-          this.ctx.console.broadcast('bot-message-sent-event', botMessageEvent);
+          const chatData = this.fileManager.readChatDataFromFile()
+          const channelKey = `${data.selfId}:${data.channelId}`
+          const messages = chatData.messages[channelKey] || []
+          // 找到最近的一条正在发送的机器人消息
+          const msg = [...messages].reverse().find(m => m.type === 'bot' && m.sending)
+          if (msg) {
+            msg.realId = messageId;
+            msg.sending = false;
+            this.fileManager.writeChatDataToFile(chatData)
+
+            // 广播更新事件给前端
+            this.ctx.console.broadcast('bot-message-updated', {
+              channelKey,
+              tempId: msg.id,
+              realId: messageId
+            })
+          }
         }
 
         return {
-          success: true,
+          success: !!messageId,
           messageId: messageId,
           tempImageIds: data.images?.map(img => img.tempId) || []
         }
@@ -515,6 +511,77 @@ export class ApiHandlers {
       }
     })
 
+    // 获取用户信息 API
+    this.ctx.console.addListener('get-user-info' as any, async (data: { selfId: string, userId: string, guildId?: string }) => {
+      try {
+        const bot = this.ctx.bots.find(b => b.selfId === data.selfId)
+        if (!bot) return { success: false, error: '机器人不存在' }
+
+        const user = await bot.getUser(data.userId, data.guildId)
+
+        if (user) {
+          // 缓存头像到本地
+          if (user.avatar) {
+            user.avatar = await this.messageHandler.downloadAndCacheMedia(user.avatar, 'avatar')
+          }
+
+          const chatData = this.fileManager.readChatDataFromFile()
+          let changed = false
+
+          // 1. 如果是私聊，更新频道名 - 支持多种私聊ID格式
+          const botChannels = chatData.channels[data.selfId] || {}
+
+          // 查找所有可能的私聊频道ID格式
+          const possibleChannelIds = [
+            data.userId,
+            `private:${data.userId}`,
+            `direct:${data.userId}`
+          ]
+
+          for (const channelId of possibleChannelIds) {
+            const channel = botChannels[channelId]
+            if (channel && channel.isDirect) {
+              const newName = `私聊（${user.name}）`
+              if (channel.name !== newName) {
+                channel.name = newName
+                changed = true
+                this.logInfo('更新私聊频道名称:', { channelId, oldName: channel.name, newName })
+              }
+            }
+          }
+
+          // 2. 更新该用户在所有历史消息中的头像和昵称（持久化缓存）
+          const channelKeyPrefix = `${data.selfId}:`
+          for (const [key, messages] of Object.entries(chatData.messages)) {
+            if (key.startsWith(channelKeyPrefix)) {
+              messages.forEach(msg => {
+                if (msg.userId === data.userId) {
+                  if (user.name && msg.username !== user.name) {
+                    msg.username = user.name
+                    changed = true
+                  }
+                  if (user.avatar && msg.avatar !== user.avatar) {
+                    msg.avatar = user.avatar
+                    changed = true
+                  }
+                }
+              });
+            }
+          }
+
+          if (changed) {
+            this.fileManager.writeChatDataToFile(chatData)
+            // 广播更新事件，让前端实时刷新
+            this.ctx.console.broadcast('chat-data-updated', {})
+          }
+        }
+
+        return { success: true, data: user }
+      } catch (error: any) {
+        return { success: false, error: error?.message || '获取用户信息失败' }
+      }
+    })
+
     // 调试API：获取原始文件数据
     this.ctx.console.addListener('debug-get-raw-data' as any, async () => {
       try {
@@ -585,75 +652,36 @@ export class ApiHandlers {
   private setupTempFileCleanup() {
     // 每5分钟执行一次基于数量的清理
     setInterval(() => {
-      this.cleanupTempImagesByCount()
-    }, 5 * 60 * 1000) // 5分钟
+      this.cleanupMediaCache()
+    }, 5 * 60 * 1000)
   }
 
-  // 基于数量清理临时图片（保留最新的N张）
-  private async cleanupTempImagesByCount() {
-    try {
-      const tempDir = this.ctx.baseDir + '/data/chat-patch/temp'
-      if (!require('fs').existsSync(tempDir)) {
-        return
-      }
+  // 统一清理媒体缓存
+  private async cleanupMediaCache() {
+    const baseDir = this.ctx.baseDir + '/data/chat-patch/persist-media'
+    if (!require('node:fs').existsSync(baseDir)) return
 
-      const now = Date.now()
-      const protectionTime = 30 * 1000 // 30秒保护期，刚上传的图片不会被删除
+    const cleanupDir = (dirName: string, limit: number) => {
+      const dirPath = require('node:path').join(baseDir, dirName)
+      if (!require('node:fs').existsSync(dirPath)) return
 
-      const files = require('fs').readdirSync(tempDir)
-        .filter((file: string) => file.startsWith('temp_'))
-        .map((file: string) => {
-          const filePath = `${tempDir}/${file}`
-          try {
-            const stats = require('fs').statSync(filePath)
-            const fileAge = now - stats.mtime.getTime()
-            return {
-              name: file,
-              path: filePath,
-              mtime: stats.mtime.getTime(),
-              age: fileAge,
-              protected: fileAge < protectionTime // 是否在保护期内
-            }
-          } catch (error) {
-            return null
-          }
+      const files = require('node:fs').readdirSync(dirPath)
+        .map(file => {
+          const filePath = require('node:path').join(dirPath, file)
+          const stats = require('node:fs').statSync(filePath)
+          return { name: file, path: filePath, mtime: stats.mtimeMs }
         })
-        .filter((file: any) => file !== null)
-        .sort((a: any, b: any) => b.mtime - a.mtime) // 按修改时间降序排列（最新的在前）
-
-      const keepCount = this.config.keepTempImages
-      let cleanedCount = 0
-      let protectedCount = 0
-
-      // 分离受保护的文件和可删除的文件
-      const protectedFiles = files.filter((file: any) => file.protected)
-      const deletableFiles = files.filter((file: any) => !file.protected)
+        .sort((a, b) => b.mtime - a.mtime)
 
-      protectedCount = protectedFiles.length
-
-      // 如果可删除的文件数量超过了保留数量，则删除多余的
-      if (deletableFiles.length > keepCount) {
-        const filesToDelete = deletableFiles.slice(keepCount) // 保留前N个，删除其余的
-
-        for (const file of filesToDelete) {
-          try {
-            if (require('fs').existsSync(file.path)) {
-              require('fs').unlinkSync(file.path)
-              cleanedCount++
-              this.logInfo('清理多余临时图片:', file.path)
-            }
-          } catch (fileError) {
-            this.logger.warn('删除临时文件失败:', { file: file.name, error: fileError })
-          }
-        }
-      }
-
-      if (cleanedCount > 0 || protectedCount > 0) {
-        this.logInfo(`基于数量清理完成，保留最新 ${keepCount} 张图片，清理了 ${cleanedCount} 张多余图片，保护了 ${protectedCount} 张新上传图片`)
+      if (files.length > limit) {
+        files.slice(limit).forEach(f => {
+          try { require('node:fs').unlinkSync(f.path) } catch { }
+        })
       }
-    } catch (error) {
-      this.logger.warn('基于数量清理临时文件失败:', error)
     }
+
+    cleanupDir('images', 100) // 图片缓存100个
+    cleanupDir('media', 20)   // 富媒体缓存20个
   }
 
   private logInfo(...args: any[]) {

package/src/message-handler.ts

@@ -50,55 +50,75 @@ export class MessageHandler {
 
   // 更新频道信息到JSON文件
   async updateChannelInfoToFile(session: Session): Promise<string> {
+    const isDirect = session.isDirect || session.channelId?.includes('private')
     let guildName = session.channelId
-    let directUserName = session.username || session.userId || '未知用户'
+    // 直接使用 session.username 作为私聊用户名
+    const directUserName = session.username || session.event?.user?.name || session.userId
 
     const data = this.fileManager.readChatDataFromFile()
 
-    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
-      }
-
-      // 获取私聊用户昵称
-      if (session.userId && session.isDirect && session.bot.getUser && typeof session.bot.getUser === 'function') {
-        try {
-          const user = await session.bot.getUser(session.userId)
-          directUserName = user?.name || session.username || session.userId || '未知用户'
-        } catch (userError) {
-          this.logInfo('获取用户信息失败，使用备用名称:', userError)
+    if (!isDirect) {
+      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
         }
-      }
 
-      // 如果没有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)
+        // 如果没有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)
+          }
         }
+      } catch (error) {
+        this.logInfo('获取频道信息失败，使用频道ID作为备用:', error)
+        guildName = session.channelId
       }
-    } catch (error) {
-      this.logInfo('获取频道信息失败，使用频道ID作为备用:', error)
-      guildName = session.channelId
     }
 
     if (!data.channels[session.selfId]) {
       data.channels[session.selfId] = {}
     }
 
+    // 获取现有频道信息
+    const existingChannel = data.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 {
+        finalName = '私聊（未知用户）'
+      }
+    } else {
+      finalName = guildName || session.channelId
+    }
+
     const channelInfo: ChannelInfo = {
       id: session.channelId,
-      name: session.isDirect
-        ? `私聊（${session.username || session.userId || directUserName}）`
-        : `${guildName} (${session.channelId})`,
+      name: finalName,
       type: session.type || 0,
       channelId: session.channelId,
       guildName: guildName,
-      isDirect: session.isDirect
+      isDirect: !!isDirect
+    }
+
+    // 记录名称变化
+    if (existingChannel && existingChannel.name !== finalName) {
+      this.logInfo('更新频道名称:', {
+        channelId: session.channelId,
+        oldName: existingChannel.name,
+        newName: finalName
+      })
     }
 
     data.channels[session.selfId][session.channelId] = channelInfo
@@ -108,13 +128,72 @@ export class MessageHandler {
     return guildName
   }
 
+  // 下载并缓存媒体文件
+  public async downloadAndCacheMedia(url: string, type: 'image' | 'media' | 'avatar') {
+    try {
+      if (!url || url.startsWith('data:') || url.startsWith('file:')) return url
+
+      let folder = 'media'
+      if (type === 'image') folder = 'images'
+      else if (type === 'avatar') folder = 'avatars'
+
+      const dir = require('node:path').join(this.ctx.baseDir, 'data', 'chat-patch', 'persist-media', folder)
+      if (!require('node:fs').existsSync(dir)) {
+        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')
+      const filename = `${hash}${ext}`
+      const filePath = require('node:path').join(dir, filename)
+
+      if (require('node:fs').existsSync(filePath)) {
+        return require('node:url').pathToFileURL(filePath).href
+      }
+
+      const buffer = await this.ctx.http.get(url, { responseType: 'arraybuffer' })
+      require('node:fs').writeFileSync(filePath, Buffer.from(buffer))
+
+      return require('node:url').pathToFileURL(filePath).href
+    } catch (e) {
+      return url
+    }
+  }
+
+  // 处理消息中的媒体元素并缓存
+  private async processMediaElements(elements: h[]) {
+    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)) {
+        const src = el.attrs.src || el.attrs.url || el.attrs.file
+        if (src) el.attrs.src = await this.downloadAndCacheMedia(src, 'media')
+      }
+      if (el.children) await this.processMediaElements(el.children)
+    }
+    return elements
+  }
+
   async broadcastMessageEvent(session: Session) {
     try {
       await this.updateBotInfoToFile(session)
       const guildName = await this.updateChannelInfoToFile(session)
+      const isDirect = session.isDirect || session.channelId?.includes('private')
 
       const timestamp = Date.now()
 
+      // 处理媒体缓存
+      if (session.elements) {
+        await this.processMediaElements(session.elements)
+      }
+      if (session.quote?.elements) {
+        await this.processMediaElements(session.quote.elements)
+      }
+
       // 处理 quote 信息
       let quoteInfo: QuoteInfo | undefined = undefined
       if (session.quote) {
@@ -169,7 +248,7 @@ export class MessageHandler {
         guildName: guildName,
         platform: session.platform || 'unknown',
         quote: quoteInfo ? this.utils.cleanBase64Content(quoteInfo) : undefined,
-        isDirect: session.isDirect
+        isDirect: !!isDirect
       }
 
       await this.fileManager.addMessageToFile(messageInfo)
@@ -208,24 +287,20 @@ export class MessageHandler {
       // 获取正确的 channelId
       const correctChannelId = this.getCorrectChannelId(session.selfId)
 
-      // 调试日志：检查 channelId 的来源
-      this.logInfo('机器人发送消息调试信息:', {
-        'session.channelId': session.channelId,
-        'session.event?.channel?.id': session.event?.channel?.id,
-        'correctChannelId': correctChannelId,
-        'session.guildId': session.guildId,
-        'session.isDirect': session.isDirect,
-        'session.platform': session.platform
-      })
-
       // 使用正确的 channelId，如果没有则使用 session.channelId
       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()
 
+      // 处理媒体缓存
+      if (session.event?.message?.elements) {
+        await this.processMediaElements(session.event.message.elements)
+      }
+
       // 优先使用 session.content，它包含了完整的消息内容（含标签）
       let content = session.content || ''
 
@@ -233,6 +308,44 @@ export class MessageHandler {
         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 = {
+            messageId: realId || quotedMsg.id,
+            id: realId || quotedMsg.id,
+            content: quotedMsg.content,
+            elements: quotedMsg.elements,
+            user: {
+              id: quotedMsg.userId,
+              name: quotedMsg.username,
+              userId: quotedMsg.userId,
+              avatar: quotedMsg.avatar,
+              username: quotedMsg.username
+            },
+            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}"`)
+          }
+        }
+      }
+
       // 创建机器人消息信息对象
       const messageInfo: MessageInfo = {
         id: `bot-msg-${timestamp}`,
@@ -247,7 +360,9 @@ export class MessageHandler {
         type: 'bot',
         guildName: guildName,
         platform: session.platform || 'unknown',
-        isDirect: session.isDirect
+        quote: quoteInfo,
+        isDirect: !!isDirect,
+        sending: true // 标记为正在发送
       }
 
       await this.fileManager.addMessageToFile(messageInfo)
@@ -266,32 +381,15 @@ export class MessageHandler {
         guildName: guildName,
         channelType: session.event?.channel?.type || session.type || 0,
         elements: this.utils.cleanBase64Content(session.event?.message?.elements),
-        isDirect: session.isDirect,
+        quote: quoteInfo,
+        isDirect: !!isDirect,
+        sending: true,
         bot: {
           avatar: session.bot.user?.avatar,
           name: session.bot.user?.name,
         }
       }
 
-      // 检查是否包含图片元素
-      const imageElements = session.event?.message?.elements?.filter((element: any) =>
-        element.type === 'img' || element.type === 'image' || element.type === 'mface'
-      ) || []
-
-      this.logInfo('机器人发送消息 (before-send):', {
-        selfId: session.selfId,
-        channelId: messageEvent.channelId,
-        originalChannelId: session.channelId,
-        correctedChannelId: finalChannelId,
-        content: content,
-        platform: session.platform,
-        imageCount: imageElements.length,
-        imageUrls: imageElements.map((el: any) => el.attrs?.src || el.attrs?.url || el.attrs?.file)
-      })
-
-      // 清理已使用的 channelId 映射
-      this.correctChannelIds.delete(session.selfId)
-
       this.ctx.console.broadcast('chat-bot-message-event', messageEvent)
     } catch (error) {
       this.logger.error('广播机器人消息事件失败:', error)

package/src/types.ts

@@ -48,6 +48,8 @@ export interface MessageInfo {
   platform: string
   quote?: QuoteInfo
   isDirect?: boolean
+  sending?: boolean
+  realId?: string
 }
 
 export interface ChatData {