@chrrxs/robloxstudio-mcp 2.15.2 → 2.16.1

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

@@ -5,39 +5,62 @@
 // `plugin:SetSetting` / `plugin:GetSetting` is a per-plugin persistent store
 // shared across every DataModel the plugin runs in (edit DMs, play-server
 // DMs, play-client DMs). For each connected place we use a dedicated key
-// "MCP_STOP_PLAY_<instanceId>" as a single-bit mailbox:
+// "MCP_STOP_PLAY_<instanceId>" as a tiny request/result mailbox:
 //
-//   * The edit DM's stopPlaytest handler writes `true` into its own key
+//   * The edit DM's handler writes a tokenized stop request into its own key
 //     (computed from its placeId / ServerStorage anon UUID).
 //   * Each play-server DM's monitor loop polls the key matching its own
-//     instanceId at 0.1Hz; on `true` it clears the key and calls
-//     StudioTestService:EndTest. Play-server DMs for other places never
-//     touch this key.
-//   * The edit DM waits up to ~8s for its key to be cleared, confirming a
-//     matching play-server actually consumed the request.
+//     instanceId at 1Hz. On a fresh token, it calls StudioTestService:EndTest
+//     and writes a matching result token. Play-server DMs for other places
+//     never touch this key.
+//   * The edit DM waits up to ~8s for its result token, confirming a matching
+//     play-server actually consumed the request.
 //
 // Earlier versions used a single shared boolean flag, which let any
 // play-server DM in the same Studio process consume any place's stop
 // request — silently yanking teammates' playtests. The per-key scoping
 // below is the fix.
 
-import { HttpService, ServerStorage } from "@rbxts/services";
+import { HttpService, RunService, ServerStorage } from "@rbxts/services";
 
 const StudioTestService = game.GetService("StudioTestService");
 
 const SETTING_KEY_PREFIX = "MCP_STOP_PLAY_";
-// Monitor checks the key at this cadence. 0.1s keeps worst-case detection
-// lag tight so the consumption-confirmation window doesn't have to absorb
-// polling jitter on top of EndTest's teardown time.
-const POLL_INTERVAL_SEC = 0.1;
+// Keep this conservative. plugin:GetSetting is backed by Studio's plugin
+// settings store, and this monitor runs during every play session, including
+// manually-started Play. The official reference implementation polls at 1s.
+const POLL_INTERVAL_SEC = 1;
 // Total time we wait for the matching play-server DM to consume the
 // signal. Must cover: monitor detection (<= POLL_INTERVAL_SEC) +
 // StudioTestService:EndTest teardown (several seconds on heavier places).
-// 8s is comfortable; the tighter poll above keeps real cases well under.
+// 8s is intentionally shorter than the MCP request timeout but long enough
+// for the 1s monitor cadence plus ordinary Studio teardown latency.
 const WAIT_FOR_CONSUMPTION_TIMEOUT_SEC = 8.0;
 const WAIT_POLL_SEC = 0.1;
+const REQUEST_TTL_SEC = 12.0;
 
 let pluginRef: Plugin | undefined;
+let endTestIssued = false;
+
+interface StopPayload {
+	kind?: string;
+	id?: string;
+	requestedAt?: number;
+	consumedAt?: number;
+	ok?: boolean;
+	error?: string;
+}
+
+interface StopRequestResult {
+	ok: boolean;
+	requestId?: string;
+}
+
+interface StopConsumptionResult {
+	ok: boolean;
+	consumed: boolean;
+	error?: string;
+}
 
 function init(p: Plugin): void {
 	pluginRef = p;
@@ -65,53 +88,147 @@ function settingKey(instanceId: string): string {
 	return SETTING_KEY_PREFIX + instanceId;
 }
 
+function readSetting(key: string): unknown {
+	if (!pluginRef) return undefined;
+	const [ok, value] = pcall(() => pluginRef!.GetSetting(key));
+	return ok ? value : undefined;
+}
+
+function writeSetting(key: string, value: unknown): boolean {
+	if (!pluginRef) return false;
+	const [ok] = pcall(() => pluginRef!.SetSetting(key, value));
+	return ok;
+}
+
+function decodePayload(value: unknown): StopPayload | undefined {
+	let decoded = value;
+	if (typeIs(value, "string")) {
+		const [ok, result] = pcall(() => HttpService.JSONDecode(value as string));
+		if (!ok) return undefined;
+		decoded = result;
+	}
+	if (!typeIs(decoded, "table")) return undefined;
+	const payload = decoded as StopPayload;
+	if (!typeIs(payload.kind, "string") || !typeIs(payload.id, "string")) {
+		return undefined;
+	}
+	return payload;
+}
+
+function writePayload(key: string, payload: StopPayload): boolean {
+	const [encodedOk, encoded] = pcall(() => HttpService.JSONEncode(payload));
+	if (!encodedOk || !typeIs(encoded, "string")) return false;
+	return writeSetting(key, encoded);
+}
+
+function writeResult(key: string, request: StopPayload, ok: boolean, errText?: string): void {
+	writePayload(key, {
+		kind: "result",
+		id: request.id,
+		requestedAt: request.requestedAt,
+		consumedAt: tick(),
+		ok,
+		error: errText,
+	});
+}
+
+function handleStopRequest(key: string, request: StopPayload): void {
+	if (request.kind !== "request" || !typeIs(request.id, "string")) return;
+	if (!typeIs(request.requestedAt, "number")) {
+		writeSetting(key, false);
+		return;
+	}
+
+	const age = tick() - request.requestedAt;
+	if (age < -5 || age > REQUEST_TTL_SEC) {
+		writeSetting(key, false);
+		return;
+	}
+
+	if (endTestIssued) {
+		writeResult(key, request, true);
+		return;
+	}
+
+	if (!RunService.IsRunning() || !RunService.IsServer()) {
+		writeResult(key, request, false, "StopPlayMonitor is not running in the server DataModel.");
+		return;
+	}
+
+	endTestIssued = true;
+	const [endOk, endErr] = pcall(() => StudioTestService.EndTest("stopped_by_mcp"));
+	writeResult(key, request, endOk, endOk ? undefined : tostring(endErr));
+	if (!endOk) {
+		endTestIssued = false;
+	}
+}
+
 function startMonitor(): void {
 	if (!pluginRef) {
-		warn("[MCP] StopPlayMonitor.startMonitor called before init; skipping");
+		warn("[robloxstudio-mcp] StopPlayMonitor.startMonitor called before init; skipping");
 		return;
 	}
 	const myKey = settingKey(computeInstanceId());
-	// Clear any stale value left from a prior session. If a real stop
-	// request is in-flight when this runs, the requesting edit DM will
-	// write again within its consumption-confirmation window.
-	pcall(() => pluginRef!.SetSetting(myKey, false));
 	task.spawn(() => {
 		while (true) {
-			const [okGet, val] = pcall(() => pluginRef!.GetSetting(myKey));
-			if (okGet && val === true) {
-				// Consume the flag first so requestStop's
-				// waitForConsumption returns success, then end the test.
-				pcall(() => pluginRef!.SetSetting(myKey, false));
-				pcall(() => StudioTestService.EndTest("stopped_by_mcp"));
+			const value = readSetting(myKey);
+			if (value === true) {
+				// Legacy boolean requests are ambiguous and may be stale from
+				// a prior crashed session. New stop requests use token payloads.
+				writeSetting(myKey, false);
+			} else {
+				const payload = decodePayload(value);
+				if (payload) {
+					handleStopRequest(myKey, payload);
+				}
 			}
 			task.wait(POLL_INTERVAL_SEC);
 		}
 	});
 }
 
-function requestStop(): boolean {
-	if (!pluginRef) return false;
+function requestStop(): StopRequestResult {
+	if (!pluginRef) return { ok: false };
 	const myKey = settingKey(computeInstanceId());
-	const [ok] = pcall(() => pluginRef!.SetSetting(myKey, true));
-	return ok;
+	const requestId = HttpService.GenerateGUID(false);
+	const ok = writePayload(myKey, {
+		kind: "request",
+		id: requestId,
+		requestedAt: tick(),
+	});
+	return { ok, requestId: ok ? requestId : undefined };
 }
 
-function waitForConsumption(): boolean {
-	if (!pluginRef) return false;
+function waitForConsumption(requestId: string): StopConsumptionResult {
+	if (!pluginRef) return { ok: false, consumed: false, error: "Plugin reference is not initialized." };
 	const myKey = settingKey(computeInstanceId());
 	const start = tick();
 	while (tick() - start < WAIT_FOR_CONSUMPTION_TIMEOUT_SEC) {
-		const [okGet, val] = pcall(() => pluginRef!.GetSetting(myKey));
-		if (okGet && val !== true) return true;
+		const payload = decodePayload(readSetting(myKey));
+		if (payload && payload.kind === "result" && payload.id === requestId) {
+			return {
+				ok: payload.ok === true,
+				consumed: true,
+				error: payload.error,
+			};
+		}
 		task.wait(WAIT_POLL_SEC);
 	}
-	return false;
+	return {
+		ok: false,
+		consumed: false,
+		error: "Timed out waiting for the play-server DataModel to acknowledge stop_playtest.",
+	};
 }
 
-function clearPending(): void {
+function clearPending(requestId?: string): void {
 	if (!pluginRef) return;
 	const myKey = settingKey(computeInstanceId());
-	pcall(() => pluginRef!.SetSetting(myKey, false));
+	if (requestId !== undefined) {
+		const payload = decodePayload(readSetting(myKey));
+		if (payload && payload.id !== requestId) return;
+	}
+	writeSetting(myKey, false);
 }
 
 export = {

package/studio-plugin/src/modules/handlers/EvalRuntimeHandlers.ts

@@ -1,5 +1,5 @@
 import { LogService, ReplicatedStorage, RunService, ServerScriptService } from "@rbxts/services";
-import { BRIDGE_NAMES } from "../EvalBridges";
+import { BRIDGE_NAMES, ensureRuntimeBridgeInstalled } from "../EvalBridges";
 import LuauExec from "../LuauExec";
 
 const PAYLOAD_INSTANCE_NAME = "__MCPEvalPayload";
@@ -15,6 +15,21 @@ interface WrapperResult {
 	output?: unknown;
 }
 
+function findBridge(config: { service: Instance; bridgeName: string }): BindableFunction | undefined {
+	const bridge = config.service.FindFirstChild(config.bridgeName);
+	return bridge && bridge.IsA("BindableFunction") ? bridge : undefined;
+}
+
+function waitForBridge(config: { service: Instance; bridgeName: string }, timeoutSec = 2): BindableFunction | undefined {
+	const deadline = tick() + timeoutSec;
+	let bridge = findBridge(config);
+	while (!bridge && tick() < deadline) {
+		task.wait(0.05);
+		bridge = findBridge(config);
+	}
+	return bridge;
+}
+
 function getBridgeConfig() {
 	if (!RunService.IsRunning()) {
 		return {
@@ -25,13 +40,13 @@ function getBridgeConfig() {
 		return {
 			service: ServerScriptService,
 			bridgeName: BRIDGE_NAMES.serverLocal,
-			missingError: "ServerEvalBridge not found. The bridge runs inside the play DM, so a playtest must be running. The bridge installs automatically (including for manually-started playtests); if a playtest is running and you still see this, reconnect the plugin in the edit window so the bridge reinstalls, then start the playtest again.",
+			missingError: "ServerEvalBridge not found. The bridge runs inside the play DM, so a playtest must be running. The bridge installs automatically in the runtime server peer, including for manually-started playtests.",
 		};
 	}
 	return {
 		service: ReplicatedStorage,
 		bridgeName: BRIDGE_NAMES.clientLocal,
-		missingError: "ClientEvalBridge not found. The bridge runs inside the play DM, so a playtest must be running. The bridge installs automatically (including for manually-started playtests); if a playtest is running and you still see this, reconnect the plugin in the edit window so the bridge reinstalls, then start the playtest again.",
+		missingError: "ClientEvalBridge not found. The bridge runs inside the play DM, so a playtest must be running. The bridge installs automatically in the runtime client peer, including for manually-started playtests.",
 	};
 }
 
@@ -44,9 +59,22 @@ function evalRuntime(requestData: Record<string, unknown>) {
 		return { bridge: "missing", error: config.error };
 	}
 
-	const bridge = config.service.FindFirstChild(config.bridgeName);
-	if (!bridge || !bridge.IsA("BindableFunction")) {
-		return { bridge: "missing", error: config.missingError };
+	let bridge = findBridge(config);
+	if (!bridge) {
+		const install = ensureRuntimeBridgeInstalled();
+		if (!install.installed) {
+			return {
+				bridge: "missing",
+				error: `${config.missingError} Runtime bridge install failed: ${install.error}`,
+			};
+		}
+		bridge = waitForBridge(config);
+	}
+	if (!bridge) {
+		return {
+			bridge: "missing",
+			error: `${config.missingError} Runtime bridge was installed but did not become ready.`,
+		};
 	}
 
 	const m = new Instance("ModuleScript");

package/studio-plugin/src/modules/handlers/TestHandlers.ts

@@ -1,5 +1,4 @@
 import { HttpService, LogService, Players, RunService } from "@rbxts/services";
-import { installBridges, ensureBridgesInstalled } from "../EvalBridges";
 import StopPlayMonitor from "../StopPlayMonitor";
 
 interface StudioTestServiceMultiplayer extends StudioTestService {
@@ -200,9 +199,8 @@ function startPlaytest(requestData: Record<string, unknown>) {
 			logConnection = undefined;
 		}
 		cleanupStopListener();
-		// Note: eval bridges are intentionally NOT cleaned up — they live
-		// permanently in the edit DM so manual playtests also get them. See
-		// EvalBridges.ts lifecycle comment.
+		// Runtime eval bridges are created by the play server/client plugin
+		// peers and disappear with the play DataModels.
 	}
 
 	if (testRunning) {
@@ -233,16 +231,7 @@ function startPlaytest(requestData: Record<string, unknown>) {
 
 	const [injected, injErr] = pcall(() => injectStopListener());
 	if (!injected) {
-		warn(`[MCP] Failed to inject stop listener: ${injErr}`);
-	}
-
-	// Force-refresh the game-VM eval bridges (ServerEvalBridge + ClientEvalBridge)
-	// right before cloning so the play DMs get the current source. They also
-	// live permanently in the edit DM (installed on connect) so manually-started
-	// playtests get them too; here we just ensure they're fresh.
-	const bridgeInstall = installBridges();
-	if (!bridgeInstall.installed) {
-		warn(`[MCP] Eval bridge install failed: ${bridgeInstall.error}`);
+		warn(`[robloxstudio-mcp] Failed to inject stop listener: ${injErr}`);
 	}
 
 	task.spawn(() => {
@@ -266,36 +255,26 @@ function startPlaytest(requestData: Record<string, unknown>) {
 		testRunning = false;
 
 		cleanupStopListener();
-		// Eval bridges persist in the edit DM (see EvalBridges.ts) — do not
-		// clean up here, so the next manual playtest still gets them.
-		ensureBridgesInstalled();
 	});
 
 	const response: Record<string, unknown> = {
 		success: true,
 		message: `Playtest started in ${mode} mode.`,
 	};
-	// Only mention eval bridges when they failed — when they're fine, the
-	// detail is noise. eval_server_runtime / eval_client_runtime will surface
-	// their own clear errors if the caller tries to use them after a failed
-	// install.
-	if (!bridgeInstall.installed) {
-		response.evalBridgesError = bridgeInstall.error;
-	}
 
 	return response;
 }
 
 function stopPlaytest(_requestData: Record<string, unknown>) {
-	// Signal the play-server DM's StopPlayMonitor via plugin:SetSetting (a
-	// cross-DM persistent store). The monitor polls at 1Hz, sees the flag,
-	// calls StudioTestService:EndTest, then resets the flag. We wait up to
-	// 2.5s for the reset to confirm a play DM actually consumed the request,
-	// which avoids returning success when nothing is running.
-	if (!StopPlayMonitor.requestStop()) {
+	// Signal the play-server DM's StopPlayMonitor via plugin:SetSetting.
+	// The monitor acknowledges with the matching request id only after its
+	// StudioTestService:EndTest call returns from pcall.
+	const stopRequest = StopPlayMonitor.requestStop();
+	if (!stopRequest.ok || stopRequest.requestId === undefined) {
 		return { error: "Plugin not ready. Try again in a moment." };
 	}
-	if (!StopPlayMonitor.waitForConsumption()) {
+	const consumption = StopPlayMonitor.waitForConsumption(stopRequest.requestId);
+	if (!consumption.ok) {
 		// Two distinct failure modes collapse here, distinguished by whether
 		// THIS edit DM has a playtest tracked:
 		//
@@ -307,19 +286,24 @@ function stopPlaytest(_requestData: Record<string, unknown>) {
 		//   from the caller's perspective — playtest may actually have ended).
 		//   Tell the caller it's a timing issue and they can retry.
 		//
-		// Either way clean up the pending flag so a future playtest's monitor
+		// Either way clean up the pending request so a future playtest's monitor
 		// doesn't fire EndTest on startup against a stale signal.
-		StopPlayMonitor.clearPending();
+		StopPlayMonitor.clearPending(stopRequest.requestId);
 		if (testRunning) {
 			return {
 				error:
-					"Playtest stop signal sent but consumption confirmation timed out. " +
+					"Playtest stop signal failed or was not acknowledged. " +
 					"The playtest may have ended anyway; check get_connected_instances.",
+				detail: consumption.error,
 			};
 		}
-		return { error: "No active playtest to stop." };
+		if (consumption.consumed) {
+			return { error: "Playtest stop request reached the play server, but EndTest failed.", detail: consumption.error };
+		}
+		return { error: "No active playtest to stop.", detail: consumption.error };
 	}
-	// Flag was consumed (EndTest called). ExecutePlayModeAsync in our
+	StopPlayMonitor.clearPending(stopRequest.requestId);
+	// Request was consumed (EndTest called). ExecutePlayModeAsync in our
 	// startPlaytest task.spawn is still unwinding though — testRunning stays
 	// true until that yield completes and the post-block runs. Wait so
 	// back-to-back stop -> start sequences don't race against the prior
@@ -370,11 +354,6 @@ function multiplayerTestStart(requestData: Record<string, unknown>) {
 	const testArgs = requestData.testArgs !== undefined ? requestData.testArgs : {};
 	const testId = HttpService.GenerateGUID(false);
 
-	const bridgeInstall = installBridges();
-	if (!bridgeInstall.installed) {
-		warn(`[MCP] Eval bridge install failed: ${bridgeInstall.error}`);
-	}
-
 	multiplayerState = {
 		phase: "starting",
 		testId,
@@ -400,8 +379,6 @@ function multiplayerTestStart(requestData: Record<string, unknown>) {
 			multiplayerState.result = undefined;
 			multiplayerState.error = tostring(result);
 		}
-
-		ensureBridgesInstalled();
 	});
 
 	const response: Record<string, unknown> = {
@@ -412,9 +389,6 @@ function multiplayerTestStart(requestData: Record<string, unknown>) {
 		numPlayers,
 		testArgs,
 	};
-	if (!bridgeInstall.installed) {
-		response.evalBridgesError = bridgeInstall.error;
-	}
 	return response;
 }
 

package/studio-plugin/src/server/index.server.ts

@@ -2,6 +2,8 @@ import State from "../modules/State";
 import UI from "../modules/UI";
 import Communication from "../modules/Communication";
 import ClientBroker from "../modules/ClientBroker";
+import ServerUrlSettings from "../modules/ServerUrlSettings";
+import { cleanupLegacyEditBridges, ensureRuntimeBridgeInstalled } from "../modules/EvalBridges";
 import RuntimeLogBuffer from "../modules/RuntimeLogBuffer";
 import StopPlayMonitor from "../modules/StopPlayMonitor";
 import * as RenderMonitor from "../modules/RenderMonitor";
@@ -20,6 +22,7 @@ RuntimeLogBuffer.install();
 // edit DM (write the flag) and the play-server DM (read+act on the flag) can
 // access plugin:SetSetting/GetSetting.
 StopPlayMonitor.init(plugin);
+ServerUrlSettings.init(plugin);
 
 UI.init(plugin);
 const elements = UI.getElements();
@@ -28,10 +31,24 @@ const elements = UI.getElements();
 const ICON_DISCONNECTED = "rbxassetid://__BUTTON_ICON_DISCONNECTED__";
 const ICON_CONNECTING = "rbxassetid://__BUTTON_ICON_CONNECTING__";
 const ICON_CONNECTED = "rbxassetid://__BUTTON_ICON_CONNECTED__";
+const TOOLBAR_REGISTRATION_DELAY_SECONDS = 1;
 
-const toolbar = plugin.CreateToolbar("__TOOLBAR_NAME__");
-const button = toolbar.CreateButton("__BUTTON_TITLE__", "__BUTTON_TOOLTIP__", ICON_DISCONNECTED);
-UI.setToolbarButton(button, { disconnected: ICON_DISCONNECTED, connecting: ICON_CONNECTING, connected: ICON_CONNECTED });
+let toolbarButtonRegistered = false;
+
+function registerToolbarButton() {
+	if (toolbarButtonRegistered) {
+		return;
+	}
+	toolbarButtonRegistered = true;
+
+	const toolbar = plugin.CreateToolbar("__TOOLBAR_NAME__");
+	const button = toolbar.CreateButton("__BUTTON_TITLE__", "__BUTTON_TOOLTIP__", ICON_DISCONNECTED);
+	UI.setToolbarButton(button, { disconnected: ICON_DISCONNECTED, connecting: ICON_CONNECTING, connected: ICON_CONNECTED });
+
+	button.Click.Connect(() => {
+		elements.screenGui.Enabled = !elements.screenGui.Enabled;
+	});
+}
 
 
 elements.connectButton.Activated.Connect(() => {
@@ -44,11 +61,6 @@ elements.connectButton.Activated.Connect(() => {
 });
 
 
-button.Click.Connect(() => {
-	elements.screenGui.Enabled = !elements.screenGui.Enabled;
-});
-
-
 plugin.Unloading.Connect(() => {
 	Communication.deactivateAll();
 });
@@ -56,6 +68,7 @@ plugin.Unloading.Connect(() => {
 
 UI.updateUIState();
 Communication.checkForUpdates();
+task.delay(TOOLBAR_REGISTRATION_DELAY_SECONDS, registerToolbarButton);
 
 // Auto-activate per peer. The boshyxd plugin only registers with MCP when the
 // user clicks Connect in its UI, but that UI is invisible in play DMs - so
@@ -63,15 +76,32 @@ Communication.checkForUpdates();
 // short delay so the UI/State have a chance to initialize first.
 task.delay(2, () => {
 	const role = ClientBroker.forkRole();
+	if (role === "edit") {
+		cleanupLegacyEditBridges();
+	} else {
+		const result = ensureRuntimeBridgeInstalled();
+		if (!result.installed) {
+			warn(`[robloxstudio-mcp] Runtime eval bridge install failed: ${result.error}`);
+		}
+	}
 	if (role === "edit" || role === "server") {
 		pcall(() => {
 			const idx = State.getActiveTabIndex();
 			const conn = State.getConnection(idx);
 			if (conn && !conn.isActive) {
+				if (role === "server") {
+					const inheritedServerUrl = ServerUrlSettings.readServerUrl() ?? ClientBroker.DEFAULT_MCP_URL;
+					conn.serverUrl = inheritedServerUrl;
+					elements.urlInput.Text = inheritedServerUrl;
+					const [portStr] = conn.serverUrl.match(":(%d+)$");
+					if (portStr) conn.port = tonumber(portStr) ?? conn.port;
+					ClientBroker.setServerUrl(inheritedServerUrl);
+				}
 				// Defensive default: in invisible play-DM UIs, the input field
 				// may not be populated by the time we activate.
 				if (conn.serverUrl === undefined || conn.serverUrl === "") {
-					conn.serverUrl = ClientBroker.MCP_URL;
+					conn.serverUrl = ClientBroker.DEFAULT_MCP_URL;
+					elements.urlInput.Text = conn.serverUrl;
 				}
 				Communication.activatePlugin(idx);
 			}
@@ -80,8 +110,8 @@ task.delay(2, () => {
 	if (role === "server") {
 		ClientBroker.setupServerBroker();
 		// The play-server DM is the only one where StudioTestService:EndTest is
-		// legal, so the stop-play monitor lives here. Reads MCP_STOP_PLAY_SIGNAL
-		// at 1Hz and calls EndTest when the edit DM sets it.
+		// legal, so the stop-play monitor lives here. It consumes tokenized
+		// stop requests from plugin settings and acknowledges EndTest results.
 		StopPlayMonitor.startMonitor();
 	} else if (role === "client") {
 		ClientBroker.setupClientBroker();