koishi-plugin-chat-analyse 0.5.5 → 0.6.1

package/lib/Analyse.d.ts

@@ -0,0 +1,22 @@
+import { Context, Command } from 'koishi';
+import { Config } from './index';
+export interface WordCloudData {
+    title: string;
+    time: Date;
+    words: [string, number][];
+}
+export declare class Analyse {
+    private ctx;
+    private config;
+    private renderer;
+    private nlp;
+    private isNlpReady;
+    constructor(ctx: Context, config: Config);
+    /**
+     * @private
+     * @method initializeNlp
+     * @description 异步加载并训练 NLP 模型。
+     */
+    private initializeNlp;
+    registerCommands(cmd: Command): void;
+}

package/lib/Renderer.d.ts

@@ -1,4 +1,5 @@
 import { Context } from 'koishi';
+import { WordCloudData } from './Analyse';
 /**
  * @interface ListRenderData
  * @description 定义了调用 `renderList` 方法所需的数据结构。
@@ -74,4 +75,12 @@ export declare class Renderer {
      * @returns {Promise<string | Buffer[]>} - 成功时返回包含图片 Buffer 的数组，失败或无数据时返回提示字符串。
      */
     renderCircadianChart(data: CircadianChartData): Promise<string | Buffer[]>;
+    /**
+     * @public
+     * @method renderWordCloud
+     * @description 将词频数据渲染成一张词云图片，使用 Puppeteer 和 wordcloud2.js。
+     * @param {WordCloudData} data - 包含标题、时间和词汇列表的对象。
+     * @returns {Promise<string | Buffer[]>} - 成功时返回图片 Buffer 数组，否则返回提示。
+     */
+    renderWordCloud(data: WordCloudData): Promise<string | Buffer[]>;
 }

package/lib/Stat.d.ts

@@ -20,18 +20,4 @@ export declare class Stat {
      * @param cmd - 主命令实例。
      */
     registerCommands(cmd: Command): void;
-    /**
-     * @private @method parseQueryScope
-     * @description 解析命令选项，转换为包含 UIDs 和描述性信息的统一查询范围对象。
-     * @param session - 当前会话对象。
-     * @param options - 命令选项。
-     * @returns 包含 uids、错误或范围描述的查询范围对象。
-     */
-    private parseQueryScope;
-    /**
-     * @private @method generateTitle
-     * @description 根据查询范围和类型动态生成易于理解的图片标题。
-     * @returns 生成的标题字符串。
-     */
-    private generateTitle;
 }

package/lib/index.d.ts

@@ -1,4 +1,4 @@
-import { Context, Schema } from 'koishi';
+import { Context, Schema, Session } from 'koishi';
 /** @name 插件使用说明 */
 export declare const usage = "\n<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);\">\n  <h2 style=\"margin-top: 0; color: #4a6ee0;\">\uD83D\uDCCC \u63D2\u4EF6\u8BF4\u660E</h2>\n  <p>\uD83D\uDCD6 <strong>\u4F7F\u7528\u6587\u6863</strong>\uFF1A\u8BF7\u70B9\u51FB\u5DE6\u4E0A\u89D2\u7684 <strong>\u63D2\u4EF6\u4E3B\u9875</strong> \u67E5\u770B\u63D2\u4EF6\u4F7F\u7528\u6587\u6863</p>\n  <p>\uD83D\uDD0D <strong>\u66F4\u591A\u63D2\u4EF6</strong>\uFF1A\u53EF\u8BBF\u95EE <a href=\"https://github.com/YisRime\" style=\"color:#4a6ee0;text-decoration:none;\">\u82E1\u6DDE\u7684 GitHub</a> \u67E5\u770B\u672C\u4EBA\u7684\u6240\u6709\u63D2\u4EF6</p>\n</div>\n<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);\">\n  <h2 style=\"margin-top: 0; color: #e0574a;\">\u2764\uFE0F \u652F\u6301\u4E0E\u53CD\u9988</h2>\n  <p>\uD83C\uDF1F \u559C\u6B22\u8FD9\u4E2A\u63D2\u4EF6\uFF1F\u8BF7\u5728 <a href=\"https://github.com/YisRime\" style=\"color:#e0574a;text-decoration:none;\">GitHub</a> \u4E0A\u7ED9\u6211\u4E00\u4E2A Star\uFF01</p>\n  <p>\uD83D\uDC1B \u9047\u5230\u95EE\u9898\uFF1F\u8BF7\u901A\u8FC7 <strong>Issues</strong> \u63D0\u4EA4\u53CD\u9988\uFF0C\u6216\u52A0\u5165 QQ \u7FA4 <a href=\"https://qm.qq.com/q/PdLMx9Jowq\" style=\"color:#e0574a;text-decoration:none;\"><strong>855571375</strong></a> \u8FDB\u884C\u4EA4\u6D41</p>\n</div>\n";
 export declare const name = "chat-analyse";
@@ -18,9 +18,43 @@ export interface Config {
     enableDataIO: boolean;
     atRetentionDays: number;
     rankRetentionDays: number;
+    enableWordCloud: boolean;
+    enableVocabulary: boolean;
 }
 /** @description 插件的配置项定义 */
 export declare const Config: Schema<Config>;
+/**
+ * @private @method parseQueryScope
+ * @description 解析命令选项，转换为包含 UIDs 和描述性信息的统一查询范围对象。
+ * @param session - 当前会话对象。
+ * @param options - 命令选项。
+ * @returns 包含 uids、错误或范围描述的查询范围对象。
+ */
+export declare function parseQueryScope(ctx: Context, session: Session, options: {
+    user?: string;
+    guild?: string;
+    all?: boolean;
+}): Promise<{
+    uids?: number[];
+    error?: string;
+    scopeDesc: {
+        guildId?: string;
+        userId?: string;
+    };
+}>;
+/**
+ * @private @method generateTitle
+ * @description 根据查询范围和类型动态生成易于理解的图片标题。
+ * @returns 生成的标题字符串。
+ */
+export declare function generateTitle(ctx: Context, scopeDesc: {
+    guildId?: string;
+    userId?: string;
+}, options: {
+    main: string;
+    subtype?: string;
+    timeRange?: number;
+}): Promise<string>;
 /**
  * @function apply
  * @description Koishi 插件的主入口函数，负责初始化和注册所有功能模块。

package/lib/index.js

@@ -30,14 +30,16 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var src_exports = {};
 __export(src_exports, {
-  Config: () => Config,
+  Config: () => Config3,
   apply: () => apply,
+  generateTitle: () => generateTitle,
   name: () => name,
+  parseQueryScope: () => parseQueryScope,
   usage: () => usage,
   using: () => using
 });
 module.exports = __toCommonJS(src_exports);
-var import_koishi6 = require("koishi");
+var import_koishi7 = require("koishi");
 
 // src/Collector.ts
 var import_koishi = require("koishi");
@@ -301,7 +303,7 @@ var Renderer = class {
   async htmlToImage(fullHtmlContent) {
     const page = await this.ctx.puppeteer.page();
     try {
-      await page.setViewport({ width: 720, height: 10, deviceScaleFactor: 2 });
+      await page.setViewport({ width: 850, height: 10, deviceScaleFactor: 2 });
       await page.setContent(fullHtmlContent, { waitUntil: "networkidle0" });
       const { width, height } = await page.evaluate(() => ({
         width: document.body.scrollWidth,
@@ -348,7 +350,7 @@ var Renderer = class {
     const CHUNK_SIZE = 100;
     const imageBuffers = [];
     const totalItems = list.length;
-    const countHeaderIndex = headers?.findIndex((h4) => ["总计发言", "条数", "次数", "数量"].includes(h4)) ?? -1;
+    const countHeaderIndex = headers?.findIndex((h6) => ["总计发言", "条数", "次数", "数量"].includes(h6)) ?? -1;
     const totalCount = data.total || (countHeaderIndex > -1 ? list.reduce((sum, row) => sum + (Number(row[countHeaderIndex]) || 0), 0) : totalItems);
     const renderCell = /* @__PURE__ */ __name((cell, i) => {
       const headerText = headers?.[i] || "";
@@ -396,7 +398,7 @@ var Renderer = class {
                 <thead>
                   <tr>
                     <th class="rank-cell">#</th>
-                    ${headers.map((h4) => `<th>${h4}</th>`).join("")}
+                    ${headers.map((h6) => `<th>${h6}</th>`).join("")}
                   </tr>
                 </thead>` : ""}
               <tbody>
@@ -457,6 +459,54 @@ var Renderer = class {
     const imageBuffer = await this.htmlToImage(fullHtml);
     return imageBuffer ? [imageBuffer] : "图片渲染失败";
   }
+  /**
+   * @public
+   * @method renderWordCloud
+   * @description 将词频数据渲染成一张词云图片，使用 Puppeteer 和 wordcloud2.js。
+   * @param {WordCloudData} data - 包含标题、时间和词汇列表的对象。
+   * @returns {Promise<string | Buffer[]>} - 成功时返回图片 Buffer 数组，否则返回提示。
+   */
+  async renderWordCloud(data) {
+    const { title, time, words } = data;
+    if (!words?.length) return "暂无数据可供渲染";
+    const wordListJson = JSON.stringify(words);
+    const cardHtml = `
+      <div class="container">
+        <div class="header">
+          <div class="stat-chip">词数: <span>${words.length}</span></div>
+          <h1 class="title-text">${title}</h1>
+          <div class="time-label">${time.toLocaleString("zh-CN", { hour12: false })}</div>
+        </div>
+        <div id="wordcloud-container" style="width: 800px; height: 600px; margin: auto;"></div>
+        <script src="https://cdn.jsdelivr.net/npm/wordcloud@1.2.2/src/wordcloud2.js"></script>
+        <script>
+          WordCloud(document.getElementById('wordcloud-container'), {
+            list: ${wordListJson},
+            gridSize: 16,
+            weightFactor: (size) => Math.pow(size, 1.2) * 2.5,
+            fontFamily: "-apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, 'Helvetica Neue', Arial, sans-serif",
+            color: 'random-dark',
+            backgroundColor: 'transparent',
+            rotateRatio: 0.5,
+            minRotation: -Math.PI / 6,
+            maxRotation: Math.PI / 6,
+            shuffle: false,
+          });
+        </script>
+      </div>`;
+    const fullHtml = `<!DOCTYPE html>
+      <html>
+        <head>
+          <meta charset="UTF-8">
+          <style>${this.COMMON_STYLE}</style>
+        </head>
+        <body>
+          ${cardHtml}
+        </body>
+      </html>`;
+    const imageBuffer = await this.htmlToImage(fullHtml);
+    return imageBuffer ? [imageBuffer] : "图片渲染失败";
+  }
 };
 
 // src/Stat.ts
@@ -488,7 +538,7 @@ var Stat = class {
   registerCommands(cmd) {
     const createHandler = /* @__PURE__ */ __name((handler) => {
       return async ({ session, options }) => {
-        const scope = await this.parseQueryScope(session, options);
+        const scope = await parseQueryScope(this.ctx, session, options);
         if (scope.error) return scope.error;
         try {
           const result = await handler(scope, options);
@@ -527,7 +577,7 @@ var Stat = class {
         }
         const total = processedStats.reduce((sum, record) => sum + record.count, 0);
         const list = processedStats.map((item) => [item.command, item.count, item.lastUsed]);
-        const title = await this.generateTitle(scope.scopeDesc, { main: "命令" });
+        const title = await generateTitle(this.ctx, scope.scopeDesc, { main: "命令" });
         return this.renderer.renderList({ title, time: /* @__PURE__ */ new Date(), total, list }, ["命令", "次数", "最后使用"]);
       }));
     }
@@ -542,7 +592,7 @@ var Stat = class {
         if (stats.length === 0) return "暂无统计数据";
         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 title = await this.generateTitle(scope.scopeDesc, { main: "发言", subtype: type });
+        const title = await generateTitle(this.ctx, scope.scopeDesc, { main: "发言", subtype: type });
         const headers = type ? ["用户", "条数", "最后发言"] : ["用户", "总计发言", "最后发言"];
         return this.renderer.renderList({ title, time: /* @__PURE__ */ new Date(), total, list }, headers);
       }));
@@ -560,7 +610,7 @@ var Stat = class {
         const total = rankStats.reduce((sum, record) => sum + record.count, 0);
         const list = rankStats.map((item) => [userNameMap.get(item.uid) || `UID ${item.uid}`, item.count]);
         const listWithPercentage = list.map((row) => [...row, total > 0 ? `${(row[1] / total * 100).toFixed(2)}%` : "0.00%"]);
-        const title = await this.generateTitle(scope.scopeDesc, { main: "发言排行", timeRange: hours, subtype: type });
+        const title = await generateTitle(this.ctx, scope.scopeDesc, { main: "发言排行", timeRange: hours, subtype: type });
         return this.renderer.renderList({ title, time: /* @__PURE__ */ new Date(), total, list: listWithPercentage }, ["用户", "总计发言", "占比"]);
       }));
     }
@@ -574,57 +624,11 @@ var Stat = class {
           hourlyCounts[stat.timestamp.getHours()] += stat.count;
           totalMessages += stat.count;
         });
-        const title = await this.generateTitle(scope.scopeDesc, { main: "活跃" });
+        const title = await generateTitle(this.ctx, scope.scopeDesc, { main: "活跃" });
         return this.renderer.renderCircadianChart({ title, time: /* @__PURE__ */ new Date(), total: totalMessages, data: hourlyCounts });
       }));
     }
   }
-  /**
-   * @private @method parseQueryScope
-   * @description 解析命令选项，转换为包含 UIDs 和描述性信息的统一查询范围对象。
-   * @param session - 当前会话对象。
-   * @param options - 命令选项。
-   * @returns 包含 uids、错误或范围描述的查询范围对象。
-   */
-  async parseQueryScope(session, options) {
-    const scopeDesc = { guildId: options.guild, userId: void 0 };
-    if (options.user) scopeDesc.userId = import_koishi3.h.select(options.user, "at")[0]?.attrs.id ?? options.user.trim();
-    if (!options.all && !scopeDesc.guildId && !scopeDesc.userId) scopeDesc.guildId = session.guildId;
-    if (!options.all && !scopeDesc.guildId && !scopeDesc.userId) return { error: "请指定查询范围", scopeDesc };
-    const query = {};
-    if (scopeDesc.guildId) query.channelId = scopeDesc.guildId;
-    if (scopeDesc.userId) query.userId = scopeDesc.userId;
-    if (Object.keys(query).length === 0) return { uids: void 0, scopeDesc };
-    const users = await this.ctx.database.get("analyse_user", query, ["uid"]);
-    if (users.length === 0) return { error: "暂无统计数据", scopeDesc };
-    return { uids: users.map((u) => u.uid), scopeDesc };
-  }
-  /**
-   * @private @method generateTitle
-   * @description 根据查询范围和类型动态生成易于理解的图片标题。
-   * @returns 生成的标题字符串。
-   */
-  async generateTitle(scopeDesc, options) {
-    let guildName = "", userName = "", scopeText = "全局";
-    if (scopeDesc.guildId) {
-      const [guild] = await this.ctx.database.get("analyse_user", { channelId: scopeDesc.guildId }, ["channelName"]);
-      guildName = guild?.channelName || scopeDesc.guildId;
-    }
-    if (scopeDesc.userId) {
-      const [user] = await this.ctx.database.get("analyse_user", { userId: scopeDesc.userId }, ["userName"]);
-      userName = user?.userName || scopeDesc.userId;
-    }
-    const typeText = options.subtype ? `“${options.subtype}”` : "";
-    const mainText = options.main;
-    if (mainText.includes("排行")) {
-      scopeText = guildName || "全局";
-      return `${options.timeRange}小时${scopeText}${typeText}${mainText}`;
-    }
-    if (userName && guildName) scopeText = `${guildName} ${userName}`;
-    else if (userName) scopeText = userName;
-    else if (guildName) scopeText = guildName;
-    return `${scopeText}${typeText}${mainText}统计`;
-  }
 };
 
 // src/WhoAt.ts
@@ -825,6 +829,114 @@ ${commandOutput}`;
   }
 };
 
+// src/Analyse.ts
+var import_koishi6 = require("koishi");
+var import_basic = require("@nlpjs/basic");
+var import_lang_zh = require("@nlpjs/lang-zh");
+var Analyse = class {
+  constructor(ctx, config) {
+    this.ctx = ctx;
+    this.config = config;
+    this.renderer = new Renderer(ctx);
+    this.nlp = new import_basic.Nlp({ languages: ["zh"], nlu: { log: false } });
+    this.nlp.container.register("extract-lang-zh", new import_lang_zh.LangZh());
+    this.initializeNlp().catch((err) => {
+      this.ctx.logger.error("NLP 语言模型加载失败:", err);
+    });
+  }
+  static {
+    __name(this, "Analyse");
+  }
+  renderer;
+  nlp;
+  isNlpReady = false;
+  /**
+   * @private
+   * @method initializeNlp
+   * @description 异步加载并训练 NLP 模型。
+   */
+  async initializeNlp() {
+    await this.nlp.train();
+    this.isNlpReady = true;
+  }
+  registerCommands(cmd) {
+    if (this.config.enableWordCloud) {
+      cmd.subcommand(".wordcloud", "生成词云").usage("基于指定范围内的聊天记录生成词云图。").option("guild", "-g <guildId:string> 指定群组").option("user", "-u <user:string> 指定用户").option("all", "-a 全局").option("hours", "-h <hours:number> 指定时长", { fallback: 24 }).action(async ({ session, options }) => {
+        if (!this.isNlpReady) return "文本分析尚未就绪，请稍后再试";
+        const scope = await parseQueryScope(this.ctx, session, options);
+        if (scope.error) return scope.error;
+        const since = new Date(Date.now() - options.hours * import_koishi6.Time.hour);
+        const records = await this.ctx.database.select("analyse_cache").where({ uid: { $in: scope.uids }, timestamp: { $gte: since } }).project(["content"]).execute();
+        if (records.length === 0) return "暂无统计数据";
+        const allText = records.map((r) => r.content).join(" ");
+        const result = await this.nlp.process("zh", allText);
+        const words = result.stems.filter((stem) => stem.length > 1);
+        const wordCounts = words.reduce((map, word) => {
+          map.set(word, (map.get(word) || 0) + 1);
+          return map;
+        }, /* @__PURE__ */ new Map());
+        if (wordCounts.size === 0) return "暂无有效词语";
+        const wordList = Array.from(wordCounts.entries()).sort((a, b) => b[1] - a[1]).slice(0, 150);
+        const title = await generateTitle(this.ctx, scope.scopeDesc, { main: "词云" });
+        const renderResult = await this.renderer.renderWordCloud({ title, time: /* @__PURE__ */ new Date(), words: wordList });
+        if (typeof renderResult === "string") return renderResult;
+        if (Array.isArray(renderResult) && renderResult.length > 0) {
+          for (const buffer of renderResult) await session.sendQueued(import_koishi6.h.image(buffer, "image/png"));
+        }
+      });
+    }
+    if (this.config.enableVocabulary) {
+      cmd.subcommand(".vocabulary", "词汇排行").usage("根据不重复词汇量占总词汇量的比例进行排行。").option("guild", "-g <guildId:string> 指定群组").option("all", "-a 全局").option("hours", "-h <hours:number> 指定时长", { fallback: 24 }).action(async ({ session, options }) => {
+        if (!this.isNlpReady) return "文本分析尚未就绪，请稍后再试";
+        const scope = await parseQueryScope(this.ctx, session, options);
+        if (scope.error) return scope.error;
+        const users = await this.ctx.database.get("analyse_user", { uid: { $in: scope.uids } }, ["uid", "userName"]);
+        const userNameMap = new Map(users.map((u) => [u.uid, u.userName]));
+        const since = new Date(Date.now() - options.hours * import_koishi6.Time.hour);
+        const allRecords = await this.ctx.database.get("analyse_cache", { uid: { $in: scope.uids }, timestamp: { $gte: since } }, ["uid", "content"]);
+        if (allRecords.length === 0) return "暂无统计数据";
+        const messagesByUid = /* @__PURE__ */ new Map();
+        for (const record of allRecords) {
+          if (!messagesByUid.has(record.uid)) messagesByUid.set(record.uid, []);
+          messagesByUid.get(record.uid).push(record.content);
+        }
+        const richnessData = [];
+        for (const [uid, messages] of messagesByUid.entries()) {
+          const allText = messages.join(" ");
+          const result = await this.nlp.process("zh", allText);
+          const words = result.stems.filter((stem) => stem.length > 1);
+          if (words.length < 50) continue;
+          const uniqueWords = new Set(words);
+          const richness = uniqueWords.size / words.length;
+          richnessData.push({
+            name: userNameMap.get(uid) || `UID ${uid}`,
+            total: words.length,
+            unique: uniqueWords.size,
+            richness
+          });
+        }
+        if (richnessData.length === 0) return "暂无有效词语";
+        richnessData.sort((a, b) => b.richness - a.richness);
+        const list = richnessData.map((item) => [
+          item.name,
+          item.unique,
+          item.total,
+          `${(item.richness * 100).toFixed(2)}%`
+        ]);
+        const title = await generateTitle(this.ctx, scope.scopeDesc, { main: "词汇排行" });
+        const renderResult = await this.renderer.renderList(
+          { title, time: /* @__PURE__ */ new Date(), total: richnessData.length, list },
+          ["用户", "不重复词数", "总词数", "丰富度"]
+        );
+        if (typeof renderResult === "string") return renderResult;
+        if (Array.isArray(renderResult) && renderResult.length > 0) {
+          for (const buffer of renderResult) await session.sendQueued(import_koishi6.h.image(buffer, "image/png"));
+        }
+      });
+    }
+  }
+};
+
 // 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);">
@@ -840,37 +952,78 @@ var usage = `
 `;
 var name = "chat-analyse";
 var using = ["database", "puppeteer", "cron"];
-var Config = import_koishi6.Schema.intersect([
-  import_koishi6.Schema.object({
-    enableListener: import_koishi6.Schema.boolean().default(true).description("启用消息监听"),
-    enableDataIO: import_koishi6.Schema.boolean().default(true).description("启用数据管理")
+var Config3 = import_koishi7.Schema.intersect([
+  import_koishi7.Schema.object({
+    enableListener: import_koishi7.Schema.boolean().default(true).description("启用消息监听"),
+    enableDataIO: import_koishi7.Schema.boolean().default(true).description("启用数据管理")
   }).description("杂项配置"),
-  import_koishi6.Schema.object({
-    enableCmdStat: import_koishi6.Schema.boolean().default(true).description("启用命令统计"),
-    enableMsgStat: import_koishi6.Schema.boolean().default(true).description("启用消息统计"),
-    enableActivity: import_koishi6.Schema.boolean().default(true).description("启用活跃统计"),
-    enableRankStat: import_koishi6.Schema.boolean().default(true).description("启用发言排行"),
-    rankRetentionDays: import_koishi6.Schema.number().min(0).default(31).description("排行保留天数"),
-    enableWhoAt: import_koishi6.Schema.boolean().default(true).description("启用提及记录"),
-    atRetentionDays: import_koishi6.Schema.number().min(0).default(7).description("提及保留天数")
+  import_koishi7.Schema.object({
+    enableCmdStat: import_koishi7.Schema.boolean().default(true).description("启用命令统计"),
+    enableMsgStat: import_koishi7.Schema.boolean().default(true).description("启用消息统计"),
+    enableActivity: import_koishi7.Schema.boolean().default(true).description("启用活跃统计"),
+    enableRankStat: import_koishi7.Schema.boolean().default(true).description("启用发言排行"),
+    rankRetentionDays: import_koishi7.Schema.number().min(0).default(31).description("排行保留天数"),
+    enableWhoAt: import_koishi7.Schema.boolean().default(true).description("启用提及记录"),
+    atRetentionDays: import_koishi7.Schema.number().min(0).default(7).description("提及保留天数")
   }).description("基础分析配置"),
-  import_koishi6.Schema.object({
-    enableOriRecord: import_koishi6.Schema.boolean().default(true).description("启用原始记录")
+  import_koishi7.Schema.object({
+    enableOriRecord: import_koishi7.Schema.boolean().default(true).description("启用原始记录"),
+    enableWordCloud: import_koishi7.Schema.boolean().default(true).description("启用词云生成"),
+    enableVocabulary: import_koishi7.Schema.boolean().default(true).description("启用词汇排行")
   }).description("高级分析配置")
 ]);
+async function parseQueryScope(ctx, session, options) {
+  const scopeDesc = { guildId: options.guild, userId: void 0 };
+  if (options.user) scopeDesc.userId = import_koishi7.h.select(options.user, "at")[0]?.attrs.id ?? options.user.trim();
+  if (!options.all && !scopeDesc.guildId && !scopeDesc.userId) scopeDesc.guildId = session.guildId;
+  if (!options.all && !scopeDesc.guildId && !scopeDesc.userId) return { error: "请指定查询范围", scopeDesc };
+  const query = {};
+  if (scopeDesc.guildId) query.channelId = scopeDesc.guildId;
+  if (scopeDesc.userId) query.userId = scopeDesc.userId;
+  if (Object.keys(query).length === 0) return { uids: void 0, scopeDesc };
+  const users = await ctx.database.get("analyse_user", query, ["uid"]);
+  if (users.length === 0) return { error: "暂无统计数据", scopeDesc };
+  return { uids: users.map((u) => u.uid), scopeDesc };
+}
+__name(parseQueryScope, "parseQueryScope");
+async function generateTitle(ctx, scopeDesc, options) {
+  let guildName = "", userName = "", scopeText = "全局";
+  if (scopeDesc.guildId) {
+    const [guild] = await ctx.database.get("analyse_user", { channelId: scopeDesc.guildId }, ["channelName"]);
+    guildName = guild?.channelName || scopeDesc.guildId;
+  }
+  if (scopeDesc.userId) {
+    const [user] = await ctx.database.get("analyse_user", { userId: scopeDesc.userId }, ["userName"]);
+    userName = user?.userName || scopeDesc.userId;
+  }
+  const typeText = options.subtype ? `“${options.subtype}”` : "";
+  const mainText = options.main;
+  if (mainText.includes("排行")) {
+    scopeText = guildName || "全局";
+    return `${options.timeRange}小时${scopeText}${typeText}${mainText}`;
+  }
+  if (userName && guildName) scopeText = `${guildName} ${userName}`;
+  else if (userName) scopeText = userName;
+  else if (guildName) scopeText = guildName;
+  return `${scopeText}${typeText}${mainText}统计`;
+}
+__name(generateTitle, "generateTitle");
 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);
   if (config.enableDataIO) new Data(ctx).registerCommands(analyse);
+  if (config.enableOriRecord && (config.enableWordCloud || config.enableVocabulary)) new Analyse(ctx, config).registerCommands(analyse);
 }
 __name(apply, "apply");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   Config,
   apply,
+  generateTitle,
   name,
+  parseQueryScope,
   usage,
   using
 });

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-chat-analyse",
   "description": "聊天记录分析",
-  "version": "0.5.5",
+  "version": "0.6.1",
   "contributors": [
     "Yis_Rime <yis_rime@outlook.com>"
   ],
@@ -32,10 +32,14 @@
     "platform"
   ],
   "devDependencies": {
-    "koishi-plugin-puppeteer": "^3.5.2",
-    "koishi-plugin-cron": "^3.0.0"
+    "koishi-plugin-cron": "^3.0.0",
+    "koishi-plugin-puppeteer": "^3.5.2"
   },
   "peerDependencies": {
     "koishi": "4.18.8"
+  },
+  "dependencies": {
+    "@nlpjs/basic": "^4.26.1",
+    "@nlpjs/lang-zh": "^4.26.1"
   }
 }