@yu_robotics/remote-cli-router 1.0.0

package/dist/websocket/ConnectionHub.js

@@ -0,0 +1,177 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ConnectionHub = void 0;
+/**
+ * WebSocket Connection Hub
+ * Manages all device WebSocket connections, responsible for message routing and connection management
+ */
+class ConnectionHub {
+    // Store deviceId -> WebSocket mapping
+    connections;
+    // Store deviceId -> last active time mapping
+    lastActiveMap;
+    constructor() {
+        this.connections = new Map();
+        this.lastActiveMap = new Map();
+    }
+    /**
+     * Register device connection
+     * @param deviceId Device unique identifier
+     * @param ws WebSocket connection
+     */
+    registerConnection(deviceId, ws) {
+        // If device already has a connection, close the old connection first
+        if (this.connections.has(deviceId)) {
+            const oldWs = this.connections.get(deviceId);
+            try {
+                oldWs.close();
+            }
+            catch (error) {
+                // Ignore close errors
+            }
+        }
+        // Register new connection
+        this.connections.set(deviceId, ws);
+        this.lastActiveMap.set(deviceId, Date.now());
+    }
+    /**
+     * Unregister device connection
+     * @param deviceId Device unique identifier
+     */
+    unregisterConnection(deviceId) {
+        this.connections.delete(deviceId);
+        this.lastActiveMap.delete(deviceId);
+    }
+    /**
+     * Send message to specified device
+     * @param deviceId Device unique identifier
+     * @param message Message object
+     * @returns Whether sending was successful
+     */
+    async sendToDevice(deviceId, message) {
+        const ws = this.connections.get(deviceId);
+        if (!ws) {
+            return false;
+        }
+        try {
+            const messageStr = JSON.stringify(message);
+            ws.send(messageStr);
+            // Update last active time
+            this.lastActiveMap.set(deviceId, Date.now());
+            return true;
+        }
+        catch (error) {
+            // Sending failed, connection may be disconnected
+            return false;
+        }
+    }
+    /**
+     * Check if device is online
+     * @param deviceId Device unique identifier
+     * @returns Whether it is online
+     */
+    isDeviceOnline(deviceId) {
+        return this.connections.has(deviceId);
+    }
+    /**
+     * Get device last active time
+     * @param deviceId Device unique identifier
+     * @returns Last active timestamp, returns undefined if device does not exist
+     */
+    getLastActiveTime(deviceId) {
+        return this.lastActiveMap.get(deviceId);
+    }
+    /**
+     * Get list of all online device IDs
+     * @returns Device ID array
+     */
+    getOnlineDevices() {
+        return Array.from(this.connections.keys());
+    }
+    /**
+     * Update device last active time
+     * @param deviceId Device unique identifier
+     */
+    updateLastActive(deviceId) {
+        if (this.connections.has(deviceId)) {
+            this.lastActiveMap.set(deviceId, Date.now());
+        }
+    }
+    /**
+     * Get connection statistics
+     * @returns Connection statistics
+     */
+    getConnectionStats() {
+        const deviceIds = this.getOnlineDevices();
+        return {
+            totalConnections: deviceIds.length,
+            deviceIds
+        };
+    }
+    /**
+     * Broadcast message to all devices
+     * @param message Message object
+     */
+    async broadcast(message) {
+        const messageStr = JSON.stringify(message);
+        const promises = [];
+        for (const [deviceId, ws] of this.connections.entries()) {
+            const promise = new Promise((resolve) => {
+                try {
+                    ws.send(messageStr);
+                    this.lastActiveMap.set(deviceId, Date.now());
+                }
+                catch (error) {
+                    // Sending failed, ignore this device
+                }
+                finally {
+                    resolve();
+                }
+            });
+            promises.push(promise);
+        }
+        await Promise.all(promises);
+    }
+    /**
+     * Clean up stale connections
+     * @param timeoutMs Timeout duration (milliseconds)
+     */
+    cleanupStaleConnections(timeoutMs) {
+        const now = Date.now();
+        const staleDevices = [];
+        for (const [deviceId, lastActive] of this.lastActiveMap.entries()) {
+            if (now - lastActive > timeoutMs) {
+                staleDevices.push(deviceId);
+            }
+        }
+        for (const deviceId of staleDevices) {
+            const ws = this.connections.get(deviceId);
+            if (ws) {
+                try {
+                    ws.close();
+                }
+                catch (error) {
+                    // Ignore close errors
+                }
+            }
+            this.unregisterConnection(deviceId);
+        }
+    }
+    /**
+     * Close all connections
+     */
+    closeAllConnections() {
+        for (const ws of this.connections.values()) {
+            try {
+                ws.close();
+            }
+            catch (error) {
+                // Ignore close errors
+            }
+        }
+        this.connections.clear();
+        this.lastActiveMap.clear();
+    }
+}
+exports.ConnectionHub = ConnectionHub;
+//# sourceMappingURL=ConnectionHub.js.map

package/dist/websocket/ConnectionHub.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ConnectionHub.js","sourceRoot":"","sources":["../../src/websocket/ConnectionHub.ts"],"names":[],"mappings":";;;AAEA;;;GAGG;AACH,MAAa,aAAa;IACxB,sCAAsC;IAC9B,WAAW,CAAyB;IAC5C,6CAA6C;IACrC,aAAa,CAAsB;IAE3C;QACE,IAAI,CAAC,WAAW,GAAG,IAAI,GAAG,EAAE,CAAC;QAC7B,IAAI,CAAC,aAAa,GAAG,IAAI,GAAG,EAAE,CAAC;IACjC,CAAC;IAED;;;;OAIG;IACH,kBAAkB,CAAC,QAAgB,EAAE,EAAa;QAChD,qEAAqE;QACrE,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YACnC,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC;YAC9C,IAAI,CAAC;gBACH,KAAK,CAAC,KAAK,EAAE,CAAC;YAChB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,sBAAsB;YACxB,CAAC;QACH,CAAC;QAED,0BAA0B;QAC1B,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QACnC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;IAC/C,CAAC;IAED;;;OAGG;IACH,oBAAoB,CAAC,QAAgB;QACnC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAClC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACtC,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,YAAY,CAAC,QAAgB,EAAE,OAAY;QAC/C,MAAM,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAE1C,IAAI,CAAC,EAAE,EAAE,CAAC;YACR,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAC3C,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAEpB,0BAA0B;YAC1B,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;YAE7C,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,iDAAiD;YACjD,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,cAAc,CAAC,QAAgB;QAC7B,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAED;;;;OAIG;IACH,iBAAiB,CAAC,QAAgB;QAChC,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;IAED;;;OAGG;IACH,gBAAgB;QACd,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED;;;OAGG;IACH,gBAAgB,CAAC,QAAgB;QAC/B,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YACnC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,kBAAkB;QAIhB,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC1C,OAAO;YACL,gBAAgB,EAAE,SAAS,CAAC,MAAM;YAClC,SAAS;SACV,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,SAAS,CAAC,OAAY;QAC1B,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAC3C,MAAM,QAAQ,GAAoB,EAAE,CAAC;QAErC,KAAK,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,EAAE,CAAC;YACxD,MAAM,OAAO,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;gBAC5C,IAAI,CAAC;oBACH,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;oBACpB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBAC/C,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,qCAAqC;gBACvC,CAAC;wBAAS,CAAC;oBACT,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC,CAAC;YACH,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACzB,CAAC;QAED,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC9B,CAAC;IAED;;;OAGG;IACH,uBAAuB,CAAC,SAAiB;QACvC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,YAAY,GAAa,EAAE,CAAC;QAElC,KAAK,MAAM,CAAC,QAAQ,EAAE,UAAU,CAAC,IAAI,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,EAAE,CAAC;YAClE,IAAI,GAAG,GAAG,UAAU,GAAG,SAAS,EAAE,CAAC;gBACjC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;QAED,KAAK,MAAM,QAAQ,IAAI,YAAY,EAAE,CAAC;YACpC,MAAM,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC1C,IAAI,EAAE,EAAE,CAAC;gBACP,IAAI,CAAC;oBACH,EAAE,CAAC,KAAK,EAAE,CAAC;gBACb,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,sBAAsB;gBACxB,CAAC;YACH,CAAC;YACD,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,mBAAmB;QACjB,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,IAAI,CAAC;gBACH,EAAE,CAAC,KAAK,EAAE,CAAC;YACb,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,sBAAsB;YACxB,CAAC;QACH,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;IAC7B,CAAC;CACF;AA1LD,sCA0LC"}

package/package.json

@@ -0,0 +1,70 @@
+{
+  "name": "@yu_robotics/remote-cli-router",
+  "version": "1.0.0",
+  "description": "Router server for remote CLI - message forwarding between Feishu and local clients",
+  "main": "dist/cli.js",
+  "types": "dist/cli.d.ts",
+  "bin": {
+    "remote-cli-router": "./bin/remote-cli-router.js"
+  },
+  "files": [
+    "dist",
+    "bin",
+    "README.md",
+    "LICENSE"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsx watch src/cli.ts -- start",
+    "start": "node dist/cli.js start",
+    "test": "vitest",
+    "test:coverage": "vitest --coverage",
+    "prepublishOnly": "npm run build && npm test"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/xiaoyu/remote-cli.git",
+    "directory": "packages/router"
+  },
+  "homepage": "https://github.com/xiaoyu/remote-cli#readme",
+  "bugs": {
+    "url": "https://github.com/xiaoyu/remote-cli/issues"
+  },
+  "keywords": [
+    "claude",
+    "cli",
+    "remote",
+    "feishu",
+    "router",
+    "websocket"
+  ],
+  "author": "xiaoyu",
+  "license": "MIT",
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "dependencies": {
+    "@koa/router": "^12.0.1",
+    "@larksuiteoapi/node-sdk": "^1.59.0",
+    "axios": "^1.6.7",
+    "commander": "^12.0.0",
+    "dotenv": "^16.4.1",
+    "koa": "^2.15.0",
+    "koa-bodyparser": "^4.4.1",
+    "uuid": "^13.0.0",
+    "ws": "^8.18.0"
+  },
+  "devDependencies": {
+    "@types/koa": "^2.15.0",
+    "@types/koa__router": "^12.0.4",
+    "@types/koa-bodyparser": "^4.3.12",
+    "@types/uuid": "^10.0.0",
+    "@types/ws": "^8.5.10",
+    "@vitest/coverage-v8": "^1.2.0",
+    "tsx": "^4.7.0",
+    "vitest": "^1.2.0"
+  }
+}