@cydm/magic-shell-agent-node 0.1.0

package/dist/ws-client.js

@@ -0,0 +1,193 @@
+import WebSocket from "ws";
+/**
+ * WebSocket 客户端
+ * 负责与 Relay Server 保持连接，自动重连
+ */
+export class WebSocketClient {
+    options;
+    ws;
+    messageHandlers = [];
+    statusHandlers = [];
+    reconnectAttempts = 0;
+    reconnectTimer;
+    isConnected = false;
+    isConnecting = false;
+    manualDisconnect = false;
+    constructor(options) {
+        this.options = {
+            autoReconnect: true,
+            reconnectDelay: 1000,
+            maxReconnectDelay: 30000,
+            ...options,
+        };
+    }
+    /**
+     * 连接到 Relay Server
+     */
+    async connect() {
+        if (this.isConnected || this.isConnecting) {
+            return;
+        }
+        if (this.reconnectTimer) {
+            clearTimeout(this.reconnectTimer);
+            this.reconnectTimer = undefined;
+        }
+        this.manualDisconnect = false;
+        this.isConnecting = true;
+        console.log(`[WebSocketClient] Connecting to ${this.options.url}...`);
+        return new Promise((resolve, reject) => {
+            try {
+                const socket = new WebSocket(this.options.url);
+                this.ws = socket;
+                socket.on("open", () => {
+                    this.isConnected = true;
+                    this.isConnecting = false;
+                    this.reconnectAttempts = 0;
+                    console.log("[WebSocketClient] Connected");
+                    // 发送认证
+                    this.send({
+                        type: "auth",
+                        clientType: "node",
+                        nodeId: this.options.nodeId,
+                        password: this.options.password,
+                    });
+                    this.emitStatus("connected");
+                    resolve();
+                });
+                socket.on("message", (data) => {
+                    try {
+                        const message = JSON.parse(data.toString());
+                        this.emitMessage(message);
+                    }
+                    catch (err) {
+                        console.error("[WebSocketClient] Failed to parse message:", err);
+                    }
+                });
+                socket.on("close", (code, reason) => {
+                    console.log(`[WebSocketClient] Closed: code=${code}, reason=${reason.toString()}`);
+                    if (this.ws === socket) {
+                        this.ws = undefined;
+                    }
+                    this.handleDisconnect();
+                });
+                socket.on("error", (err) => {
+                    console.error("[WebSocketClient] Error:", err.message);
+                    this.isConnecting = false;
+                    this.emitStatus("error", err);
+                    if (!this.isConnected) {
+                        reject(err);
+                    }
+                });
+            }
+            catch (err) {
+                this.isConnecting = false;
+                reject(err);
+            }
+        });
+    }
+    /**
+     * 断开连接
+     */
+    disconnect() {
+        this.manualDisconnect = true;
+        if (this.reconnectTimer) {
+            clearTimeout(this.reconnectTimer);
+            this.reconnectTimer = undefined;
+        }
+        if (this.ws) {
+            this.ws.close();
+            this.ws = undefined;
+        }
+        this.handleDisconnect();
+    }
+    /**
+     * 发送消息
+     */
+    send(message) {
+        if (!this.isConnected || !this.ws) {
+            console.warn("[WebSocketClient] Cannot send, not connected");
+            return;
+        }
+        this.ws.send(JSON.stringify(message));
+    }
+    /**
+     * 注册消息处理器
+     */
+    onMessage(handler) {
+        this.messageHandlers.push(handler);
+    }
+    /**
+     * 注册状态处理器
+     */
+    onStatus(handler) {
+        this.statusHandlers.push(handler);
+    }
+    /**
+     * 获取连接状态
+     */
+    getStatus() {
+        if (this.isConnected)
+            return "connected";
+        if (this.isConnecting)
+            return "connecting";
+        return "disconnected";
+    }
+    /**
+     * 处理断开连接
+     */
+    handleDisconnect() {
+        const wasConnected = this.isConnected;
+        this.isConnected = false;
+        this.isConnecting = false;
+        if (wasConnected) {
+            this.emitStatus("disconnected");
+        }
+        // 自动重连
+        if (!this.manualDisconnect && this.options.autoReconnect && !this.reconnectTimer) {
+            this.scheduleReconnect();
+        }
+    }
+    /**
+     * 安排重连
+     */
+    scheduleReconnect() {
+        const delay = Math.min(this.options.reconnectDelay * Math.pow(2, this.reconnectAttempts), this.options.maxReconnectDelay);
+        this.reconnectAttempts++;
+        console.log(`[WebSocketClient] Reconnecting in ${delay}ms (attempt ${this.reconnectAttempts})...`);
+        this.reconnectTimer = setTimeout(() => {
+            this.reconnectTimer = undefined;
+            this.connect().catch(() => {
+                // 重连失败，继续尝试
+                if (this.options.autoReconnect) {
+                    this.scheduleReconnect();
+                }
+            });
+        }, delay);
+    }
+    /**
+     * 触发消息事件
+     */
+    emitMessage(message) {
+        for (const handler of this.messageHandlers) {
+            try {
+                handler(message);
+            }
+            catch (err) {
+                console.error("[WebSocketClient] Message handler error:", err);
+            }
+        }
+    }
+    /**
+     * 触发状态事件
+     */
+    emitStatus(status, error) {
+        for (const handler of this.statusHandlers) {
+            try {
+                handler(status, error);
+            }
+            catch (err) {
+                console.error("[WebSocketClient] Status handler error:", err);
+            }
+        }
+    }
+}

package/package.json

@@ -0,0 +1,38 @@
+{
+  "name": "@cydm/magic-shell-agent-node",
+  "version": "0.1.0",
+  "description": "Magic Shell Agent Node - Local agent connector",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "bin": {
+    "agent-node": "./dist/node.js"
+  },
+  "files": [
+    "dist"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "tsc && node ./scripts/copy-default-plugins.mjs",
+    "dev": "tsc --watch",
+    "start": "node dist/node.js"
+  },
+  "dependencies": {
+    "@cydm/magic-shell-protocol": "0.1.0",
+    "node-pty": "^1.1.0",
+    "ws": "^8.18.0"
+  },
+  "devDependencies": {
+    "@types/node": "^20.10.0",
+    "@types/ws": "^8.5.13",
+    "typescript": "^5.3.0"
+  },
+  "exports": {
+    ".": {
+      "import": "./dist/index.js",
+      "types": "./dist/index.d.ts"
+    }
+  }
+}