koishi-plugin-chat-patch 2.0.1 → 2.1.1

package/lib/types.d.ts

@@ -0,0 +1,57 @@
+import { h } from 'koishi';
+export interface BotInfo {
+    selfId: string;
+    platform: string;
+    username: string;
+    avatar?: string;
+    status: 'online' | 'offline';
+}
+export interface ChannelInfo {
+    id: string;
+    name: string;
+    type: number | string;
+    channelId?: string;
+    guildName?: string;
+    isDirect?: boolean;
+}
+export interface QuoteInfo {
+    messageId: string;
+    id: string;
+    content: string;
+    elements?: h[];
+    user: {
+        id: string;
+        name: string;
+        userId: string;
+        avatar?: string;
+        username: string;
+    };
+    timestamp: number;
+}
+export interface MessageInfo {
+    id: string;
+    content: string;
+    userId: string;
+    username: string;
+    avatar?: string;
+    timestamp: number;
+    channelId: string;
+    selfId: string;
+    elements?: h[];
+    type: 'user' | 'bot';
+    guildId?: string;
+    guildName?: string;
+    platform: string;
+    quote?: QuoteInfo;
+    isDirect?: boolean;
+    sending?: boolean;
+    realId?: string;
+}
+export interface ChatData {
+    bots: Record<string, BotInfo>;
+    channels: Record<string, Record<string, ChannelInfo>>;
+    messages: Record<string, MessageInfo[]>;
+    pinnedBots: string[];
+    pinnedChannels: string[];
+    lastSaveTime?: number;
+}

package/package.json

@@ -1,7 +1,7 @@
 {
   "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.0.1",
+  "version": "2.1.1",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [

package/src/api-handlers.ts

@@ -1,6 +1,6 @@
 import { FileManager } from './file-manager'
 import { MessageHandler } from './message-handler'
-import { Context, h, Logger } from 'koishi'
+import { Context, h, Logger, Universal } from 'koishi'
 import { Config } from './config'
 import { } from '@koishijs/plugin-console'
 import { URL, pathToFileURL, fileURLToPath } from 'node:url'
@@ -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) {
@@ -222,10 +219,17 @@ export class ApiHandlers {
       try {
         this.logInfo('收到发送消息请求:', data)
 
-        const bot = this.ctx.bots.find((bot: any) => bot.selfId === data.selfId)
+        // 优化机器人查找逻辑
+        const bot = this.ctx.bots.find((bot: any) => bot.selfId === data.selfId || bot.user?.id === data.selfId)
         if (!bot) {
-          this.logger.error('未找到机器人:', data.selfId)
-          return { success: false, error: '未找到指定的机器人' }
+          this.logger.error('未找到机器人:', data.selfId, '当前可用机器人:', this.ctx.bots.map((b: any) => b.selfId))
+          return { success: false, error: `未找到机器人 ${data.selfId}，请检查机器人是否在线` }
+        }
+
+        // 检查机器人状态
+        if (bot.status !== Universal.Status.ONLINE) {
+          this.logger.error('机器人离线:', data.selfId, '状态:', bot.status)
+          return { success: false, error: `机器人 ${data.selfId} 当前离线` }
         }
 
         let messageContent = data.content
@@ -257,32 +261,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) || []
         }
@@ -503,7 +506,7 @@ export class ApiHandlers {
           config: {
             maxMessagesPerChannel: this.config.maxMessagesPerChannel,
             keepMessagesOnClear: this.config.keepMessagesOnClear,
-            keepTempImages: this.config.keepTempImages,
+            maxPersistImages: this.config.maxPersistImages,
             loggerinfo: this.config.loggerinfo,
             blockedPlatforms: this.config.blockedPlatforms || [],
             clearIndexedDBOnStart: this.config.clearIndexedDBOnStart
@@ -515,6 +518,82 @@ 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: '机器人不存在' }
+
+        // 检查平台是否支持 getUser 方法
+        if (!bot.getUser || typeof bot.getUser !== 'function') {
+          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 +664,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
+        .sort((a, b) => b.mtime - a.mtime)
 
-      // 分离受保护的文件和可删除的文件
-      const protectedFiles = files.filter((file: any) => file.protected)
-      const deletableFiles = files.filter((file: any) => !file.protected)
-
-      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/config.ts

@@ -5,7 +5,6 @@ export interface Config {
   clearIndexedDBOnStart: boolean
   maxMessagesPerChannel: number
   keepMessagesOnClear: number
-  keepTempImages: number
   maxPersistImages: number
   blockedPlatforms: Array<{
     platformName: string
@@ -15,10 +14,9 @@ export interface Config {
 
 export const Config: Schema<Config> = Schema.intersect([
   Schema.object({
-    maxMessagesPerChannel: Schema.number().default(500).description('每个群组最大保存消息数量').min(50).max(1500),
-    keepMessagesOnClear: Schema.number().default(50).description('手动清理历史记录时保留的消息数量').min(0).max(1000),
-    keepTempImages: Schema.number().default(50).description('发送消息保留的临时图片数量（最新的N张）').min(10).max(200),
-    maxPersistImages: Schema.number().default(100).description('持久化存储的机器人发送图片数量').min(10).max(500),
+    maxMessagesPerChannel: Schema.number().default(500).description('每个群组最大保存消息数量').min(50).max(1500).step(1),
+    keepMessagesOnClear: Schema.number().default(50).description('手动清理历史记录时保留的消息数量').min(0).max(1000).step(1),
+    maxPersistImages: Schema.number().default(100).description('持久化存储的图片缓存数量').min(10).max(500).step(1),
     blockedPlatforms: Schema.array(Schema.object({
       platformName: Schema.string().description('平台名称或关键词'),
       exactMatch: Schema.boolean().default(false).description('完全匹配？如果关闭，包含关键词即屏蔽').default(true)