robloxstudio-mcp 2.2.7 → 2.2.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "robloxstudio-mcp",
-  "version": "2.2.7",
+  "version": "2.2.8",
   "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/MCPPlugin.rbxmx

@@ -3027,11 +3027,47 @@ return {
 local TS = require(script.Parent.Parent.Parent.include.RuntimeLib)
 local LogService = TS.import(script, script.Parent.Parent.Parent, "node_modules", "@rbxts", "services").LogService
 local StudioTestService = game:GetService("StudioTestService")
+local ServerScriptService = game:GetService("ServerScriptService")
+local ScriptEditorService = game:GetService("ScriptEditorService")
+local STOP_SIGNAL = "__MCP_STOP__"
 local testRunning = false
 local outputBuffer = {}
 local logConnection
 local testResult
 local testError
+local stopListenerScript
+local function buildStopListenerSource()
+	return `local LogService = game:GetService("LogService")\
+local StudioTestService = game:GetService("StudioTestService")\
+LogService.MessageOut:Connect(function(message)\
+	if message == "{STOP_SIGNAL}" then\
+		pcall(function() StudioTestService:EndTest("stopped_by_mcp") end)\
+	end\
+end)`
+end
+local function injectStopListener()
+	local listener = Instance.new("Script")
+	listener.Name = "__MCP_StopListener"
+	listener.Parent = ServerScriptService
+	local source = buildStopListenerSource()
+	local seOk = pcall(function()
+		ScriptEditorService:UpdateSourceAsync(listener, function()
+			return source
+		end)
+	end)
+	if not seOk then
+		listener.Source = source
+	end
+	stopListenerScript = listener
+end
+local function cleanupStopListener()
+	if stopListenerScript then
+		pcall(function()
+			return stopListenerScript:Destroy()
+		end)
+		stopListenerScript = nil
+	end
+end
 local function startPlaytest(requestData)
 	local mode = requestData.mode
 	if mode ~= "play" and mode ~= "run" then
@@ -3048,7 +3084,11 @@ local function startPlaytest(requestData)
 	outputBuffer = {}
 	testResult = nil
 	testError = nil
+	cleanupStopListener()
 	logConnection = LogService.MessageOut:Connect(function(message, messageType)
+		if message == STOP_SIGNAL then
+			return nil
+		end
 		local _outputBuffer = outputBuffer
 		local _arg0 = {
 			message = message,
@@ -3057,6 +3097,12 @@ local function startPlaytest(requestData)
 		}
 		table.insert(_outputBuffer, _arg0)
 	end)
+	local injected, injErr = pcall(function()
+		return injectStopListener()
+	end)
+	if not injected then
+		warn(`[MCP] Failed to inject stop listener: {injErr}`)
+	end
 	task.spawn(function()
 		local ok, result = pcall(function()
 			if mode == "play" then
@@ -3074,6 +3120,7 @@ local function startPlaytest(requestData)
 			logConnection = nil
 		end
 		testRunning = false
+		cleanupStopListener()
 	end)
 	return {
 		success = true,
@@ -3086,14 +3133,7 @@ local function stopPlaytest(_requestData)
 			error = "No test is currently running",
 		}
 	end
-	local ok, err = pcall(function()
-		StudioTestService:EndTest("stopped_by_mcp")
-	end)
-	if not ok then
-		return {
-			error = `EndTest failed: {err}`,
-		}
-	end
+	warn(STOP_SIGNAL)
 	local _object = {
 		success = true,
 	}
@@ -3103,7 +3143,7 @@ local function stopPlaytest(_requestData)
 	table.move(outputBuffer, 1, #outputBuffer, _length + 1, _array)
 	_object[_left] = _array
 	_object.outputCount = #outputBuffer
-	_object.message = "Playtest stopped."
+	_object.message = "Playtest stop signal sent."
 	return _object
 end
 local function getPlaytestOutput(_requestData)
@@ -3133,7 +3173,7 @@ return {
         <Properties>
           <string name="Name">State</string>
           <string name="Source"><![CDATA[-- Compiled with roblox-ts v3.0.0
-local CURRENT_VERSION = "2.2.7"
+local CURRENT_VERSION = "2.2.8"
 local MAX_CONNECTIONS = 5
 local BASE_PORT = 58741
 local activeTabIndex = 0

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

@@ -1,6 +1,10 @@
 import { LogService } from "@rbxts/services";
 
 const StudioTestService = game.GetService("StudioTestService");
+const ServerScriptService = game.GetService("ServerScriptService");
+const ScriptEditorService = game.GetService("ScriptEditorService");
+
+const STOP_SIGNAL = "__MCP_STOP__";
 
 interface OutputEntry {
 	message: string;
@@ -13,6 +17,40 @@ let outputBuffer: OutputEntry[] = [];
 let logConnection: RBXScriptConnection | undefined;
 let testResult: unknown;
 let testError: string | undefined;
+let stopListenerScript: Script | undefined;
+
+function buildStopListenerSource(): string {
+	return `local LogService = game:GetService("LogService")
+local StudioTestService = game:GetService("StudioTestService")
+LogService.MessageOut:Connect(function(message)
+	if message == "${STOP_SIGNAL}" then
+		pcall(function() StudioTestService:EndTest("stopped_by_mcp") end)
+	end
+end)`;
+}
+
+function injectStopListener() {
+	const listener = new Instance("Script");
+	listener.Name = "__MCP_StopListener";
+	listener.Parent = ServerScriptService;
+
+	const source = buildStopListenerSource();
+	const [seOk] = pcall(() => {
+		ScriptEditorService.UpdateSourceAsync(listener, () => source);
+	});
+	if (!seOk) {
+		(listener as unknown as { Source: string }).Source = source;
+	}
+
+	stopListenerScript = listener;
+}
+
+function cleanupStopListener() {
+	if (stopListenerScript) {
+		pcall(() => stopListenerScript!.Destroy());
+		stopListenerScript = undefined;
+	}
+}
 
 function startPlaytest(requestData: Record<string, unknown>) {
 	const mode = requestData.mode as string | undefined;
@@ -30,7 +68,10 @@ function startPlaytest(requestData: Record<string, unknown>) {
 	testResult = undefined;
 	testError = undefined;
 
+	cleanupStopListener();
+
 	logConnection = LogService.MessageOut.Connect((message, messageType) => {
+		if (message === STOP_SIGNAL) return;
 		outputBuffer.push({
 			message,
 			messageType: messageType.Name,
@@ -38,6 +79,11 @@ function startPlaytest(requestData: Record<string, unknown>) {
 		});
 	});
 
+	const [injected, injErr] = pcall(() => injectStopListener());
+	if (!injected) {
+		warn(`[MCP] Failed to inject stop listener: ${injErr}`);
+	}
+
 	task.spawn(() => {
 		const [ok, result] = pcall(() => {
 			if (mode === "play") {
@@ -57,6 +103,8 @@ function startPlaytest(requestData: Record<string, unknown>) {
 			logConnection = undefined;
 		}
 		testRunning = false;
+
+		cleanupStopListener();
 	});
 
 	return { success: true, message: `Playtest started in ${mode} mode` };
@@ -67,19 +115,13 @@ function stopPlaytest(_requestData: Record<string, unknown>) {
 		return { error: "No test is currently running" };
 	}
 
-	const [ok, err] = pcall(() => {
-		(StudioTestService as unknown as { EndTest(reason: string): void }).EndTest("stopped_by_mcp");
-	});
-
-	if (!ok) {
-		return { error: `EndTest failed: ${err}` };
-	}
+	warn(STOP_SIGNAL);
 
 	return {
 		success: true,
 		output: [...outputBuffer],
 		outputCount: outputBuffer.size(),
-		message: "Playtest stopped.",
+		message: "Playtest stop signal sent.",
 	};
 }