node-lab-mcp 0.1.2 → 0.1.3

package/dist/index.js

@@ -25315,7 +25315,7 @@ function getPartyInfo(partySocketOptions, defaultProtocol, defaultParams = {}) {
     host: rawHost,
     path: rawPath,
     protocol: rawProtocol,
-    room: room2,
+    room,
     party,
     basePath,
     prefix,
@@ -25328,13 +25328,13 @@ function getPartyInfo(partySocketOptions, defaultProtocol, defaultParams = {}) {
   const name = party ?? "main";
   const path = rawPath ? `/${rawPath}` : "";
   const protocol = rawProtocol || (host.startsWith("localhost:") || host.startsWith("127.0.0.1:") || host.startsWith("192.168.") || host.startsWith("10.") || host.startsWith("172.") && host.split(".")[1] >= "16" && host.split(".")[1] <= "31" || host.startsWith("[::ffff:7f00:1]:") ? defaultProtocol : `${defaultProtocol}s`);
-  const baseUrl = `${protocol}://${host}/${basePath || `${prefix || "parties"}/${name}/${room2}`}${path}`;
+  const baseUrl = `${protocol}://${host}/${basePath || `${prefix || "parties"}/${name}/${room}`}${path}`;
   const makeUrl = (query2 = {}) => `${baseUrl}?${new URLSearchParams([...Object.entries(defaultParams), ...Object.entries(query2).filter(valueIsNotNil)])}`;
   const urlProvider = typeof query === "function" ? async () => makeUrl(await query()) : makeUrl(query);
   return {
     host,
     path,
-    room: room2,
+    room,
     name,
     protocol,
     partyUrl: baseUrl,
@@ -25386,11 +25386,11 @@ var PartySocket = class extends ReconnectingWebSocket {
     this.setWSProperties(wsOptions);
   }
   setWSProperties(wsOptions) {
-    const { _pk, _pkurl, name, room: room2, host, path, basePath } = wsOptions;
+    const { _pk, _pkurl, name, room, host, path, basePath } = wsOptions;
     this._pk = _pk;
     this._pkurl = _pkurl;
     this.name = name;
-    this.room = room2;
+    this.room = room;
     this.host = host;
     this.path = path;
     this.basePath = basePath;
@@ -25453,14 +25453,34 @@ function getWSOptions(partySocketOptions) {
 // party-bridge.ts
 import { randomUUID as randomUUID2 } from "node:crypto";
 var PartyBridge = class {
-  constructor(host, room2) {
+  constructor(host, room) {
     this.host = host;
-    this.room = room2;
-    this.ps = new PartySocket({ host, room: room2 });
-    this.ps.addEventListener("message", (e) => this.onMessage(typeof e.data === "string" ? e.data : ""));
-    this.ps.addEventListener("open", () => process.stderr.write(`[party] joined room "${room2}" on ${host}
+    if (room) this.openRoom(room);
+  }
+  ps = null;
+  pending = /* @__PURE__ */ new Map();
+  peers = 0;
+  room;
+  openRoom(room) {
+    if (this.ps) {
+      try {
+        this.ps.close();
+      } catch {
+      }
+    }
+    for (const [id, p] of this.pending) {
+      clearTimeout(p.timer);
+      p.reject(new Error("relay room switched mid-command"));
+      this.pending.delete(id);
+    }
+    this.peers = 0;
+    this.room = room;
+    const ps = new PartySocket({ host: this.host, room });
+    this.ps = ps;
+    ps.addEventListener("message", (e) => this.onMessage(typeof e.data === "string" ? e.data : ""));
+    ps.addEventListener("open", () => process.stderr.write(`[party] joined room "${room}" on ${this.host}
 `));
-    this.ps.addEventListener("close", () => {
+    ps.addEventListener("close", () => {
       this.peers = 0;
       for (const [id, p] of this.pending) {
         clearTimeout(p.timer);
@@ -25469,18 +25489,27 @@ var PartyBridge = class {
       }
     });
   }
-  ps;
-  pending = /* @__PURE__ */ new Map();
-  peers = 0;
+  // GraphBridge: runtime room switching. The promise resolves immediately after the new socket is
+  // requested — the `open` event follows async; check status()/connected() to know when the editor pairs.
+  async setRoom(room) {
+    this.openRoom(room);
+  }
+  currentRoom() {
+    return this.room;
+  }
   // peers counts everyone in the room including us, so >1 means an editor has joined.
   connected() {
-    return this.ps.readyState === PartySocket.OPEN && this.peers > 1;
+    return this.ps?.readyState === PartySocket.OPEN && this.peers > 1;
   }
   status() {
+    if (!this.ps || !this.room) return `no relay room set \u2014 call set_room with a pairing code from the lab's MCP modal`;
     if (this.ps.readyState !== PartySocket.OPEN) return `connecting to relay ${this.host}\u2026`;
     return this.peers > 1 ? `editor connected via relay (room "${this.room}")` : `in room "${this.room}", waiting for the editor to pair`;
   }
   send(cmd, timeoutMs = 8e3) {
+    if (!this.ps || !this.room) {
+      return Promise.reject(new Error(`No relay room set. Call set_room with the pairing code from the lab's MCP modal.`));
+    }
     if (this.ps.readyState !== PartySocket.OPEN) {
       return Promise.reject(new Error(`Not connected to the relay (${this.host}) yet \u2014 retry in a moment.`));
     }
@@ -27184,17 +27213,23 @@ function gradeHeadless(spec, inputs, expected, ticks = 16) {
 }
 
 // server.ts
-var room = process.env.NODE_LAB_ROOM;
+var initialRoom = process.env.NODE_LAB_ROOM;
+var useLocal = process.env.NODE_LAB_LOCAL === "1";
 var bridge;
-if (room) {
-  const host = process.env.NODE_LAB_PARTY_HOST ?? DEFAULT_PARTY_HOST;
-  bridge = new PartyBridge(host, room);
-  process.stderr.write(`[node-lab-mcp] relay mode \u2014 room "${room}" @ ${host}
-`);
-} else {
+if (useLocal) {
   const local = new Bridge();
   local.start(BRIDGE_PORT);
   bridge = local;
+  process.stderr.write(`[node-lab-mcp] local-WS mode \u2014 ws://localhost:${BRIDGE_PORT}
+`);
+} else {
+  const host = process.env.NODE_LAB_PARTY_HOST ?? DEFAULT_PARTY_HOST;
+  bridge = new PartyBridge(host, initialRoom);
+  process.stderr.write(
+    initialRoom ? `[node-lab-mcp] relay mode \u2014 room "${initialRoom}" @ ${host}
+` : `[node-lab-mcp] relay mode (no room yet) \u2014 agent must call set_room with the pairing code @ ${host}
+`
+  );
 }
 var server = new McpServer({ name: "node-lab", version: "0.1.0" });
 var json = (data) => ({ content: [{ type: "text", text: JSON.stringify(data, null, 2) }] });
@@ -27202,6 +27237,30 @@ var fail = (e) => ({
   content: [{ type: "text", text: `ERROR: ${e instanceof Error ? e.message : String(e)}` }],
   isError: true
 });
+server.registerTool(
+  "set_room",
+  {
+    description: `Switch the PartyKit relay room this MCP connects to. Pair with any browser tab on the fly without restarting the MCP server. Pass the 4-4 pairing code from the lab's MCP modal (e.g. "138n-f2le"). Returns the new status \u2014 call bridge_status again after a short delay to confirm the editor joined.`,
+    inputSchema: { room: external_exports.string() }
+  },
+  async ({ room }) => {
+    try {
+      if (!bridge.setRoom) throw new Error("set_room is only available in relay mode. Start the MCP without NODE_LAB_LOCAL=1.");
+      await bridge.setRoom(room);
+      return json({ room, status: bridge.status() });
+    } catch (e) {
+      return fail(e);
+    }
+  }
+);
+server.registerTool(
+  "get_room",
+  {
+    description: "Read the current PartyKit relay room id (the 4-4 pairing code), plus the bridge status.",
+    inputSchema: {}
+  },
+  async () => json({ room: bridge.currentRoom?.() ?? null, status: bridge.status() })
+);
 server.registerTool(
   "bridge_status",
   {
@@ -27321,5 +27380,5 @@ live(
 );
 var transport = new StdioServerTransport();
 await server.connect(transport);
-process.stderr.write(`[node-lab-mcp] ready \u2014 ${room ? `relay room "${room}"` : `local bridge ws://localhost:${BRIDGE_PORT}`}
+process.stderr.write(`[node-lab-mcp] ready \u2014 ${bridge.status()}
 `);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "node-lab-mcp",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "MCP server to drive the IFR Node Lab graph from any MCP client (Claude / Codex / Gemini …) — locally or, via a PartyKit relay, against the deployed app.",
   "type": "module",
   "bin": {