koishi-plugin-chat-analyse 0.4.4 → 0.4.6

package/lib/Data.d.ts

@@ -0,0 +1,19 @@
+import { Context, Command } from 'koishi';
+/**
+ * @class Data
+ * @description
+ * 提供数据备份、恢复和清理的管理功能，恢复逻辑采用分批处理以优化性能。
+ * 清理功能支持按表、时间范围和用户范围进行精确操作。
+ */
+export declare class Data {
+    private ctx;
+    private dataDir;
+    constructor(ctx: Context);
+    /**
+     * @public
+     * @method registerCommands
+     * @description 在 'analyse.admin' 命令下注册所有数据管理相关的子命令。
+     * @param {Command} analyse - '.admin' 命令实例。
+     */
+    registerCommands(analyse: Command): void;
+}

package/lib/index.d.ts

@@ -17,6 +17,7 @@ export interface Config {
     enableRankStat: boolean;
     enableOriRecord: boolean;
     enableWhoAt: boolean;
+    enableData: boolean;
     atRetentionDays: number;
     rankRetentionDays: number;
 }

package/lib/index.js

@@ -1,6 +1,8 @@
+var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
 var __export = (target, all) => {
@@ -15,6 +17,14 @@ var __copyProps = (to, from, except, desc) => {
   }
   return to;
 };
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
 // src/index.ts
@@ -27,7 +37,7 @@ __export(src_exports, {
   using: () => using
 });
 module.exports = __toCommonJS(src_exports);
-var import_koishi5 = require("koishi");
+var import_koishi6 = require("koishi");
 
 // src/Collector.ts
 var import_koishi = require("koishi");
@@ -198,29 +208,40 @@ var Collector = class _Collector {
     if (this.pendingUserRequests.has(cacheKey)) return this.pendingUserRequests.get(cacheKey);
     const promise = (async () => {
       try {
-        const existing = await this.ctx.database.get("analyse_user", { channelId, userId });
-        if (existing.length > 0) {
-          const { uid: uid2, userName: userName2, channelName: channelName2 } = existing[0];
-          const cachedUser2 = { uid: uid2, userName: userName2, channelName: channelName2 };
-          this.userCache.set(cacheKey, cachedUser2);
-          return cachedUser2;
+        const [dbUser] = await this.ctx.database.get("analyse_user", { channelId, userId });
+        if (dbUser && dbUser.userName && dbUser.channelName) {
+          this.userCache.set(cacheKey, dbUser);
+          return dbUser;
         }
         const [guild, member] = await Promise.all([
           guildId ? bot.getGuild(guildId).catch(() => null) : Promise.resolve(null),
           guildId ? bot.getGuildMember(guildId, userId).catch(() => null) : Promise.resolve(null)
         ]);
         const user = !member ? await bot.getUser(userId).catch(() => null) : null;
-        const newUserRecord = {
-          channelId,
-          userId,
-          channelName: guild?.name || channelId,
-          userName: member?.nick || member?.name || user?.name || userId
-        };
-        const createdUser = await this.ctx.database.create("analyse_user", newUserRecord);
-        const { uid, userName, channelName } = createdUser;
-        const cachedUser = { uid, userName, channelName };
-        this.userCache.set(cacheKey, cachedUser);
-        return cachedUser;
+        const fetchedUserName = member?.nick || member?.name || user?.name || "";
+        const fetchedChannelName = guild?.name || "";
+        if (dbUser) {
+          const needsUpdate = !dbUser.userName && fetchedUserName || !dbUser.channelName && fetchedChannelName;
+          if (needsUpdate) {
+            dbUser.userName = dbUser.userName || fetchedUserName;
+            dbUser.channelName = dbUser.channelName || fetchedChannelName;
+            await this.ctx.database.set("analyse_user", { uid: dbUser.uid }, {
+              userName: dbUser.userName,
+              channelName: dbUser.channelName
+            });
+          }
+          if (dbUser.userName && dbUser.channelName) this.userCache.set(cacheKey, dbUser);
+          return dbUser;
+        } else {
+          const createdUser = await this.ctx.database.create("analyse_user", {
+            channelId,
+            userId,
+            userName: fetchedUserName,
+            channelName: fetchedChannelName
+          });
+          if (createdUser.userName && createdUser.channelName) this.userCache.set(cacheKey, createdUser);
+          return createdUser;
+        }
       } catch (error) {
         this.ctx.logger.error(`创建或获取用户(${cacheKey})失败:`, error);
         return null;
@@ -884,8 +905,9 @@ var WhoAt = class {
         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 userId = senderInfo.id;
+          const userName = senderInfo.name || userId;
+          const authorElement = (0, import_koishi4.h)("author", { userId, name: userName });
           const contentElement = import_koishi4.h.text(record.content);
           return (0, import_koishi4.h)("message", {}, [authorElement, contentElement]);
         });
@@ -900,6 +922,165 @@ var WhoAt = class {
   }
 };
 
+// src/Data.ts
+var import_koishi5 = require("koishi");
+var fs = __toESM(require("fs/promises"));
+var path = __toESM(require("path"));
+var ALL_TABLES = [
+  "analyse_user",
+  "analyse_cmd",
+  "analyse_msg",
+  "analyse_rank",
+  "analyse_at",
+  "analyse_cache"
+];
+var Data = class {
+  constructor(ctx) {
+    this.ctx = ctx;
+    this.dataDir = path.join(this.ctx.baseDir, "data", "chat-analyse");
+  }
+  static {
+    __name(this, "Data");
+  }
+  dataDir;
+  /**
+   * @public
+   * @method registerCommands
+   * @description 在 'analyse.admin' 命令下注册所有数据管理相关的子命令。
+   * @param {Command} analyse - '.admin' 命令实例。
+   */
+  registerCommands(analyse) {
+    analyse.subcommand(".backup", "备份统计数据", { authority: 4 }).action(async () => {
+      try {
+        await fs.mkdir(this.dataDir, { recursive: true });
+        const allUsers = await this.ctx.database.get("analyse_user", {});
+        const uidToUserInfoMap = new Map(allUsers.map((u) => [u.uid, u]));
+        for (const tableName of ALL_TABLES) {
+          const filepath = path.join(this.dataDir, `${tableName}.json`);
+          let dataToExport;
+          if (tableName === "analyse_user") {
+            dataToExport = allUsers.map(({ uid, ...rest }) => rest);
+          } else {
+            const records = await this.ctx.database.get(tableName, {});
+            dataToExport = records.map((record) => {
+              const userInfo = uidToUserInfoMap.get(record.uid);
+              if (!userInfo) return null;
+              const { uid, ...restOfRecord } = record;
+              return {
+                userId: userInfo.userId,
+                channelId: userInfo.channelId,
+                ...restOfRecord
+              };
+            }).filter(Boolean);
+          }
+          await fs.writeFile(filepath, JSON.stringify(dataToExport, null, 2));
+        }
+        return `数据备份成功`;
+      } catch (error) {
+        this.ctx.logger.error("数据备份失败:", error);
+        return "数据备份失败";
+      }
+    });
+    analyse.subcommand(".restore", "恢复统计数据", { authority: 4 }).action(async () => {
+      const BATCH_SIZE = 100;
+      try {
+        const userTablePath = path.join(this.dataDir, "analyse_user.json");
+        try {
+          const usersToImport = JSON.parse(await fs.readFile(userTablePath, "utf-8"));
+          if (Array.isArray(usersToImport) && usersToImport.length > 0) {
+            for (let i = 0; i < usersToImport.length; i += BATCH_SIZE) {
+              const batch = usersToImport.slice(i, i + BATCH_SIZE);
+              await this.ctx.database.upsert("analyse_user", batch);
+            }
+          }
+        } catch (e) {
+          if (e.code !== "ENOENT") throw e;
+          this.ctx.logger.warn("无用户数据可恢复");
+        }
+        const allUsers = await this.ctx.database.get("analyse_user", {});
+        const userToUidMap = new Map(allUsers.map((u) => [`${u.channelId}:${u.userId}`, u.uid]));
+        for (const tableName of ALL_TABLES.filter((t) => t !== "analyse_user")) {
+          const filepath = path.join(this.dataDir, `${tableName}.json`);
+          try {
+            const recordsToImport = JSON.parse(await fs.readFile(filepath, "utf-8"));
+            if (Array.isArray(recordsToImport) && recordsToImport.length > 0) {
+              const recordsWithUid = recordsToImport.map((r) => {
+                const uid = userToUidMap.get(`${r.channelId}:${r.userId}`);
+                if (!uid) return null;
+                const { userId, channelId, ...rest } = r;
+                return { uid, ...rest };
+              }).filter(Boolean);
+              if (recordsWithUid.length > 0) {
+                for (let i = 0; i < recordsWithUid.length; i += BATCH_SIZE) {
+                  const batch = recordsWithUid.slice(i, i + BATCH_SIZE);
+                  await this.ctx.database.upsert(tableName, batch);
+                }
+              }
+            }
+          } catch (e) {
+            if (e.code !== "ENOENT") throw e;
+          }
+        }
+        return `数据恢复成功`;
+      } catch (error) {
+        this.ctx.logger.error("数据恢复失败:", error);
+        return "数据恢复失败";
+      }
+    });
+    analyse.subcommand(".clear", "清理统计数据", { authority: 4 }).option("table", "-t <table:string> 指定表").option("guild", "-g <guildId:string> 指定群组").option("user", "-u <user:string> 指定用户").option("days", "-d <days:number> 指定天数").option("all", "-a 清理全部数据").action(async ({ options }) => {
+      if (!options.table && !options.guild && !options.user && !options.all && !options.days) return "请提供清理条件";
+      try {
+        if (options.all) {
+          for (const tableName of ALL_TABLES) await this.ctx.database.remove(tableName, {});
+          return "已清除所有聊天分析数据";
+        }
+        let tablesToClear;
+        if (options.table) {
+          if (!ALL_TABLES.includes(options.table)) return `无效表名: ${options.table}`;
+          tablesToClear = [options.table];
+        } else {
+          tablesToClear = ALL_TABLES.filter((t) => t !== "analyse_user");
+        }
+        const queryParts = { query: {}, desc: "" };
+        const descParts = [];
+        if (options.guild || options.user) {
+          const uidsToClear = [];
+          const scopeDesc = [];
+          if (options.guild) {
+            uidsToClear.push(...(await this.ctx.database.get("analyse_user", { channelId: options.guild })).map((u) => u.uid));
+            scopeDesc.push(`群组 ${options.guild} `);
+          }
+          if (options.user) {
+            const userId = import_koishi5.Element.select(options.user, "at")[0]?.attrs.id || options.user;
+            uidsToClear.push(...(await this.ctx.database.get("analyse_user", { userId })).map((u) => u.uid));
+            scopeDesc.push(`用户 ${userId} `);
+          }
+          const uniqueUids = [...new Set(uidsToClear)];
+          if (uniqueUids.length === 0) return "未找到该用户";
+          queryParts.query.uid = { $in: uniqueUids };
+          descParts.push(scopeDesc.join("、"));
+        }
+        if (options.days && options.days > 0) {
+          queryParts.query.timestamp = { $lt: new Date(Date.now() - options.days * import_koishi5.Time.day) };
+          descParts.push(`超过 ${options.days} 天`);
+        }
+        for (const tableName of tablesToClear) {
+          const finalQuery = { ...queryParts.query };
+          if (tableName === "analyse_user" && finalQuery.timestamp) delete finalQuery.timestamp;
+          await this.ctx.database.remove(tableName, finalQuery);
+        }
+        const targetStr = options.table ? `表 ${options.table} ` : "所有表";
+        const conditionStr = descParts.join(" 且 ");
+        const finalDescription = conditionStr ? `${targetStr} 中${conditionStr}的数据` : `${targetStr}的全部统计数据`;
+        return `已成功清理${finalDescription}`;
+      } 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);">
@@ -915,22 +1096,23 @@ var usage = `
 `;
 var name = "chat-analyse";
 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("启用原始记录")
+var Config = import_koishi6.Schema.intersect([
+  import_koishi6.Schema.object({
+    enableListener: import_koishi6.Schema.boolean().default(true).description("启用消息监听"),
+    enableOriRecord: import_koishi6.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("启用消息统计")
-  }).description("统计配置"),
-  import_koishi5.Schema.object({
-    enableRankStat: import_koishi5.Schema.boolean().default(true).description("启用发言排行"),
-    rankRetentionDays: import_koishi5.Schema.number().min(0).default(31).description("记录保留天数")
+  import_koishi6.Schema.object({
+    enableCmdStat: import_koishi6.Schema.boolean().default(true).description("启用命令统计"),
+    enableMsgStat: import_koishi6.Schema.boolean().default(true).description("启用消息统计"),
+    enableData: import_koishi6.Schema.boolean().default(false).description("启用数据管理")
+  }).description("功能配置"),
+  import_koishi6.Schema.object({
+    enableRankStat: import_koishi6.Schema.boolean().default(true).description("启用发言排行"),
+    rankRetentionDays: import_koishi6.Schema.number().min(0).default(31).description("记录保留天数")
   }).description("发言排行配置"),
-  import_koishi5.Schema.object({
-    enableWhoAt: import_koishi5.Schema.boolean().default(true).description("启用 @ 记录"),
-    atRetentionDays: import_koishi5.Schema.number().min(0).default(7).description("记录保留天数")
+  import_koishi6.Schema.object({
+    enableWhoAt: import_koishi6.Schema.boolean().default(true).description("启用 @ 记录"),
+    atRetentionDays: import_koishi6.Schema.number().min(0).default(7).description("记录保留天数")
   }).description("@ 记录配置")
 ]);
 function apply(ctx, config) {
@@ -938,6 +1120,7 @@ function apply(ctx, config) {
   const analyse = ctx.command("analyse", "聊天记录分析");
   new Stat(ctx, config).registerCommands(analyse);
   if (config.enableWhoAt) new WhoAt(ctx, config).registerCommand(analyse);
+  if (config.enableData) new Data(ctx).registerCommands(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.4.4",
+  "version": "0.4.6",
   "contributors": [
     "Yis_Rime <yis_rime@outlook.com>"
   ],