koishi-plugin-chat-analyse 0.3.5 → 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);
     }
@@ -298,13 +320,13 @@ var Renderer = class {
             <style>
               :root {
                 --card-bg: #ffffff; --text-color: #111827; --header-color: #111827;
-                --sub-text-color: #6b7280; --border-color: #e5e7eb; --accent-color: #3b82f6;
+                --sub-text-color: #6b7280; --border-color: #e5e7eb; --accent-color: #4a6ee0;
                 --chip-bg: #f3f4f6; --stripe-bg: #f9fafb; --gold: #f59e0b; --silver: #9ca3af; --bronze: #a16207;
               }
               body {
                 display: inline-block; /* Crucial for shrink-wrapping */
                 font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, 'Helvetica Neue', Arial, sans-serif;
-                background: transparent; margin: 0; padding: 10px;
+                background: transparent; margin: 0; padding: 8px;
                 -webkit-font-smoothing: antialiased;
               }
               .container {
@@ -312,22 +334,22 @@ var Renderer = class {
                 border-radius: 12px; padding: 0; overflow: hidden;
                 box-shadow: 0 2px 4px rgba(0,0,0,0.05);
               }
-              .header { padding: 12px 16px; }
-              .header-table { border-collapse: collapse; table-layout: auto; width: 100%; }
+              .header { padding: 10px 14px; }
+              .header-table { border-collapse: collapse; width: 100%; }
               .header-table-left, .header-table-right { width: 1%; white-space: nowrap; }
               .header-table-left { text-align: left; }
               .header-table-center { text-align: center; }
               .header-table-right { text-align: right; }
               .title-text { font-size: 18px; font-weight: 600; color: var(--header-color); margin: 0; }
               .stat-chip, .time-label {
-                display: inline-flex; align-items: baseline; padding: 5px 10px; border-radius: 8px;
+                display: inline-flex; align-items: baseline; padding: 4px 8px; border-radius: 8px;
                 background: var(--chip-bg); font-size: 13px; color: var(--sub-text-color);
               }
               .stat-chip span { font-weight: 600; color: var(--text-color); margin-left: 4px; }
               .table-container { border-top: 1px solid var(--border-color); }
-              .main-table { border-collapse: collapse; table-layout: auto; width: 100%; }
+              .main-table { border-collapse: collapse; width: 100%; }
               .main-table th, .main-table td {
-                padding: 10px 16px;
+                padding: 8px 14px;
                 vertical-align: middle;
               }
               .main-table th {
@@ -336,10 +358,7 @@ var Renderer = class {
               }
               .main-table td { font-size: 14px; color: var(--text-color); }
               .main-table tbody tr:nth-child(even) { background-color: var(--stripe-bg); }
-              .main-table .name-cell, .main-table .name-header {
-                text-align: left;
-                white-space: normal;
-              }
+              .main-table .name-cell, .main-table .name-header { text-align: left; }
               .main-table .rank-cell, .main-table .count-cell, .main-table .date-cell, .main-table .percent-cell, .main-table .header-right-align {
                 text-align: right;
                 white-space: nowrap;
@@ -355,7 +374,7 @@ var Renderer = class {
               .rank-silver { color: var(--silver) !important; }
               .rank-bronze { color: var(--bronze) !important; }
               .percent-cell { position: relative; }
-              .percent-bar { position: absolute; top: 0; left: 0; height: 100%; background-color: var(--accent-color); opacity: 0.1; }
+              .percent-bar { position: absolute; top: 0; right: 0; height: 100%; background-color: var(--accent-color); opacity: 0.15; }
               .percent-text { position: relative; z-index: 1; }
             </style>
           </head>
@@ -373,7 +392,6 @@ var Renderer = class {
       return await page.screenshot({ type: "png", fullPage: true, omitBackground: true });
     } catch (error) {
       this.ctx.logger.error("图片渲染出错:", error);
-      throw new Error(`图片渲染出错: ${error.message || "未知错误"}`);
     } finally {
       if (page) await page.close().catch(() => {
       });
@@ -392,17 +410,17 @@ var Renderer = class {
     const diff = Date.now() - date.getTime();
     if (diff < import_koishi2.Time.minute) return "刚刚";
     if (diff > 365 * import_koishi2.Time.day) {
-      return `${date.getFullYear()}-${String(date.getMonth() + 1).padStart(2, "0")}-${String(date.getDate()).padStart(2, "0")}`;
+      return date.toLocaleDateString("zh-CN", { year: "numeric", month: "2-digit", day: "2-digit" }).replace(/\//g, "-");
     }
     const timeUnits = [
-      { unit: "月", ms: 30 * import_koishi2.Time.day },
-      { unit: "天", ms: import_koishi2.Time.day },
-      { unit: "时", ms: import_koishi2.Time.hour },
-      { unit: "分", ms: import_koishi2.Time.minute }
+      ["月", 30 * import_koishi2.Time.day],
+      ["天", import_koishi2.Time.day],
+      ["时", import_koishi2.Time.hour],
+      ["分", import_koishi2.Time.minute]
     ];
     let remainingDiff = diff;
     const parts = [];
-    for (const { unit, ms } of timeUnits) {
+    for (const [unit, ms] of timeUnits) {
       if (remainingDiff >= ms) {
         const value = Math.floor(remainingDiff / ms);
         parts.push(`${value}${unit}`);
@@ -410,7 +428,7 @@ var Renderer = class {
       }
     }
     const result = parts.slice(0, 2).join("");
-    return result ? `${result}前` : "刚刚";
+    return `${result}前`;
   }
   /**
    * @public
@@ -427,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;
@@ -514,7 +532,7 @@ var Stat = class {
    */
   registerCommands(analyse) {
     if (this.config.enableCmdStat) {
-      analyse.subcommand(".cmd", "命令使用统计").option("user", "-u [user:user] 指定用户").option("guild", "-g [guildId:string] 指定群组").option("all", "-a 展示全局统计").action(async ({ session, options }) => {
+      analyse.subcommand(".cmd", "命令使用统计").option("user", "-u <user:string> 指定用户").option("guild", "-g <guildId:string> 指定群组").option("all", "-a 展示全局统计").action(async ({ session, options }) => {
         const scope = this.parseQueryScope(session, options);
         if (scope.error) return scope.error;
         try {
@@ -531,7 +549,7 @@ var Stat = class {
       });
     }
     if (this.config.enableMsgStat) {
-      analyse.subcommand(".msg", "消息发送统计").option("user", "-u [user:user] 指定用户").option("guild", "-g [guildId:string] 指定群组").option("type", "-t <type:string> 指定类型").option("all", "-a 展示全局统计").action(async ({ session, options }) => {
+      analyse.subcommand(".msg", "消息发送统计").option("user", "-u <user:string> 指定用户").option("guild", "-g <guildId:string> 指定群组").option("type", "-t <type:string> 指定类型").option("all", "-a 展示全局统计").action(async ({ session, options }) => {
         const scope = this.parseQueryScope(session, options);
         if (scope.error) return scope.error;
         try {
@@ -557,9 +575,8 @@ var Stat = class {
       });
     }
     if (this.config.enableRankStat) {
-      analyse.subcommand(".rank", "用户发言排行").option("guild", "-g [guildId:string] 指定群组").option("all", "-a 展示全局统计").option("hours", "-h <hours:number> 指定时长", { fallback: 24 }).action(async ({ session, options }) => {
-        let guildId = options.all ? void 0 : typeof options.guild === "string" ? options.guild : session.guildId;
-        if (!session.guildId) return "请指定群组 ID";
+      analyse.subcommand(".rank", "用户发言排行").option("guild", "-g <guildId:string> 指定群组").option("all", "-a 展示全局统计").option("hours", "-h <hours:number> 指定时长", { fallback: 24 }).action(async ({ session, options }) => {
+        const guildId = options.all ? void 0 : options.guild || session.guildId;
         if (!guildId && !options.all) return "请提供查询范围";
         try {
           const stats = await this.getActiveUserStats(options.hours, guildId);
@@ -589,15 +606,20 @@ var Stat = class {
    */
   parseQueryScope(session, options) {
     let userId, guildId;
-    if (typeof options.user === "string") userId = import_koishi3.h.select(options.user, "user")[0]?.attrs.id;
-    else if (options.user) userId = session.userId;
-    if (typeof options.guild === "string") guildId = options.guild;
-    else if (options.guild) {
-      if (!session.guildId) return { error: "请指定群组 ID" };
-      guildId = session.guildId;
+    if (options.user) {
+      const atElements = import_koishi3.h.select(options.user, "at");
+      if (atElements.length > 0) {
+        userId = atElements[0].attrs.id;
+      } else {
+        userId = options.user.trim();
+      }
     }
+    if (options.guild) guildId = options.guild;
     if (options.all) return { userId, guildId: void 0 };
-    if (!guildId && !userId) return session.guildId ? { guildId: session.guildId } : { error: "请提供查询范围" };
+    if (!userId && !guildId) {
+      if (session.guildId) return { guildId: session.guildId };
+      return { error: "请提供查询范围" };
+    }
     return { userId, guildId };
   }
   /**
@@ -643,9 +665,9 @@ var Stat = class {
       const guild = await this.ctx.database.get("analyse_user", { channelId: guildId }, ["channelName"]);
       scopeText = guild[0]?.channelName || guildId;
     }
-    if (options.main === "排行") return `${scopeText}的${options.timeRange}小时消息排行`;
-    if (options.main === "消息" && options.subtype) return `${scopeText}的"${options.subtype}"消息统计`;
-    return `${scopeText}的${options.main}统计`;
+    if (options.main === "排行") return `${scopeText}${options.timeRange}小时消息排行`;
+    if (options.main === "消息" && options.subtype) return `${scopeText}"${options.subtype}"消息统计`;
+    return `${scopeText}${options.main}统计`;
   }
   /**
    * @private
@@ -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.5",
+  "version": "0.4.0",
   "contributors": [
     "Yis_Rime <yis_rime@outlook.com>"
   ],