@dolusoft/claude-collab 1.8.4 → 1.8.6

package/dist/mcp-main.js

@@ -6,7 +6,7 @@ import { execFile, spawn } from 'child_process';
 import { unlinkSync } from 'fs';
 import { tmpdir } from 'os';
 import { join } from 'path';
-import { Bonjour } from 'bonjour-service';
+import { createSocket } from 'dgram';
 import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
 import { z } from 'zod';
@@ -666,28 +666,48 @@ var HubServer = class {
     }
   }
 };
-var SERVICE_TYPE = "claude-collab";
-var MdnsAdvertiser = class {
-  bonjour = null;
-  start(port) {
-    this.bonjour = new Bonjour();
-    this.bonjour.publish({ name: "claude-collab-hub", type: SERVICE_TYPE, port });
-    console.error(`[mdns] advertising hub on port ${port}`);
+var DISCOVERY_PORT = 9998;
+var BROADCAST_INTERVAL_MS = 3e3;
+var BROADCAST_ADDRESS = "255.255.255.255";
+var HubBroadcaster = class {
+  socket = null;
+  timer = null;
+  start(hubPort) {
+    if (this.socket) return;
+    const socket = createSocket("udp4");
+    this.socket = socket;
+    socket.on("error", (err) => {
+      console.error("[hub-broadcaster] error:", err.message);
+    });
+    socket.bind(0, () => {
+      socket.setBroadcast(true);
+      const send = () => {
+        if (!this.socket) return;
+        const msg = Buffer.from(JSON.stringify({ type: "claude-collab-hub", port: hubPort }));
+        socket.send(msg, 0, msg.length, DISCOVERY_PORT, BROADCAST_ADDRESS, (err) => {
+          if (err) console.error("[hub-broadcaster] send error:", err.message);
+        });
+      };
+      send();
+      this.timer = setInterval(send, BROADCAST_INTERVAL_MS);
+    });
   }
   stop() {
-    if (!this.bonjour) return;
-    this.bonjour.unpublishAll(() => {
-      this.bonjour?.destroy();
-    });
-    this.bonjour = null;
-    console.error("[mdns] stopped advertising");
+    if (this.timer) {
+      clearInterval(this.timer);
+      this.timer = null;
+    }
+    if (this.socket) {
+      this.socket.close();
+      this.socket = null;
+    }
   }
 };
 
 // src/infrastructure/hub/hub-manager.ts
 var HubManager = class {
   hubServer = null;
-  advertiser = null;
+  broadcaster = null;
   currentPort = null;
   get isRunning() {
     return this.hubServer !== null;
@@ -701,9 +721,9 @@ var HubManager = class {
     server.start(port);
     this.hubServer = server;
     this.currentPort = port;
-    const advertiser = new MdnsAdvertiser();
-    advertiser.start(port);
-    this.advertiser = advertiser;
+    const broadcaster = new HubBroadcaster();
+    broadcaster.start(port);
+    this.broadcaster = broadcaster;
     let firewallAdded = false;
     try {
       await addFirewallRule(port);
@@ -715,9 +735,9 @@ var HubManager = class {
   }
   async stop() {
     if (!this.isRunning) throw new Error("Hub is not running");
-    if (this.advertiser) {
-      this.advertiser.stop();
-      this.advertiser = null;
+    if (this.broadcaster) {
+      this.broadcaster.stop();
+      this.broadcaster = null;
     }
     const port = this.currentPort;
     this.hubServer.stop();
@@ -1012,6 +1032,97 @@ function registerStopHubTool(server, hubManager) {
     }
   );
 }
+function discoverHub(timeoutMs = 1e4) {
+  return new Promise((resolve) => {
+    const socket = createSocket("udp4");
+    let settled = false;
+    const finish = (result) => {
+      if (settled) return;
+      settled = true;
+      clearTimeout(timer);
+      try {
+        socket.close();
+      } catch {
+      }
+      resolve(result);
+    };
+    const timer = setTimeout(() => finish(null), timeoutMs);
+    socket.on("error", () => finish(null));
+    socket.on("message", (msg, rinfo) => {
+      try {
+        const data = JSON.parse(msg.toString());
+        if (data.type === "claude-collab-hub" && typeof data.port === "number") {
+          finish({ host: rinfo.address, port: data.port });
+        }
+      } catch {
+      }
+    });
+    socket.bind(DISCOVERY_PORT, "0.0.0.0");
+  });
+}
+function watchForHub(onFound) {
+  const socket = createSocket("udp4");
+  socket.on("error", (err) => {
+    console.error("[hub-listener] error:", err.message);
+  });
+  socket.on("message", (msg, rinfo) => {
+    try {
+      const data = JSON.parse(msg.toString());
+      if (data.type === "claude-collab-hub" && typeof data.port === "number") {
+        onFound({ host: rinfo.address, port: data.port });
+      }
+    } catch {
+    }
+  });
+  socket.bind(DISCOVERY_PORT, "0.0.0.0");
+  return () => {
+    try {
+      socket.close();
+    } catch {
+    }
+  };
+}
+
+// src/presentation/mcp/tools/connect.tool.ts
+function registerConnectTool(server, client) {
+  server.tool(
+    "connect",
+    "Find and connect to the hub on the LAN automatically. No IP or port needed.",
+    {},
+    async () => {
+      if (client.isConnected) {
+        const info = client.getInfo();
+        return {
+          content: [{ type: "text", text: `Already connected to hub as "${info.teamName}".` }]
+        };
+      }
+      const hub = await discoverHub(1e4);
+      if (!hub) {
+        return {
+          content: [{
+            type: "text",
+            text: "No hub found on the LAN. Make sure someone has called start_hub on the host machine."
+          }]
+        };
+      }
+      try {
+        await client.connectToHub(`ws://${hub.host}:${hub.port}`);
+        const info = client.getInfo();
+        return {
+          content: [{
+            type: "text",
+            text: `Connected to hub at ${hub.host}:${hub.port} as "${info.teamName}".`
+          }]
+        };
+      } catch (err) {
+        const msg = err instanceof Error ? err.message : String(err);
+        return {
+          content: [{ type: "text", text: `Hub found at ${hub.host}:${hub.port} but connection failed: ${msg}` }]
+        };
+      }
+    }
+  );
+}
 
 // src/presentation/mcp/server.ts
 function createMcpServer(options) {
@@ -1026,6 +1137,7 @@ function createMcpServer(options) {
   registerHistoryTool(server, client);
   registerStartHubTool(server, client, hubManager);
   registerStopHubTool(server, hubManager);
+  registerConnectTool(server, client);
   return server;
 }
 async function startMcpServer(options) {
@@ -1033,41 +1145,6 @@ async function startMcpServer(options) {
   const transport = new StdioServerTransport();
   await server.connect(transport);
 }
-var SERVICE_TYPE2 = "claude-collab";
-function resolveHost(svc) {
-  const ipv4 = svc.addresses?.find((a) => /^\d+\.\d+\.\d+\.\d+$/.test(a));
-  return ipv4 ?? svc.host;
-}
-function discoverHub(timeoutMs = 5e3) {
-  return new Promise((resolve) => {
-    const bonjour = new Bonjour();
-    const browser = bonjour.find({ type: SERVICE_TYPE2 });
-    let settled = false;
-    const finish = (result) => {
-      if (settled) return;
-      settled = true;
-      clearTimeout(timer);
-      browser.stop();
-      bonjour.destroy();
-      resolve(result);
-    };
-    const timer = setTimeout(() => finish(null), timeoutMs);
-    browser.on("up", (svc) => {
-      finish({ host: resolveHost(svc), port: svc.port });
-    });
-  });
-}
-function watchForHub(onFound) {
-  const bonjour = new Bonjour();
-  const browser = bonjour.find({ type: SERVICE_TYPE2 });
-  browser.on("up", (svc) => {
-    onFound({ host: resolveHost(svc), port: svc.port });
-  });
-  return () => {
-    browser.stop();
-    bonjour.destroy();
-  };
-}
 
 // src/mcp-main.ts
 function getArg(flag) {