koishi-plugin-chat-analyse 0.0.1 → 0.0.2

package/lib/collector.d.ts

@@ -1,24 +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 监听所有消息和命令，将消息内容存入 'analyse_origin_msg' 表，将命令存入 'analyse_origin_cmd' 表。
  */
-export declare class MessageCollector {
+export declare class Collector {
     private ctx;
     /**
      * @param ctx Koishi 的上下文对象，用于访问模型和事件系统

package/lib/index.js

@@ -29,13 +29,13 @@ module.exports = __toCommonJS(src_exports);
 var import_koishi = require("koishi");
 
 // src/collector.ts
-var MessageCollector = class {
+var Collector = class {
   /**
    * @param ctx Koishi 的上下文对象，用于访问模型和事件系统
    */
   constructor(ctx) {
     this.ctx = ctx;
-    ctx.model.extend("chat_origin", {
+    ctx.model.extend("analyse_origin_msg", {
       id: "unsigned",
       channelId: "string",
       userId: "string",
@@ -44,29 +44,52 @@ var MessageCollector = class {
     }, {
       autoInc: true
     });
-    ctx.on("message", async (session) => {
-      const messageParts = session.elements.map((element) => {
-        switch (element.type) {
-          case "text":
-            return element.attrs.content;
-          case "at":
-            return `[at=${element.attrs.id}]`;
-          default:
-            return "";
-        }
-      });
-      const sanitizedContent = messageParts.join("").trim();
-      if (!sanitizedContent) return;
-      await ctx.database.create("chat_origin", {
-        channelId: session.channelId,
-        userId: session.userId,
-        content: sanitizedContent,
-        timestamp: new Date(session.timestamp)
+    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) => {
+      if (!argv.command) return;
+      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", (session) => {
+      setTimeout(async () => {
+        if (session.argv) return;
+        const messageParts = session.elements.map((element) => {
+          switch (element.type) {
+            case "text":
+              return element.attrs.content;
+            case "at":
+              return `[at=${element.attrs.id}]`;
+            default:
+              return "";
+          }
+        });
+        const sanitizedContent = messageParts.join("").trim();
+        if (!sanitizedContent) return;
+        await ctx.database.create("analyse_origin_msg", {
+          channelId: session.channelId,
+          userId: session.userId,
+          content: sanitizedContent,
+          timestamp: new Date(session.timestamp)
+        });
+      }, 0);
+    });
   }
   static {
-    __name(this, "MessageCollector");
+    __name(this, "Collector");
   }
 };
 
@@ -75,7 +98,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.2",
   "contributors": [
     "Yis_Rime <yis_rime@outlook.com>"
   ],