koishi-plugin-chat-analyse 0.4.5 → 0.4.7

package/lib/Debug.d.ts

@@ -0,0 +1,22 @@
+import { Context, Command } from 'koishi';
+/**
+ * @class Debug
+ * @description
+ * 提供一系列调试工具，用于数据维护和状态检查。
+ * 包括手动补全用户信息、列出数据库中的频道和命令等功能。
+ */
+export declare class Debug {
+    private ctx;
+    /**
+     * @constructor
+     * @param {Context} ctx - Koishi 的插件上下文。
+     */
+    constructor(ctx: Context);
+    /**
+     * @public
+     * @method registerCommands
+     * @description 在 'analyse' 命令下注册所有调试相关的子命令。
+     * @param {Command} analyse - 主 'analyse' 命令实例。
+     */
+    registerCommands(analyse: Command): void;
+}

package/lib/index.d.ts

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

package/lib/index.js

@@ -208,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;
@@ -894,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]);
         });
@@ -1069,6 +1081,81 @@ var Data = class {
   }
 };
 
+// src/Debug.ts
+var Debug = class {
+  /**
+   * @constructor
+   * @param {Context} ctx - Koishi 的插件上下文。
+   */
+  constructor(ctx) {
+    this.ctx = ctx;
+  }
+  static {
+    __name(this, "Debug");
+  }
+  /**
+   * @public
+   * @method registerCommands
+   * @description 在 'analyse' 命令下注册所有调试相关的子命令。
+   * @param {Command} analyse - 主 'analyse' 命令实例。
+   */
+  registerCommands(analyse) {
+    analyse.subcommand(".fill", "手动补全用户信息", { authority: 4 }).action(async ({ session }) => {
+      const bots = this.ctx.bots;
+      if (bots.length === 0) return "暂无可用机器人";
+      const usersToUpdate = await this.ctx.database.get("analyse_user", {
+        $or: [{ userName: "" }, { channelName: "" }]
+      });
+      if (usersToUpdate.length === 0) return "暂无用户信息需要补全";
+      const usersByChannel = usersToUpdate.reduce((acc, user) => {
+        (acc[user.channelId] = acc[user.channelId] || []).push(user);
+        return acc;
+      }, {});
+      let updatedCount = 0;
+      const bot = bots.find((b) => b.platform === session.platform) || bots[0];
+      for (const channelId in usersByChannel) {
+        const usersInChannel = usersByChannel[channelId];
+        let channelName = usersInChannel.find((u) => u.channelName)?.channelName || "";
+        if (!channelName && channelId) {
+          try {
+            channelName = (await bot.getGuild(channelId))?.name || "";
+          } catch (e) {
+            this.ctx.logger.warn(`获取频道 ${channelId} 信息失败:`, e);
+          }
+        }
+        for (const user of usersInChannel) {
+          if (user.userName && user.channelName) continue;
+          let userName = user.userName;
+          if (!userName && user.userId && channelId) {
+            try {
+              const member = await bot.getGuildMember(channelId, user.userId);
+              userName = member?.nick || member?.name || "";
+              if (!userName) userName = (await bot.getUser(user.userId))?.name || "";
+            } catch (e) {
+              this.ctx.logger.warn(`获取频道 ${channelId} 的用户 ${user.userId} 信息失败:`, e);
+            }
+          }
+          await this.ctx.database.set("analyse_user", { uid: user.uid }, {
+            userName: userName || user.userName,
+            channelName: channelName || user.channelName
+          });
+          updatedCount++;
+        }
+      }
+      return `已补全 ${updatedCount} 条用户信息`;
+    });
+    analyse.subcommand(".list", "列出频道及命令", { authority: 4 }).action(async () => {
+      const allChannelInfo = await this.ctx.database.get("analyse_user", {}, ["channelId", "channelName"]);
+      const uniqueChannels = [...new Map(allChannelInfo.map((item) => [item.channelId, item])).values()];
+      const channelOutput = uniqueChannels.length > 0 ? "频道列表:\n" + uniqueChannels.map((c) => `[${c.channelId}] ${c.channelName}`).join("\n") : "暂无频道记录";
+      const commands = await this.ctx.database.select("analyse_cmd").distinct("command").execute();
+      const commandOutput = commands.length > 0 ? "命令列表:\n" + commands.map((c) => c.command).join(", ") : "暂无命令记录";
+      return `${channelOutput}
+${commandOutput}`;
+    });
+  }
+};
+
 // 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);">
@@ -1091,8 +1178,7 @@ var Config = import_koishi6.Schema.intersect([
   }).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("启用数据管理")
+    enableMsgStat: import_koishi6.Schema.boolean().default(true).description("启用消息统计")
   }).description("功能配置"),
   import_koishi6.Schema.object({
     enableRankStat: import_koishi6.Schema.boolean().default(true).description("启用发言排行"),
@@ -1101,7 +1187,11 @@ var Config = import_koishi6.Schema.intersect([
   import_koishi6.Schema.object({
     enableWhoAt: import_koishi6.Schema.boolean().default(true).description("启用 @ 记录"),
     atRetentionDays: import_koishi6.Schema.number().min(0).default(7).description("记录保留天数")
-  }).description("@ 记录配置")
+  }).description("@ 记录配置"),
+  import_koishi6.Schema.object({
+    enableData: import_koishi6.Schema.boolean().default(false).description("启用数据管理"),
+    enableDebug: import_koishi6.Schema.boolean().default(false).description("启用调试工具")
+  }).description("高级功能")
 ]);
 function apply(ctx, config) {
   if (config.enableListener) new Collector(ctx, config);
@@ -1109,6 +1199,7 @@ function apply(ctx, config) {
   new Stat(ctx, config).registerCommands(analyse);
   if (config.enableWhoAt) new WhoAt(ctx, config).registerCommand(analyse);
   if (config.enableData) new Data(ctx).registerCommands(analyse);
+  if (config.enableDebug) new Debug(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.5",
+  "version": "0.4.7",
   "contributors": [
     "Yis_Rime <yis_rime@outlook.com>"
   ],