@autolabz/mcp-bridge 1.0.0 → 1.0.2

package/dist/bridge.d.ts

@@ -4,6 +4,7 @@ export declare class Bridge {
     private ws;
     private mcpClient;
     private reconnectTimer;
+    private heartbeatTimer;
     private shouldReconnect;
     constructor(config: BridgeConfig);
     /**
@@ -17,6 +18,8 @@ export declare class Bridge {
      * Rotate the key hash on the cloud relay.
      */
     rotateKey(newKeyHash: string): void;
+    private startHeartbeat;
+    private stopHeartbeat;
     /**
      * Gracefully stop the bridge.
      */

package/dist/bridge.js

@@ -1,10 +1,12 @@
 import WebSocket from 'ws';
 import { MCPClient } from './mcp-client.js';
+const HEARTBEAT_INTERVAL = 30_000; // 30 seconds
 export class Bridge {
     config;
     ws = null;
     mcpClient;
     reconnectTimer = null;
+    heartbeatTimer = null;
     shouldReconnect = true;
     constructor(config) {
         this.config = config;
@@ -38,6 +40,7 @@ export class Bridge {
                 clearTimeout(this.reconnectTimer);
                 this.reconnectTimer = null;
             }
+            this.startHeartbeat();
         });
         this.ws.on('message', async (data) => {
             try {
@@ -53,6 +56,11 @@ export class Bridge {
                 else if (msg.type === 'error') {
                     console.error(`❌ Cloud error: ${msg.message}`);
                 }
+                else if (msg.type === 'pong') {
+                    // Heartbeat response received
+                    // You could update a "lastActivity" timestamp here if you wanted
+                    // to implement a watchdog that reconnects if the server is silent for too long.
+                }
             }
             catch (err) {
                 console.error('❌ Failed to handle cloud message:', err);
@@ -60,10 +68,12 @@ export class Bridge {
         });
         this.ws.on('close', (code, reason) => {
             console.log(`🔌 Disconnected from cloud (${code}: ${reason.toString()})`);
+            this.stopHeartbeat();
             this.scheduleReconnect();
         });
         this.ws.on('error', (err) => {
             console.error('❌ WebSocket error:', err.message);
+            this.stopHeartbeat();
         });
     }
     async handleRequest(msg) {
@@ -119,6 +129,22 @@ export class Bridge {
         }));
         this.config.keyHash = newKeyHash;
     }
+    startHeartbeat() {
+        this.stopHeartbeat();
+        // Initial ping
+        // this.sendPing(); // Optional: send one immediately
+        this.heartbeatTimer = setInterval(() => {
+            if (this.ws?.readyState === WebSocket.OPEN) {
+                this.ws.send(JSON.stringify({ type: 'ping' }));
+            }
+        }, HEARTBEAT_INTERVAL);
+    }
+    stopHeartbeat() {
+        if (this.heartbeatTimer) {
+            clearInterval(this.heartbeatTimer);
+            this.heartbeatTimer = null;
+        }
+    }
     /**
      * Gracefully stop the bridge.
      */
@@ -127,6 +153,7 @@ export class Bridge {
         if (this.reconnectTimer) {
             clearTimeout(this.reconnectTimer);
         }
+        this.stopHeartbeat();
         if (this.ws) {
             this.ws.close(1000, 'bridge stopped');
         }

package/dist/index.js

@@ -50,10 +50,19 @@ program
     };
     saveConfig(config);
     console.log(`\n✅ 配置已保存到: ${getConfigPath()}`);
-    console.log(`📋 Node ID: ${nodeId}`);
-    console.log(`\n📌 远程调用示例:`);
-    console.log(`   curl http://localhost:8787/mcp/${nodeId}/tools \\`);
-    console.log(`     -H "Authorization: Bearer ${rawKey}"`);
+    // Construct full bridge URL
+    const fullBridgeUrl = `${cloudUrl}/mcp/${nodeId}`;
+    console.log('\n🎉 Bridge Initialized Successfully!');
+    console.log('----------------------------------------');
+    console.log(`🌍 Bridge URL: ${fullBridgeUrl}`);
+    console.log(`� API Key:    ${rawKey}`);
+    console.log('----------------------------------------');
+    console.log('⚠️  请妥善保管 API Key，它不会再次显示！');
+    console.log(`\n👉 Next steps:`);
+    console.log(`   1. Start the bridge:`);
+    console.log(`      mcp-bridge start`);
+    console.log(`   2. Test connection:`);
+    console.log(`      mcp-bridge client --url ${fullBridgeUrl} --key ${rawKey}`);
 });
 /**
  * start: Connect the bridge to the cloud relay.
@@ -67,10 +76,15 @@ program
         console.error('❌ No configuration found. Run `mcp-bridge init` first.');
         process.exit(1);
     }
+    const fullBridgeUrl = `${config.cloudUrl}/mcp/${config.nodeId}`;
     console.log('🚀 Starting MCP Bridge...');
     console.log(`   Cloud:  ${config.cloudUrl}`);
     console.log(`   Pieces: ${config.piecesEndpoint}`);
-    console.log(`   NodeID: ${config.nodeId}\n`);
+    console.log(`   NodeID: ${config.nodeId}`);
+    console.log(`   ---------------------------------------------------`);
+    console.log(`   🌍 Bridge URL: ${fullBridgeUrl}`);
+    console.log(`   ---------------------------------------------------`);
+    console.log('\n');
     const bridge = new Bridge(config);
     // Graceful shutdown
     process.on('SIGINT', () => {
@@ -116,23 +130,134 @@ program
     console.log('⚠️  如果 bridge 正在运行，请重启以生效');
 });
 /**
- * status: Show current configuration.
+ * client: Interactive client to test remote bridge
  */
 program
-    .command('status')
-    .description('Show current bridge configuration')
-    .action(() => {
-    const config = loadConfig();
-    if (!config) {
-        console.log('❌ No configuration found. Run `mcp-bridge init` first.');
-        return;
+    .command('client')
+    .description('Connect to a remote bridge and test tools interactively')
+    .option('--url <url>', 'Full bridge URL (e.g. https://cloud.com/mcp/{NODE_ID})')
+    .option('--key <key>', 'API Key')
+    .action(async (opts) => {
+    const readline = await import('readline/promises');
+    // 1. Get URL
+    let baseUrl = opts.url;
+    if (!baseUrl) {
+        const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+        baseUrl = await rl.question('Bridge URL: ');
+        rl.close();
+    }
+    baseUrl = baseUrl.replace(/\/+$/, '');
+    // 2. Get Key (Hidden)
+    let key = opts.key;
+    if (!key) {
+        const { Writable } = await import('node:stream');
+        let muted = false;
+        const mutableStdout = new Writable({
+            write: function (chunk, encoding, callback) {
+                if (!muted)
+                    process.stdout.write(chunk, encoding);
+                callback();
+            }
+        });
+        const secureRl = readline.createInterface({
+            input: process.stdin,
+            output: mutableStdout,
+            terminal: true
+        });
+        process.stdout.write('API Key: ');
+        muted = true;
+        key = await secureRl.question('');
+        muted = false;
+        secureRl.close();
+        process.stdout.write('\n');
+    }
+    const headers = {
+        'Authorization': `Bearer ${key}`,
+        'Content-Type': 'application/json'
+    };
+    console.log(`\n🔌 Connecting to ${baseUrl}...\n`);
+    try {
+        // Check Status
+        process.stdout.write('Checking status... ');
+        const statusRes = await fetch(`${baseUrl}/status`, { headers });
+        if (!statusRes.ok)
+            throw new Error(`Status check failed: ${statusRes.status}`);
+        const status = await statusRes.json();
+        console.log('✅ Online');
+        console.log(status);
+        // List Tools
+        console.log('\nFetching tools...');
+        const toolsRes = await fetch(`${baseUrl}/tools`, { headers });
+        if (!toolsRes.ok)
+            throw new Error(`List tools failed: ${toolsRes.status}`);
+        const toolsData = (await toolsRes.json());
+        const tools = toolsData.result?.tools || [];
+        if (tools.length === 0) {
+            console.log('⚠️  No tools found.');
+            return;
+        }
+        console.table(tools.map((t) => ({
+            Name: t.name,
+            Description: t.description?.slice(0, 50) + (t.description?.length > 50 ? '...' : '')
+        })));
+        // Interactive Loop
+        const rlLoop = readline.createInterface({
+            input: process.stdin,
+            output: process.stdout
+        });
+        console.log('\n💡 Enter a tool name to call it, or "exit" to quit.');
+        while (true) {
+            const name = await rlLoop.question('\n> ');
+            if (name.trim() === 'exit')
+                break;
+            if (!name.trim())
+                continue;
+            const tool = tools.find((t) => t.name === name.trim());
+            if (!tool) {
+                console.log(`❌ Tool "${name}" not found.`);
+                continue;
+            }
+            console.log(`\nCalling ${name}...`);
+            console.log('Arguments (JSON, optional):');
+            console.log(`Schema: ${JSON.stringify(tool.inputSchema, null, 2)}`);
+            const argsStr = await rlLoop.question('Enter args ({}): ');
+            let args = {};
+            try {
+                args = argsStr.trim() ? JSON.parse(argsStr) : {};
+            }
+            catch (e) {
+                console.log('❌ Invalid JSON arguments');
+                continue;
+            }
+            const startTime = Date.now();
+            try {
+                const callRes = await fetch(`${baseUrl}/call`, {
+                    method: 'POST',
+                    headers,
+                    body: JSON.stringify({
+                        method: 'tools/call',
+                        params: { name, arguments: args }
+                    })
+                });
+                if (!callRes.ok) {
+                    const errText = await callRes.text();
+                    console.log(`❌ Call failed (${callRes.status}): ${errText}`);
+                }
+                else {
+                    const result = await callRes.json();
+                    console.log(`✅ Success (${Date.now() - startTime}ms)`);
+                    console.dir(result, { depth: null, colors: true });
+                }
+            }
+            catch (err) {
+                console.log(`❌ Error: ${err.message}`);
+            }
+        }
+        rlLoop.close();
+    }
+    catch (error) {
+        console.error(`\n❌ Error: ${error.message}`);
+        process.exit(1);
     }
-    console.log('📋 MCP Bridge Status');
-    console.log('-'.repeat(40));
-    console.log(`Config: ${getConfigPath()}`);
-    console.log(`Cloud:  ${config.cloudUrl}`);
-    console.log(`Pieces: ${config.piecesEndpoint}`);
-    console.log(`NodeID: ${config.nodeId}`);
-    console.log(`KeyHash: ${config.keyHash.slice(0, 12)}...`);
 });
 program.parse();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@autolabz/mcp-bridge",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "Bridge local Pieces MCP to cloud relay for remote AI agent access",
   "type": "module",
   "main": "dist/index.js",