koishi-plugin-chat-patch 2.3.0 → 2.4.0

package/lib/file-manager.d.ts

@@ -4,16 +4,23 @@ import { Config } from './config';
 export declare class FileManager {
     private ctx;
     private config;
-    private dataFilePath;
-    private fileOperationLock;
+    private chatHistoryDir;
+    private metadataFilePath;
     private logger;
     private utils;
     private memoryCache;
     private pendingMessages;
-    private writeTimer;
+    private writeTimers;
     private readonly WRITE_DEBOUNCE_MS;
     constructor(ctx: Context, config: Config);
-    private ensureDataDir;
+    private cleanupOldDataFiles;
+    private ensureDir;
+    private getChannelFilePath;
+    private readChannelMessages;
+    private writeChannelMessages;
+    private readMetadata;
+    private writeMetadata;
+    private loadAllChannelMessages;
     readChatDataFromFile(): ChatData;
     writeChatDataToFile(data: ChatData): void;
     private scheduleWrite;

package/lib/index.js

@@ -540,27 +540,149 @@ var FileManager = class {
   constructor(ctx, config) {
     this.ctx = ctx;
     this.config = config;
-    this.dataFilePath = import_node_path2.default.resolve(ctx.baseDir, "data", "chat-patch", "chat-data.json");
+    const baseDir = import_node_path2.default.resolve(ctx.baseDir, "data", "chat-patch");
+    this.chatHistoryDir = import_node_path2.default.join(baseDir, "chat-history");
+    this.metadataFilePath = import_node_path2.default.join(baseDir, "metadata.json");
     this.logger = ctx.logger("chat-patch");
     this.utils = new Utils(config);
+    this.cleanupOldDataFiles(baseDir);
     this.memoryCache = this.readChatDataFromFile();
   }
   static {
     __name(this, "FileManager");
   }
-  dataFilePath;
-  fileOperationLock = Promise.resolve();
+  chatHistoryDir;
+  // 聊天记录根目录
+  metadataFilePath;
+  // 元数据文件路径（存储bots、channels、pinned等信息）
   logger;
   utils;
   memoryCache = null;
-  pendingMessages = [];
-  writeTimer = null;
+  pendingMessages = /* @__PURE__ */ new Map();
+  // 按channelKey分组的待写入消息
+  writeTimers = /* @__PURE__ */ new Map();
+  // 每个频道独立的写入定时器
   WRITE_DEBOUNCE_MS = 1e3;
-  ensureDataDir() {
-    const dir = import_node_path2.default.dirname(this.dataFilePath);
-    if (!import_node_fs2.default.existsSync(dir)) {
-      import_node_fs2.default.mkdirSync(dir, { recursive: true });
+  // 清理旧版本的JSON文件
+  cleanupOldDataFiles(baseDir) {
+    const oldFiles = ["chat-data.json", "data.json", "messages.json"];
+    for (const fileName of oldFiles) {
+      const filePath = import_node_path2.default.join(baseDir, fileName);
+      if (import_node_fs2.default.existsSync(filePath)) {
+        try {
+          import_node_fs2.default.unlinkSync(filePath);
+          this.logger.info(`已删除旧版本数据文件: ${fileName}`);
+        } catch (error) {
+          this.logger.warn(`删除旧版本数据文件失败: ${fileName}`, error);
+        }
+      }
+    }
+  }
+  // 确保目录存在
+  ensureDir(dirPath) {
+    if (!import_node_fs2.default.existsSync(dirPath)) {
+      import_node_fs2.default.mkdirSync(dirPath, { recursive: true });
+    }
+  }
+  // 获取频道消息文件路径
+  getChannelFilePath(selfId, channelId) {
+    const botDir = import_node_path2.default.join(this.chatHistoryDir, selfId);
+    this.ensureDir(botDir);
+    return import_node_path2.default.join(botDir, `${channelId}.json`);
+  }
+  // 读取单个频道的消息
+  readChannelMessages(selfId, channelId) {
+    const filePath = this.getChannelFilePath(selfId, channelId);
+    if (!import_node_fs2.default.existsSync(filePath)) {
+      return [];
+    }
+    try {
+      const jsonData = import_node_fs2.default.readFileSync(filePath, "utf8");
+      const messages = JSON.parse(jsonData);
+      return Array.isArray(messages) ? messages : [];
+    } catch (error) {
+      this.logger.error(`读取频道消息失败 [${selfId}:${channelId}]:`, error);
+      return [];
+    }
+  }
+  // 写入单个频道的消息
+  writeChannelMessages(selfId, channelId, messages) {
+    const filePath = this.getChannelFilePath(selfId, channelId);
+    try {
+      const jsonData = JSON.stringify(messages, null, 2);
+      import_node_fs2.default.writeFileSync(filePath, jsonData, "utf8");
+    } catch (error) {
+      this.logger.error(`写入频道消息失败 [${selfId}:${channelId}]:`, error);
+    }
+  }
+  // 读取元数据（bots、channels、pinned等）
+  readMetadata() {
+    if (!import_node_fs2.default.existsSync(this.metadataFilePath)) {
+      return {
+        bots: {},
+        channels: {},
+        pinnedBots: [],
+        pinnedChannels: []
+      };
+    }
+    try {
+      const jsonData = import_node_fs2.default.readFileSync(this.metadataFilePath, "utf8");
+      const data = JSON.parse(jsonData);
+      return {
+        bots: data.bots || {},
+        channels: data.channels || {},
+        pinnedBots: data.pinnedBots || [],
+        pinnedChannels: data.pinnedChannels || [],
+        lastSaveTime: data.lastSaveTime
+      };
+    } catch (error) {
+      this.logger.error("读取元数据失败:", error);
+      return {
+        bots: {},
+        channels: {},
+        pinnedBots: [],
+        pinnedChannels: []
+      };
+    }
+  }
+  // 写入元数据
+  writeMetadata(metadata) {
+    try {
+      this.ensureDir(import_node_path2.default.dirname(this.metadataFilePath));
+      const dataToWrite = {
+        ...metadata,
+        lastSaveTime: Date.now()
+      };
+      const jsonData = JSON.stringify(dataToWrite, null, 2);
+      import_node_fs2.default.writeFileSync(this.metadataFilePath, jsonData, "utf8");
+    } catch (error) {
+      this.logger.error("写入元数据失败:", error);
+    }
+  }
+  // 扫描所有频道消息文件并加载到内存
+  loadAllChannelMessages() {
+    const messages = {};
+    if (!import_node_fs2.default.existsSync(this.chatHistoryDir)) {
+      return messages;
+    }
+    try {
+      const botDirs = import_node_fs2.default.readdirSync(this.chatHistoryDir);
+      for (const botId of botDirs) {
+        const botDir = import_node_path2.default.join(this.chatHistoryDir, botId);
+        const stat = import_node_fs2.default.statSync(botDir);
+        if (!stat.isDirectory()) continue;
+        const channelFiles = import_node_fs2.default.readdirSync(botDir);
+        for (const fileName of channelFiles) {
+          if (!fileName.endsWith(".json")) continue;
+          const channelId = fileName.replace(".json", "");
+          const channelKey = `${botId}:${channelId}`;
+          messages[channelKey] = this.readChannelMessages(botId, channelId);
+        }
+      }
+    } catch (error) {
+      this.logger.error("加载频道消息失败:", error);
     }
+    return messages;
   }
   readChatDataFromFile() {
     if (this.memoryCache) {
@@ -568,18 +690,12 @@ var FileManager = class {
     }
     process.nextTick(() => {
       try {
-        if (import_node_fs2.default.existsSync(this.dataFilePath)) {
-          const jsonData = import_node_fs2.default.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
-          };
-        }
+        const metadata = this.readMetadata();
+        const messages = this.loadAllChannelMessages();
+        this.memoryCache = {
+          ...metadata,
+          messages
+        };
       } catch (error) {
         this.logger.error("读取聊天数据失败:", error);
       }
@@ -598,36 +714,45 @@ var FileManager = class {
     this.memoryCache = data;
     process.nextTick(() => {
       try {
-        this.ensureDataDir();
-        const jsonData = JSON.stringify(data, null, 2);
-        import_node_fs2.default.writeFileSync(this.dataFilePath, jsonData, "utf8");
+        const { messages, ...metadata } = data;
+        this.writeMetadata(metadata);
+        for (const [channelKey, channelMessages] of Object.entries(messages)) {
+          const [selfId, channelId] = channelKey.split(":");
+          if (selfId && channelId) {
+            this.writeChannelMessages(selfId, channelId, channelMessages);
+          }
+        }
       } catch (error) {
         this.logger.error("写入聊天数据失败:", error);
       }
     });
   }
-  scheduleWrite() {
-    if (this.writeTimer) {
-      this.writeTimer();
-      this.writeTimer = null;
+  // 为特定频道安排写入
+  scheduleWrite(channelKey) {
+    const existingTimer = this.writeTimers.get(channelKey);
+    if (existingTimer) {
+      existingTimer();
     }
-    this.writeTimer = this.ctx.setTimeout(() => {
+    const timer = this.ctx.setTimeout(() => {
       process.nextTick(() => {
-        this.flushPendingMessages();
+        this.flushPendingMessages(channelKey);
       });
-      this.writeTimer = null;
+      this.writeTimers.delete(channelKey);
     }, this.WRITE_DEBOUNCE_MS);
+    this.writeTimers.set(channelKey, timer);
   }
-  flushPendingMessages() {
-    if (this.pendingMessages.length === 0) return;
-    const messagesToWrite = [...this.pendingMessages];
-    this.pendingMessages = [];
+  // 刷新特定频道的待写入消息
+  flushPendingMessages(channelKey) {
+    const messagesToWrite = this.pendingMessages.get(channelKey);
+    if (!messagesToWrite || messagesToWrite.length === 0) return;
+    this.pendingMessages.delete(channelKey);
     const data = this.memoryCache || this.readChatDataFromFile();
+    const [selfId, channelId] = channelKey.split(":");
+    if (!selfId || !channelId) return;
+    if (!data.messages[channelKey]) {
+      data.messages[channelKey] = [];
+    }
     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;
@@ -637,13 +762,14 @@ var FileManager = class {
       }
       const cleanedMessageInfo = this.utils.cleanBase64Content(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} 条消息`);
+    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.writeChannelMessages(selfId, channelId, data.messages[channelKey]);
+    this.memoryCache = data;
+    this.logInfo(`批量写入 ${messagesToWrite.length} 条消息到频道 ${channelKey}`);
   }
   cleanExcessMessages(data) {
     let cleanedCount = 0;
@@ -668,9 +794,12 @@ var FileManager = class {
     };
   }
   async addMessageToFile(messageInfo) {
-    this.pendingMessages.push(messageInfo);
-    const data = this.memoryCache || this.readChatDataFromFile();
     const channelKey = `${messageInfo.selfId}:${messageInfo.channelId}`;
+    if (!this.pendingMessages.has(channelKey)) {
+      this.pendingMessages.set(channelKey, []);
+    }
+    this.pendingMessages.get(channelKey).push(messageInfo);
+    const data = this.memoryCache || this.readChatDataFromFile();
     if (!data.messages[channelKey]) {
       data.messages[channelKey] = [];
     }
@@ -687,14 +816,14 @@ var FileManager = class {
       }
       this.memoryCache = data;
     }
-    this.scheduleWrite();
+    this.scheduleWrite(channelKey);
   }
   dispose() {
-    if (this.writeTimer) {
-      this.writeTimer();
-      this.writeTimer = null;
+    for (const [channelKey, timer] of this.writeTimers.entries()) {
+      timer();
+      this.flushPendingMessages(channelKey);
     }
-    this.flushPendingMessages();
+    this.writeTimers.clear();
   }
   logInfo(...args) {
     if (this.config.loggerinfo) {

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.3.0",
+  "version": "2.4.0",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [