npm - koishi-plugin-chat-analyse - Versions diffs - 0.3.3 → 0.3.5 - Mend

koishi-plugin-chat-analyse 0.3.3 → 0.3.5

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (4) hide show

package/lib/Renderer.d.ts CHANGED Viewed

@@ -1,12 +1,12 @@
 import { Context } from 'koishi';
 /**
- * @typedef {Array<string | number | Date>} RenderListItem
- * @description 定义了统计列表中单行数据的格式，它是一个由字符串、数字或日期组成的元组。
+ * 定义了渲染列表中单行数据的格式。它是一个由字符串、数字或 `Date` 对象组成的数组。
  */
 export type RenderListItem = (string | number | Date)[];
 /**
  * @interface ListRenderData
- * @description 定义了调用渲染器生成列表图片时所需的完整数据结构。
+ * @description 定义了调用 `renderList` 方法所需的数据结构。
+ * 它包含了渲染一张完整列表图片所必需的所有信息。
  */
 export interface ListRenderData {
     title: string;
@@ -16,41 +16,45 @@ export interface ListRenderData {
 }
 /**
  * @class Renderer
- * @description 一个通用的列表渲染器。它使用 Koishi 的 Puppeteer 服务将结构化的 `ListRenderData` 数据
- *              渲染为一张包含精美表格的图片。
+ * @classdesc
+ * 负责将结构化的数据（特别是列表）转换为设计精美的PNG图片。
+ * 其核心特性是能够动态计算内容尺寸，生成布局紧凑、自适应的图片。
  */
 export declare class Renderer {
     private ctx;
     /**
-     * @constructor
-     * @param {Context} ctx - Koishi 的插件上下文，用于访问 puppeteer 服务。
+     * @param {Context} ctx - Koishi 的插件上下文，用于访问核心服务如 `puppeteer` 和 `logger`。
      */
     constructor(ctx: Context);
     /**
-     * @public
-     * @async
-     * @method renderList
-     * @description 将列表数据渲染为图片。
-     * @param {ListRenderData} data - 待渲染的完整列表数据。
-     * @param {string[]} [headers] - (可选) 表头文案数组。如果提供，将会在表格顶部渲染表头。
-     * @returns {Promise<string | Buffer>} 渲染成功时返回图片的 Buffer 数据；如果输入数据为空，则返回提示文本。
+     * @private
+     * @method htmlToImage
+     * @description
+     * 负责将任意HTML字符串转换为PNG图片Buffer。
+     * @param {string} html - 要渲染的HTML主体内容（不包含 `<html>` 和 `<body>` 标签）。
+     * @returns {Promise<Buffer>} 返回一个包含PNG图片数据的 Buffer 对象。
+     * @throws {Error} 如果 Puppeteer 截图过程中发生错误，将抛出异常。
      */
-    renderList(data: ListRenderData, headers?: string[]): Promise<string | Buffer>;
+    private htmlToImage;
     /**
      * @private
      * @method formatDate
-     * @description 智能格式化日期。对于近期的时间，提供更人性化的相对时间描述（如“刚刚”，“x 分钟前”）；对于较早的时间，则显示标准的“年-月-日”格式。
-     * @param {Date} date - 待格式化的 Date 对象。
-     * @returns {string} 格式化后的日期字符串。
+     * @description
+     * 将 `Date` 对象格式化为人类友好的相对时间字符串。
+     * @param {Date} date - 需要格式化的日期对象。
+     * @returns {string} - 格式化后的时间字符串。
      */
     private formatDate;
     /**
-     * @private
-     * @method generateListHtml
-     * @description 根据传入的结构化数据和表头，动态生成用于 Puppeteer 渲染的完整 HTML 字符串。
-     * @param {ListRenderData} data - 列表数据对象。
-     * @param {string[]} [headers] - (可选) 表头数组。
-     * @returns {string | null} 返回生成的 HTML 字符串。如果列表数据为空，则返回 `null`。
+     * @public
+     * @method renderList
+     * @description
+     * 接收一个标准化的 `ListRenderData` 对象和可选的表头数组，
+     * 然后构建一个包含标题、统计信息和数据表格的完整HTML卡片。
+     * @param {ListRenderData} data - 包含渲染所需全部信息的对象。
+     * @param {string[]} [headers] - (可选) 表格的表头字符串数组。如果提供，将渲染表头。
+     * @returns {Promise<string | Buffer>}
+     * 如果成功，返回包含PNG图片的 Buffer。如果输入的数据列表为空，则返回一个提示性字符串。
      */
-    private generateListHtml;
+    renderList(data: ListRenderData, headers?: string[]): Promise<string | Buffer>;
 }

package/lib/Stat.d.ts CHANGED Viewed

@@ -3,7 +3,7 @@ import { Renderer } from './Renderer';
 import { Config } from './index';
 /**
  * @class Stat
- * @description 提供统一的统计查询服务。它负责注册查询命令，根据用户输入从数据库中获取数据，并调用渲染器生成统计图表。
+ * @description 提供统一的统计查询服务。负责注册查询命令，从数据库获取数据，并调用渲染器生成图表。
  */
 export declare class Stat {
     private ctx;
@@ -16,11 +16,31 @@ export declare class Stat {
      */
     constructor(ctx: Context, config: Config);
     /**
+     * @public
      * @method registerCommands
      * @description 根据插件配置，动态地将 `.cmd`, `.msg`, `.rank` 子命令注册到主 `analyse` 命令下。
      * @param {Command} analyse - 主 `analyse` 命令实例。
      */
     registerCommands(analyse: Command): void;
+    /**
+     * @private
+     * @method parseQueryScope
+     * @description 解析命令的选项，将其转换为统一的查询范围对象（userId 和 guildId）。
+     * @param {Session} session - 当前会话对象。
+     * @param {QueryScopeOptions} options - 命令传入的选项。
+     * @returns {QueryScopeResult} 包含 userId、guildId 或 error 信息的查询范围对象。
+     */
+    private parseQueryScope;
+    /**
+     * @private
+     * @async
+     * @method getUidsInScope
+     * @description 根据查询范围（guildId, userId）获取匹配用户的 UID 列表。
+     * @param {string} [guildId] - (可选) 群组 ID。
+     * @param {string} [userId] - (可选) 用户 ID。
+     * @returns {Promise<{ uids?: number[], error?: string }>} 包含 UID 数组或错误信息的对象。
+     */
+    private getUidsInScope;
     /**
      * @private
      * @async
@@ -48,13 +68,13 @@ export declare class Stat {
     /**
      * @private
      * @async
-     * @method getMessageStats
-     * @description 从数据库中获取并聚合所有消息类型的统计数据。
+     * @method getUserMessageStats
+     * @description 从数据库中获取并聚合每个用户的消息统计数据。
      * @param {string} [guildId] - (可选) 若提供，则将范围限制在此群组。
      * @param {string} [userId] - (可选) 若提供，则将范围限制在此用户。
      * @returns {Promise<{ list: RenderListItem[], total: number } | string>} 返回一个包含列表和总数的对象，或在无数据时返回提示字符串。
      */
-    private getMessageStats;
+    private getUserMessageStats;
     /**
      * @private
      * @async
@@ -63,17 +83,17 @@ export declare class Stat {
      * @param {string} type - 要查询的消息类型。
      * @param {string} [guildId] - (可选) 若提供，则将范围限制在此群组。
      * @param {string} [userId] - (可选) 若提供，则将范围限制在此用户。
-     * @returns {Promise<{ list: RenderListItem[], total: number } | string>}
+     * @returns {Promise<{ list: RenderListItem[], total: number } | string>} 返回一个包含列表和总数的对象，或在无数据时返回提示字符串。
      */
     private getMessageStatsByType;
     /**
      * @private
      * @async
      * @method getActiveUserStats
-     * @description 从数据库中获取并聚合活跃用户排行数据。
-     * @param {string} guildId - 要查询的群组 ID。
+     * @description 从数据库中获取并聚合指定时间范围内的活跃用户排行数据。
      * @param {number} hours - 查询过去的小时数。
-     * @returns {Promise<{ list: RenderListItem[], total: number } | string>}
+     * @param {string} [guildId] - (可选) 要查询的群组 ID。若不提供，则进行全局排行。
+     * @returns {Promise<{ list: RenderListItem[], total: number } | string>} 返回一个包含列表和总数的对象，或在无数据时返回提示字符串。
      */
     private getActiveUserStats;
 }

package/lib/index.js CHANGED Viewed

@@ -267,8 +267,7 @@ var import_koishi3 = require("koishi");
 var import_koishi2 = require("koishi");
 var Renderer = class {
   /**
-   * @constructor
-   * @param {Context} ctx - Koishi 的插件上下文，用于访问 puppeteer 服务。
+   * @param {Context} ctx - Koishi 的插件上下文，用于访问核心服务如 `puppeteer` 和 `logger`。
    */
   constructor(ctx) {
     this.ctx = ctx;
@@ -277,102 +276,217 @@ var Renderer = class {
     __name(this, "Renderer");
   }
   /**
-   * @public
-   * @async
-   * @method renderList
-   * @description 将列表数据渲染为图片。
-   * @param {ListRenderData} data - 待渲染的完整列表数据。
-   * @param {string[]} [headers] - (可选) 表头文案数组。如果提供，将会在表格顶部渲染表头。
-   * @returns {Promise<string | Buffer>} 渲染成功时返回图片的 Buffer 数据；如果输入数据为空，则返回提示文本。
+   * @private
+   * @method htmlToImage
+   * @description
+   * 负责将任意HTML字符串转换为PNG图片Buffer。
+   * @param {string} html - 要渲染的HTML主体内容（不包含 `<html>` 和 `<body>` 标签）。
+   * @returns {Promise<Buffer>} 返回一个包含PNG图片数据的 Buffer 对象。
+   * @throws {Error} 如果 Puppeteer 截图过程中发生错误，将抛出异常。
    */
-  async renderList(data, headers) {
-    const htmlContent = this.generateListHtml(data, headers);
-    if (!htmlContent) return "暂无数据可供渲染";
-    return this.ctx.puppeteer.render(htmlContent);
+  async htmlToImage(html) {
+    let page = null;
+    try {
+      page = await this.ctx.puppeteer.page();
+      await page.setViewport({ width: 720, height: 1080, deviceScaleFactor: 2 });
+      await page.setContent(`
+        <!DOCTYPE html>
+        <html>
+          <head>
+            <meta charset="UTF-8">
+            <meta name="viewport" content="width=device-width, initial-scale=1.0">
+            <style>
+              :root {
+                --card-bg: #ffffff; --text-color: #111827; --header-color: #111827;
+                --sub-text-color: #6b7280; --border-color: #e5e7eb; --accent-color: #3b82f6;
+                --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;
+                -webkit-font-smoothing: antialiased;
+              }
+              .container {
+                display: inline-block; background: var(--card-bg);
+                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-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;
+                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 th, .main-table td {
+                padding: 10px 16px;
+                vertical-align: middle;
+              }
+              .main-table th {
+                font-size: 12px; font-weight: 500; color: var(--sub-text-color);
+                text-transform: uppercase; letter-spacing: 0.05em; background: var(--stripe-bg);
+              }
+              .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 .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;
+                width: 1%;
+                font-variant-numeric: tabular-nums;
+              }
+              .name-cell { font-weight: 500; }
+              .rank-cell { font-weight: 500; color: var(--sub-text-color); }
+              .count-cell { font-weight: 600; color: var(--accent-color); }
+              .date-cell { color: var(--sub-text-color); }
+              .rank-gold, .rank-silver, .rank-bronze { font-weight: 600; }
+              .rank-gold { color: var(--gold) !important; }
+              .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-text { position: relative; z-index: 1; }
+            </style>
+          </head>
+          <body>${html}</body>
+        </html>
+      `, { waitUntil: "networkidle0" });
+      const dimensions = await page.evaluate(() => {
+        const el = document.body;
+        return {
+          width: el.scrollWidth,
+          height: el.scrollHeight
+        };
+      });
+      await page.setViewport({ ...dimensions, deviceScaleFactor: 2 });
+      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(() => {
+      });
+    }
   }
   /**
    * @private
    * @method formatDate
-   * @description 智能格式化日期。对于近期的时间，提供更人性化的相对时间描述（如“刚刚”，“x 分钟前”）；对于较早的时间，则显示标准的“年-月-日”格式。
-   * @param {Date} date - 待格式化的 Date 对象。
-   * @returns {string} 格式化后的日期字符串。
+   * @description
+   * 将 `Date` 对象格式化为人类友好的相对时间字符串。
+   * @param {Date} date - 需要格式化的日期对象。
+   * @returns {string} - 格式化后的时间字符串。
    */
   formatDate(date) {
     if (!date) return "未知";
     const diff = Date.now() - date.getTime();
     if (diff < import_koishi2.Time.minute) return "刚刚";
-    if (diff < import_koishi2.Time.hour) return `${Math.floor(diff / import_koishi2.Time.minute)} 分钟前`;
-    if (diff < import_koishi2.Time.day) return `${Math.floor(diff / import_koishi2.Time.hour)} 小时前`;
-    return `${date.getFullYear()}-${String(date.getMonth() + 1).padStart(2, "0")}-${String(date.getDate()).padStart(2, "0")}`;
+    if (diff > 365 * import_koishi2.Time.day) {
+      return `${date.getFullYear()}-${String(date.getMonth() + 1).padStart(2, "0")}-${String(date.getDate()).padStart(2, "0")}`;
+    }
+    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 }
+    ];
+    let remainingDiff = diff;
+    const parts = [];
+    for (const { unit, ms } of timeUnits) {
+      if (remainingDiff >= ms) {
+        const value = Math.floor(remainingDiff / ms);
+        parts.push(`${value}${unit}`);
+        remainingDiff %= ms;
+      }
+    }
+    const result = parts.slice(0, 2).join("");
+    return result ? `${result}前` : "刚刚";
   }
   /**
-   * @private
-   * @method generateListHtml
-   * @description 根据传入的结构化数据和表头，动态生成用于 Puppeteer 渲染的完整 HTML 字符串。
-   * @param {ListRenderData} data - 列表数据对象。
-   * @param {string[]} [headers] - (可选) 表头数组。
-   * @returns {string | null} 返回生成的 HTML 字符串。如果列表数据为空，则返回 `null`。
+   * @public
+   * @method renderList
+   * @description
+   * 接收一个标准化的 `ListRenderData` 对象和可选的表头数组，
+   * 然后构建一个包含标题、统计信息和数据表格的完整HTML卡片。
+   * @param {ListRenderData} data - 包含渲染所需全部信息的对象。
+   * @param {string[]} [headers] - (可选) 表格的表头字符串数组。如果提供，将渲染表头。
+   * @returns {Promise<string | Buffer>}
+   * 如果成功，返回包含PNG图片的 Buffer。如果输入的数据列表为空，则返回一个提示性字符串。
    */
-  generateListHtml(data, headers) {
-    const { title, time, total, list } = data;
-    if (!list?.length) return null;
-    const tableHeadHtml = headers?.length > 0 ? `<thead><tr><th class="rank-cell">排名</th>${headers.map((h2) => `<th>${h2}</th>`).join("")}</tr></thead>` : "";
+  async renderList(data, headers) {
+    const { title, time, list } = data;
+    if (!list?.length) return "暂无数据可供渲染";
+    let totalValueForPercent = 0;
+    const countHeaderIndex = headers?.findIndex((h2) => ["总计发言", "条数", "次数", "数量"].includes(h2));
+    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 firstCell = list[0]?.[i];
+      const isRightAlign = typeof firstCell === "number" || firstCell instanceof Date || h2.includes("占比");
+      const alignClass = isRightAlign ? "header-right-align" : "name-header";
+      return `<th class="${alignClass}">${h2}</th>`;
+    }).join("")}</tr></thead>` : "";
     const tableRowsHtml = list.map((row, index) => {
       const rank = index + 1;
       const rankClass = rank === 1 ? "rank-gold" : rank === 2 ? "rank-silver" : rank === 3 ? "rank-bronze" : "";
-      const rankCell = `<td class="rank-cell"><span class="rank-badge ${rankClass}">${rank}</span></td>`;
-      const dataCells = row.map((cell) => {
-        if (cell instanceof Date) return `<td class="data-cell date-cell">${this.formatDate(cell)}</td>`;
-        if (typeof cell === "number") return `<td class="data-cell count-cell">${cell}</td>`;
-        return `<td class="data-cell name-cell">${String(cell)}</td>`;
+      const rankCell = `<td class="rank-cell ${rankClass}">${rank}</td>`;
+      const dataCells = row.map((cell, i) => {
+        let className = "";
+        let content;
+        const headerText = headers?.[i] || "";
+        if (headerText.includes("占比")) {
+          className = "percent-cell";
+          const percentValue = parseFloat(String(cell).replace("%", ""));
+          content = `<div class="percent-bar" style="width: ${percentValue}%;"></div><span class="percent-text">${cell}</span>`;
+        } else if (cell instanceof Date) {
+          className = "date-cell";
+          content = this.formatDate(cell);
+        } else if (typeof cell === "number") {
+          className = "count-cell";
+          content = cell.toLocaleString();
+        } else {
+          className = "name-cell";
+          content = String(cell);
+        }
+        return `<td class="${className}">${content}</td>`;
       }).join("");
       return `<tr>${rankCell}${dataCells}</tr>`;
     }).join("");
-    const metaInfoHtml = `
-      <div class="meta-group">
-        ${total !== void 0 ? `<div class="total-count">总计: ${total}</div>` : ""}
-        <div class="time-label">生成于 ${time.toLocaleString("zh-CN", { hour12: false })}</div>
+    const cardHtml = `
+      <div class="container">
+        <div class="header">
+          <table class="header-table">
+            <tr>
+              <td class="header-table-left">
+                <div class="stat-chip">总计: <span>${typeof totalCount === "number" ? totalCount.toLocaleString() : totalCount}</span></div>
+              </td>
+              <td class="header-table-center">
+                <h1 class="title-text">${title}</h1>
+              </td>
+              <td class="header-table-right">
+                <div class="time-label">${time.toLocaleString("zh-CN", { hour12: false }).replace(/\//g, "-")}</div>
+              </td>
+            </tr>
+          </table>
+        </div>
+        <div class="table-container">
+          <table class="main-table">${tableHeadHtml}<tbody>${tableRowsHtml}</tbody></table>
+        </div>
       </div>
     `;
-    const styles = `
-      :root {
-        --bg-color: #f7f8fa; --card-bg: #ffffff; --text-color: #333; --header-color: #1f2329;
-        --sub-text-color: #646a73; --border-color: #e4e6eb; --accent-color: #4a6ee0;
-        --gold: #ffc327; --silver: #a8b5c1; --bronze: #d69864;
-      }
-      body { font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, 'Helvetica Neue', Arial, sans-serif; background: var(--bg-color); margin: 0; padding: 20px; width: 700px; box-sizing: border-box; -webkit-font-smoothing: antialiased; }
-      .container { background: var(--card-bg); border-radius: 12px; box-shadow: 0 6px 16px rgba(0,0,0,0.08); padding: 24px; }
-      .header { display: flex; justify-content: space-between; align-items: flex-start; border-bottom: 1px solid var(--border-color); padding-bottom: 16px; margin-bottom: 16px; }
-      .title-group h1 { font-size: 24px; font-weight: 700; color: var(--header-color); margin: 0; }
-      .meta-group { text-align: right; }
-      .meta-group .total-count { font-size: 22px; font-weight: 700; color: var(--accent-color); }
-      .meta-group .time-label { font-size: 13px; color: var(--sub-text-color); margin-top: 4px; }
-      table { width: 100%; border-collapse: collapse; color: var(--text-color); }
-      th, td { padding: 12px 8px; text-align: left; border-bottom: 1px solid var(--border-color); vertical-align: middle; }
-      th { font-size: 13px; font-weight: 600; color: var(--sub-text-color); }
-      td { font-size: 15px; }
-      tr:last-child td { border-bottom: none; }
-      .rank-cell { width: 50px; text-align: center; }
-      .rank-badge { display: inline-block; width: 24px; height: 24px; line-height: 24px; border-radius: 50%; font-weight: 600; font-size: 14px; color: var(--header-color); background-color: #eef0f3; }
-      .rank-gold, .rank-silver, .rank-bronze { color: #fff; }
-      .rank-gold { background-color: var(--gold); } .rank-silver { background-color: var(--silver); } .rank-bronze { background-color: var(--bronze); }
-      .data-cell { word-break: break-all; }
-      .name-cell { font-weight: 600; color: var(--header-color); }
-      .count-cell { text-align: right; font-weight: 600; color: var(--accent-color); }
-      .date-cell { text-align: right; font-size: 13px; color: var(--sub-text-color); }
-    `;
-    return `
-      <!DOCTYPE html><html lang="zh-CN">
-      <head><meta charset="UTF-8"><title>${title}</title><style>${styles}</style></head>
-      <body>
-        <div class="container">
-          <div class="header">
-            <div class="title-group"><h1>${title}</h1></div>
-            ${metaInfoHtml}
-          </div>
-          <table>${tableHeadHtml}<tbody>${tableRowsHtml}</tbody></table>
-        </div>
-      </body></html>`;
+    return this.htmlToImage(cardHtml);
   }
 };
@@ -393,32 +507,22 @@ var Stat = class {
   }
   renderer;
   /**
+   * @public
    * @method registerCommands
    * @description 根据插件配置，动态地将 `.cmd`, `.msg`, `.rank` 子命令注册到主 `analyse` 命令下。
    * @param {Command} analyse - 主 `analyse` 命令实例。
    */
   registerCommands(analyse) {
     if (this.config.enableCmdStat) {
-      analyse.subcommand(".cmd", "命令使用统计").option("user", "-u [user:user] 指定用户").option("guild", "-g [guildId:string] 指定群组").usage("查询用户或群组的命令使用统计，默认展示全局统计。").action(async ({ session, options }) => {
-        const userId = options.user ? import_koishi3.h.select(options.user, "user")[0]?.attrs.id : void 0;
-        let guildId = options.guild;
-        if (!userId && !guildId && session.guildId) {
-          guildId = session.guildId;
-        } else if (!userId && !guildId && !session.guildId) {
-          return "请指定查询范围";
-        }
+      analyse.subcommand(".cmd", "命令使用统计").option("user", "-u [user:user] 指定用户").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 {
-          const stats = await this.getCommandStats(guildId, userId);
+          const stats = await this.getCommandStats(scope.guildId, scope.userId);
           if (typeof stats === "string") return stats;
-          const title = await this.generateTitle(guildId, userId, { main: "命令" });
-          const renderData = {
-            title,
-            time: /* @__PURE__ */ new Date(),
-            total: stats.total,
-            list: stats.list
-          };
-          const headers = ["命令", "次数", "最后使用"];
-          const result = await this.renderer.renderList(renderData, headers);
+          const title = await this.generateTitle(scope.guildId, scope.userId, { main: "命令" });
+          const renderData = { title, time: /* @__PURE__ */ new Date(), total: stats.total, list: stats.list };
+          const result = await this.renderer.renderList(renderData, ["命令", "次数", "最后使用"]);
           return Buffer.isBuffer(result) ? import_koishi3.Element.image(result, "image/png") : result;
         } catch (error) {
           this.ctx.logger.error("渲染命令统计图片失败:", error);
@@ -427,40 +531,23 @@ 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> 指定类型").usage("查询用户或群组的消息发送统计，默认展示全局统计。").action(async ({ session, options }) => {
-        const userId = options.user ? import_koishi3.h.select(options.user, "user")[0]?.attrs.id : void 0;
-        let guildId = options.guild;
-        if (!userId && !guildId && !options.type && session.guildId) {
-          guildId = session.guildId;
-        } else if (!userId && !guildId && !session.guildId) {
-          return "请指定查询范围";
-        }
+      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 }) => {
+        const scope = this.parseQueryScope(session, options);
+        if (scope.error) return scope.error;
         try {
           if (options.type) {
-            const stats = await this.getMessageStatsByType(options.type, guildId, userId);
+            const stats = await this.getMessageStatsByType(options.type, scope.guildId, scope.userId);
             if (typeof stats === "string") return stats;
-            const title = await this.generateTitle(guildId, void 0, { main: "消息", subtype: options.type });
-            const renderData = {
-              title,
-              time: /* @__PURE__ */ new Date(),
-              total: stats.total,
-              list: stats.list
-            };
-            const headers = ["用户", "条数", "最后发言"];
-            const result = await this.renderer.renderList(renderData, headers);
+            const title = await this.generateTitle(scope.guildId, scope.userId, { main: "消息", subtype: options.type });
+            const renderData = { title, time: /* @__PURE__ */ new Date(), total: stats.total, list: stats.list };
+            const result = await this.renderer.renderList(renderData, ["用户", "条数", "最后发言"]);
             return Buffer.isBuffer(result) ? import_koishi3.Element.image(result, "image/png") : result;
           } else {
-            const stats = await this.getMessageStats(guildId, userId);
+            const stats = await this.getUserMessageStats(scope.guildId, scope.userId);
             if (typeof stats === "string") return stats;
-            const title = await this.generateTitle(guildId, userId, { main: "消息" });
-            const renderData = {
-              title,
-              time: /* @__PURE__ */ new Date(),
-              total: stats.total,
-              list: stats.list
-            };
-            const headers = ["类型", "条数", "最后发言"];
-            const result = await this.renderer.renderList(renderData, headers);
+            const title = await this.generateTitle(scope.guildId, scope.userId, { main: "消息" });
+            const renderData = { title, time: /* @__PURE__ */ new Date(), total: stats.total, list: stats.list };
+            const result = await this.renderer.renderList(renderData, ["用户", "总计发言", "最后发言"]);
             return Buffer.isBuffer(result) ? import_koishi3.Element.image(result, "image/png") : result;
           }
         } catch (error) {
@@ -470,27 +557,20 @@ var Stat = class {
       });
     }
     if (this.config.enableRankStat) {
-      analyse.subcommand(".rank", "用户发言排行").option("guild", "-g [guildId:string] 指定群组").option("hours", "-h <hours:number> 指定时长", { fallback: 24 }).usage("查询用户或群组的用户发言排行，默认展示全局统计。").action(async ({ session, options }) => {
-        let guildId = options.guild;
-        if (!guildId && session.guildId) guildId = session.guildId;
-        if (!guildId) return "请指定查询范围";
+      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";
+        if (!guildId && !options.all) return "请提供查询范围";
         try {
-          const stats = await this.getActiveUserStats(guildId, options.hours);
+          const stats = await this.getActiveUserStats(options.hours, guildId);
           if (typeof stats === "string") return stats;
-          const listWithPercentage = stats.list.map((row) => {
-            const count = row[1];
-            const percentage = stats.total > 0 ? `${(count / stats.total * 100).toFixed(2)}%` : "0.00%";
-            return [...row, percentage];
-          });
+          const listWithPercentage = stats.list.map((row) => [
+            ...row,
+            stats.total > 0 ? `${(row[1] / stats.total * 100).toFixed(2)}%` : "0.00%"
+          ]);
           const title = await this.generateTitle(guildId, void 0, { main: "排行", timeRange: options.hours });
-          const renderData = {
-            title,
-            time: /* @__PURE__ */ new Date(),
-            total: stats.total,
-            list: listWithPercentage
-          };
-          const headers = ["用户", "总计发言", "占比"];
-          const result = await this.renderer.renderList(renderData, headers);
+          const renderData = { title, time: /* @__PURE__ */ new Date(), total: stats.total, list: listWithPercentage };
+          const result = await this.renderer.renderList(renderData, ["用户", "总计发言", "占比"]);
           return Buffer.isBuffer(result) ? import_koishi3.Element.image(result, "image/png") : result;
         } catch (error) {
           this.ctx.logger.error("渲染发言排行图片失败:", error);
@@ -499,6 +579,44 @@ var Stat = class {
       });
     }
   }
+  /**
+   * @private
+   * @method parseQueryScope
+   * @description 解析命令的选项，将其转换为统一的查询范围对象（userId 和 guildId）。
+   * @param {Session} session - 当前会话对象。
+   * @param {QueryScopeOptions} options - 命令传入的选项。
+   * @returns {QueryScopeResult} 包含 userId、guildId 或 error 信息的查询范围对象。
+   */
+  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.all) return { userId, guildId: void 0 };
+    if (!guildId && !userId) return session.guildId ? { guildId: session.guildId } : { error: "请提供查询范围" };
+    return { userId, guildId };
+  }
+  /**
+   * @private
+   * @async
+   * @method getUidsInScope
+   * @description 根据查询范围（guildId, userId）获取匹配用户的 UID 列表。
+   * @param {string} [guildId] - (可选) 群组 ID。
+   * @param {string} [userId] - (可选) 用户 ID。
+   * @returns {Promise<{ uids?: number[], error?: string }>} 包含 UID 数组或错误信息的对象。
+   */
+  async getUidsInScope(guildId, userId) {
+    const query = {};
+    if (guildId) query.channelId = guildId;
+    if (userId) query.userId = userId;
+    const users = await this.ctx.database.get("analyse_user", query, ["uid"]);
+    if (users.length === 0) return { error: "暂无统计数据" };
+    return { uids: users.map((u) => u.uid) };
+  }
   /**
    * @private
    * @async
@@ -513,34 +631,21 @@ var Stat = class {
    * @returns {Promise<string>} 生成的标题字符串。
    */
   async generateTitle(guildId, userId, options) {
-    let scopeText;
+    let scopeText = "全局";
     if (userId && guildId) {
       const user = await this.ctx.database.get("analyse_user", { channelId: guildId, userId }, ["userName"]);
       const guild = await this.ctx.database.get("analyse_user", { channelId: guildId }, ["channelName"]);
-      const userName = user[0]?.userName || userId;
-      const guildName = guild[0]?.channelName || guildId;
-      scopeText = `${userName} 在 ${guildName}`;
+      scopeText = `${user[0]?.userName || userId} 在 ${guild[0]?.channelName || guildId}`;
     } else if (userId) {
       const user = await this.ctx.database.get("analyse_user", { userId }, ["userName"]);
-      const userName = user[0]?.userName || userId;
-      scopeText = `${userName}的全局`;
+      scopeText = `${user[0]?.userName || userId}的全局`;
     } else if (guildId) {
       const guild = await this.ctx.database.get("analyse_user", { channelId: guildId }, ["channelName"]);
       scopeText = guild[0]?.channelName || guildId;
-    } else {
-      scopeText = "全局";
-    }
-    switch (options.main) {
-      case "命令":
-        return `${scopeText}的命令统计`;
-      case "消息":
-        if (options.subtype) return `${scopeText}的"${options.subtype}"消息统计`;
-        return `${scopeText}的消息统计`;
-      case "排行":
-        return `${scopeText}的${options.timeRange}小时消息排行`;
-      default:
-        return scopeText;
     }
+    if (options.main === "排行") return `${scopeText}的${options.timeRange}小时消息排行`;
+    if (options.main === "消息" && options.subtype) return `${scopeText}的"${options.subtype}"消息统计`;
+    return `${scopeText}的${options.main}统计`;
   }
   /**
    * @private
@@ -552,45 +657,39 @@ var Stat = class {
    * @returns {Promise<{ list: RenderListItem[], total: number } | string>} 返回一个包含列表和总数的对象，或在无数据时返回提示字符串。
    */
   async getCommandStats(guildId, userId) {
-    const userQuery = {};
-    if (guildId) userQuery.channelId = guildId;
-    if (userId) userQuery.userId = userId;
-    const users = await this.ctx.database.get("analyse_user", userQuery, ["uid"]);
-    if (users.length === 0) return "暂无目标用户统计数据";
-    const uids = users.map((u) => u.uid);
-    const aggregatedStats = await this.ctx.database.select("analyse_cmd").where({ uid: { $in: uids } }).groupBy(["command"], {
-      count: /* @__PURE__ */ __name((row) => import_koishi3.$.sum(row.count), "count"),
-      lastUsed: /* @__PURE__ */ __name((row) => import_koishi3.$.max(row.timestamp), "lastUsed")
-    }).orderBy("count", "desc").execute();
-    if (aggregatedStats.length === 0) return "暂无统计数据";
-    const totalCount = aggregatedStats.reduce((sum, record) => sum + record.count, 0);
-    const list = aggregatedStats.map((item) => [item.command, item.count, item.lastUsed]);
-    return { list, total: totalCount };
+    const { uids, error } = await this.getUidsInScope(guildId, userId);
+    if (error) return error;
+    const stats = await this.ctx.database.select("analyse_cmd").where({ uid: { $in: uids } }).groupBy("command", { count: /* @__PURE__ */ __name((row) => import_koishi3.$.sum(row.count), "count"), lastUsed: /* @__PURE__ */ __name((row) => import_koishi3.$.max(row.timestamp), "lastUsed") }).orderBy("count", "desc").execute();
+    if (stats.length === 0) return "暂无统计数据";
+    const total = stats.reduce((sum, record) => sum + record.count, 0);
+    const list = stats.map((item) => [item.command, item.count, item.lastUsed]);
+    return { list, total };
   }
   /**
    * @private
    * @async
-   * @method getMessageStats
-   * @description 从数据库中获取并聚合所有消息类型的统计数据。
+   * @method getUserMessageStats
+   * @description 从数据库中获取并聚合每个用户的消息统计数据。
    * @param {string} [guildId] - (可选) 若提供，则将范围限制在此群组。
    * @param {string} [userId] - (可选) 若提供，则将范围限制在此用户。
    * @returns {Promise<{ list: RenderListItem[], total: number } | string>} 返回一个包含列表和总数的对象，或在无数据时返回提示字符串。
    */
-  async getMessageStats(guildId, userId) {
-    const userQuery = {};
-    if (guildId) userQuery.channelId = guildId;
-    if (userId) userQuery.userId = userId;
-    const users = await this.ctx.database.get("analyse_user", userQuery, ["uid"]);
-    if (users.length === 0) return "暂无目标用户统计数据";
+  async getUserMessageStats(guildId, userId) {
+    const query = {};
+    if (guildId) query.channelId = guildId;
+    if (userId) query.userId = userId;
+    const users = await this.ctx.database.get("analyse_user", query, ["uid", "userName"]);
+    if (users.length === 0) return "暂无统计数据";
     const uids = users.map((u) => u.uid);
-    const aggregatedStats = await this.ctx.database.select("analyse_msg").where({ uid: { $in: uids } }).groupBy(["type"], {
+    const userNameMap = new Map(users.map((u) => [u.uid, u.userName]));
+    const stats = await this.ctx.database.select("analyse_msg").where({ uid: { $in: uids } }).groupBy("uid", {
       count: /* @__PURE__ */ __name((row) => import_koishi3.$.sum(row.count), "count"),
       lastUsed: /* @__PURE__ */ __name((row) => import_koishi3.$.max(row.timestamp), "lastUsed")
     }).orderBy("count", "desc").execute();
-    if (aggregatedStats.length === 0) return "暂无统计数据";
-    const totalCount = aggregatedStats.reduce((sum, record) => sum + record.count, 0);
-    const list = aggregatedStats.map((item) => [item.type, item.count, item.lastUsed]);
-    return { list, total: totalCount };
+    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, item.lastUsed]);
+    return { list, total };
   }
   /**
    * @private
@@ -600,60 +699,62 @@ var Stat = class {
    * @param {string} type - 要查询的消息类型。
    * @param {string} [guildId] - (可选) 若提供，则将范围限制在此群组。
    * @param {string} [userId] - (可选) 若提供，则将范围限制在此用户。
-   * @returns {Promise<{ list: RenderListItem[], total: number } | string>}
+   * @returns {Promise<{ list: RenderListItem[], total: number } | string>} 返回一个包含列表和总数的对象，或在无数据时返回提示字符串。
    */
   async getMessageStatsByType(type, guildId, userId) {
-    const userQuery = {};
-    if (guildId) userQuery.channelId = guildId;
-    if (userId) userQuery.userId = userId;
-    const users = await this.ctx.database.get("analyse_user", userQuery, ["uid", "userName"]);
-    if (users.length === 0) return "暂无目标用户统计数据";
+    const query = {};
+    if (guildId) query.channelId = guildId;
+    if (userId) query.userId = userId;
+    const users = await this.ctx.database.get("analyse_user", query, ["uid", "userName"]);
+    if (users.length === 0) return "暂无统计数据";
     const uids = users.map((u) => u.uid);
     const userNameMap = new Map(users.map((u) => [u.uid, u.userName]));
-    const aggregatedStats = await this.ctx.database.select("analyse_msg").where({
-      uid: { $in: uids },
-      type
-    }).groupBy(["uid"], {
-      count: /* @__PURE__ */ __name((row) => import_koishi3.$.sum(row.count), "count"),
-      lastUsed: /* @__PURE__ */ __name((row) => import_koishi3.$.max(row.timestamp), "lastUsed")
-    }).orderBy("count", "desc").execute();
-    if (aggregatedStats.length === 0) return `暂无统计数据`;
-    const totalCount = aggregatedStats.reduce((sum, record) => sum + record.count, 0);
-    const list = aggregatedStats.map((item) => [
-      userNameMap.get(item.uid) || `UID ${item.uid}`,
-      item.count,
-      item.lastUsed
-    ]);
-    return { list, total: totalCount };
+    const stats = await this.ctx.database.select("analyse_msg").where({ uid: { $in: uids }, type }).groupBy("uid", { count: /* @__PURE__ */ __name((row) => import_koishi3.$.sum(row.count), "count"), lastUsed: /* @__PURE__ */ __name((row) => import_koishi3.$.max(row.timestamp), "lastUsed") }).orderBy("count", "desc").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, item.lastUsed]);
+    return { list, total };
   }
   /**
    * @private
    * @async
    * @method getActiveUserStats
-   * @description 从数据库中获取并聚合活跃用户排行数据。
-   * @param {string} guildId - 要查询的群组 ID。
+   * @description 从数据库中获取并聚合指定时间范围内的活跃用户排行数据。
    * @param {number} hours - 查询过去的小时数。
-   * @returns {Promise<{ list: RenderListItem[], total: number } | string>}
+   * @param {string} [guildId] - (可选) 要查询的群组 ID。若不提供，则进行全局排行。
+   * @returns {Promise<{ list: RenderListItem[], total: number } | string>} 返回一个包含列表和总数的对象，或在无数据时返回提示字符串。
    */
-  async getActiveUserStats(guildId, hours) {
+  async getActiveUserStats(hours, guildId) {
     const since = new Date(Date.now() - hours * 3600 * 1e3);
-    const usersInGuild = await this.ctx.database.get("analyse_user", { channelId: guildId }, ["uid", "userId", "userName"]);
-    if (usersInGuild.length === 0) return "暂无用户统计数据";
-    const uids = usersInGuild.map((u) => u.uid);
-    const userNameMap = new Map(usersInGuild.map((u) => [u.uid, u.userName]));
-    const aggregatedStats = 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();
-    if (aggregatedStats.length === 0) return "暂无统计数据";
-    const totalCount = aggregatedStats.reduce((sum, record) => sum + record.count, 0);
-    const list = aggregatedStats.map((item) => [
-      userNameMap.get(item.uid) || `UID ${item.uid}`,
-      item.count
-    ]);
-    return { list, total: totalCount };
+    if (guildId) {
+      const usersInGuild = await this.ctx.database.get("analyse_user", { channelId: guildId }, ["uid", "userName"]);
+      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();
+      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();
+      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 }]));
+      const userCounts = /* @__PURE__ */ new Map();
+      for (const msg of msgStats) {
+        const userInfo = uidToUserMap.get(msg.uid);
+        if (userInfo) {
+          const existing = userCounts.get(userInfo.userId);
+          userCounts.set(userInfo.userId, { count: (existing?.count || 0) + msg.count, name: userInfo.userName });
+        }
+      }
+      if (userCounts.size === 0) return "暂无统计数据";
+      const grandTotal = Array.from(userCounts.values()).reduce((sum, data) => sum + data.count, 0);
+      const sortedUsers = Array.from(userCounts.entries()).sort(([, a], [, b]) => b.count - a.count).slice(0, 100);
+      const list = sortedUsers.map(([userId, data]) => [data.name || userId, data.count]);
+      return { list, total: grandTotal };
+    }
   }
 };

package/package.json CHANGED Viewed

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