koishi-plugin-chat-analyse 0.3.6 → 0.4.1

package/lib/Collector.d.ts

@@ -18,6 +18,11 @@ declare module 'koishi' {
         analyse_msg: {
             uid: number;
             type: string;
+            count: number;
+            timestamp: Date;
+        };
+        analyse_rank: {
+            uid: number;
             hour: Date;
             count: number;
             timestamp: Date;
@@ -28,6 +33,12 @@ declare module 'koishi' {
             content: string;
             timestamp: Date;
         };
+        analyse_at: {
+            uid: number;
+            target: string;
+            content: string;
+            timestamp: Date;
+        };
     }
 }
 /**
@@ -42,8 +53,10 @@ export declare class Collector {
     /** @const {number} BUFFER_THRESHOLD - 内存缓存区触发自动刷新的消息数量阈值。 */
     private static readonly BUFFER_THRESHOLD;
     private msgStatBuffer;
+    private rankStatBuffer;
     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");
@@ -55,10 +55,12 @@ var Collector = class _Collector {
   static FLUSH_INTERVAL = 60 * 1e3;
   /** @const {number} BUFFER_THRESHOLD - 内存缓存区触发自动刷新的消息数量阈值。 */
   static BUFFER_THRESHOLD = 100;
-  // 统一的缓冲区
+  // 分离的缓冲区
   msgStatBuffer = /* @__PURE__ */ new Map();
+  rankStatBuffer = /* @__PURE__ */ new Map();
   cmdStatBuffer = /* @__PURE__ */ new Map();
   oriCacheBuffer = [];
+  whoAtBuffer = [];
   // 用户缓存
   userCache = /* @__PURE__ */ new Map();
   pendingUserRequests = /* @__PURE__ */ new Map();
@@ -85,10 +87,15 @@ var Collector = class _Collector {
     this.ctx.model.extend("analyse_msg", {
       uid: "unsigned",
       type: "string",
+      count: "unsigned",
+      timestamp: "timestamp"
+    }, { primary: ["uid", "type"] });
+    this.ctx.model.extend("analyse_rank", {
+      uid: "unsigned",
       hour: "timestamp",
       count: "unsigned",
       timestamp: "timestamp"
-    }, { primary: ["uid", "type", "hour"] });
+    }, { primary: ["uid", "hour"] });
     if (this.config.enableOriRecord) {
       this.ctx.model.extend("analyse_cache", {
         id: "unsigned",
@@ -97,6 +104,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
@@ -125,15 +140,33 @@ var Collector = class _Collector {
         }
       }
       const hourStart = new Date(messageTime.getFullYear(), messageTime.getMonth(), messageTime.getDate(), messageTime.getHours());
+      const rankKey = `${uid}:${hourStart.toISOString()}`;
+      const existingRank = this.rankStatBuffer.get(rankKey);
+      if (existingRank) {
+        existingRank.count++;
+        existingRank.timestamp = messageTime;
+      } else {
+        this.rankStatBuffer.set(rankKey, { uid, hour: hourStart, count: 1, timestamp: messageTime });
+      }
       const uniqueElementTypes = new Set(elements.map((e) => e.type));
       for (const type of uniqueElementTypes) {
-        const key = `${uid}:${type}:${hourStart.toISOString()}`;
+        const key = `${uid}:${type}`;
         const existing = this.msgStatBuffer.get(key);
         if (existing) {
           existing.count++;
           existing.timestamp = messageTime;
         } else {
-          this.msgStatBuffer.set(key, { uid, type, hour: hourStart, count: 1, timestamp: messageTime });
+          this.msgStatBuffer.set(key, { uid, type, 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) {
@@ -225,10 +258,14 @@ 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 rankBufferToFlush = Array.from(this.rankStatBuffer.values());
+    const oriCacheBufferToFlush = this.oriCacheBuffer;
+    const whoAtBufferToFlush = this.whoAtBuffer;
     this.cmdStatBuffer.clear();
     this.msgStatBuffer.clear();
+    this.rankStatBuffer.clear();
     this.oriCacheBuffer = [];
+    this.whoAtBuffer = [];
     try {
       if (cmdBufferToFlush.length > 0) {
         await this.ctx.database.upsert(
@@ -247,13 +284,24 @@ var Collector = class _Collector {
           (row) => msgBufferToFlush.map((item) => ({
             uid: item.uid,
             type: item.type,
+            count: import_koishi.$.add(import_koishi.$.ifNull(row.count, 0), item.count),
+            timestamp: item.timestamp
+          }))
+        );
+      }
+      if (rankBufferToFlush.length > 0) {
+        await this.ctx.database.upsert(
+          "analyse_rank",
+          (row) => rankBufferToFlush.map((item) => ({
+            uid: item.uid,
             hour: item.hour,
             count: import_koishi.$.add(import_koishi.$.ifNull(row.count, 0), item.count),
             timestamp: item.timestamp
           }))
         );
       }
-      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 +471,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;
@@ -731,13 +779,13 @@ var Stat = class {
       if (usersInGuild.length === 0) return "暂无统计数据";
       const uids = usersInGuild.map((u) => u.uid);
       const userNameMap = new Map(usersInGuild.map((u) => [u.uid, u.userName]));
-      const stats = await this.ctx.database.select("analyse_msg").where({ uid: { $in: uids }, hour: { $gte: since } }).groupBy("uid", { count: /* @__PURE__ */ __name((row) => import_koishi3.$.sum(row.count), "count") }).orderBy("count", "desc").limit(100).execute();
+      const stats = await this.ctx.database.select("analyse_rank").where({ uid: { $in: uids }, hour: { $gte: since } }).groupBy("uid", { count: /* @__PURE__ */ __name((row) => import_koishi3.$.sum(row.count), "count") }).orderBy("count", "desc").limit(100).execute();
       if (stats.length === 0) return "暂无统计数据";
       const total = stats.reduce((sum, record) => sum + record.count, 0);
       const list = stats.map((item) => [userNameMap.get(item.uid) || `UID ${item.uid}`, item.count]);
       return { list, total };
     } else {
-      const msgStats = await this.ctx.database.select("analyse_msg").where({ hour: { $gte: since } }).project(["uid", "count"]).execute();
+      const msgStats = await this.ctx.database.select("analyse_rank").where({ hour: { $gte: since } }).project(["uid", "count"]).execute();
       if (msgStats.length === 0) return "暂无统计数据";
       const allUsers = await this.ctx.database.get("analyse_user", {}, ["uid", "userId", "userName"]);
       const uidToUserMap = new Map(allUsers.map((u) => [u.uid, { userId: u.userId, userName: u.userName }]));
@@ -758,6 +806,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 +887,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.1",
   "contributors": [
     "Yis_Rime <yis_rime@outlook.com>"
   ],