koishi-plugin-chat-analyse 0.0.1 → 0.0.3

package/lib/collector.d.ts

@@ -1,27 +1,32 @@
 import { Context } from 'koishi';
 /**
  * @file collector.ts
- * @description 用于将收到的消息持久化存储到数据库中。
+ * @description 通过独立的事件监听器，分别持久化存储收到的普通消息和已确认的命令。
  */
 declare module 'koishi' {
     interface Tables {
-        chat_origin: {
+        analyse_origin_msg: {
             id: number;
             channelId: string;
             userId: string;
             content: string;
             timestamp: Date;
         };
+        analyse_origin_cmd: {
+            id: number;
+            channelId: string;
+            userId: string;
+            command: string;
+            content: string;
+            timestamp: Date;
+        };
     }
 }
 /**
- * @class MessageCollector
- * @description 监听所有消息，将其内容序列化后存入名为 'chat_origin' 的数据库表中。
+ * @class Collector
+ * @description 使用双监听器，精确捕获并存储消息和命令。
  */
-export declare class MessageCollector {
+export declare class Collector {
     private ctx;
-    /**
-     * @param ctx Koishi 的上下文对象，用于访问模型和事件系统
-     */
     constructor(ctx: Context);
 }

package/lib/index.js

@@ -29,35 +29,48 @@ module.exports = __toCommonJS(src_exports);
 var import_koishi = require("koishi");
 
 // src/collector.ts
-var MessageCollector = class {
-  /**
-   * @param ctx Koishi 的上下文对象，用于访问模型和事件系统
-   */
+var Collector = class {
   constructor(ctx) {
     this.ctx = ctx;
-    ctx.model.extend("chat_origin", {
+    ctx.model.extend("analyse_origin_msg", {
       id: "unsigned",
       channelId: "string",
       userId: "string",
       content: "text",
       timestamp: "timestamp"
-    }, {
-      autoInc: true
+    }, { autoInc: true });
+    ctx.model.extend("analyse_origin_cmd", {
+      id: "unsigned",
+      channelId: "string",
+      userId: "string",
+      command: "string",
+      content: "text",
+      timestamp: "timestamp"
+    }, { autoInc: true });
+    ctx.on("command/before-execute", async (argv) => {
+      await ctx.database.create("analyse_origin_cmd", {
+        channelId: argv.session.channelId,
+        userId: argv.session.userId,
+        command: argv.command.name,
+        content: argv.session.content,
+        timestamp: new Date(argv.session.timestamp)
+      });
     });
     ctx.on("message", async (session) => {
-      const messageParts = session.elements.map((element) => {
+      if (session.argv?.command) return;
+      const content = session.elements.map((element) => {
         switch (element.type) {
           case "text":
             return element.attrs.content;
-          case "at":
-            return `[at=${element.attrs.id}]`;
+          case "img":
+            return element.attrs.summary === "[动画表情]" ? "[gif]" : `[${element.type}]`;
           default:
-            return "";
+            return `[${element.type}]`;
         }
-      });
-      const sanitizedContent = messageParts.join("").trim();
+      }).join("");
+      const sanitizedContent = content.trim();
       if (!sanitizedContent) return;
-      await ctx.database.create("chat_origin", {
+      await ctx.database.create("analyse_origin_msg", {
         channelId: session.channelId,
         userId: session.userId,
         content: sanitizedContent,
@@ -66,7 +79,7 @@ var MessageCollector = class {
     });
   }
   static {
-    __name(this, "MessageCollector");
+    __name(this, "Collector");
   }
 };
 
@@ -75,7 +88,7 @@ var name = "chat-analyse";
 var Config = import_koishi.Schema.object({});
 var using = ["database"];
 function apply(ctx) {
-  new MessageCollector(ctx);
+  new Collector(ctx);
 }
 __name(apply, "apply");
 // Annotate the CommonJS export names for ESM import in node:

package/package.json

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