koishi-plugin-chat-analyse 0.3.6 → 0.4.0

package/lib/Collector.d.ts

@@ -28,6 +28,12 @@ declare module 'koishi' {
             content: string;
             timestamp: Date;
         };
+        analyse_at: {
+            uid: number;
+            target: string;
+            content: string;
+            timestamp: Date;
+        };
     }
 }
 /**
@@ -44,6 +50,7 @@ export declare class Collector {
     private msgStatBuffer;
     private cmdStatBuffer;
     private oriCacheBuffer;
+    private whoAtBuffer;
     private userCache;
     private pendingUserRequests;
     private flushInterval;

package/lib/WhoAt.d.ts

@@ -0,0 +1,34 @@
+import { Context, Command } from 'koishi';
+import { Config } from './index';
+/**
+ * @class WhoAt
+ * @description
+ * 负责处理与“谁@我”相关的功能。
+ * 该类会注册一个 'whoatme' 子命令，允许用户查询在何时被谁提及。
+ * 查询结果将以合并转发的形式发送给用户。
+ * 此外，该类还包含一个定时任务，用于定期清理数据库中旧的@记录。
+ */
+export declare class WhoAt {
+    private ctx;
+    private config;
+    /**
+     * WhoAt 类的构造函数。
+     * @param {Context} ctx - Koishi 的插件上下文，用于访问框架核心功能和数据库等服务。
+     * @param {Config} config - 插件的配置对象，包含如记录保留天数等设置。
+     */
+    constructor(ctx: Context, config: Config);
+    /**
+     * @private
+     * @method setupCleanupTask
+     * @description 设置一个定时清理任务。
+     * 此任务会根据配置中的 `retentionDays` 定期删除过期的@记录，以防止数据库膨胀。
+     */
+    private setupCleanupTask;
+    /**
+     * @public
+     * @method registerCommand
+     * @description 在主 `analyse` 命令下注册 `whoatme` 子命令。
+     * @param {Command} analyse - 用户传入的主 `analyse` 命令实例，`whoatme` 将作为其子命令。
+     */
+    registerCommand(analyse: Command): void;
+}

package/lib/index.d.ts

@@ -16,6 +16,8 @@ export interface Config {
     enableMsgStat: boolean;
     enableRankStat: boolean;
     enableOriRecord: boolean;
+    enableWhoAt: boolean;
+    retentionDays: number;
 }
 /**
  * @const {Schema<Config>} Config

package/lib/index.js

@@ -27,7 +27,7 @@ __export(src_exports, {
   using: () => using
 });
 module.exports = __toCommonJS(src_exports);
-var import_koishi4 = require("koishi");
+var import_koishi5 = require("koishi");
 
 // src/Collector.ts
 var import_koishi = require("koishi");
@@ -59,6 +59,7 @@ var Collector = class _Collector {
   msgStatBuffer = /* @__PURE__ */ new Map();
   cmdStatBuffer = /* @__PURE__ */ new Map();
   oriCacheBuffer = [];
+  whoAtBuffer = [];
   // 用户缓存
   userCache = /* @__PURE__ */ new Map();
   pendingUserRequests = /* @__PURE__ */ new Map();
@@ -97,6 +98,14 @@ var Collector = class _Collector {
         timestamp: "timestamp"
       }, { primary: "id", autoInc: true, indexes: ["uid", "timestamp"] });
     }
+    if (this.config.enableWhoAt) {
+      this.ctx.model.extend("analyse_at", {
+        uid: "unsigned",
+        target: "string",
+        content: "text",
+        timestamp: "timestamp"
+      }, { indexes: ["target", "uid"] });
+    }
   }
   /**
    * @private
@@ -136,6 +145,16 @@ var Collector = class _Collector {
           this.msgStatBuffer.set(key, { uid, type, hour: hourStart, count: 1, timestamp: messageTime });
         }
       }
+      if (this.config.enableWhoAt) {
+        const atElements = elements.filter((e) => e.type === "at");
+        if (atElements.length > 0) {
+          const sanitizedContent = this.sanitizeContent(elements);
+          for (const atElement of atElements) {
+            const targetId = atElement.attrs.id;
+            if (targetId && targetId !== userId) this.whoAtBuffer.push({ uid, target: targetId, content: sanitizedContent, timestamp: messageTime });
+          }
+        }
+      }
       if (this.config.enableOriRecord) {
         this.oriCacheBuffer.push({
           uid,
@@ -225,10 +244,12 @@ var Collector = class _Collector {
   async flushBuffers() {
     const cmdBufferToFlush = Array.from(this.cmdStatBuffer.values());
     const msgBufferToFlush = Array.from(this.msgStatBuffer.values());
-    const advancedBufferToFlush = this.oriCacheBuffer;
+    const oriCacheBufferToFlush = this.oriCacheBuffer;
+    const whoAtBufferToFlush = this.whoAtBuffer;
     this.cmdStatBuffer.clear();
     this.msgStatBuffer.clear();
     this.oriCacheBuffer = [];
+    this.whoAtBuffer = [];
     try {
       if (cmdBufferToFlush.length > 0) {
         await this.ctx.database.upsert(
@@ -253,7 +274,8 @@ var Collector = class _Collector {
           }))
         );
       }
-      if (advancedBufferToFlush.length > 0) await this.ctx.database.upsert("analyse_cache", advancedBufferToFlush);
+      if (whoAtBufferToFlush.length > 0) await this.ctx.database.upsert("analyse_at", whoAtBufferToFlush);
+      if (oriCacheBufferToFlush.length > 0) await this.ctx.database.upsert("analyse_cache", oriCacheBufferToFlush);
     } catch (error) {
       this.ctx.logger.error("写入数据出错:", error);
     }
@@ -423,16 +445,16 @@ var Renderer = class {
     const { title, time, list } = data;
     if (!list?.length) return "暂无数据可供渲染";
     let totalValueForPercent = 0;
-    const countHeaderIndex = headers?.findIndex((h2) => ["总计发言", "条数", "次数", "数量"].includes(h2));
+    const countHeaderIndex = headers?.findIndex((h3) => ["总计发言", "条数", "次数", "数量"].includes(h3));
     if (countHeaderIndex > -1) {
       totalValueForPercent = list.reduce((sum, row) => sum + (Number(row[countHeaderIndex]) || 0), 0);
     }
     const totalCount = data.total || totalValueForPercent;
-    const tableHeadHtml = headers?.length > 0 ? `<thead><tr><th class="rank-cell">#</th>${headers.map((h2, i) => {
+    const tableHeadHtml = headers?.length > 0 ? `<thead><tr><th class="rank-cell">#</th>${headers.map((h3, i) => {
       const firstCell = list[0]?.[i];
-      const isRightAlign = typeof firstCell === "number" || firstCell instanceof Date || h2.includes("占比");
+      const isRightAlign = typeof firstCell === "number" || firstCell instanceof Date || h3.includes("占比");
       const alignClass = isRightAlign ? "header-right-align" : "name-header";
-      return `<th class="${alignClass}">${h2}</th>`;
+      return `<th class="${alignClass}">${h3}</th>`;
     }).join("")}</tr></thead>` : "";
     const tableRowsHtml = list.map((row, index) => {
       const rank = index + 1;
@@ -758,6 +780,73 @@ var Stat = class {
   }
 };
 
+// src/WhoAt.ts
+var import_koishi4 = require("koishi");
+var WhoAt = class {
+  /**
+   * WhoAt 类的构造函数。
+   * @param {Context} ctx - Koishi 的插件上下文，用于访问框架核心功能和数据库等服务。
+   * @param {Config} config - 插件的配置对象，包含如记录保留天数等设置。
+   */
+  constructor(ctx, config) {
+    this.ctx = ctx;
+    this.config = config;
+    this.setupCleanupTask();
+  }
+  static {
+    __name(this, "WhoAt");
+  }
+  /**
+   * @private
+   * @method setupCleanupTask
+   * @description 设置一个定时清理任务。
+   * 此任务会根据配置中的 `retentionDays` 定期删除过期的@记录，以防止数据库膨胀。
+   */
+  setupCleanupTask() {
+    if (this.config.retentionDays > 0) {
+      this.ctx.cron("0 0 * * *", async () => {
+        try {
+          const cutoffDate = new Date(Date.now() - this.config.retentionDays * import_koishi4.Time.day);
+          await this.ctx.database.remove("analyse_at", { timestamp: { $lt: cutoffDate } });
+        } catch (error) {
+          this.ctx.logger.error("清理 @ 历史记录出错:", error);
+        }
+      });
+    }
+  }
+  /**
+   * @public
+   * @method registerCommand
+   * @description 在主 `analyse` 命令下注册 `whoatme` 子命令。
+   * @param {Command} analyse - 用户传入的主 `analyse` 命令实例，`whoatme` 将作为其子命令。
+   */
+  registerCommand(analyse) {
+    analyse.subcommand("whoatme", "谁 @ 我").action(async ({ session }) => {
+      if (!session.userId) return "无法获取用户信息";
+      try {
+        const records = await this.ctx.database.select("analyse_at").where({ target: session.userId }).orderBy("timestamp", "asc").limit(100).execute();
+        if (records.length === 0) return "暂无 @ 记录";
+        const uids = [...new Set(records.map((r) => r.uid))];
+        const users = await this.ctx.database.select("analyse_user", { uid: { $in: uids } }).project(["uid", "userName", "userId"]).execute();
+        const userInfoMap = new Map(users.map((u) => [u.uid, { name: u.userName, id: u.userId }]));
+        const messageElements = records.map((record) => {
+          const senderInfo = userInfoMap.get(record.uid);
+          const userId = senderInfo?.id;
+          const authorElement = (0, import_koishi4.h)("author", { userId, name: userId });
+          const contentElement = import_koishi4.h.text(record.content);
+          return (0, import_koishi4.h)("message", {}, [authorElement, contentElement]);
+        });
+        if (messageElements.length === 0) return "暂无有效 @ 记录";
+        const forwardMessage = (0, import_koishi4.h)("message", { forward: true }, messageElements);
+        await session.send(forwardMessage);
+      } catch (error) {
+        this.ctx.logger.error("查询 @ 记录时失败:", error);
+        return "查询失败，请稍后再试";
+      }
+    });
+  }
+};
+
 // src/index.ts
 var usage = `
 <div style="border-radius: 10px; border: 1px solid #ddd; padding: 16px; margin-bottom: 20px; box-shadow: 0 2px 5px rgba(0,0,0,0.1);">
@@ -772,22 +861,27 @@ var usage = `
 </div>
 `;
 var name = "chat-analyse";
-var using = ["database", "puppeteer"];
-var Config = import_koishi4.Schema.intersect([
-  import_koishi4.Schema.object({
-    enableListener: import_koishi4.Schema.boolean().default(true).description("启用消息监听"),
-    enableOriRecord: import_koishi4.Schema.boolean().default(true).description("启用原始记录")
+var using = ["database", "puppeteer", "cron"];
+var Config = import_koishi5.Schema.intersect([
+  import_koishi5.Schema.object({
+    enableListener: import_koishi5.Schema.boolean().default(true).description("启用消息监听"),
+    enableOriRecord: import_koishi5.Schema.boolean().default(true).description("启用原始记录")
   }).description("监听配置"),
-  import_koishi4.Schema.object({
-    enableCmdStat: import_koishi4.Schema.boolean().default(true).description("启用命令统计"),
-    enableMsgStat: import_koishi4.Schema.boolean().default(true).description("启用消息统计"),
-    enableRankStat: import_koishi4.Schema.boolean().default(true).description("启用发言排行")
-  }).description("命令配置")
+  import_koishi5.Schema.object({
+    enableCmdStat: import_koishi5.Schema.boolean().default(true).description("启用命令统计"),
+    enableMsgStat: import_koishi5.Schema.boolean().default(true).description("启用消息统计"),
+    enableRankStat: import_koishi5.Schema.boolean().default(true).description("启用发言排行")
+  }).description("命令配置"),
+  import_koishi5.Schema.object({
+    enableWhoAt: import_koishi5.Schema.boolean().default(true).description("启用 @ 记录"),
+    retentionDays: import_koishi5.Schema.number().min(0).default(7).description("保留天数")
+  }).description("@ 记录配置")
 ]);
 function apply(ctx, config) {
   if (config.enableListener) new Collector(ctx, config);
   const analyse = ctx.command("analyse", "聊天记录分析");
   new Stat(ctx, config).registerCommands(analyse);
+  if (config.enableWhoAt) new WhoAt(ctx, config).registerCommand(analyse);
 }
 __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.3.6",
+  "version": "0.4.0",
   "contributors": [
     "Yis_Rime <yis_rime@outlook.com>"
   ],