@feng3d/ctc 0.0.6

package/dist/controller.js

@@ -0,0 +1,340 @@
+/**
+ * @module controller
+ *
+ * 客户端控制器模块。
+ *
+ * 管理客户端与穿透服务器之间的 WebSocket 控制连接，
+ * 负责身份认证、心跳保活、消息收发、断线重连等核心通信逻辑。
+ */
+import { WebSocket } from 'ws';
+import { EventEmitter } from 'events';
+import { MessageType, createMessage, logger, } from '@feng3d/chuantou-shared';
+/**
+ * 客户端控制器类，管理与穿透服务器的 WebSocket 控制连接。
+ *
+ * 继承自 {@link EventEmitter}，提供以下事件：
+ * - `connected` - 成功连接到服务器时触发
+ * - `disconnected` - 与服务器断开连接时触发
+ * - `authenticated` - 身份认证成功时触发
+ * - `maxReconnectAttemptsReached` - 达到最大重连次数时触发
+ * - `newConnection` - 收到新的代理连接请求时触发
+ * - `connectionClose` - 收到连接关闭通知时触发
+ * - `connectionError` - 收到连接错误通知时触发
+ *
+ * @example
+ * ```typescript
+ * const controller = new Controller(config);
+ * controller.on('authenticated', () => logger.log('已认证'));
+ * await controller.connect();
+ * ```
+ */
+export class Controller extends EventEmitter {
+    /**
+     * 创建控制器实例。
+     *
+     * @param config - 客户端配置对象
+     */
+    constructor(config) {
+        super();
+        /** 与服务器的 WebSocket 连接实例 */
+        this.ws = null;
+        /** 是否已连接到服务器 */
+        this.connected = false;
+        /** 是否已通过身份认证 */
+        this.authenticated = false;
+        /** 重连定时器 */
+        this.reconnectTimer = null;
+        /** 心跳定时器 */
+        this.heartbeatTimer = null;
+        /** 当前重连尝试次数 */
+        this.reconnectAttempts = 0;
+        /** 待响应的请求映射表，键为消息 ID */
+        this.pendingRequests = new Map();
+        this.config = config;
+    }
+    /**
+     * 连接到穿透服务器。
+     *
+     * 建立 WebSocket 连接后会自动进行身份认证和启动心跳。
+     * 连接断开时会自动安排重连。
+     *
+     * @returns 连接并认证成功后解析的 Promise
+     * @throws {Error} 连接失败或认证失败时抛出错误
+     */
+    async connect() {
+        return new Promise((resolve, reject) => {
+            logger.log(`正在连接 ${this.config.serverUrl}...`);
+            this.ws = new WebSocket(this.config.serverUrl);
+            this.ws.on('open', async () => {
+                logger.log('已连接到服务器');
+                this.connected = true;
+                this.reconnectAttempts = 0;
+                this.emit('connected');
+                // 认证
+                try {
+                    await this.authenticate();
+                    this.startHeartbeat();
+                    resolve();
+                }
+                catch (error) {
+                    reject(error);
+                }
+            });
+            this.ws.on('message', (data) => {
+                this.handleMessage(data);
+            });
+            this.ws.on('close', () => {
+                logger.log('连接已关闭');
+                this.connected = false;
+                this.authenticated = false;
+                this.stopHeartbeat();
+                this.emit('disconnected');
+                this.scheduleReconnect();
+            });
+            this.ws.on('error', (error) => {
+                logger.error('WebSocket 错误:', error.message);
+                if (!this.connected) {
+                    reject(error);
+                }
+            });
+        });
+    }
+    /**
+     * 向服务器发送身份认证请求。
+     *
+     * 使用配置中的 token 进行认证，认证成功后触发 `authenticated` 事件。
+     *
+     * @throws {Error} 认证失败时抛出错误，包含服务器返回的错误信息
+     */
+    async authenticate() {
+        logger.log('正在认证...');
+        const authMsg = createMessage(MessageType.AUTH, {
+            token: this.config.token,
+        });
+        const response = await this.sendRequest(authMsg);
+        if (!response.payload.success) {
+            throw new Error(`认证失败: ${response.payload.error}`);
+        }
+        this.authenticated = true;
+        logger.log('认证成功');
+        this.emit('authenticated');
+    }
+    /**
+     * 启动心跳定时器。
+     *
+     * 每 30 秒向服务器发送一次心跳消息，仅在已连接且已认证状态下发送。
+     */
+    startHeartbeat() {
+        this.heartbeatTimer = setInterval(() => {
+            if (this.connected && this.authenticated) {
+                const heartbeatMsg = createMessage(MessageType.HEARTBEAT, {
+                    timestamp: Date.now(),
+                });
+                this.sendMessage(heartbeatMsg);
+            }
+        }, 30000); // 30秒
+    }
+    /**
+     * 停止心跳定时器。
+     *
+     * 清除心跳定时器并将其设置为 null。
+     */
+    stopHeartbeat() {
+        if (this.heartbeatTimer) {
+            clearInterval(this.heartbeatTimer);
+            this.heartbeatTimer = null;
+        }
+    }
+    /**
+     * 安排断线重连。
+     *
+     * 使用指数退避策略计算重连延迟时间。如果已达到最大重连次数，
+     * 则触发 `maxReconnectAttemptsReached` 事件并停止重连。
+     */
+    scheduleReconnect() {
+        if (this.reconnectTimer) {
+            return; // 已经安排了重连
+        }
+        if (this.reconnectAttempts >= this.config.maxReconnectAttempts) {
+            logger.error('已达到最大重连次数');
+            this.emit('maxReconnectAttemptsReached');
+            return;
+        }
+        const delay = this.calculateReconnectDelay();
+        logger.log(`将在 ${delay}ms 后重连... (第 ${this.reconnectAttempts + 1}/${this.config.maxReconnectAttempts} 次尝试)`);
+        this.reconnectTimer = setTimeout(() => {
+            this.reconnectTimer = null;
+            this.reconnectAttempts++;
+            this.connect().catch((error) => {
+                const errorMessage = error instanceof Error ? error.message : String(error);
+                logger.error('重连失败:', errorMessage);
+                this.scheduleReconnect();
+            });
+        }, delay);
+    }
+    /**
+     * 计算重连延迟时间（指数退避算法）。
+     *
+     * 基于重连间隔和当前重连次数计算延迟，最大不超过 60 秒，
+     * 并添加 0~1 秒的随机抖动以避免多客户端同时重连造成服务器压力。
+     *
+     * @returns 重连延迟时间（毫秒）
+     */
+    calculateReconnectDelay() {
+        const baseDelay = this.config.reconnectInterval;
+        const maxDelay = 60000; // 最大60秒
+        const delay = Math.min(baseDelay * Math.pow(2, this.reconnectAttempts), maxDelay);
+        // 添加随机抖动
+        return delay + Math.random() * 1000;
+    }
+    /**
+     * 处理从服务器接收到的消息。
+     *
+     * 根据消息类型进行分发：
+     * - 响应类消息（AUTH_RESP、REGISTER_RESP、HEARTBEAT_RESP）交由 {@link handleResponse} 处理
+     * - 新连接、连接关闭、连接错误等消息通过事件触发通知上层
+     *
+     * @param data - 接收到的原始消息数据
+     */
+    handleMessage(data) {
+        try {
+            const message = JSON.parse(data.toString());
+            const msgType = message.type;
+            switch (msgType) {
+                case MessageType.AUTH_RESP:
+                case MessageType.REGISTER_RESP:
+                case MessageType.HEARTBEAT_RESP:
+                    // 响应消息，由pendingRequests处理
+                    this.handleResponse(message);
+                    break;
+                case MessageType.NEW_CONNECTION:
+                    this.emit('newConnection', message);
+                    break;
+                case MessageType.CONNECTION_CLOSE:
+                    this.emit('connectionClose', message);
+                    break;
+                case MessageType.CONNECTION_ERROR:
+                    this.emit('connectionError', message);
+                    break;
+                default:
+                    logger.warn(`未知消息类型: ${msgType}`);
+            }
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            logger.error('处理消息时出错:', errorMessage);
+        }
+    }
+    /**
+     * 处理响应类消息。
+     *
+     * 根据消息 ID 查找对应的待处理请求，清除超时定时器并解析 Promise。
+     *
+     * @param message - 服务器返回的响应消息对象
+     */
+    handleResponse(message) {
+        const pending = this.pendingRequests.get(message.id);
+        if (pending) {
+            clearTimeout(pending.timeout);
+            this.pendingRequests.delete(message.id);
+            pending.resolve(message);
+        }
+    }
+    /**
+     * 发送请求消息并等待服务器响应。
+     *
+     * 将消息发送到服务器，并在指定超时时间内等待对应的响应消息。
+     * 如果超时未收到响应，Promise 将被拒绝。
+     *
+     * @typeParam T - 期望的响应消息类型
+     * @param message - 要发送的请求消息对象，必须包含 `id` 字段
+     * @param timeout - 请求超时时间（毫秒），默认为 30000
+     * @returns 服务器响应消息的 Promise
+     * @throws {Error} 请求超时时抛出 "Request timeout" 错误
+     */
+    sendRequest(message, timeout = 30000) {
+        return new Promise((resolve, reject) => {
+            const timer = setTimeout(() => {
+                this.pendingRequests.delete(message.id);
+                reject(new Error('请求超时'));
+            }, timeout);
+            this.pendingRequests.set(message.id, { resolve, reject, timeout: timer });
+            this.sendMessage(message);
+        });
+    }
+    /**
+     * 向服务器发送消息。
+     *
+     * 将消息对象序列化为 JSON 并通过 WebSocket 发送。
+     * 仅在 WebSocket 连接处于 OPEN 状态时才能发送。
+     *
+     * @param message - 要发送的消息对象
+     * @returns 发送成功返回 `true`，未连接时返回 `false`
+     */
+    sendMessage(message) {
+        if (this.ws && this.ws.readyState === WebSocket.OPEN) {
+            this.ws.send(JSON.stringify(message));
+            return true;
+        }
+        logger.error('无法发送消息: 未连接');
+        return false;
+    }
+    /**
+     * 检查是否已连接到服务器。
+     *
+     * @returns 已连接返回 `true`，否则返回 `false`
+     */
+    isConnected() {
+        return this.connected;
+    }
+    /**
+     * 检查是否已通过身份认证。
+     *
+     * @returns 已认证返回 `true`，否则返回 `false`
+     */
+    isAuthenticated() {
+        return this.authenticated;
+    }
+    /**
+     * 获取当前重连尝试次数。
+     *
+     * @returns 当前重连次数
+     */
+    getReconnectAttempts() {
+        return this.reconnectAttempts;
+    }
+    /**
+     * 断开与服务器的连接。
+     *
+     * 清除重连定时器、停止心跳、关闭 WebSocket 连接，
+     * 并重置连接和认证状态。
+     */
+    disconnect() {
+        if (this.reconnectTimer) {
+            clearTimeout(this.reconnectTimer);
+            this.reconnectTimer = null;
+        }
+        this.stopHeartbeat();
+        if (this.ws) {
+            this.ws.close();
+            this.ws = null;
+        }
+        this.connected = false;
+        this.authenticated = false;
+    }
+    /**
+     * 销毁控制器实例，释放所有资源。
+     *
+     * 断开连接、清除所有待处理请求的超时定时器、
+     * 清空待处理请求映射表并移除所有事件监听器。
+     */
+    destroy() {
+        this.disconnect();
+        for (const { timeout } of this.pendingRequests.values()) {
+            clearTimeout(timeout);
+        }
+        this.pendingRequests.clear();
+        this.removeAllListeners();
+    }
+}
+//# sourceMappingURL=controller.js.map

package/dist/controller.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"controller.js","sourceRoot":"","sources":["../src/controller.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC/B,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EACL,WAAW,EAEX,aAAa,EAQb,MAAM,GACP,MAAM,yBAAyB,CAAC;AAGjC;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAM,OAAO,UAAW,SAAQ,YAAY;IA6B1C;;;;OAIG;IACH,YAAY,MAAc;QACxB,KAAK,EAAE,CAAC;QA/BV,2BAA2B;QACnB,OAAE,GAAqB,IAAI,CAAC;QAEpC,gBAAgB;QACR,cAAS,GAAY,KAAK,CAAC;QAEnC,gBAAgB;QACR,kBAAa,GAAY,KAAK,CAAC;QAEvC,YAAY;QACJ,mBAAc,GAA0B,IAAI,CAAC;QAErD,YAAY;QACJ,mBAAc,GAA0B,IAAI,CAAC;QAErD,eAAe;QACP,sBAAiB,GAAW,CAAC,CAAC;QAEtC,wBAAwB;QAChB,oBAAe,GAIlB,IAAI,GAAG,EAAE,CAAC;QASb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,OAAO;QACX,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC3C,MAAM,CAAC,GAAG,CAAC,QAAQ,IAAI,CAAC,MAAM,CAAC,SAAS,KAAK,CAAC,CAAC;YAE/C,IAAI,CAAC,EAAE,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAE/C,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE;gBAC5B,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBACtB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;gBACtB,IAAI,CAAC,iBAAiB,GAAG,CAAC,CAAC;gBAC3B,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;gBAEvB,KAAK;gBACL,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;oBAC1B,IAAI,CAAC,cAAc,EAAE,CAAC;oBACtB,OAAO,EAAE,CAAC;gBACZ,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,CAAC,KAAK,CAAC,CAAC;gBAChB,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAY,EAAE,EAAE;gBACrC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;YAC3B,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBACvB,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACpB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;gBACvB,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;gBAC3B,IAAI,CAAC,aAAa,EAAE,CAAC;gBACrB,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;gBAC1B,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBAC5B,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;gBAC7C,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;oBACpB,MAAM,CAAC,KAAK,CAAC,CAAC;gBAChB,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;OAMG;IACK,KAAK,CAAC,YAAY;QACxB,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAEtB,MAAM,OAAO,GAAgB,aAAa,CAAC,WAAW,CAAC,IAAI,EAAE;YAC3D,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;SACzB,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAkB,OAAO,CAAC,CAAC;QAClE,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,SAAS,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAC1B,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACnB,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAC7B,CAAC;IAED;;;;OAIG;IACK,cAAc;QACpB,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;YACrC,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;gBACzC,MAAM,YAAY,GAAqB,aAAa,CAAC,WAAW,CAAC,SAAS,EAAE;oBAC1E,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;iBACtB,CAAC,CAAC;gBACH,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;YACjC,CAAC;QACH,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,MAAM;IACnB,CAAC;IAED;;;;OAIG;IACK,aAAa;QACnB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACK,iBAAiB;QACvB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,OAAO,CAAC,UAAU;QACpB,CAAC;QAED,IAAI,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;YAC/D,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAC1B,IAAI,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YACzC,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAC7C,MAAM,CAAC,GAAG,CAAC,MAAM,KAAK,gBAAgB,IAAI,CAAC,iBAAiB,GAAG,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,OAAO,CAAC,CAAC;QAE7G,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,GAAG,EAAE;YACpC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAC3B,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACzB,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC7B,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC5E,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;gBACpC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,KAAK,CAAC,CAAC;IACZ,CAAC;IAED;;;;;;;OAOG;IACK,uBAAuB;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC;QAChD,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,QAAQ;QAChC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,iBAAiB,CAAC,EAAE,QAAQ,CAAC,CAAC;QAClF,SAAS;QACT,OAAO,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC;IACtC,CAAC;IAED;;;;;;;;OAQG;IACK,aAAa,CAAC,IAAY;QAChC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;YAC5C,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;YAE7B,QAAQ,OAAO,EAAE,CAAC;gBAChB,KAAK,WAAW,CAAC,SAAS,CAAC;gBAC3B,KAAK,WAAW,CAAC,aAAa,CAAC;gBAC/B,KAAK,WAAW,CAAC,cAAc;oBAC7B,0BAA0B;oBAC1B,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;oBAC7B,MAAM;gBAER,KAAK,WAAW,CAAC,cAAc;oBAC7B,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,OAA+B,CAAC,CAAC;oBAC5D,MAAM;gBAER,KAAK,WAAW,CAAC,gBAAgB;oBAC/B,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,OAAiC,CAAC,CAAC;oBAChE,MAAM;gBAER,KAAK,WAAW,CAAC,gBAAgB;oBAC/B,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,OAAiC,CAAC,CAAC;oBAChE,MAAM;gBAER;oBACE,MAAM,CAAC,IAAI,CAAC,WAAW,OAAO,EAAE,CAAC,CAAC;YACtC,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACK,cAAc,CAAC,OAAY;QACjC,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACrD,IAAI,OAAO,EAAE,CAAC;YACZ,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC9B,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACxC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IAED;;;;;;;;;;;OAWG;IACH,WAAW,CAAI,OAAY,EAAE,UAAkB,KAAK;QAClD,OAAO,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACxC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC5B,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACxC,MAAM,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;YAC5B,CAAC,EAAE,OAAO,CAAC,CAAC;YAEZ,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;YAC1E,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;OAQG;IACH,WAAW,CAAC,OAAY;QACtB,IAAI,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YACrD,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;YACtC,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;QAC5B,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;;;OAIG;IACH,WAAW;QACT,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED;;;;OAIG;IACH,eAAe;QACb,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED;;;;OAIG;IACH,oBAAoB;QAClB,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAED;;;;;OAKG;IACH,UAAU;QACR,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAClC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;QACD,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACZ,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;QACD,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;IAC7B,CAAC;IAED;;;;;OAKG;IACH,OAAO;QACL,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,KAAK,MAAM,EAAE,OAAO,EAAE,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,EAAE,CAAC;YACxD,YAAY,CAAC,OAAO,CAAC,CAAC;QACxB,CAAC;QACD,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAC7B,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC5B,CAAC;CACF"}

package/dist/handlers/unified-handler.d.ts

@@ -0,0 +1,110 @@
+/**
+ * @module handlers/unified-handler
+ *
+ * 统一代理连接处理器模块。
+ *
+ * 负责处理通过穿透隧道传输的 HTTP 和 WebSocket 连接。
+ * 同时支持 HTTP 请求转发和 WebSocket 连接桥接，无需区分协议类型。
+ */
+import { EventEmitter } from 'events';
+import { Controller } from '../controller.js';
+import { ProxyConfig } from '@feng3d/chuantou-shared';
+/**
+ * 统一代理连接处理器类。
+ *
+ * 继承自 {@link EventEmitter}，负责在远程客户端和本地服务之间建立双向数据转发通道。
+ * 同时支持 HTTP 请求转发和 WebSocket 连接桥接。
+ *
+ * 工作流程：
+ * 1. 监听控制器的 `newConnection` 事件，接收新的连接请求
+ * 2. 根据协议类型（http/websocket）建立相应的本地连接
+ * 3. 将本地服务发送的数据转发给服务器
+ * 4. 将服务器转发的远程客户端数据转发到本地服务
+ * 5. 处理连接关闭和错误情况
+ *
+ * 触发的事件：
+ * - `error` - 处理连接时发生错误
+ */
+export declare class UnifiedHandler extends EventEmitter {
+    /** 控制器实例，用于与服务器通信 */
+    private controller;
+    /** 代理配置，包含本地服务地址和端口信息 */
+    private config;
+    /** 正在处理的 HTTP 连接映射表 */
+    private pendingConnections;
+    /** 本地 WebSocket 连接映射表 */
+    private localWsConnections;
+    /** 流式 HTTP 响应映射表（如 SSE） */
+    private streamingResponses;
+    /**
+     * 创建统一处理器实例。
+     *
+     * 自动监听控制器的 `newConnection` 和 `connectionClose` 事件，
+     * 并设置数据转发监听器。
+     *
+     * @param controller - 控制器实例，用于与服务器通信
+     * @param config - 代理配置对象，包含本地服务地址和端口
+     */
+    constructor(controller: Controller, config: ProxyConfig);
+    /**
+     * 设置从服务器接收数据的监听器。
+     */
+    private setupDataListener;
+    /**
+     * 处理来自服务器的新 HTTP 连接请求。
+     */
+    private handleHttpConnection;
+    /**
+     * 处理来自服务器的新 WebSocket 连接请求。
+     */
+    private handleWebSocketConnection;
+    /**
+     * 将本地数据转发到服务器。
+     */
+    private forwardToServer;
+    /**
+     * 处理从服务器接收到的远程客户端数据。
+     */
+    private handleClientData;
+    /**
+     * 将 HTTP 响应数据发送回服务器。
+     */
+    private sendResponse;
+    /**
+     * 发送流式响应头（用于 SSE）。
+     */
+    private sendResponseHeaders;
+    /**
+     * 发送流式响应数据块（用于 SSE）。
+     */
+    private forwardStreamData;
+    /**
+     * 通知服务器流式响应结束（用于 SSE）。
+     */
+    private notifyStreamEnd;
+    /**
+     * 向服务器发送错误响应。
+     */
+    private sendError;
+    /**
+     * 通知服务器某个 WebSocket 连接已关闭。
+     */
+    private notifyServerClose;
+    /**
+     * 处理来自服务器的连接关闭通知。
+     */
+    private handleConnectionClose;
+    /**
+     * 清理 WebSocket 连接资源。
+     */
+    private cleanupWsConnection;
+    /**
+     * 过滤请求头，移除逐跳头部字段。
+     */
+    private filterHeaders;
+    /**
+     * 销毁处理器，释放所有资源。
+     */
+    destroy(): void;
+}
+//# sourceMappingURL=unified-handler.d.ts.map

package/dist/handlers/unified-handler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"unified-handler.d.ts","sourceRoot":"","sources":["../../src/handlers/unified-handler.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAKtC,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAuB,MAAM,yBAAyB,CAAC;AAS3E;;;;;;;;;;;;;;;GAeG;AACH,qBAAa,cAAe,SAAQ,YAAY;IAC9C,qBAAqB;IACrB,OAAO,CAAC,UAAU,CAAa;IAE/B,yBAAyB;IACzB,OAAO,CAAC,MAAM,CAAc;IAE5B,uBAAuB;IACvB,OAAO,CAAC,kBAAkB,CAGX;IAEf,yBAAyB;IACzB,OAAO,CAAC,kBAAkB,CAAqC;IAE/D,2BAA2B;IAC3B,OAAO,CAAC,kBAAkB,CAA+B;IAEzD;;;;;;;;OAQG;gBACS,UAAU,EAAE,UAAU,EAAE,MAAM,EAAE,WAAW;IAuBvD;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAgBzB;;OAEG;YACW,oBAAoB;IA4FlC;;OAEG;IACH,OAAO,CAAC,yBAAyB;IAoCjC;;OAEG;IACH,OAAO,CAAC,eAAe;IAWvB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAOxB;;OAEG;IACH,OAAO,CAAC,YAAY;IAYpB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAW3B;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAWzB;;OAEG;IACH,OAAO,CAAC,eAAe;IAWvB;;OAEG;IACH,OAAO,CAAC,SAAS;IASjB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAOzB;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAyB7B;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAQ3B;;OAEG;IACH,OAAO,CAAC,aAAa;IA0BrB;;OAEG;IACH,OAAO,IAAI,IAAI;CAqBhB"}