koishi-plugin-chat-analyse 1.6.13 → 1.7.0

package/lib/index.js

@@ -1605,7 +1605,7 @@ var Renderer = class {
    */
   async *renderList(data, headers) {
     const { title, time, list } = data;
-    const CHUNK_SIZE = 500;
+    const CHUNK_SIZE = 200;
     const totalItems = list.length;
     const countHeaderIndex = headers?.findIndex((h7) => ["总计发言", "条数", "次数", "数量"].includes(h7)) ?? -1;
     const totalCount = data.total || (countHeaderIndex > -1 ? list.reduce((sum, row) => sum + (Number(row[countHeaderIndex]) || 0), 0) : totalItems);
@@ -1907,7 +1907,7 @@ var Stat = class {
       }
     }, "handleAction");
     if (this.config.enableCmdStat) {
-      cmd.subcommand("cmdstat", "命令统计").usage("查询命令统计，可指定查询范围，默认当前群组。").option("user", "-u <user:string> 指定用户").option("guild", "-g <guildId:string> 指定群组").option("separate", "-p 分离子命令").option("sortByTime", "-s 以时间排序").option("all", "-a 全局统计").action(({ session, options }) => handleAction(session, (async () => {
+      cmd.subcommand("cmdstat", "命令统计").usage("查询命令统计，可指定查询范围，默认当前群组。").option("user", "-u <user:string> 指定用户").option("guild", "-g <guildId:string> 指定群组").option("limit", "-l <count:number> 限制数量").option("separate", "-p 分离子命令").option("sortByTime", "-s 以时间排序").option("all", "-a 全局统计").action(({ session, options }) => handleAction(session, (async () => {
         const scope = await this.parseScope(session, options);
         if (scope.error) return scope.error;
         const query = scope.uids ? { uid: { $in: scope.uids } } : {};
@@ -1932,14 +1932,15 @@ var Stat = class {
         } else {
           processedStats.sort((a, b) => b.count - a.count);
         }
+        const limitedStats = options.limit > 0 ? processedStats.slice(0, options.limit) : processedStats;
         const total = processedStats.reduce((sum, r) => sum + r.count, 0);
-        const list = processedStats.map((item) => [item.command, item.count, item.lastUsed]);
+        const list = limitedStats.map((item) => [item.command, item.count, item.lastUsed]);
         const title = await generateTitle(this.ctx, scope.scopeDesc, { main: "命令" });
         return this.renderer.renderList({ title, time: /* @__PURE__ */ new Date(), total, list }, ["命令", "次数", "最后使用"]);
       })()));
     }
     if (this.config.enableMsgStat) {
-      cmd.subcommand("msgstat", "发言统计").usage("查询发言统计，可指定查询范围，默认当前群组。").option("user", "-u <user:string> 指定用户").option("guild", "-g <guildId:string> 指定群组").option("type", "-t <type:string> 指定类型").option("sortByTime", "-s 以时间排序").option("all", "-a 全局统计").action(({ session, options }) => handleAction(session, (async () => {
+      cmd.subcommand("msgstat", "发言统计").usage("查询发言统计，可指定查询范围，默认当前群组。").option("user", "-u <user:string> 指定用户").option("guild", "-g <guildId:string> 指定群组").option("type", "-t <type:string> 指定类型").option("limit", "-l <count:number> 限制数量").option("sortByTime", "-s 以时间排序").option("all", "-a 全局统计").action(({ session, options }) => handleAction(session, (async () => {
         const scope = await this.parseScope(session, options);
         if (scope.error) return scope.error;
         const query = scope.uids ? { uid: { $in: scope.uids } } : {};
@@ -1956,8 +1957,9 @@ var Stat = class {
           const stats2 = await this.ctx.database.select("analyse_msg").where(query).groupBy("type", { count: /* @__PURE__ */ __name((row) => import_koishi3.$.sum(row.count), "count"), lastUsed: /* @__PURE__ */ __name((row) => import_koishi3.$.max(row.timestamp), "lastUsed") }).execute();
           if (stats2.length === 0) return "暂无统计数据";
           applySort(stats2);
+          const limitedStats2 = options.limit > 0 ? stats2.slice(0, options.limit) : stats2;
           const total2 = stats2.reduce((sum, r) => sum + r.count, 0);
-          const list2 = stats2.map((item) => [item.type, item.count, item.lastUsed]);
+          const list2 = limitedStats2.map((item) => [item.type, item.count, item.lastUsed]);
           return this.renderer.renderList({ title, time: /* @__PURE__ */ new Date(), total: total2, list: list2 }, ["类型", "条数", "最后发言"]);
         }
         if (options.user) {
@@ -1966,14 +1968,16 @@ var Stat = class {
           const stats2 = await this.ctx.database.select("analyse_msg").where(query).groupBy("uid", { count: /* @__PURE__ */ __name((row) => import_koishi3.$.sum(row.count), "count"), lastUsed: /* @__PURE__ */ __name((row) => import_koishi3.$.max(row.timestamp), "lastUsed") }).execute();
           if (stats2.length === 0) return "暂无统计数据";
           applySort(stats2);
+          const limitedStats2 = options.limit > 0 ? stats2.slice(0, options.limit) : stats2;
           const total2 = stats2.reduce((sum, r) => sum + r.count, 0);
-          const list2 = stats2.map((item) => [uidToChannelMap.get(item.uid) || `未知群组`, item.count, item.lastUsed]);
+          const list2 = limitedStats2.map((item) => [uidToChannelMap.get(item.uid) || `未知群组`, item.count, item.lastUsed]);
           return this.renderer.renderList({ title, time: /* @__PURE__ */ new Date(), total: total2, list: list2 }, ["群组", "条数", "最后发言"]);
         }
         const stats = await this.ctx.database.select("analyse_msg").where(query).groupBy("uid", { count: /* @__PURE__ */ __name((row) => import_koishi3.$.sum(row.count), "count"), lastUsed: /* @__PURE__ */ __name((row) => import_koishi3.$.max(row.timestamp), "lastUsed") }).execute();
         if (stats.length === 0) return "暂无统计数据";
         applySort(stats);
-        const allUids = stats.map((s) => s.uid);
+        const limitedStats = options.limit > 0 ? stats.slice(0, options.limit) : stats;
+        const allUids = limitedStats.map((s) => s.uid);
         const userNameMap = /* @__PURE__ */ new Map();
         const BATCH_SIZE2 = 4096;
         for (let i = 0; i < allUids.length; i += BATCH_SIZE2) {
@@ -1984,12 +1988,12 @@ var Stat = class {
           }
         }
         const total = stats.reduce((sum, r) => sum + r.count, 0);
-        const list = stats.map((item) => [userNameMap.get(item.uid) || `UID ${item.uid}`, item.count, item.lastUsed]);
+        const list = limitedStats.map((item) => [userNameMap.get(item.uid) || `UID ${item.uid}`, item.count, item.lastUsed]);
         return this.renderer.renderList({ title, time: /* @__PURE__ */ new Date(), total, list }, ["用户", "条数", "最后发言"]);
       })()));
     }
     if (this.config.enableRankStat) {
-      cmd.subcommand("rankstat", "发言排行").usage("查询发言排行，可指定查询范围，默认当前群组。").option("user", "-u <user:string> 指定用户").option("guild", "-g <guildId:string> 指定群组").option("type", "-t <type:string> 指定类型").option("duration", "-n <hours:number> 指定时长", { fallback: 24 }).option("offset", "-o <hours:number> 指定偏移", { fallback: 0 }).option("all", "-a 全局统计").action(({ session, options }) => handleAction(session, (async () => {
+      cmd.subcommand("rankstat", "发言排行").usage("查询发言排行，可指定查询范围，默认当前群组。").option("user", "-u <user:string> 指定用户").option("guild", "-g <guildId:string> 指定群组").option("type", "-t <type:string> 指定类型").option("duration", "-n <hours:number> 指定时长", { fallback: 24 }).option("offset", "-o <hours:number> 指定偏移", { fallback: 0 }).option("limit", "-l <count:number> 限制数量").option("all", "-a 全局统计").action(({ session, options }) => handleAction(session, (async () => {
         const scope = await this.parseScope(session, options);
         if (scope.error) return scope.error;
         const until = new Date(Date.now() - options.offset * import_koishi3.Time.hour);
@@ -2001,8 +2005,9 @@ var Stat = class {
         if (options.user && options.guild) {
           const stats2 = await this.ctx.database.select("analyse_rank").where(query).groupBy("type", { count: /* @__PURE__ */ __name((row) => import_koishi3.$.sum(row.count), "count") }).orderBy("count", "desc").execute();
           if (stats2.length === 0) return "暂无统计数据";
+          const limitedStats2 = options.limit > 0 ? stats2.slice(0, options.limit) : stats2;
           const total2 = stats2.reduce((sum, r) => sum + r.count, 0);
-          const list2 = stats2.map((r) => [r.type, r.count, total2 > 0 ? `${(r.count / total2 * 100).toFixed(2)}%` : "0.00%"]);
+          const list2 = limitedStats2.map((r) => [r.type, r.count, total2 > 0 ? `${(r.count / total2 * 100).toFixed(2)}%` : "0.00%"]);
           return this.renderer.renderList({ title, time: /* @__PURE__ */ new Date(), total: total2, list: list2 }, ["类型", "条数", "占比"]);
         }
         if (options.user) {
@@ -2010,13 +2015,15 @@ var Stat = class {
           const uidToChannelMap = new Map(userRecords.map((u) => [u.uid, u.channelName || u.channelId]));
           const stats2 = await this.ctx.database.select("analyse_rank").where(query).groupBy("uid", { count: /* @__PURE__ */ __name((row) => import_koishi3.$.sum(row.count), "count") }).orderBy("count", "desc").execute();
           if (stats2.length === 0) return "暂无统计数据";
+          const limitedStats2 = options.limit > 0 ? stats2.slice(0, options.limit) : stats2;
           const total2 = stats2.reduce((sum, r) => sum + r.count, 0);
-          const list2 = stats2.map((r) => [uidToChannelMap.get(r.uid) || "未知群组", r.count, total2 > 0 ? `${(r.count / total2 * 100).toFixed(2)}%` : "0.00%"]);
+          const list2 = limitedStats2.map((r) => [uidToChannelMap.get(r.uid) || "未知群组", r.count, total2 > 0 ? `${(r.count / total2 * 100).toFixed(2)}%` : "0.00%"]);
           return this.renderer.renderList({ title, time: /* @__PURE__ */ new Date(), total: total2, list: list2 }, ["群组", "条数", "占比"]);
         }
         const stats = await this.ctx.database.select("analyse_rank").where(query).groupBy("uid", { count: /* @__PURE__ */ __name((row) => import_koishi3.$.sum(row.count), "count") }).orderBy("count", "desc").execute();
         if (stats.length === 0) return "暂无统计数据";
-        const allUids = stats.map((s) => s.uid);
+        const limitedStats = options.limit > 0 ? stats.slice(0, options.limit) : stats;
+        const allUids = limitedStats.map((s) => s.uid);
         const userNameMap = /* @__PURE__ */ new Map();
         const BATCH_SIZE2 = 4096;
         for (let i = 0; i < allUids.length; i += BATCH_SIZE2) {
@@ -2027,7 +2034,7 @@ var Stat = class {
           }
         }
         const total = stats.reduce((sum, r) => sum + r.count, 0);
-        const list = stats.map((r) => [userNameMap.get(r.uid) || `UID ${r.uid}`, r.count, total > 0 ? `${(r.count / total * 100).toFixed(2)}%` : "0.00%"]);
+        const list = limitedStats.map((r) => [userNameMap.get(r.uid) || `UID ${r.uid}`, r.count, total > 0 ? `${(r.count / total * 100).toFixed(2)}%` : "0.00%"]);
         return this.renderer.renderList({ title, time: /* @__PURE__ */ new Date(), total, list }, ["用户", "条数", "占比"]);
       })()));
     }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-chat-analyse",
   "description": "强大而全面的聊天数据分析插件。支持多维度统计（命令、发言、消息类型、活跃度），可生成发言排行、词云图，并提供完善的数据管理。",
-  "version": "1.6.13",
+  "version": "1.7.0",
   "contributors": [
     "Yis_Rime <yis_rime@outlook.com>"
   ],