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

package/lib/index.js

@@ -894,6 +894,13 @@ var WsClient = class extends import_koishi5.Adapter.WsClient {
   static {
     __name(this, "WsClient");
   }
+  constructor(ctx, bot) {
+    super(ctx, bot);
+    ctx.on("http/websocket-init", (url, options) => {
+      ;
+      options.maxPayload = 500 * 1024 * 1024;
+    });
+  }
   accept(socket) {
     accept(socket, this.bot);
   }
@@ -975,9 +982,18 @@ function accept(socket, bot) {
       delete listeners[parsed.echo];
     }
   });
-  socket.addEventListener("close", () => {
+  socket.addEventListener("error", (event) => {
+    bot.logger.warn(`WebSocket 错误: ${event.message || "未知错误"}`);
+  });
+  socket.addEventListener("close", (event) => {
     delete bot.internal._request;
-    bot.offline();
+    const closeInfo = `code: ${event.code}, reason: ${event.reason || "无"}`;
+    if (event.code === 1009) {
+      bot.logger.error(`WebSocket 因消息过大断开 (${closeInfo})，协议端可能发送了超大消息`);
+    } else if (event.code !== 1e3 && event.code !== 1001) {
+      bot.logger.warn(`WebSocket 连接关闭 (${closeInfo})`);
+    }
+    ;
     bot.ctx.emit("onebot-multi/bot-offline", bot);
   });
   bot.internal._request = (action, params) => {
@@ -1051,7 +1067,7 @@ var HeartbeatMonitor = class {
         }
       } else {
         if (this.bot.status === import_koishi6.Universal.Status.ONLINE) {
-          this.bot.offline();
+          this.logger.warn(`Bot ${this.bot.selfId} 协议端报告离线，等待恢复...`);
           this.ctx.emit("onebot-multi/bot-offline", this.bot);
         }
       }
@@ -1064,10 +1080,7 @@ var HeartbeatMonitor = class {
       }
       if (this.lastOnlineState !== false) {
         this.lastOnlineState = false;
-        if (this.bot.status === import_koishi6.Universal.Status.ONLINE) {
-          this.bot.offline(error);
-          this.ctx.emit("onebot-multi/bot-offline", this.bot);
-        }
+        this.ctx.emit("onebot-multi/bot-offline", this.bot);
       }
     }
   }
@@ -3353,6 +3366,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 +3463,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/lib/ws.d.ts

@@ -6,6 +6,7 @@ export interface SharedConfig<T = 'ws' | 'ws-reverse'> {
     responseTimeout?: number;
 }
 export declare class WsClient<C extends Context = Context> extends Adapter.WsClient<C, OneBotBot<C, any>> {
+    constructor(ctx: C, bot: OneBotBot<C>);
     accept(socket: Universal.WebSocket): void;
     prepare(): any;
 }

package/package.json

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