koishi-plugin-chat-analyse 0.1.0 → 0.1.1

package/lib/collector.d.ts

@@ -1,7 +1,7 @@
 import { Context } from 'koishi';
 /**
  * @file collector.ts
- * @description 通过独立的事件监听器，分别持久化存储收到的普通消息和已确认的命令。
+ * @description 通过独立的事件监听器，分别持久化存储收到的普通消息和已确认的命令，并高效地维护ID与名称的映射。
  */
 declare module 'koishi' {
     interface Tables {
@@ -20,13 +20,23 @@ declare module 'koishi' {
             content: string;
             timestamp: Date;
         };
+        analyse_name_map: {
+            id: number;
+            channelId: string;
+            channelName: string;
+            userId: string;
+            userName: string;
+            timestamp: Date;
+        };
     }
 }
 /**
  * @class Collector
- * @description 使用双监听器，精确捕获并存储消息和命令。
+ * @description 使用双监听器，精确捕获并存储消息和命令，并维护一个ID到名称的映射表。
  */
 export declare class Collector {
     private ctx;
+    private nameCache;
     constructor(ctx: Context);
+    private updateName;
 }

package/lib/index.js

@@ -39,7 +39,10 @@ var Collector = class {
       userId: "string",
       content: "text",
       timestamp: "timestamp"
-    }, { autoInc: true });
+    }, {
+      autoInc: true,
+      indexes: ["channelId", "userId", "timestamp"]
+    });
     ctx.model.extend("analyse_origin_cmd", {
       id: "unsigned",
       channelId: "string",
@@ -47,10 +50,26 @@ var Collector = class {
       command: "string",
       content: "text",
       timestamp: "timestamp"
-    }, { autoInc: true });
+    }, {
+      autoInc: true,
+      indexes: ["channelId", "userId", "command", "timestamp"]
+    });
+    ctx.model.extend("analyse_name_map", {
+      id: "unsigned",
+      channelId: "string",
+      channelName: "string",
+      userId: "string",
+      userName: "string",
+      timestamp: "timestamp"
+    }, {
+      autoInc: true,
+      unique: [["channelId", "userId"]]
+    });
     ctx.on("command/before-execute", async (argv) => {
+      const effectiveId = argv.session.channelId || argv.session.guildId;
+      if (!effectiveId) return;
       await ctx.database.create("analyse_origin_cmd", {
-        channelId: argv.session.channelId,
+        channelId: effectiveId,
         userId: argv.session.userId,
         command: argv.command.name,
         content: argv.session.content,
@@ -58,7 +77,27 @@ var Collector = class {
       });
     });
     ctx.on("message", async (session) => {
+      const { userId, author } = session;
+      const effectiveId = session.channelId || session.guildId;
+      const { channelName, guildName } = session;
+      const effectiveName = channelName || guildName;
+      const currentName = author?.name || author?.nick;
+      if (currentName && effectiveName && effectiveId && userId) {
+        const cacheKey = `${effectiveId}:${userId}`;
+        const cachedName = this.nameCache.get(cacheKey);
+        if (cachedName && currentName !== cachedName) await this.updateName(effectiveId, effectiveName, userId, currentName);
+        else if (!cachedName) {
+          const dbRecord = await ctx.database.get("analyse_name_map", {
+            channelId: effectiveId,
+            userId
+          });
+          const dbName = dbRecord[0]?.userName;
+          if (!dbName || currentName !== dbName) await this.updateName(effectiveId, effectiveName, userId, currentName);
+          else this.nameCache.set(cacheKey, currentName);
+        }
+      }
       if (session.argv?.command) return;
+      if (!effectiveId) return;
       const content = session.elements.map((element) => {
         switch (element.type) {
           case "text":
@@ -74,7 +113,7 @@ var Collector = class {
       const sanitizedContent = content.trim();
       if (!sanitizedContent) return;
       await ctx.database.create("analyse_origin_msg", {
-        channelId: session.channelId,
+        channelId: effectiveId,
         userId: session.userId,
         content: sanitizedContent,
         timestamp: new Date(session.timestamp)
@@ -84,6 +123,17 @@ var Collector = class {
   static {
     __name(this, "Collector");
   }
+  nameCache = /* @__PURE__ */ new Map();
+  async updateName(channelId, channelName, userId, userName) {
+    await this.ctx.database.upsert("analyse_name_map", [{
+      channelId,
+      channelName,
+      userId,
+      userName,
+      timestamp: /* @__PURE__ */ new Date()
+    }]);
+    this.nameCache.set(`${channelId}:${userId}`, userName);
+  }
 };
 
 // src/index.ts

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-chat-analyse",
   "description": "聊天记录分析",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "contributors": [
     "Yis_Rime <yis_rime@outlook.com>"
   ],