koishi-plugin-adapter-onebot-multi 0.0.16 → 0.0.17

package/lib/index.js

@@ -3353,6 +3353,54 @@ var LoadBalancer = class {
       botStats
     };
   }
+  /**
+   * 获取详细的负载均衡状态（供指令使用）
+   */
+  getDetailedStatus() {
+    const onlineBots = this.getOnlineBots();
+    const unassignedValue = this.config.unassignedValue || "";
+    let assignedChannels = 0;
+    let unassignedChannels = 0;
+    for (const [, assignee] of this.lastAssignees.entries()) {
+      if (assignee && assignee !== unassignedValue) {
+        assignedChannels++;
+      } else {
+        unassignedChannels++;
+      }
+    }
+    const bots = [];
+    for (const botId of onlineBots) {
+      const channels = this.botChannels.get(botId) || /* @__PURE__ */ new Set();
+      let assignedCount = 0;
+      for (const [, assignee] of this.lastAssignees.entries()) {
+        if (assignee === botId) {
+          assignedCount++;
+        }
+      }
+      const maxLoad = this.getBotMaxLoad(botId);
+      const remaining = maxLoad === 0 ? -1 : Math.max(0, maxLoad - assignedCount);
+      const botInstance = this.ctx.bots.find((b) => b.selfId === botId && b.platform === "onebot");
+      const nickname = botInstance?.user?.name || botId;
+      const avatarUrl = `http://q.qlogo.cn/headimg_dl?dst_uin=${botId}&spec=640`;
+      bots.push({
+        selfId: botId,
+        nickname,
+        avatarUrl,
+        channelCount: channels.size,
+        assignedCount,
+        maxLoad,
+        remaining
+      });
+    }
+    bots.sort((a, b) => b.assignedCount - a.assignedCount);
+    return {
+      onlineBots: onlineBots.length,
+      totalChannels: this.allChannels.size,
+      assignedChannels,
+      unassignedChannels,
+      bots
+    };
+  }
 };
 
 // src/index.ts
@@ -3402,10 +3450,51 @@ function apply(ctx, config) {
   if (config.panel?.enabled) {
     new StatusPanel(ctx, config.panel, statusManager, configManager);
   }
+  let loadBalancer = null;
   if (config.loadBalance?.enabled) {
-    new LoadBalancer(ctx, config.loadBalance, statusManager);
+    loadBalancer = new LoadBalancer(ctx, config.loadBalance, statusManager);
     logger.info("负载均衡已启用");
   }
+  ctx.command("onebot.balance", "查看 OneBot 负载均衡状态").alias("ob.balance").alias("负载均衡").action(async () => {
+    if (!loadBalancer || !config.loadBalance?.enabled) {
+      return "负载均衡未启用。";
+    }
+    const status = loadBalancer.getDetailedStatus();
+    const elements = [];
+    elements.push(import_koishi12.h.text("═══ OneBot 负载均衡状态 ═══\n\n"));
+    elements.push(import_koishi12.h.text(`📊 总览
+`));
+    elements.push(import_koishi12.h.text(`  在线 Bot: ${status.onlineBots} 个
+`));
+    elements.push(import_koishi12.h.text(`  管理群数: ${status.totalChannels} 个
+`));
+    elements.push(import_koishi12.h.text(`  已分配群: ${status.assignedChannels} 个
+`));
+    elements.push(import_koishi12.h.text(`  未分配群: ${status.unassignedChannels} 个
+
+`));
+    elements.push(import_koishi12.h.text(`🤖 Bot 负载详情
+`));
+    for (const bot of status.bots) {
+      const maxLoadStr = bot.maxLoad === 0 ? "无限制" : `${bot.maxLoad}`;
+      const remainingStr = bot.maxLoad === 0 ? "∞" : `${bot.remaining}`;
+      const barLength = 20;
+      const usedLength = bot.maxLoad === 0 ? Math.min(Math.round(bot.assignedCount / Math.max(bot.channelCount, 1) * barLength), barLength) : Math.round(bot.assignedCount / bot.maxLoad * barLength);
+      const bar = "█".repeat(usedLength) + "░".repeat(barLength - usedLength);
+      elements.push(import_koishi12.h.image(bot.avatarUrl));
+      elements.push(import_koishi12.h.text("\n"));
+      elements.push(import_koishi12.h.text(`  ${bot.nickname} (${bot.selfId})
+`));
+      elements.push(import_koishi12.h.text(`  所在群: ${bot.channelCount} | 负责群: ${bot.assignedCount} | 上限: ${maxLoadStr}
+`));
+      elements.push(import_koishi12.h.text(`  剩余容量: ${remainingStr}
+`));
+      elements.push(import_koishi12.h.text(`  [${bar}]
+
+`));
+    }
+    return elements;
+  });
   ctx.inject(["console"], (ctx2) => {
     ctx2.console.addEntry({
       dev: (0, import_path.resolve)(__dirname, "../client/index.ts"),

package/lib/load-balancer.d.ts

@@ -85,4 +85,22 @@ export declare class LoadBalancer {
             assignedCount: number;
         }>;
     };
+    /**
+     * 获取详细的负载均衡状态（供指令使用）
+     */
+    getDetailedStatus(): {
+        onlineBots: number;
+        totalChannels: number;
+        assignedChannels: number;
+        unassignedChannels: number;
+        bots: {
+            selfId: string;
+            nickname: string;
+            avatarUrl: string;
+            channelCount: number;
+            assignedCount: number;
+            maxLoad: number;
+            remaining: number;
+        }[];
+    };
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-adapter-onebot-multi",
   "description": "奶龙bot定制版onebot适配器，支持自动负载均衡，适配器级黑名单/白名单，提供webui，可指定端口",
-  "version": "0.0.16",
+  "version": "0.0.17",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [