koishi-plugin-chat-patch 2.2.0 → 2.3.0

package/src/file-manager.ts

@@ -1,169 +1,215 @@
-import { ChatData, MessageInfo } from './types'
-import { Context, Logger } from 'koishi'
-import { Config } from './config'
-import { Utils } from './utils'
-
-import path from 'node:path'
-import fs from 'node:fs'
-
-export class FileManager {
-  private dataFilePath: string
-  private fileOperationLock = Promise.resolve()
-  private logger: Logger
-  private utils: Utils
-
-  constructor(
-    private ctx: Context,
-    private config: Config
-  ) {
-    this.dataFilePath = path.resolve(ctx.baseDir, 'data', 'chat-patch', 'chat-data.json')
-    this.logger = ctx.logger('chat-patch')
-    this.utils = new Utils(config)
-  }
-
-  // 确保目录存在
-  private ensureDataDir() {
-    const dir = path.dirname(this.dataFilePath)
-    if (!fs.existsSync(dir)) {
-      fs.mkdirSync(dir, { recursive: true })
-    }
-  }
-
-  // 从JSON文件读取数据
-  readChatDataFromFile(): ChatData {
-    try {
-      if (fs.existsSync(this.dataFilePath)) {
-        const jsonData = fs.readFileSync(this.dataFilePath, 'utf8')
-        const data = JSON.parse(jsonData)
-        return {
-          bots: data.bots || {},
-          channels: data.channels || {},
-          messages: data.messages || {},
-          pinnedBots: data.pinnedBots || [],
-          pinnedChannels: data.pinnedChannels || [],
-          lastSaveTime: data.lastSaveTime
-        }
-      }
-    } catch (error) {
-      this.logger.error('读取聊天数据失败:', error)
-    }
-    return {
-      bots: {},
-      channels: {},
-      messages: {},
-      pinnedBots: [],
-      pinnedChannels: []
-    }
-  }
-
-  // 写入数据到JSON文件
-  writeChatDataToFile(data: ChatData) {
-    try {
-      this.ensureDataDir()
-      data.lastSaveTime = Date.now()
-      const jsonData = JSON.stringify(data, null, 2)
-      fs.writeFileSync(this.dataFilePath, jsonData, 'utf8')
-    } catch (error) {
-      this.logger.error('写入聊天数据失败:', error)
-    }
-  }
-
-  // 清理超量消息
-  cleanExcessMessages(data: ChatData): ChatData {
-    let cleanedCount = 0
-    const cleanedMessages: Record<string, MessageInfo[]> = {}
-
-    for (const [channelKey, messages] of Object.entries(data.messages)) {
-      if (messages.length > this.config.maxMessagesPerChannel) {
-        const sortedMessages = [...messages].sort((a, b) => a.timestamp - b.timestamp)
-        const keptMessages = sortedMessages.slice(-this.config.maxMessagesPerChannel)
-        cleanedCount += messages.length - keptMessages.length
-        cleanedMessages[channelKey] = keptMessages
-        this.logInfo(`频道 ${channelKey} 清理了 ${messages.length - keptMessages.length} 条旧消息，保留最新 ${keptMessages.length} 条`)
-      } else {
-        cleanedMessages[channelKey] = messages
-      }
-    }
-
-    if (cleanedCount > 0) {
-      this.logInfo('总共清理超量消息:', cleanedCount, '条')
-    }
-
-    return {
-      ...data,
-      messages: cleanedMessages
-    }
-  }
-
-  // 添加消息到JSON文件（使用锁机制防止并发冲突）
-  async addMessageToFile(messageInfo: MessageInfo) {
-    this.fileOperationLock = this.fileOperationLock.then(async () => {
-      const data = this.readChatDataFromFile()
-      const channelKey = `${messageInfo.selfId}:${messageInfo.channelId}`
-
-      if (!data.messages[channelKey]) {
-        data.messages[channelKey] = []
-      }
-
-      // 检查消息是否已存在
-      const existingMessage = data.messages[channelKey].find(m => m.id === messageInfo.id)
-      if (existingMessage) {
-        this.logInfo('消息已存在，跳过保存:', {
-          channelKey: channelKey,
-          messageId: messageInfo.id,
-          existingType: existingMessage.type,
-          existingContent: existingMessage.content,
-          newType: messageInfo.type,
-          newContent: messageInfo.content
-        })
-        return
-      }
-
-      if (!messageInfo.timestamp) {
-        messageInfo.timestamp = Date.now()
-      }
-
-      // 清理消息中的base64内容
-      const cleanedMessageInfo = this.utils.cleanBase64Content(messageInfo) as MessageInfo
-
-      const beforeCount = data.messages[channelKey].length
-      data.messages[channelKey].push(cleanedMessageInfo)
-      const afterCount = data.messages[channelKey].length
-
-      // 限制消息数量 - 保留最新的消息
-      if (data.messages[channelKey].length > this.config.maxMessagesPerChannel) {
-        data.messages[channelKey].sort((a, b) => a.timestamp - b.timestamp)
-        const removedCount = data.messages[channelKey].length - this.config.maxMessagesPerChannel
-        data.messages[channelKey] = data.messages[channelKey].slice(-this.config.maxMessagesPerChannel)
-        this.logInfo(`频道 ${channelKey} 达到消息上限，清理了 ${removedCount} 条旧消息`)
-      }
-
-      this.writeChatDataToFile(data)
-
-      const isCommandMessage = messageInfo.content?.startsWith('++') || messageInfo.content?.startsWith('.')
-
-      this.logInfo('添加消息到文件:', {
-        channelKey: channelKey,
-        messageId: messageInfo.id,
-        content: messageInfo.content,
-        type: messageInfo.type,
-        userId: messageInfo.userId,
-        username: messageInfo.username,
-        timestamp: messageInfo.timestamp,
-        isCommandMessage: isCommandMessage,
-        消息数变化: `${beforeCount} -> ${afterCount} -> ${data.messages[channelKey].length}`
-      })
-
-    }).catch(error => {
-      this.logger.error('保存消息时发生错误:', error)
-    })
-
-    await this.fileOperationLock
-  }
-
-  private logInfo(...args: any[]) {
-    if (this.config.loggerinfo) {
-      (this.logger.info as (...args: any[]) => void)(...args)
-    }
-  }
-}
+import { ChatData, MessageInfo } from './types'
+import { Context, Logger } from 'koishi'
+import { Config } from './config'
+import { Utils } from './utils'
+
+import path from 'node:path'
+import fs from 'node:fs'
+
+export class FileManager {
+  private dataFilePath: string
+  private fileOperationLock = Promise.resolve()
+  private logger: Logger
+  private utils: Utils
+
+  private memoryCache: ChatData | null = null
+
+  private pendingMessages: MessageInfo[] = []
+
+  private writeTimer: (() => void) | null = null
+
+  private readonly WRITE_DEBOUNCE_MS = 1000
+
+  constructor(
+    private ctx: Context,
+    private config: Config
+  ) {
+    this.dataFilePath = path.resolve(ctx.baseDir, 'data', 'chat-patch', 'chat-data.json')
+    this.logger = ctx.logger('chat-patch')
+    this.utils = new Utils(config)
+
+    this.memoryCache = this.readChatDataFromFile()
+  }
+
+  private ensureDataDir() {
+    const dir = path.dirname(this.dataFilePath)
+    if (!fs.existsSync(dir)) {
+      fs.mkdirSync(dir, { recursive: true })
+    }
+  }
+
+  readChatDataFromFile(): ChatData {
+
+    if (this.memoryCache) {
+      return this.memoryCache
+    }
+
+    process.nextTick(() => {
+      try {
+        if (fs.existsSync(this.dataFilePath)) {
+          const jsonData = fs.readFileSync(this.dataFilePath, 'utf8')
+          const data = JSON.parse(jsonData)
+          this.memoryCache = {
+            bots: data.bots || {},
+            channels: data.channels || {},
+            messages: data.messages || {},
+            pinnedBots: data.pinnedBots || [],
+            pinnedChannels: data.pinnedChannels || [],
+            lastSaveTime: data.lastSaveTime
+          }
+        }
+      } catch (error) {
+        this.logger.error('读取聊天数据失败:', error)
+      }
+    })
+
+    this.memoryCache = {
+      bots: {},
+      channels: {},
+      messages: {},
+      pinnedBots: [],
+      pinnedChannels: []
+    }
+    return this.memoryCache
+  }
+
+  writeChatDataToFile(data: ChatData) {
+
+    data.lastSaveTime = Date.now()
+    this.memoryCache = data
+
+    process.nextTick(() => {
+      try {
+        this.ensureDataDir()
+        const jsonData = JSON.stringify(data, null, 2)
+        fs.writeFileSync(this.dataFilePath, jsonData, 'utf8')
+      } catch (error) {
+        this.logger.error('写入聊天数据失败:', error)
+      }
+    })
+  }
+
+  private scheduleWrite() {
+
+    if (this.writeTimer) {
+      this.writeTimer()
+      this.writeTimer = null
+    }
+
+    this.writeTimer = this.ctx.setTimeout(() => {
+
+      process.nextTick(() => {
+        this.flushPendingMessages()
+      })
+      this.writeTimer = null
+    }, this.WRITE_DEBOUNCE_MS)
+  }
+
+  private flushPendingMessages() {
+    if (this.pendingMessages.length === 0) return
+
+    const messagesToWrite = [...this.pendingMessages]
+    this.pendingMessages = []
+
+    const data = this.memoryCache || this.readChatDataFromFile()
+
+    for (const messageInfo of messagesToWrite) {
+      const channelKey = `${messageInfo.selfId}:${messageInfo.channelId}`
+
+      if (!data.messages[channelKey]) {
+        data.messages[channelKey] = []
+      }
+
+      const existingMessage = data.messages[channelKey].find(m => m.id === messageInfo.id)
+      if (existingMessage) {
+        continue
+      }
+
+      if (!messageInfo.timestamp) {
+        messageInfo.timestamp = Date.now()
+      }
+
+      const cleanedMessageInfo = this.utils.cleanBase64Content(messageInfo) as MessageInfo
+      data.messages[channelKey].push(cleanedMessageInfo)
+
+      if (data.messages[channelKey].length > this.config.maxMessagesPerChannel) {
+        data.messages[channelKey].sort((a, b) => a.timestamp - b.timestamp)
+        data.messages[channelKey] = data.messages[channelKey].slice(-this.config.maxMessagesPerChannel)
+      }
+    }
+
+    this.writeChatDataToFile(data)
+    this.logInfo(`批量写入 ${messagesToWrite.length} 条消息`)
+  }
+
+  cleanExcessMessages(data: ChatData): ChatData {
+    let cleanedCount = 0
+    const cleanedMessages: Record<string, MessageInfo[]> = {}
+
+    for (const [channelKey, messages] of Object.entries(data.messages)) {
+      if (messages.length > this.config.maxMessagesPerChannel) {
+        const sortedMessages = [...messages].sort((a, b) => a.timestamp - b.timestamp)
+        const keptMessages = sortedMessages.slice(-this.config.maxMessagesPerChannel)
+        cleanedCount += messages.length - keptMessages.length
+        cleanedMessages[channelKey] = keptMessages
+        this.logInfo(`频道 ${channelKey} 清理了 ${messages.length - keptMessages.length} 条旧消息，保留最新 ${keptMessages.length} 条`)
+      } else {
+        cleanedMessages[channelKey] = messages
+      }
+    }
+
+    if (cleanedCount > 0) {
+      this.logInfo('总共清理超量消息:', cleanedCount, '条')
+    }
+
+    return {
+      ...data,
+      messages: cleanedMessages
+    }
+  }
+
+  async addMessageToFile(messageInfo: MessageInfo) {
+
+    this.pendingMessages.push(messageInfo)
+
+    const data = this.memoryCache || this.readChatDataFromFile()
+    const channelKey = `${messageInfo.selfId}:${messageInfo.channelId}`
+
+    if (!data.messages[channelKey]) {
+      data.messages[channelKey] = []
+    }
+
+    const existingMessage = data.messages[channelKey].find(m => m.id === messageInfo.id)
+    if (!existingMessage) {
+      if (!messageInfo.timestamp) {
+        messageInfo.timestamp = Date.now()
+      }
+      const cleanedMessageInfo = this.utils.cleanBase64Content(messageInfo) as MessageInfo
+      data.messages[channelKey].push(cleanedMessageInfo)
+
+      if (data.messages[channelKey].length > this.config.maxMessagesPerChannel) {
+        data.messages[channelKey].sort((a, b) => a.timestamp - b.timestamp)
+        data.messages[channelKey] = data.messages[channelKey].slice(-this.config.maxMessagesPerChannel)
+      }
+
+      this.memoryCache = data
+    }
+
+    this.scheduleWrite()
+  }
+
+  dispose() {
+    if (this.writeTimer) {
+      this.writeTimer()
+      this.writeTimer = null
+    }
+
+    this.flushPendingMessages()
+  }
+
+  private logInfo(...args: any[]) {
+    if (this.config.loggerinfo) {
+      (this.logger.info as (...args: any[]) => void)(...args)
+    }
+  }
+}

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('插件已卸载，所有待处理的消息已写入')
+  })
 }