dominus-cli 0.5.8 → 0.6.0

package/dist/index.js

@@ -252,7 +252,12 @@ var init_ws_server = __esm({
     init_protocol();
     DominusServer = class extends EventEmitter {
       wss = null;
-      studioClient = null;
+      /** All connected Studio instances, keyed by placeId (0 for unknown) */
+      studioClients = /* @__PURE__ */ new Map();
+      /** Reverse lookup: ws → placeId */
+      wsToPlaceId = /* @__PURE__ */ new Map();
+      /** Which placeId to target. 0 = auto-select if only one is connected. */
+      activePlaceId = 0;
       controllers = /* @__PURE__ */ new Set();
       pendingRequests = /* @__PURE__ */ new Map();
       relayRequests = /* @__PURE__ */ new Map();
@@ -301,7 +306,9 @@ var init_ws_server = __esm({
                 this.addController(ws);
                 ws.send(serializeMessage(createMessage("controller:welcome", {
                   studioConnected: this.isConnected(),
-                  port: this.port
+                  port: this.port,
+                  connections: this.listConnections(),
+                  activePlaceId: this.activePlaceId
                 })));
                 return;
               }
@@ -309,7 +316,7 @@ var init_ws_server = __esm({
             if (this.controllers.has(ws)) {
               this.handleControllerMessage(ws, msg);
             } else {
-              this.handleStudioMessage(msg);
+              this.handleStudioMessage(ws, msg);
             }
           } catch (err) {
             this.emit("server:error", err);
@@ -321,14 +328,22 @@ var init_ws_server = __esm({
             for (const [id, controllerWs] of this.relayRequests) {
               if (controllerWs === ws) this.relayRequests.delete(id);
             }
-          } else if (this.studioClient?.ws === ws) {
-            this.studioClient = null;
-            this.emit("studio:disconnected");
-            this.broadcastToControllers("studio:disconnected", {});
-            for (const [id, pending] of this.pendingRequests) {
-              clearTimeout(pending.timer);
-              pending.reject(new Error("Studio disconnected"));
-              this.pendingRequests.delete(id);
+          } else {
+            const placeId = this.wsToPlaceId.get(ws);
+            if (placeId !== void 0) {
+              const conn = this.studioClients.get(placeId);
+              this.studioClients.delete(placeId);
+              this.wsToPlaceId.delete(ws);
+              this.emit("studio:disconnected", { placeId, placeName: conn?.placeName });
+              this.broadcastToControllers("studio:disconnected", { placeId, placeName: conn?.placeName });
+              for (const [id, pending] of this.pendingRequests) {
+                clearTimeout(pending.timer);
+                pending.reject(new Error(`Studio disconnected (place ${placeId})`));
+                this.pendingRequests.delete(id);
+              }
+              if (this.activePlaceId === placeId) {
+                this.activePlaceId = 0;
+              }
             }
           }
         });
@@ -337,19 +352,24 @@ var init_ws_server = __esm({
         });
       }
       setStudioClient(ws, msg) {
-        if (this.studioClient) {
-          this.studioClient.ws.close();
-        }
         const payload = msg.payload;
-        this.studioClient = {
+        const placeId = payload.placeId ?? 0;
+        const existing = this.studioClients.get(placeId);
+        if (existing && existing.ws !== ws) {
+          existing.ws.close();
+          this.wsToPlaceId.delete(existing.ws);
+        }
+        const conn = {
           ws,
           connectedAt: Date.now(),
           studioVersion: payload.studioVersion,
-          placeId: payload.placeId,
+          placeId,
           placeName: payload.placeName
         };
-        this.emit("studio:connected", payload);
-        this.broadcastToControllers("studio:connected", payload);
+        this.studioClients.set(placeId, conn);
+        this.wsToPlaceId.set(ws, placeId);
+        this.emit("studio:connected", { ...payload, placeId });
+        this.broadcastToControllers("studio:connected", { ...payload, placeId });
       }
       addController(ws) {
         this.controllers.add(ws);
@@ -364,25 +384,30 @@ var init_ws_server = __esm({
         }
       }
       handleControllerMessage(controllerWs, msg) {
-        if (!this.studioClient || this.studioClient.ws.readyState !== WebSocket.OPEN) {
+        const target = this.resolveTarget();
+        if (!target || target.ws.readyState !== WebSocket.OPEN) {
           controllerWs.send(serializeMessage({
             id: msg.id,
             type: StudioMsg.RESPONSE,
-            payload: { error: "Studio not connected" },
+            payload: { error: this.getTargetError() },
             ts: Date.now()
           }));
           return;
         }
         this.relayRequests.set(msg.id, controllerWs);
-        this.studioClient.ws.send(serializeMessage(msg));
+        target.ws.send(serializeMessage(msg));
       }
-      handleStudioMessage(msg) {
+      handleStudioMessage(ws, msg) {
         if (msg.type === StudioMsg.CONNECTED) {
           const payload = msg.payload;
-          if (this.studioClient) {
-            this.studioClient.studioVersion = payload.studioVersion;
-            this.studioClient.placeId = payload.placeId;
-            this.studioClient.placeName = payload.placeName;
+          const placeId = this.wsToPlaceId.get(ws);
+          if (placeId !== void 0) {
+            const conn = this.studioClients.get(placeId);
+            if (conn) {
+              conn.studioVersion = payload.studioVersion;
+              conn.placeId = payload.placeId;
+              conn.placeName = payload.placeName;
+            }
           }
           this.emit("studio:connected", payload);
           this.broadcastToControllers("studio:connected", payload);
@@ -407,10 +432,35 @@ var init_ws_server = __esm({
         this.emit("message", msg);
         this.broadcastToControllers(msg.type, msg.payload);
       }
+      /** Resolve which Studio connection to target */
+      resolveTarget() {
+        if (this.studioClients.size === 0) return null;
+        if (this.activePlaceId !== 0) {
+          const conn = this.studioClients.get(this.activePlaceId);
+          if (conn && conn.ws.readyState === WebSocket.OPEN) return conn;
+          return null;
+        }
+        if (this.studioClients.size === 1) {
+          const conn = this.studioClients.values().next().value;
+          if (conn.ws.readyState === WebSocket.OPEN) return conn;
+          return null;
+        }
+        return null;
+      }
+      /** Generate an appropriate error message when no target found */
+      getTargetError() {
+        if (this.studioClients.size === 0) return "No Studio connection";
+        if (this.activePlaceId !== 0) {
+          return `Active place ${this.activePlaceId} is not connected. Use set_active_place or list_connections.`;
+        }
+        const places = [...this.studioClients.values()].map((c) => `${c.placeName ?? "Unknown"} (${c.placeId})`);
+        return `Multiple Studio instances connected: ${places.join(", ")}. Use set_active_place to choose one.`;
+      }
       sendRequest(type, payload, timeoutMs = 1e4) {
         return new Promise((resolve, reject) => {
-          if (!this.studioClient || this.studioClient.ws.readyState !== WebSocket.OPEN) {
-            reject(new Error("No Studio connection"));
+          const target = this.resolveTarget();
+          if (!target || target.ws.readyState !== WebSocket.OPEN) {
+            reject(new Error(this.getTargetError()));
             return;
           }
           const msg = createMessage(type, payload);
@@ -423,31 +473,45 @@ var init_ws_server = __esm({
             reject,
             timer
           });
-          this.studioClient.ws.send(serializeMessage(msg));
+          target.ws.send(serializeMessage(msg));
         });
       }
       sendNotification(type, payload) {
-        if (!this.studioClient || this.studioClient.ws.readyState !== WebSocket.OPEN) {
-          return;
-        }
+        const target = this.resolveTarget();
+        if (!target || target.ws.readyState !== WebSocket.OPEN) return;
         const msg = createMessage(type, payload);
-        this.studioClient.ws.send(serializeMessage(msg));
+        target.ws.send(serializeMessage(msg));
       }
       isConnected() {
-        return this.studioClient !== null && this.studioClient.ws.readyState === WebSocket.OPEN;
+        return this.resolveTarget() !== null;
       }
       getConnectionInfo() {
-        return this.studioClient;
+        return this.resolveTarget();
+      }
+      listConnections() {
+        return [...this.studioClients.values()].filter((c) => c.ws.readyState === WebSocket.OPEN).map((c) => ({
+          placeId: c.placeId ?? 0,
+          placeName: c.placeName,
+          studioVersion: c.studioVersion,
+          connectedAt: c.connectedAt
+        }));
+      }
+      getActivePlaceId() {
+        return this.activePlaceId;
+      }
+      setActivePlaceId(placeId) {
+        this.activePlaceId = placeId;
       }
       getPort() {
         return this.port;
       }
       stop() {
         return new Promise((resolve) => {
-          if (this.studioClient) {
-            this.studioClient.ws.close();
-            this.studioClient = null;
+          for (const conn of this.studioClients.values()) {
+            conn.ws.close();
           }
+          this.studioClients.clear();
+          this.wsToPlaceId.clear();
           for (const ws of this.controllers) {
             ws.close();
           }
@@ -494,7 +558,8 @@ var init_ws_client = __esm({
       ws = null;
       port;
       studioConnected = false;
-      connectionInfo = null;
+      connections = /* @__PURE__ */ new Map();
+      activePlaceId = 0;
       pendingRequests = /* @__PURE__ */ new Map();
       constructor(port = 18088) {
         super();
@@ -538,6 +603,20 @@ var init_ws_client = __esm({
         if (msg.type === "controller:welcome") {
           const p = msg.payload;
           this.studioConnected = p.studioConnected;
+          if (p.connections) {
+            this.connections.clear();
+            for (const c of p.connections) {
+              this.connections.set(c.placeId, {
+                connectedAt: c.connectedAt,
+                studioVersion: c.studioVersion,
+                placeId: c.placeId,
+                placeName: c.placeName
+              });
+            }
+          }
+          if (p.activePlaceId !== void 0) {
+            this.activePlaceId = p.activePlaceId;
+          }
           if (this.studioConnected) {
             this.emit("studio:connected", {});
           }
@@ -546,19 +625,27 @@ var init_ws_client = __esm({
         if (msg.type === "studio:connected") {
           this.studioConnected = true;
           const payload = msg.payload;
-          this.connectionInfo = {
+          const placeId = payload.placeId ?? 0;
+          this.connections.set(placeId, {
             connectedAt: Date.now(),
             studioVersion: payload.studioVersion,
-            placeId: payload.placeId,
+            placeId,
             placeName: payload.placeName
-          };
+          });
           this.emit("studio:connected", payload);
           return;
         }
         if (msg.type === "studio:disconnected") {
-          this.studioConnected = false;
-          this.connectionInfo = null;
-          this.emit("studio:disconnected");
+          const payload = msg.payload;
+          const placeId = payload.placeId ?? 0;
+          this.connections.delete(placeId);
+          if (this.connections.size === 0) {
+            this.studioConnected = false;
+          }
+          if (this.activePlaceId === placeId) {
+            this.activePlaceId = 0;
+          }
+          this.emit("studio:disconnected", payload);
           return;
         }
         if (msg.type === StudioMsg.RESPONSE) {
@@ -602,11 +689,27 @@ var init_ws_client = __esm({
         this.ws.send(serializeMessage(msg));
       }
       isConnected() {
-        return this.studioConnected;
+        return this.studioConnected && this.connections.size > 0;
       }
       getConnectionInfo() {
-        if (!this.connectionInfo) return null;
-        return { ...this.connectionInfo, ws: this.ws };
+        if (this.connections.size === 0) return null;
+        const target = this.activePlaceId !== 0 ? this.connections.get(this.activePlaceId) : this.connections.values().next().value;
+        if (!target) return null;
+        return { ...target, ws: this.ws };
+      }
+      listConnections() {
+        return [...this.connections.values()].map((c) => ({
+          placeId: c.placeId ?? 0,
+          placeName: c.placeName,
+          studioVersion: c.studioVersion,
+          connectedAt: c.connectedAt
+        }));
+      }
+      getActivePlaceId() {
+        return this.activePlaceId;
+      }
+      setActivePlaceId(placeId) {
+        this.activePlaceId = placeId;
       }
       getPort() {
         return this.port;
@@ -683,6 +786,7 @@ These are built-in lessons that ship with Dominus. Follow them exactly.
 - **Inspecting user's selection \u2192 \`get_selection\`, then \`get_properties\` or \`serialize_ui\`**
 - **Reading properties \u2192 \`get_properties\` (NEVER \`run_code\` with dump scripts)**
 - **Browsing the tree \u2192 \`get_explorer\`**
+- **Multiple Studios open \u2192 \`list_connections\` to see all, \`set_active_place\` to target one**
 - Anything exotic or procedural \u2192 \`run_code\` as last resort
 
 ### NEVER use run_code for:
@@ -2605,7 +2709,7 @@ async function startMcpServer() {
   }
   const mcp = new McpServer({
     name: "dominus",
-    version: "0.5.8"
+    version: "0.6.0"
   }, {
     instructions: INTERNAL_MEMORY
   });
@@ -2618,6 +2722,11 @@ async function startMcpServer() {
     for (const p of parts) r += `[${luaStr(p)}]`;
     return r;
   }
+  function cleanNum(n) {
+    if (Number.isInteger(n)) return String(n);
+    const rounded = Math.round(n * 1e3) / 1e3;
+    return String(rounded);
+  }
   function generateComponentCode(tree, componentName, framework) {
     const isReact = framework === "react";
     const lib = isReact ? "React" : "Roact";
@@ -2675,7 +2784,7 @@ async function startMcpServer() {
   function luaValue(val, _depth) {
     if (val === null || val === void 0) return "nil";
     if (typeof val === "boolean") return String(val);
-    if (typeof val === "number") return String(val);
+    if (typeof val === "number") return cleanNum(val);
     if (typeof val === "string") {
       if (val.startsWith("#") && val.length === 7) {
         return `Color3.fromHex("${val}")`;
@@ -2687,15 +2796,15 @@ async function startMcpServer() {
     }
     if (Array.isArray(val)) {
       if (val.length === 4 && val.every((v) => typeof v === "number")) {
-        return `UDim2.new(${val.join(", ")})`;
+        return `UDim2.new(${val.map(cleanNum).join(", ")})`;
       }
       if (val.length === 2 && val.every((v) => typeof v === "number")) {
-        return `Vector2.new(${val.join(", ")})`;
+        return `Vector2.new(${val.map(cleanNum).join(", ")})`;
       }
       if (val.length === 3 && val.every((v) => typeof v === "number")) {
         const allSmall = val.every((v) => v <= 1);
-        if (allSmall) return `Color3.new(${val.join(", ")})`;
-        return `Color3.fromRGB(${val.join(", ")})`;
+        if (allSmall) return `Color3.new(${val.map(cleanNum).join(", ")})`;
+        return `Color3.fromRGB(${val.map((v) => String(Math.round(v))).join(", ")})`;
       }
       return `{${val.map((v) => luaValue(v, _depth)).join(", ")}}`;
     }
@@ -3188,6 +3297,57 @@ ${"	".repeat(_depth)}}`;
       return { content: [{ type: "text", text: code }] };
     }
   );
+  mcp.tool(
+    "list_connections",
+    "List all Roblox Studio instances currently connected to Dominus. Shows placeId, placeName, and connection time for each. Use this when working with multiple Studio windows.",
+    {},
+    async () => {
+      const connections = bridge.listConnections();
+      const activePlaceId = bridge.getActivePlaceId();
+      return {
+        content: [{
+          type: "text",
+          text: JSON.stringify({
+            activePlaceId: activePlaceId || "auto",
+            connections: connections.map((c) => ({
+              ...c,
+              active: activePlaceId === 0 && connections.length === 1 ? true : c.placeId === activePlaceId
+            }))
+          }, null, 2)
+        }]
+      };
+    }
+  );
+  mcp.tool(
+    "set_active_place",
+    "Set which Roblox Studio instance to target when multiple are connected. Pass placeId=0 to reset to auto mode (works when only one is connected). Call list_connections first to see available placeIds.",
+    {
+      placeId: z.number().describe("The placeId to target. Use 0 to reset to auto-select.")
+    },
+    async ({ placeId }) => {
+      const connections = bridge.listConnections();
+      if (placeId !== 0 && !connections.some((c) => c.placeId === placeId)) {
+        const available = connections.map((c) => `${c.placeName ?? "Unknown"} (${c.placeId})`).join(", ");
+        return {
+          content: [{
+            type: "text",
+            text: `Place ${placeId} is not connected. Available: ${available || "none"}`
+          }]
+        };
+      }
+      bridge.setActivePlaceId(placeId);
+      if (placeId === 0) {
+        return { content: [{ type: "text", text: "Active place reset to auto-select." }] };
+      }
+      const target = connections.find((c) => c.placeId === placeId);
+      return {
+        content: [{
+          type: "text",
+          text: `Active place set to ${target?.placeName ?? "Unknown"} (${placeId})`
+        }]
+      };
+    }
+  );
   mcp.tool(
     "recall_memory",
     "Search persistent memory for facts about the current Roblox project",
@@ -3482,11 +3642,20 @@ var Setup = ({ onComplete }) => {
 var StatusBar = React6.memo(({
   connected,
   placeName,
+  connectionCount,
   model,
   sessionTokens
 }) => {
   const statusColor = connected ? "green" : "red";
-  const statusText = connected ? `\u25CF Connected${placeName ? ` \u2014 ${placeName}` : ""}` : "\u25CB Disconnected";
+  const count = connectionCount ?? (connected ? 1 : 0);
+  let statusText;
+  if (!connected) {
+    statusText = "\u25CB Disconnected";
+  } else if (count > 1) {
+    statusText = `\u25CF ${count} Studios${placeName ? ` \u2014 active: ${placeName}` : ""}`;
+  } else {
+    statusText = `\u25CF Connected${placeName ? ` \u2014 ${placeName}` : ""}`;
+  }
   const shortModel = model.split("/").pop() ?? model;
   return /* @__PURE__ */ jsxs(
     Box,
@@ -3904,6 +4073,7 @@ var ToolCallView = React6.memo(({ calls }) => {
 var Layout = React6.memo(({
   connected,
   placeName,
+  connectionCount,
   model,
   chatEntries,
   outputEntries,
@@ -3917,7 +4087,7 @@ var Layout = React6.memo(({
   onClearImage
 }) => {
   return /* @__PURE__ */ jsxs(Box, { flexDirection: "column", width: "100%", children: [
-    /* @__PURE__ */ jsx(StatusBar, { connected, placeName, model }),
+    /* @__PURE__ */ jsx(StatusBar, { connected, placeName, connectionCount, model }),
     /* @__PURE__ */ jsxs(Box, { flexGrow: 1, children: [
       /* @__PURE__ */ jsx(ExplorerPane, { tree: explorerTree }),
       /* @__PURE__ */ jsxs(Box, { flexDirection: "column", flexGrow: 1, children: [
@@ -3943,10 +4113,12 @@ var Layout = React6.memo(({
 init_protocol();
 function useStudio(server) {
   const connInfo = server.getConnectionInfo();
+  const connections = server.listConnections();
   const [state, setState] = useState({
     connected: server.isConnected(),
     placeName: connInfo?.placeName,
     placeId: connInfo?.placeId,
+    connectionCount: connections.length,
     explorerTree: [],
     outputLog: []
   });
@@ -3955,27 +4127,36 @@ function useStudio(server) {
   useEffect(() => {
     if (server.isConnected() && !stateRef.current.connected) {
       const info = server.getConnectionInfo();
+      const conns = server.listConnections();
       setState((prev) => ({
         ...prev,
         connected: true,
         placeName: info?.placeName,
-        placeId: info?.placeId
+        placeId: info?.placeId,
+        connectionCount: conns.length
       }));
     }
     const onConnected = (payload) => {
+      const conns = server.listConnections();
+      const info = server.getConnectionInfo();
       setState((prev) => ({
         ...prev,
         connected: true,
-        placeName: payload.placeName ?? prev.placeName,
-        placeId: payload.placeId ?? prev.placeId
+        placeName: info?.placeName ?? payload.placeName ?? prev.placeName,
+        placeId: info?.placeId ?? payload.placeId ?? prev.placeId,
+        connectionCount: conns.length
       }));
     };
     const onDisconnected = () => {
+      const conns = server.listConnections();
+      const info = server.getConnectionInfo();
+      const stillConnected = conns.length > 0;
       setState((prev) => ({
         ...prev,
-        connected: false,
-        placeName: void 0,
-        placeId: void 0
+        connected: stillConnected,
+        placeName: stillConnected ? info?.placeName : void 0,
+        placeId: stillConnected ? info?.placeId : void 0,
+        connectionCount: conns.length
       }));
     };
     const onExplorer = (payload) => {
@@ -4277,6 +4458,7 @@ var App = ({ server, agent: initialAgent, version }) => {
     {
       connected: studio.connected,
       placeName: studio.placeName,
+      connectionCount: studio.connectionCount,
       model: agent?.getModel() ?? resolveProviderSettings(config).model,
       chatEntries,
       outputEntries: studio.outputLog,