aws-runtime-bridge 1.9.79 → 1.9.81

package/package/aws-client-agent-mcp/dist/websocket-client.js

@@ -1,677 +1,2 @@
-/**
- * WebSocket 客户端模块
- * 负责与调度中心建立和维护 WebSocket 连接
- */
-import { EventEmitter } from 'node:events';
-import WebSocket from 'ws';
-import { logger } from './logger.js';
-const DEFAULT_HEARTBEAT_TIMEOUT_MS = 9000;
-/** 指数退避最大间隔（毫秒） */
-const DEFAULT_RECONNECT_MAX_INTERVAL = 30000;
-/** 指数退避随机抖动范围（毫秒） */
-const DEFAULT_RECONNECT_JITTER_MS = 1000;
-/** 日志批处理节流窗口（毫秒） */
-const LOG_BATCH_THROTTLE_MS = 100;
-/** 日志发送重试次数 */
-const LOG_SEND_MAX_RETRIES = 3;
-/** 1008 token 刷新连续失败上限：达到后终止重连，避免用旧 token 死循环 */
-const MAX_REFRESH_RETRIES = 3;
-/** 单次发送最大日志数 */
-const MAX_LOGS_PER_BATCH = 50;
-/**
- * 已处理 reliable 消息 ID 上限。
- * 服务端 ReliableMessageDeliveryService 在 ACK 超时/丢失时会重发同一 messageId 的消息，
- * 接收方需保留近期已处理 ID 以跳过重复业务处理；超出上限时淘汰最旧记录。
- */
-const PROCESSED_MSG_ID_MAX_SIZE = 1000;
-export class WebSocketClient extends EventEmitter {
-    ws = null;
-    config;
-    agentId = null;
-    heartbeatTimer = null;
-    heartbeatTimeoutTimer = null;
-    reconnectAttempts = 0;
-    reconnectTimer = null;
-    isManualClose = false;
-    websocketFactory;
-    /** 已尝试刷新过的 token，仅当真正拿到新 token 时记录，用于 hammer 保护 */
-    lastRefreshedToken = null;
-    /** 1008 token 刷新连续失败次数，达到上限后终止重连 */
-    consecutiveRefreshFailures = 0;
-    // 日志批处理相关
-    logBatchQueue = [];
-    logBatchTimer = null;
-    logSequenceNumber = 0;
-    pendingLogBatches = new Map();
-    isProcessingLogBatch = false;
-    /**
-     * 已处理的 reliable 消息 ID（msgId → 处理时间戳）。
-     * 用于对冲服务端 ACK 超时/丢失导致的消息重投，覆盖所有 reliable 事件类型，
-     * 而非仅靠下游 MessageBuffer 对私信/群聊的去重。
-     */
-    processedMsgIds = new Map();
-    constructor(config, websocketFactory) {
-        super();
-        this.config = config;
-        this.websocketFactory = websocketFactory ?? ((url) => new WebSocket(url));
-    }
-    /**
-       * 建立 WebSocket 连接
-       * 主流程：准备连接 -> 创建连接 -> 设置事件监听
-       */
-    connect(agentId) {
-        return new Promise((resolve, reject) => {
-            this.prepareConnection(agentId);
-            const wsUrl = this.buildWebSocketUrl(agentId);
-            logger.info(`[WebSocket] 正在连接: ${this.maskRuntimeTokenInUrl(wsUrl)}`);
-            const { rejectOnce, resolveOnce } = this.createSettledHandlers(resolve, reject);
-            try {
-                this.ws = this.websocketFactory(wsUrl);
-                this.setupWebSocketEventHandlers(this.ws, rejectOnce, resolveOnce);
-            }
-            catch (error) {
-                rejectOnce(error instanceof Error ? error : new Error(String(error)));
-            }
-        });
-    }
-    /**
-     * 准备连接前的状态清理
-     */
-    prepareConnection(agentId) {
-        this.agentId = agentId;
-        this.isManualClose = false;
-        if (this.ws && this.ws.readyState !== WebSocket.CLOSED) {
-            this.ws.removeAllListeners();
-            this.ws.terminate();
-        }
-    }
-    /**
-     * 构建 WebSocket URL
-     */
-    buildWebSocketUrl(agentId) {
-        const url = new URL(this.config.serverUrl);
-        url.searchParams.set('agentId', agentId);
-        const runtimeAccessToken = this.config.runtimeAccessToken?.trim();
-        if (runtimeAccessToken) {
-            url.searchParams.set('runtimeAccessToken', runtimeAccessToken);
-        }
-        return url.toString();
-    }
-    maskRuntimeTokenInUrl(url) {
-        try {
-            const parsedUrl = new URL(url);
-            if (parsedUrl.searchParams.has('runtimeAccessToken')) {
-                parsedUrl.searchParams.set('runtimeAccessToken', '***');
-            }
-            return parsedUrl.toString();
-        }
-        catch {
-            return url.replace(/([?&]runtimeAccessToken=)[^&]*/u, '$1***');
-        }
-    }
-    /**
-     * 创建一次性 resolve/reject 处理器
-     */
-    createSettledHandlers(resolve, reject) {
-        let settled = false;
-        const rejectOnce = (error) => {
-            if (!settled) {
-                settled = true;
-                reject(error);
-            }
-        };
-        const resolveOnce = () => {
-            if (!settled) {
-                settled = true;
-                resolve();
-            }
-        };
-        return { rejectOnce, resolveOnce };
-    }
-    /**
-     * 设置 WebSocket 事件监听器
-     */
-    setupWebSocketEventHandlers(ws, rejectOnce, resolveOnce) {
-        let opened = false;
-        ws.on('open', () => {
-            opened = true;
-            this.handleWebSocketOpen(resolveOnce);
-        });
-        ws.on('message', (data) => this.handleWebSocketMessage(data));
-        ws.on('pong', () => this.markHeartbeatReceived());
-        ws.on('close', (code, reason) => this.handleWebSocketClose(code, reason, opened, rejectOnce));
-        ws.on('error', (error) => this.handleWebSocketError(error, opened, rejectOnce));
-    }
-    /**
-     * 处理 WebSocket 打开事件
-     */
-    handleWebSocketOpen(resolveOnce) {
-        logger.info('[WebSocket] 连接已建立');
-        this.reconnectAttempts = 0;
-        this.consecutiveRefreshFailures = 0;
-        this.lastRefreshedToken = null;
-        this.clearReconnectTimer();
-        this.startHeartbeat();
-        this.emit('connected');
-        resolveOnce();
-    }
-    /**
-     * 处理 WebSocket 消息
-     *
-     * 可靠消息去重：服务端 ReliableMessageDeliveryService 在 ACK 超时（5s）或丢失时会
-     * 重发同一 messageId 的消息（最多 3 次 + 用户上线重投）。接收方若不区分重投与
-     * 首次投递，会对同一事件（如 agentStatus、profileInfo 等）重复处理。
-     * 这里在 ACK 之前检查 messageId 是否已处理过：
-     *   - 重复消息：仍回 ACK（避免服务端继续重试堆积），但跳过 handleMessage；
-     *   - 新消息：记录 messageId，回 ACK，正常处理。
-     */
-    handleWebSocketMessage(data) {
-        try {
-            const message = JSON.parse(data.toString());
-            const reliableMsgId = this.extractReliableMessageId(message);
-            const isDuplicate = reliableMsgId !== null && this.markMsgIdProcessed(reliableMsgId);
-            // 始终回 ACK，即使是重复消息，避免服务端重试队列堆积
-            this.acknowledgeReliableMessage(message);
-            if (isDuplicate) {
-                logger.info('[WebSocket] 忽略重复投递的可靠消息: msgId=%s, type=%s', reliableMsgId, message.type);
-                return;
-            }
-            this.handleMessage(message);
-        }
-        catch (error) {
-            const rawText = data.toString();
-            const preview = rawText.length > 200 ? `${rawText.slice(0, 200)}...(truncated, ${rawText.length} total)` : rawText;
-            logger.warn('[WebSocket] 消息解析失败, dataLength=%s, preview=%s:', rawText.length, preview, error);
-        }
-    }
-    /**
-     * 提取 reliable 消息的 messageId。
-     * 仅当 data 为对象、含非空 _msgId 且 _requireAck 为真时返回 messageId，否则返回 null。
-     */
-    extractReliableMessageId(message) {
-        const payload = message.data;
-        if (!payload || typeof payload !== 'object') {
-            return null;
-        }
-        const data = payload;
-        const messageId = String(data._msgId || '').trim();
-        const requireAck = data._requireAck === true || data._requireAck === 'true';
-        if (!messageId || !requireAck) {
-            return null;
-        }
-        return messageId;
-    }
-    /**
-     * 标记 messageId 为已处理。
-     * 返回 true 表示该 messageId 此前已被处理过（重复投递），false 表示首次处理。
-     * 超出上限时淘汰最旧记录，避免内存无界增长。
-     */
-    markMsgIdProcessed(messageId) {
-        if (this.processedMsgIds.has(messageId)) {
-            return true;
-        }
-        this.processedMsgIds.set(messageId, Date.now());
-        if (this.processedMsgIds.size > PROCESSED_MSG_ID_MAX_SIZE) {
-            let count = 0;
-            const toDelete = this.processedMsgIds.size - PROCESSED_MSG_ID_MAX_SIZE;
-            for (const key of this.processedMsgIds.keys()) {
-                if (count >= toDelete)
-                    break;
-                this.processedMsgIds.delete(key);
-                count++;
-            }
-        }
-        return false;
-    }
-    acknowledgeReliableMessage(message) {
-        const messageId = this.extractReliableMessageId(message);
-        if (!messageId) {
-            return;
-        }
-        this.send({
-            type: 'MESSAGE_ACK',
-            data: { msgId: messageId },
-            timestamp: new Date().toISOString(),
-        });
-    }
-    /**
-     * 处理 WebSocket 关闭事件
-     */
-    async handleWebSocketClose(code, reason, opened, rejectOnce) {
-        const closeReasonPart = reason.toString();
-        const wsState = this.ws?.readyState;
-        logger.warn(`[WebSocket] 连接关闭: code=${code}, reason=${closeReasonPart}, readyState=${wsState}, opened=${opened}, reconnectAttempts=${this.reconnectAttempts}`);
-        this.stopHeartbeat();
-        this.clearHeartbeatTimeout();
-        this.emit('disconnected', code, reason);
-        if (!opened) {
-            rejectOnce(new Error(`WebSocket 连接在建立前关闭: code=${code}, reason=${closeReasonPart}`));
-        }
-        if (!this.isManualClose) {
-            if (code === 1008) {
-                await this.handleTokenExpiredAndReconnect();
-            }
-            else {
-                this.scheduleReconnect();
-            }
-        }
-    }
-    /**
-     * WebSocket 1008（token 失效）时的反应式刷新。
-     * 修复死循环：旧实现在刷新失败时仍用旧 token 立即重连，导致被 1008 反复拒绝、
-     * 200ms 无限重连且 Agent 永远 offline。新实现：
-     *  - 仅当拿到【新】token 时才视为刷新成功并立即重连；
-     *  - 服务端返回 token 不变（unchanged）说明换 token 无法解决 1008，直接终止重连，避免死循环；
-     *  - 刷新失败（网络/超时/异常）计入连续失败次数，达到上限后终止重连并上报事件，
-     *    交由上层 supervisor 重启，而非用旧 token 无限重试。
-     */
-    async handleTokenExpiredAndReconnect() {
-        const result = await this.refreshRuntimeTokenOnPolicyRejection();
-        if (result === 'refreshed') {
-            logger.info('[WebSocket] runtime token 已刷新，使用新 token 重连');
-            this.consecutiveRefreshFailures = 0;
-            this.scheduleReconnect();
-            return;
-        }
-        if (result === 'unchanged') {
-            // 服务端认为 token 仍然有效却以 1008 拒绝连接，换 token 无意义，终止重连避免死循环
-            logger.error('[WebSocket] bridge 返回 token 未变化，但服务端仍以 1008 拒绝，停止重连（需人工或 supervisor 介入）');
-            this.emit('tokenRejected', this.config.runtimeAccessToken);
-            return;
-        }
-        // result === 'failed'：刷新失败，带退避地重试，但受连续失败上限约束
-        this.consecutiveRefreshFailures += 1;
-        if (this.consecutiveRefreshFailures >= MAX_REFRESH_RETRIES) {
-            logger.error(`[WebSocket] runtime token 刷新连续失败 ${this.consecutiveRefreshFailures} 次，停止重连（交由 supervisor 重启）`);
-            this.emit('tokenRefreshFailed', this.consecutiveRefreshFailures);
-            return;
-        }
-        logger.warn(`[WebSocket] runtime token 刷新失败，第 ${this.consecutiveRefreshFailures}/${MAX_REFRESH_RETRIES} 次尝试重连`);
-        this.scheduleReconnect();
-    }
-    /**
-     * 向 bridge 请求刷新 runtimeAccessToken（forceRefresh 模式）。
-     * 返回：
-     *  - 'refreshed'：成功拿到【新】token 并已更新 config；
-     *  - 'unchanged'：服务端返回 token 与当前一致（换 token 无法解决 1008）；
-     *  - 'failed'：刷新请求失败（网络/超时/非 2xx/异常）。
-     * 仅当真正获得新 token 时才记录 lastRefreshedToken（hammer 保护），
-     * 失败时不再把当前 token 标记为"已刷新"，从而允许后续在退避约束下重试。
-     */
-    async refreshRuntimeTokenOnPolicyRejection() {
-        const bridgeBaseUrl = this.config.runtimeBridgeBaseUrl;
-        if (!bridgeBaseUrl || !this.agentId) {
-            return 'failed';
-        }
-        const currentToken = this.config.runtimeAccessToken || '';
-        const timeoutMs = this.config.mcpHttpRequestTimeoutMs ?? 120_000;
-        const controller = new AbortController();
-        const timer = setTimeout(() => controller.abort(), timeoutMs);
-        try {
-            const response = await fetch(`${bridgeBaseUrl}/runtime/binding/refresh-token`, {
-                method: 'POST',
-                headers: { 'Content-Type': 'application/json' },
-                body: JSON.stringify({
-                    agentId: this.agentId,
-                    currentToken,
-                    forceRefresh: true,
-                }),
-                signal: controller.signal,
-            });
-            if (!response.ok) {
-                logger.warn(`[WebSocket] bridge refresh-token 返回 ${response.status}，无法刷新`);
-                return 'failed';
-            }
-            const data = (await response.json());
-            const newToken = data.runtimeAccessToken;
-            if (!newToken || newToken === currentToken) {
-                // token 未变化：说明问题不在 token，换 token 无法解决 1008
-                return 'unchanged';
-            }
-            this.config.runtimeAccessToken = newToken;
-            this.lastRefreshedToken = currentToken;
-            logger.info('[WebSocket] runtimeAccessToken 已通过 bridge refresh 更新');
-            return 'refreshed';
-        }
-        catch (error) {
-            if (error instanceof Error && error.name === 'AbortError') {
-                logger.warn(`[WebSocket] bridge refresh-token 超时（${timeoutMs}ms）`);
-            }
-            else {
-                logger.warn('[WebSocket] bridge refresh-token 异常: %s', error instanceof Error ? error.message : String(error));
-                if (error instanceof Error && error.stack) {
-                    logger.warn('[WebSocket] refresh-token 异常堆栈:', error.stack.slice(0, 2000));
-                }
-            }
-            return 'failed';
-        }
-        finally {
-            clearTimeout(timer);
-        }
-    }
-    /**
-     * 处理 WebSocket 错误事件
-     */
-    handleWebSocketError(error, opened, rejectOnce) {
-        const wsState = this.ws?.readyState;
-        logger.error('[WebSocket] 连接错误: readyState=%s, opened=%s, reconnectAttempts=%s, message=%s', wsState, opened, this.reconnectAttempts, error.message);
-        logger.error('[WebSocket] 错误堆栈:', error.stack?.slice(0, 2000));
-        this.emit('error', error);
-        if (!opened) {
-            rejectOnce(error);
-        }
-    }
-    /**
-     * 断开 WebSocket 连接
-     */
-    disconnect() {
-        this.isManualClose = true;
-        this.stopHeartbeat();
-        this.clearHeartbeatTimeout();
-        this.clearReconnectTimer();
-        if (this.ws) {
-            if (this.ws.readyState === WebSocket.OPEN) {
-                this.ws.close(1000, 'Agent logout');
-            }
-            this.ws = null;
-        }
-        this.agentId = null;
-        logger.info('[WebSocket] 已断开连接');
-    }
-    /**
-     * 发送消息
-     */
-    send(message) {
-        const wsState = this.ws?.readyState;
-        if (!this.ws || wsState !== WebSocket.OPEN) {
-            logger.warn('[WebSocket] 连接未建立，无法发送消息: readyState=%s, isManualClose=%s, agentId=%s', wsState, this.isManualClose, this.agentId);
-            return false;
-        }
-        try {
-            this.ws.send(JSON.stringify(message));
-            return true;
-        }
-        catch (error) {
-            logger.error('[WebSocket] send() 发送消息异常: readyState=%s, error=%s', wsState, error);
-            return false;
-        }
-    }
-    /**
-     * 上报 MCP Node.js 进程启动后的运行日志（批处理模式）。
-     * 日志先入队，每 100ms 节流窗口内累积后批量发送，
-     * 等待 bridge 返回 LOG_ACK 确认后再发送下一批。
-     */
-    sendMcpNodeLog(data) {
-        if (!this.isConnected())
-            return false;
-        this.logBatchQueue.push(data);
-        // 启动节流定时器（如果尚未启动）
-        if (!this.logBatchTimer) {
-            this.logBatchTimer = setTimeout(() => {
-                this.logBatchTimer = null;
-                this.flushLogBatch();
-            }, LOG_BATCH_THROTTLE_MS);
-        }
-        return true;
-    }
-    /**
-     * 刷新日志批次：取出队列中的日志，按序列号打包发送。
-     * 等待 bridge 返回 LOG_ACK 后方可发送下一批。
-     */
-    flushLogBatch() {
-        if (this.logBatchQueue.length === 0 || this.isProcessingLogBatch)
-            return;
-        const batch = this.logBatchQueue.splice(0, MAX_LOGS_PER_BATCH);
-        if (batch.length === 0)
-            return;
-        this.isProcessingLogBatch = true;
-        this.logSequenceNumber += 1;
-        const seqNum = this.logSequenceNumber;
-        // 创建 Promise，等待 LOG_ACK
-        const promise = new Promise((resolve, reject) => {
-            this.pendingLogBatches.set(seqNum, {
-                logs: batch,
-                sequenceNumber: seqNum,
-                retryCount: 0,
-                resolve,
-                reject,
-            });
-        });
-        promise.then(() => {
-            // 当前批次已确认，处理下一批
-            this.isProcessingLogBatch = false;
-            if (this.logBatchQueue.length > 0) {
-                // 如果还有新日志，继续下一批次
-                this.logBatchTimer = setTimeout(() => {
-                    this.logBatchTimer = null;
-                    this.flushLogBatch();
-                }, LOG_BATCH_THROTTLE_MS);
-            }
-        }).catch(() => {
-            // 批次发送彻底失败，重置状态以允许后续日志继续发送
-            this.isProcessingLogBatch = false;
-        });
-        this.sendLogBatch(seqNum);
-    }
-    /**
-     * 发送日志批次（实际发送 + 重试逻辑）
-     */
-    sendLogBatch(seqNum) {
-        const pending = this.pendingLogBatches.get(seqNum);
-        if (!pending)
-            return;
-        const success = this.send({
-            type: 'MCP_NODE_LOG_BATCH',
-            data: {
-                sequenceNumber: seqNum,
-                logs: pending.logs,
-            },
-            timestamp: new Date().toISOString(),
-        });
-        if (!success) {
-            this.handleLogBatchSendFailure(seqNum);
-        }
-        // 如果发送成功，等待 LOG_ACK 回调
-    }
-    /**
-     * 处理日志批次发送失败（重试或放弃）
-     */
-    handleLogBatchSendFailure(seqNum) {
-        const pending = this.pendingLogBatches.get(seqNum);
-        if (!pending)
-            return;
-        pending.retryCount += 1;
-        if (pending.retryCount <= LOG_SEND_MAX_RETRIES) {
-            logger.warn(`[WebSocket] 日志批次 #${seqNum} 发送失败，第 ${pending.retryCount} 次重试`);
-            setTimeout(() => {
-                this.sendLogBatch(seqNum);
-            }, LOG_BATCH_THROTTLE_MS * pending.retryCount);
-        }
-        else {
-            logger.error(`[WebSocket] 日志批次 #${seqNum} 发送失败，已重试 ${LOG_SEND_MAX_RETRIES} 次，丢弃`);
-            this.pendingLogBatches.delete(seqNum);
-            pending.reject(new Error(`日志批次 #${seqNum} 发送失败`));
-        }
-    }
-    /**
-     * 处理 LOG_ACK 确认。
-     * 收到 bridge 对某序列号日志批次的确认后，标记该批次完成，继续发送下一批。
-     */
-    handleLogAck(data) {
-        const ackData = data;
-        const seqNum = Number(ackData?.ackSequenceNumber);
-        if (!seqNum || !this.pendingLogBatches.has(seqNum))
-            return;
-        const pending = this.pendingLogBatches.get(seqNum);
-        if (!pending)
-            return;
-        this.pendingLogBatches.delete(seqNum);
-        pending.resolve();
-    }
-    /**
-     * 是否已连接
-     */
-    isConnected() {
-        return this.ws !== null && this.ws.readyState === WebSocket.OPEN;
-    }
-    /**
-     * 处理接收到的消息
-     */
-    handleMessage(message) {
-        switch (message.type) {
-            case 'GROUP_MSG':
-                this.emit('groupMessage', message.data);
-                break;
-            case 'DM_MSG':
-                this.emit('directMessage', message.data);
-                break;
-            case 'ONLINE_CHANGE':
-                this.emit('onlineChange', message.data);
-                break;
-            case 'HEARTBEAT':
-                this.markHeartbeatReceived();
-                break;
-            case 'PROFILE_INFO':
-                logger.info('[WebSocket] 收到 PROFILE_INFO 消息');
-                this.emit('profileInfo', message.data);
-                break;
-            case 'AGENT_STATUS':
-                logger.info('[WebSocket] 收到 AGENT_STATUS 消息');
-                this.emit('agentStatus', message.data);
-                break;
-            case 'LOG_ACK':
-                this.handleLogAck(message.data);
-                break;
-            case 'GROUP_READ_RECEIPT':
-            case 'DM_READ_RECEIPT':
-            case 'TERMINAL_OUTPUT':
-            case 'TERMINAL_STATUS':
-            case 'MCP_NODE_LOG':
-            case 'AGENT_EVENTS_DELTA':
-                break;
-            default:
-                logger.warn('[WebSocket] 未知消息类型:', message.type);
-        }
-    }
-    /**
-     * 启动心跳
-     */
-    startHeartbeat() {
-        this.stopHeartbeat();
-        this.heartbeatTimer = setInterval(() => {
-            if (this.isConnected()) {
-                this.ensureHeartbeatTimeout();
-                this.sendProtocolPing();
-                this.send({
-                    type: 'HEARTBEAT',
-                    data: 'ping',
-                    timestamp: new Date().toISOString()
-                });
-            }
-        }, this.config.heartbeatInterval);
-    }
-    /**
-     * 停止心跳
-     */
-    stopHeartbeat() {
-        if (this.heartbeatTimer) {
-            clearInterval(this.heartbeatTimer);
-            this.heartbeatTimer = null;
-        }
-        this.clearHeartbeatTimeout();
-    }
-    sendProtocolPing() {
-        if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
-            return;
-        }
-        try {
-            this.ws.ping();
-        }
-        catch (error) {
-            logger.warn('[WebSocket] 协议层 ping 发送失败，终止连接并重连:', error instanceof Error ? error.message : String(error));
-            this.ws.terminate();
-        }
-    }
-    markHeartbeatReceived() {
-        this.clearHeartbeatTimeout();
-    }
-    ensureHeartbeatTimeout() {
-        if (this.heartbeatTimeoutTimer) {
-            return;
-        }
-        this.heartbeatTimeoutTimer = setTimeout(() => {
-            this.heartbeatTimeoutTimer = null;
-            if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
-                return;
-            }
-            logger.warn(`[WebSocket] ${this.resolveHeartbeatTimeout()}ms 未收到心跳响应，终止连接并触发重连`);
-            this.ws.terminate();
-        }, this.resolveHeartbeatTimeout());
-    }
-    clearHeartbeatTimeout() {
-        if (this.heartbeatTimeoutTimer) {
-            clearTimeout(this.heartbeatTimeoutTimer);
-            this.heartbeatTimeoutTimer = null;
-        }
-    }
-    resolveHeartbeatTimeout() {
-        return this.config.heartbeatTimeout ?? DEFAULT_HEARTBEAT_TIMEOUT_MS;
-    }
-    /**
-     * 安排重连（指数退避 + 随机抖动）
-     *
-     * 主流程：校验重连上限 -> 计算指数退避间隔（含抖动）-> 延迟后重连
-     *
-     * 退避公式：baseDelay = min(reconnectInterval * 2^(attempt-1), reconnectMaxInterval)
-     *          jitter    = random(0, reconnectJitterMs)
-     *          delay     = baseDelay + jitter
-     *
-     * 首次重连 (attempt=1)：baseDelay = reconnectInterval（200ms），与之前行为一致
-     * 后续重连指数递增，上限 30s + 1s 随机抖动，避免重连风暴
-     */
-    scheduleReconnect() {
-        const hasRetryLimit = this.config.maxReconnectAttempts !== -1;
-        if (hasRetryLimit && this.reconnectAttempts >= this.config.maxReconnectAttempts) {
-            logger.error('[WebSocket] 重连次数已达上限，放弃重连');
-            this.emit('reconnectFailed');
-            return;
-        }
-        if (this.reconnectTimer) {
-            return;
-        }
-        this.reconnectAttempts++;
-        // 指数退避：baseDelay = initial * 2^(attempt-1)，cap 在 maxInterval
-        const baseDelay = Math.min(this.config.reconnectInterval * Math.pow(2, this.reconnectAttempts - 1), this.config.reconnectMaxInterval ?? DEFAULT_RECONNECT_MAX_INTERVAL);
-        // 随机抖动：在 [0, jitterMs) 范围内均匀分布，避免 thundering herd
-        const jitter = Math.random() * (this.config.reconnectJitterMs ?? DEFAULT_RECONNECT_JITTER_MS);
-        const delay = Math.round(baseDelay + jitter);
-        const retryInfo = hasRetryLimit
-            ? `第 ${this.reconnectAttempts}/${this.config.maxReconnectAttempts} 次`
-            : `第 ${this.reconnectAttempts} 次（无限重试模式）`;
-        logger.warn(`[WebSocket] ${delay}ms 后尝试${retryInfo}重连 (base=${Math.round(baseDelay)}ms, jitter=+${Math.round(jitter)}ms)...`);
-        this.reconnectTimer = setTimeout(() => {
-            this.reconnectTimer = null;
-            if (this.agentId) {
-                this.connect(this.agentId).catch((error) => {
-                    const errMsg = error instanceof Error ? error.message : String(error);
-                    const errStack = error instanceof Error ? error.stack : undefined;
-                    logger.warn('[WebSocket] 重连失败 (attempt #%s), 继续重试: %s', this.reconnectAttempts, errMsg);
-                    if (errStack) {
-                        logger.warn('[WebSocket] 重连失败堆栈:', errStack.slice(0, 2000));
-                    }
-                    this.scheduleReconnect();
-                });
-            }
-        }, delay);
-    }
-    /**
-     * 清除重连定时器
-     */
-    clearReconnectTimer() {
-        if (this.reconnectTimer) {
-            clearTimeout(this.reconnectTimer);
-            this.reconnectTimer = null;
-        }
-    }
-}
-//# sourceMappingURL=websocket-client.js.map
+import{EventEmitter as m}from"node:events";import c from"ws";import{logger as r}from"./logger.js";const f=9e3,T=3e4,b=1e3,l=100,u=3,d=3,k=50,g=1e3;class w extends m{ws=null;config;agentId=null;heartbeatTimer=null;heartbeatTimeoutTimer=null;reconnectAttempts=0;reconnectTimer=null;isManualClose=!1;websocketFactory;lastRefreshedToken=null;consecutiveRefreshFailures=0;logBatchQueue=[];logBatchTimer=null;logSequenceNumber=0;pendingLogBatches=new Map;isProcessingLogBatch=!1;processedMsgIds=new Map;constructor(e,t){super(),this.config=e,this.websocketFactory=t??(s=>new c(s))}connect(e){return new Promise((t,s)=>{this.prepareConnection(e);const n=this.buildWebSocketUrl(e);r.info(`[WebSocket] \u6B63\u5728\u8FDE\u63A5: ${this.maskRuntimeTokenInUrl(n)}`);const{rejectOnce:o,resolveOnce:i}=this.createSettledHandlers(t,s);try{this.ws=this.websocketFactory(n),this.setupWebSocketEventHandlers(this.ws,o,i)}catch(a){o(a instanceof Error?a:new Error(String(a)))}})}prepareConnection(e){this.agentId=e,this.isManualClose=!1,this.ws&&this.ws.readyState!==c.CLOSED&&(this.ws.removeAllListeners(),this.ws.terminate())}buildWebSocketUrl(e){const t=new URL(this.config.serverUrl);t.searchParams.set("agentId",e);const s=this.config.runtimeAccessToken?.trim();return s&&t.searchParams.set("runtimeAccessToken",s),t.toString()}maskRuntimeTokenInUrl(e){try{const t=new URL(e);return t.searchParams.has("runtimeAccessToken")&&t.searchParams.set("runtimeAccessToken","***"),t.toString()}catch{return e.replace(/([?&]runtimeAccessToken=)[^&]*/u,"$1***")}}createSettledHandlers(e,t){let s=!1;return{rejectOnce:i=>{s||(s=!0,t(i))},resolveOnce:()=>{s||(s=!0,e())}}}setupWebSocketEventHandlers(e,t,s){let n=!1;e.on("open",()=>{n=!0,this.handleWebSocketOpen(s)}),e.on("message",o=>this.handleWebSocketMessage(o)),e.on("pong",()=>this.markHeartbeatReceived()),e.on("close",(o,i)=>this.handleWebSocketClose(o,i,n,t)),e.on("error",o=>this.handleWebSocketError(o,n,t))}handleWebSocketOpen(e){r.info("[WebSocket] \u8FDE\u63A5\u5DF2\u5EFA\u7ACB"),this.reconnectAttempts=0,this.consecutiveRefreshFailures=0,this.lastRefreshedToken=null,this.clearReconnectTimer(),this.startHeartbeat(),this.emit("connected"),e()}handleWebSocketMessage(e){try{const t=JSON.parse(e.toString()),s=this.extractReliableMessageId(t),n=s!==null&&this.markMsgIdProcessed(s);if(this.acknowledgeReliableMessage(t),n){r.info("[WebSocket] \u5FFD\u7565\u91CD\u590D\u6295\u9012\u7684\u53EF\u9760\u6D88\u606F: msgId=%s, type=%s",s,t.type);return}this.handleMessage(t)}catch(t){const s=e.toString(),n=s.length>200?`${s.slice(0,200)}...(truncated, ${s.length} total)`:s;r.warn("[WebSocket] \u6D88\u606F\u89E3\u6790\u5931\u8D25, dataLength=%s, preview=%s:",s.length,n,t)}}extractReliableMessageId(e){const t=e.data;if(!t||typeof t!="object")return null;const s=t,n=String(s._msgId||"").trim(),o=s._requireAck===!0||s._requireAck==="true";return!n||!o?null:n}markMsgIdProcessed(e){if(this.processedMsgIds.has(e))return!0;if(this.processedMsgIds.set(e,Date.now()),this.processedMsgIds.size>g){let t=0;const s=this.processedMsgIds.size-g;for(const n of this.processedMsgIds.keys()){if(t>=s)break;this.processedMsgIds.delete(n),t++}}return!1}acknowledgeReliableMessage(e){const t=this.extractReliableMessageId(e);t&&this.send({type:"MESSAGE_ACK",data:{msgId:t},timestamp:new Date().toISOString()})}async handleWebSocketClose(e,t,s,n){const o=t.toString(),i=this.ws?.readyState;r.warn(`[WebSocket] \u8FDE\u63A5\u5173\u95ED: code=${e}, reason=${o}, readyState=${i}, opened=${s}, reconnectAttempts=${this.reconnectAttempts}`),this.stopHeartbeat(),this.clearHeartbeatTimeout(),this.emit("disconnected",e,t),s||n(new Error(`WebSocket \u8FDE\u63A5\u5728\u5EFA\u7ACB\u524D\u5173\u95ED: code=${e}, reason=${o}`)),this.isManualClose||(e===1008?await this.handleTokenExpiredAndReconnect():this.scheduleReconnect())}async handleTokenExpiredAndReconnect(){const e=await this.refreshRuntimeTokenOnPolicyRejection();if(e==="refreshed"){r.info("[WebSocket] runtime token \u5DF2\u5237\u65B0\uFF0C\u4F7F\u7528\u65B0 token \u91CD\u8FDE"),this.consecutiveRefreshFailures=0,this.scheduleReconnect();return}if(e==="unchanged"){r.error("[WebSocket] bridge \u8FD4\u56DE token \u672A\u53D8\u5316\uFF0C\u4F46\u670D\u52A1\u7AEF\u4ECD\u4EE5 1008 \u62D2\u7EDD\uFF0C\u505C\u6B62\u91CD\u8FDE\uFF08\u9700\u4EBA\u5DE5\u6216 supervisor \u4ECB\u5165\uFF09"),this.emit("tokenRejected",this.config.runtimeAccessToken);return}if(this.consecutiveRefreshFailures+=1,this.consecutiveRefreshFailures>=d){r.error(`[WebSocket] runtime token \u5237\u65B0\u8FDE\u7EED\u5931\u8D25 ${this.consecutiveRefreshFailures} \u6B21\uFF0C\u505C\u6B62\u91CD\u8FDE\uFF08\u4EA4\u7531 supervisor \u91CD\u542F\uFF09`),this.emit("tokenRefreshFailed",this.consecutiveRefreshFailures);return}r.warn(`[WebSocket] runtime token \u5237\u65B0\u5931\u8D25\uFF0C\u7B2C ${this.consecutiveRefreshFailures}/${d} \u6B21\u5C1D\u8BD5\u91CD\u8FDE`),this.scheduleReconnect()}async refreshRuntimeTokenOnPolicyRejection(){const e=this.config.runtimeBridgeBaseUrl;if(!e||!this.agentId)return"failed";const t=this.config.runtimeAccessToken||"",s=this.config.mcpHttpRequestTimeoutMs??12e4,n=new AbortController,o=setTimeout(()=>n.abort(),s);try{const i=await fetch(`${e}/runtime/binding/refresh-token`,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify({agentId:this.agentId,currentToken:t,forceRefresh:!0}),signal:n.signal});if(!i.ok)return r.warn(`[WebSocket] bridge refresh-token \u8FD4\u56DE ${i.status}\uFF0C\u65E0\u6CD5\u5237\u65B0`),"failed";const h=(await i.json()).runtimeAccessToken;return!h||h===t?"unchanged":(this.config.runtimeAccessToken=h,this.lastRefreshedToken=t,r.info("[WebSocket] runtimeAccessToken \u5DF2\u901A\u8FC7 bridge refresh \u66F4\u65B0"),"refreshed")}catch(i){return i instanceof Error&&i.name==="AbortError"?r.warn(`[WebSocket] bridge refresh-token \u8D85\u65F6\uFF08${s}ms\uFF09`):(r.warn("[WebSocket] bridge refresh-token \u5F02\u5E38: %s",i instanceof Error?i.message:String(i)),i instanceof Error&&i.stack&&r.warn("[WebSocket] refresh-token \u5F02\u5E38\u5806\u6808:",i.stack.slice(0,2e3))),"failed"}finally{clearTimeout(o)}}handleWebSocketError(e,t,s){const n=this.ws?.readyState;r.error("[WebSocket] \u8FDE\u63A5\u9519\u8BEF: readyState=%s, opened=%s, reconnectAttempts=%s, message=%s",n,t,this.reconnectAttempts,e.message),r.error("[WebSocket] \u9519\u8BEF\u5806\u6808:",e.stack?.slice(0,2e3)),this.emit("error",e),t||s(e)}disconnect(){this.isManualClose=!0,this.stopHeartbeat(),this.clearHeartbeatTimeout(),this.clearReconnectTimer(),this.ws&&(this.ws.readyState===c.OPEN&&this.ws.close(1e3,"Agent logout"),this.ws=null),this.agentId=null,r.info("[WebSocket] \u5DF2\u65AD\u5F00\u8FDE\u63A5")}send(e){const t=this.ws?.readyState;if(!this.ws||t!==c.OPEN)return r.warn("[WebSocket] \u8FDE\u63A5\u672A\u5EFA\u7ACB\uFF0C\u65E0\u6CD5\u53D1\u9001\u6D88\u606F: readyState=%s, isManualClose=%s, agentId=%s",t,this.isManualClose,this.agentId),!1;try{return this.ws.send(JSON.stringify(e)),!0}catch(s){return r.error("[WebSocket] send() \u53D1\u9001\u6D88\u606F\u5F02\u5E38: readyState=%s, error=%s",t,s),!1}}sendMcpNodeLog(e){return this.isConnected()?(this.logBatchQueue.push(e),this.logBatchTimer||(this.logBatchTimer=setTimeout(()=>{this.logBatchTimer=null,this.flushLogBatch()},l)),!0):!1}flushLogBatch(){if(this.logBatchQueue.length===0||this.isProcessingLogBatch)return;const e=this.logBatchQueue.splice(0,k);if(e.length===0)return;this.isProcessingLogBatch=!0,this.logSequenceNumber+=1;const t=this.logSequenceNumber;new Promise((n,o)=>{this.pendingLogBatches.set(t,{logs:e,sequenceNumber:t,retryCount:0,resolve:n,reject:o})}).then(()=>{this.isProcessingLogBatch=!1,this.logBatchQueue.length>0&&(this.logBatchTimer=setTimeout(()=>{this.logBatchTimer=null,this.flushLogBatch()},l))}).catch(()=>{this.isProcessingLogBatch=!1}),this.sendLogBatch(t)}sendLogBatch(e){const t=this.pendingLogBatches.get(e);if(!t)return;this.send({type:"MCP_NODE_LOG_BATCH",data:{sequenceNumber:e,logs:t.logs},timestamp:new Date().toISOString()})||this.handleLogBatchSendFailure(e)}handleLogBatchSendFailure(e){const t=this.pendingLogBatches.get(e);t&&(t.retryCount+=1,t.retryCount<=u?(r.warn(`[WebSocket] \u65E5\u5FD7\u6279\u6B21 #${e} \u53D1\u9001\u5931\u8D25\uFF0C\u7B2C ${t.retryCount} \u6B21\u91CD\u8BD5`),setTimeout(()=>{this.sendLogBatch(e)},l*t.retryCount)):(r.error(`[WebSocket] \u65E5\u5FD7\u6279\u6B21 #${e} \u53D1\u9001\u5931\u8D25\uFF0C\u5DF2\u91CD\u8BD5 ${u} \u6B21\uFF0C\u4E22\u5F03`),this.pendingLogBatches.delete(e),t.reject(new Error(`\u65E5\u5FD7\u6279\u6B21 #${e} \u53D1\u9001\u5931\u8D25`))))}handleLogAck(e){const s=Number(e?.ackSequenceNumber);if(!s||!this.pendingLogBatches.has(s))return;const n=this.pendingLogBatches.get(s);n&&(this.pendingLogBatches.delete(s),n.resolve())}isConnected(){return this.ws!==null&&this.ws.readyState===c.OPEN}handleMessage(e){switch(e.type){case"GROUP_MSG":this.emit("groupMessage",e.data);break;case"DM_MSG":this.emit("directMessage",e.data);break;case"ONLINE_CHANGE":this.emit("onlineChange",e.data);break;case"HEARTBEAT":this.markHeartbeatReceived();break;case"PROFILE_INFO":r.info("[WebSocket] \u6536\u5230 PROFILE_INFO \u6D88\u606F"),this.emit("profileInfo",e.data);break;case"AGENT_STATUS":r.info("[WebSocket] \u6536\u5230 AGENT_STATUS \u6D88\u606F"),this.emit("agentStatus",e.data);break;case"LOG_ACK":this.handleLogAck(e.data);break;case"GROUP_READ_RECEIPT":case"DM_READ_RECEIPT":case"TERMINAL_OUTPUT":case"TERMINAL_STATUS":case"MCP_NODE_LOG":case"AGENT_EVENTS_DELTA":break;default:r.warn("[WebSocket] \u672A\u77E5\u6D88\u606F\u7C7B\u578B:",e.type)}}startHeartbeat(){this.stopHeartbeat(),this.heartbeatTimer=setInterval(()=>{this.isConnected()&&(this.ensureHeartbeatTimeout(),this.sendProtocolPing(),this.send({type:"HEARTBEAT",data:"ping",timestamp:new Date().toISOString()}))},this.config.heartbeatInterval)}stopHeartbeat(){this.heartbeatTimer&&(clearInterval(this.heartbeatTimer),this.heartbeatTimer=null),this.clearHeartbeatTimeout()}sendProtocolPing(){if(!(!this.ws||this.ws.readyState!==c.OPEN))try{this.ws.ping()}catch(e){r.warn("[WebSocket] \u534F\u8BAE\u5C42 ping \u53D1\u9001\u5931\u8D25\uFF0C\u7EC8\u6B62\u8FDE\u63A5\u5E76\u91CD\u8FDE:",e instanceof Error?e.message:String(e)),this.ws.terminate()}}markHeartbeatReceived(){this.clearHeartbeatTimeout()}ensureHeartbeatTimeout(){this.heartbeatTimeoutTimer||(this.heartbeatTimeoutTimer=setTimeout(()=>{this.heartbeatTimeoutTimer=null,!(!this.ws||this.ws.readyState!==c.OPEN)&&(r.warn(`[WebSocket] ${this.resolveHeartbeatTimeout()}ms \u672A\u6536\u5230\u5FC3\u8DF3\u54CD\u5E94\uFF0C\u7EC8\u6B62\u8FDE\u63A5\u5E76\u89E6\u53D1\u91CD\u8FDE`),this.ws.terminate())},this.resolveHeartbeatTimeout()))}clearHeartbeatTimeout(){this.heartbeatTimeoutTimer&&(clearTimeout(this.heartbeatTimeoutTimer),this.heartbeatTimeoutTimer=null)}resolveHeartbeatTimeout(){return this.config.heartbeatTimeout??f}scheduleReconnect(){const e=this.config.maxReconnectAttempts!==-1;if(e&&this.reconnectAttempts>=this.config.maxReconnectAttempts){r.error("[WebSocket] \u91CD\u8FDE\u6B21\u6570\u5DF2\u8FBE\u4E0A\u9650\uFF0C\u653E\u5F03\u91CD\u8FDE"),this.emit("reconnectFailed");return}if(this.reconnectTimer)return;this.reconnectAttempts++;const t=Math.min(this.config.reconnectInterval*Math.pow(2,this.reconnectAttempts-1),this.config.reconnectMaxInterval??T),s=Math.random()*(this.config.reconnectJitterMs??b),n=Math.round(t+s),o=e?`\u7B2C ${this.reconnectAttempts}/${this.config.maxReconnectAttempts} \u6B21`:`\u7B2C ${this.reconnectAttempts} \u6B21\uFF08\u65E0\u9650\u91CD\u8BD5\u6A21\u5F0F\uFF09`;r.warn(`[WebSocket] ${n}ms \u540E\u5C1D\u8BD5${o}\u91CD\u8FDE (base=${Math.round(t)}ms, jitter=+${Math.round(s)}ms)...`),this.reconnectTimer=setTimeout(()=>{this.reconnectTimer=null,this.agentId&&this.connect(this.agentId).catch(i=>{const a=i instanceof Error?i.message:String(i),h=i instanceof Error?i.stack:void 0;r.warn("[WebSocket] \u91CD\u8FDE\u5931\u8D25 (attempt #%s), \u7EE7\u7EED\u91CD\u8BD5: %s",this.reconnectAttempts,a),h&&r.warn("[WebSocket] \u91CD\u8FDE\u5931\u8D25\u5806\u6808:",h.slice(0,2e3)),this.scheduleReconnect()})},n)}clearReconnectTimer(){this.reconnectTimer&&(clearTimeout(this.reconnectTimer),this.reconnectTimer=null)}}export{w as WebSocketClient};
+