@feng3d/ctc 0.0.6

package/dist/proxy-manager.js

@@ -0,0 +1,124 @@
+/**
+ * @module proxy-manager
+ *
+ * 代理管理器模块。
+ *
+ * 负责管理所有代理隧道的生命周期，包括向服务器注册/注销代理、
+ * 创建统一的处理器（同时支持 HTTP 和 WebSocket）、
+ * 以及在连接断开时清理所有处理器。
+ */
+import { logger } from '@feng3d/chuantou-shared';
+import { UnifiedHandler } from './handlers/unified-handler.js';
+import { MessageType, createMessage } from '@feng3d/chuantou-shared';
+/**
+ * 代理管理器类，负责管理所有代理隧道的注册、注销和生命周期。
+ *
+ * 创建统一的处理器实例，每个端口同时支持 HTTP 和 WebSocket 协议。
+ * 当控制器连接断开时，自动清理所有处理器。
+ *
+ * @example
+ * ```typescript
+ * const proxyManager = new ProxyManager(controller);
+ * await proxyManager.registerProxy({ remotePort: 8080, localPort: 3000 });
+ * ```
+ */
+export class ProxyManager {
+    /**
+     * 创建代理管理器实例。
+     *
+     * @param controller - 控制器实例，用于与服务器进行通信
+     */
+    constructor(controller) {
+        this.controller = controller;
+        this.handlers = new Map();
+        // 监听控制器事件
+        this.controller.on('disconnected', () => {
+            this.onDisconnected();
+        });
+    }
+    /**
+     * 向服务器注册一个代理隧道。
+     *
+     * 发送注册消息到服务器，注册成功后创建统一的处理器
+     * （{@link UnifiedHandler}）并存储到处理器映射表中。
+     * 每个端口同时支持 HTTP 和 WebSocket 协议。
+     *
+     * @param config - 代理配置对象，包含远程端口、本地端口等信息
+     * @throws {Error} 注册失败时抛出错误，包含服务器返回的错误信息
+     */
+    async registerProxy(config) {
+        logger.log(`正在注册代理: :${config.remotePort} -> ${config.localHost || 'localhost'}:${config.localPort}`);
+        // 发送注册消息
+        const registerMsg = createMessage(MessageType.REGISTER, {
+            remotePort: config.remotePort,
+            localPort: config.localPort,
+            localHost: config.localHost,
+        });
+        const response = await this.controller.sendRequest(registerMsg);
+        if (!response.payload.success) {
+            throw new Error(`注册代理失败: ${response.payload.error}`);
+        }
+        logger.log(`代理已注册: ${response.payload.remoteUrl}`);
+        // 创建统一的处理器（同时支持 HTTP 和 WebSocket）
+        const handler = new UnifiedHandler(this.controller, config);
+        // 设置处理器事件
+        handler.on('error', (error) => {
+            logger.error(`端口 ${config.remotePort} 的处理器错误:`, error);
+        });
+        this.handlers.set(config.remotePort, handler);
+    }
+    /**
+     * 注销指定远程端口的代理隧道。
+     *
+     * 销毁对应的处理器，并向服务器发送注销请求。
+     *
+     * @param remotePort - 要注销的代理对应的远程端口号
+     */
+    async unregisterProxy(remotePort) {
+        const handler = this.handlers.get(remotePort);
+        if (handler) {
+            handler.destroy();
+            this.handlers.delete(remotePort);
+        }
+        const unregisterMsg = createMessage(MessageType.UNREGISTER, {
+            remotePort,
+        });
+        await this.controller.sendRequest(unregisterMsg);
+        logger.log(`代理已注销: 端口 ${remotePort}`);
+    }
+    /**
+     * 注销所有已注册的代理隧道。
+     *
+     * 并行注销所有代理，等待全部注销完成后返回。
+     *
+     * @returns 所有代理注销完成后解析的 Promise
+     */
+    async unregisterAll() {
+        const unregisterPromises = [];
+        for (const port of this.handlers.keys()) {
+            unregisterPromises.push(this.unregisterProxy(port));
+        }
+        await Promise.all(unregisterPromises);
+    }
+    /**
+     * 处理控制器连接断开事件。
+     *
+     * 销毁所有处理器并清空处理器映射表。
+     */
+    onDisconnected() {
+        logger.log('连接丢失，正在停止所有处理器...');
+        for (const handler of this.handlers.values()) {
+            handler.destroy();
+        }
+        this.handlers.clear();
+    }
+    /**
+     * 销毁代理管理器，注销所有代理并释放资源。
+     *
+     * @returns 销毁完成后解析的 Promise
+     */
+    async destroy() {
+        await this.unregisterAll();
+    }
+}
+//# sourceMappingURL=proxy-manager.js.map

package/dist/proxy-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"proxy-manager.js","sourceRoot":"","sources":["../src/proxy-manager.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAGH,OAAO,EAAe,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAC9D,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAE,WAAW,EAAE,aAAa,EAA2D,MAAM,yBAAyB,CAAC;AAE9H;;;;;;;;;;;GAWG;AACH,MAAM,OAAO,YAAY;IAOvB;;;;OAIG;IACH,YAAY,UAAsB;QAChC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,QAAQ,GAAG,IAAI,GAAG,EAAE,CAAC;QAE1B,UAAU;QACV,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,cAAc,EAAE,GAAG,EAAE;YACtC,IAAI,CAAC,cAAc,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;;OASG;IACH,KAAK,CAAC,aAAa,CAAC,MAAmB;QACrC,MAAM,CAAC,GAAG,CAAC,YAAY,MAAM,CAAC,UAAU,OAAO,MAAM,CAAC,SAAS,IAAI,WAAW,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC;QAEtG,SAAS;QACT,MAAM,WAAW,GAAoB,aAAa,CAAC,WAAW,CAAC,QAAQ,EAAE;YACvE,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,SAAS,EAAE,MAAM,CAAC,SAAS;SAC5B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,CAAsB,WAAW,CAAC,CAAC;QAErF,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,WAAW,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;QACvD,CAAC;QAED,MAAM,CAAC,GAAG,CAAC,UAAU,QAAQ,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;QAEnD,kCAAkC;QAClC,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAE5D,UAAU;QACV,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YAC5B,MAAM,CAAC,KAAK,CAAC,MAAM,MAAM,CAAC,UAAU,UAAU,EAAE,KAAK,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;IAChD,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,eAAe,CAAC,UAAkB;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC9C,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACnC,CAAC;QAED,MAAM,aAAa,GAAsB,aAAa,CAAC,WAAW,CAAC,UAAU,EAAE;YAC7E,UAAU;SACX,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;QACjD,MAAM,CAAC,GAAG,CAAC,aAAa,UAAU,EAAE,CAAC,CAAC;IACxC,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,aAAa;QACjB,MAAM,kBAAkB,GAAoB,EAAE,CAAC;QAC/C,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC;YACxC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC;QACtD,CAAC;QACD,MAAM,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;IACxC,CAAC;IAED;;;;OAIG;IACK,cAAc;QACpB,MAAM,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QAChC,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7C,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,OAAO;QACX,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;IAC7B,CAAC;CACF"}

package/package.json

@@ -0,0 +1,39 @@
+{
+  "name": "@feng3d/ctc",
+  "version": "0.0.6",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "bin": {
+    "feng3d-ctc": "./dist/cli.js"
+  },
+  "exports": {
+    ".": {
+      "import": "./dist/index.js",
+      "types": "./dist/index.d.ts"
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "watch": "tsc --watch",
+    "test": "vitest run",
+    "test:watch": "vitest watch"
+  },
+  "dependencies": {
+    "@feng3d/chuantou-shared": "^0.0.6",
+    "chalk": "^5.6.2",
+    "commander": "^14.0.3",
+    "uuid": "^9.0.1",
+    "ws": "^8.16.0"
+  },
+  "devDependencies": {
+    "@types/node": "^20.10.0",
+    "@types/uuid": "^9.0.7",
+    "@types/ws": "^8.5.10",
+    "typescript": "^5.3.3",
+    "vitest": "^2.1.9"
+  }
+}