robloxstudio-mcp 2.6.0-next.2 → 2.6.0-next.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "robloxstudio-mcp",
-  "version": "2.6.0-next.2",
+  "version": "2.6.0-next.3",
   "description": "MCP Server for Roblox Studio Integration - Access Studio data, scripts, and objects through AI tools",
   "main": "dist/index.js",
   "type": "module",

package/studio-plugin/src/modules/Communication.ts

@@ -194,7 +194,7 @@ function pollForRequests(connIndex: number) {
 				el.troubleshootLabel.Visible = elapsed > 8;
 				if (elapsed > 3 && elapsed % 5 < conn.pollInterval) {
 					task.spawn(() => {
-						const discovered = discoverPort();
+						const discovered = discoverPort(getUsedPorts(connIndex));
 						if (discovered !== undefined && discovered !== conn.port) {
 							conn.port = discovered;
 							conn.serverUrl = `http://localhost:${discovered}`;
@@ -231,7 +231,7 @@ function pollForRequests(connIndex: number) {
 
 		if (conn.consecutiveFailures === 5 || conn.consecutiveFailures % 20 === 0) {
 			task.spawn(() => {
-				const discovered = discoverPort();
+				const discovered = discoverPort(getUsedPorts(connIndex));
 				if (discovered !== undefined && discovered !== conn.port) {
 					conn.port = discovered;
 					conn.serverUrl = `http://localhost:${discovered}`;
@@ -304,10 +304,33 @@ function pollForRequests(connIndex: number) {
 	}
 }
 
-function discoverPort(): number | undefined {
+function getUsedPorts(excludeIndex: number): Set<number> {
+	const ports = new Set<number>();
+	const conns = State.getConnections();
+	for (let i = 0; i < conns.size(); i++) {
+		if (i !== excludeIndex && conns[i].isActive) {
+			ports.add(conns[i].port);
+		}
+	}
+	return ports;
+}
+
+function checkPort(port: number): boolean {
+	const [ok, res] = pcall(() => {
+		return HttpService.RequestAsync({
+			Url: `http://localhost:${port}/status`,
+			Method: "GET",
+			Headers: { "Content-Type": "application/json" },
+		});
+	});
+	return ok && res.Success;
+}
+
+function discoverPort(excludePorts?: Set<number>): number | undefined {
 	let firstActivePort: number | undefined;
 	for (let offset = 0; offset < 5; offset++) {
 		const port = State.BASE_PORT + offset;
+		if (excludePorts && excludePorts.has(port)) continue;
 		const [success, result] = pcall(() => {
 			return HttpService.RequestAsync({
 				Url: `http://localhost:${port}/status`,
@@ -351,13 +374,16 @@ function activatePlugin(connIndex?: number) {
 	UI.updateTabDot(idx);
 
 	task.spawn(() => {
-		const discoveredPort = discoverPort();
-		if (discoveredPort !== undefined) {
-			conn.port = discoveredPort;
-			conn.serverUrl = `http://localhost:${discoveredPort}`;
-			UI.updateTabLabel(idx);
-			if (idx === State.getActiveTabIndex()) {
-				ui.urlInput.Text = conn.serverUrl;
+		if (!checkPort(conn.port)) {
+			const usedPorts = getUsedPorts(idx);
+			const discoveredPort = discoverPort(usedPorts);
+			if (discoveredPort !== undefined) {
+				conn.port = discoveredPort;
+				conn.serverUrl = `http://localhost:${discoveredPort}`;
+				UI.updateTabLabel(idx);
+				if (idx === State.getActiveTabIndex()) {
+					ui.urlInput.Text = conn.serverUrl;
+				}
 			}
 		}
 

package/studio-plugin/src/modules/State.ts

@@ -32,8 +32,14 @@ function addConnection(port?: number): number | undefined {
 	if (connections.size() >= MAX_CONNECTIONS) {
 		return undefined;
 	}
-	const lastPort = connections[connections.size() - 1].port;
-	const conn = createConnection(port ?? lastPort + 1);
+	if (port === undefined) {
+		let maxPort = BASE_PORT - 1;
+		for (const conn of connections) {
+			if (conn.port > maxPort) maxPort = conn.port;
+		}
+		port = maxPort + 1;
+	}
+	const conn = createConnection(port);
 	connections.push(conn);
 	return connections.size() - 1;
 }