@chrrxs/robloxstudio-mcp 2.14.0 → 2.15.1

package/studio-plugin/MCPInspectorPlugin.rbxmx

@@ -102,9 +102,11 @@ local RunService = _services.RunService
 local ServerStorage = _services.ServerStorage
 local RuntimeLogBuffer = TS.import(script, script.Parent, "RuntimeLogBuffer")
 local MemoryHandlers = TS.import(script, script.Parent, "handlers", "MemoryHandlers")
+local SceneAnalysisHandlers = TS.import(script, script.Parent, "handlers", "SceneAnalysisHandlers")
 local CaptureHandlers = TS.import(script, script.Parent, "handlers", "CaptureHandlers")
 local InputHandlers = TS.import(script, script.Parent, "handlers", "InputHandlers")
 local LuauExec = TS.import(script, script.Parent, "LuauExec")
+local State = TS.import(script, script.Parent, "State")
 local StudioTestService = game:GetService("StudioTestService")
 -- Mirror of Communication.computeInstanceId() — duplicated here because the
 -- client broker runs in the play-server DM where it can't easily import from
@@ -167,6 +169,7 @@ local CLIENT_BROKER_ALLOWED_ENDPOINTS = {
 	["/api/execute-luau"] = true,
 	["/api/get-runtime-logs"] = true,
 	["/api/get-memory-breakdown"] = true,
+	["/api/get-scene-analysis"] = true,
 	["/api/multiplayer-test-state"] = true,
 	["/api/multiplayer-test-leave-client"] = true,
 	["/api/capture-begin"] = true,
@@ -199,6 +202,8 @@ local function reRegisterProxy(proxyId, role)
 			placeName = resolvePlaceName(),
 			dataModelName = game.Name,
 			isRunning = RunService:IsRunning(),
+			pluginVersion = State.CURRENT_VERSION,
+			pluginVariant = State.PLUGIN_VARIANT,
 		})
 	end)
 end
@@ -324,7 +329,7 @@ end
 local function setupClientBroker()
 	local rf = ReplicatedStorage:WaitForChild(BROKER_NAME, 10)
 	if not rf or not rf:IsA("RemoteFunction") then
-		warn(`[MCPFork] client: {BROKER_NAME} not found`)
+		warn(`[robloxstudio-mcp] client: {BROKER_NAME} not found`)
 		return nil
 	end
 	rf.OnClientInvoke = function(payload)
@@ -340,6 +345,9 @@ local function setupClientBroker()
 		if payload and payload.endpoint == "/api/get-memory-breakdown" then
 			return MemoryHandlers.getMemoryBreakdown(payload.data or {})
 		end
+		if payload and payload.endpoint == "/api/get-scene-analysis" then
+			return SceneAnalysisHandlers.getSceneAnalysis(payload.data or {})
+		end
 		if payload and payload.endpoint == "/api/multiplayer-test-state" then
 			return handleMultiplayerTestState()
 		end
@@ -451,9 +459,11 @@ local function registerProxy(player, rf)
 		placeName = resolvePlaceName(),
 		dataModelName = game.Name,
 		isRunning = RunService:IsRunning(),
+		pluginVersion = State.CURRENT_VERSION,
+		pluginVariant = State.PLUGIN_VARIANT,
 	})
 	if not ok or not res or not res.Success then
-		warn(`[MCPFork] proxy register failed for {player.Name}`)
+		warn(`[robloxstudio-mcp] proxy register failed for {player.Name}`)
 		return nil
 	end
 	local body = HttpService:JSONDecode(res.Body)
@@ -551,6 +561,7 @@ local InputHandlers = TS.import(script, script.Parent, "handlers", "InputHandler
 local LogHandlers = TS.import(script, script.Parent, "handlers", "LogHandlers")
 local SerializationHandlers = TS.import(script, script.Parent, "handlers", "SerializationHandlers")
 local MemoryHandlers = TS.import(script, script.Parent, "handlers", "MemoryHandlers")
+local SceneAnalysisHandlers = TS.import(script, script.Parent, "handlers", "SceneAnalysisHandlers")
 -- Per-plugin-load random GUID. Used as the /poll URL param so the server
 -- can tell our polls apart from any other plugin's polls. Not user-facing —
 -- MCP tools and the LLM operate on instanceId (the place identifier).
@@ -578,6 +589,8 @@ end
 local instanceId = computeInstanceId()
 local assignedRole
 local duplicateInstanceRole = false
+local hasVersionMismatch = false
+local lastVersionMismatchWarningKey
 -- Cache the published place name from MarketplaceService:GetProductInfo so
 -- /ready can carry a friendly identifier (e.g. "Natural Disasters") distinct
 -- from game.Name (the DataModel name, often "Place1" in edit). We only fetch
@@ -683,6 +696,7 @@ local routeMap = {
 	["/api/export-rbxm"] = SerializationHandlers.exportRbxm,
 	["/api/import-rbxm"] = SerializationHandlers.importRbxm,
 	["/api/get-memory-breakdown"] = MemoryHandlers.getMemoryBreakdown,
+	["/api/get-scene-analysis"] = SceneAnalysisHandlers.getSceneAnalysis,
 }
 local function processRequest(request)
 	local endpoint = request.endpoint
@@ -776,6 +790,8 @@ function sendReady(conn)
 					placeName = resolvePlaceName(),
 					dataModelName = game.Name,
 					isRunning = RunService:IsRunning(),
+					pluginVersion = State.CURRENT_VERSION,
+					pluginVariant = State.PLUGIN_VARIANT,
 					pluginReady = true,
 					timestamp = tick(),
 				}),
@@ -841,6 +857,23 @@ local function pollForRequests(connIndex)
 		local mcpConnected = data.mcpConnected == true
 		conn.lastHttpOk = true
 		conn.lastMcpOk = mcpConnected
+		local _condition = data.serverVersion
+		if _condition == nil then
+			_condition = "unknown"
+		end
+		local serverVersion = _condition
+		if data.versionMismatch == true then
+			hasVersionMismatch = true
+			local warningKey = `{State.CURRENT_VERSION}:{serverVersion}`
+			if lastVersionMismatchWarningKey ~= warningKey then
+				lastVersionMismatchWarningKey = warningKey
+				warn(`[MCPPlugin] Version mismatch: Studio plugin v{State.CURRENT_VERSION} / MCP v{serverVersion}. Run npx -y @chrrxs/robloxstudio-mcp@latest --auto-install-plugin and restart Studio.`)
+			end
+			UI.showBanner("version-mismatch", `Plugin v{State.CURRENT_VERSION} / MCP v{serverVersion} mismatch`)
+		elseif hasVersionMismatch then
+			hasVersionMismatch = false
+			UI.hideBanner("version-mismatch")
+		end
 		-- Server tells us when its in-memory instances map doesn't have us
 		-- (e.g. after an MCP process restart). Re-issue /ready immediately so
 		-- target=server/client-N start routing again. The throttle inside
@@ -853,12 +886,12 @@ local function pollForRequests(connIndex)
 			local el = ui
 			el.step1Dot.BackgroundColor3 = Color3.fromRGB(34, 197, 94)
 			el.step1Label.Text = "HTTP server (OK)"
-			local _condition = mcpConnected
-			if _condition then
+			local _condition_1 = mcpConnected
+			if _condition_1 then
 				local _value = (string.find(el.statusLabel.Text, "Connected"))
-				_condition = not (_value ~= 0 and _value == _value and _value)
+				_condition_1 = not (_value ~= 0 and _value == _value and _value)
 			end
-			if _condition then
+			if _condition_1 then
 				el.statusLabel.Text = "Connected"
 				el.statusLabel.TextColor3 = Color3.fromRGB(34, 197, 94)
 				el.statusIndicator.BackgroundColor3 = Color3.fromRGB(34, 197, 94)
@@ -889,11 +922,11 @@ local function pollForRequests(connIndex)
 					conn.mcpWaitStartTime = tick()
 				end
 				local _exp = tick()
-				local _condition_1 = conn.mcpWaitStartTime
-				if _condition_1 == nil then
-					_condition_1 = tick()
+				local _condition_2 = conn.mcpWaitStartTime
+				if _condition_2 == nil then
+					_condition_2 = tick()
 				end
-				local elapsed = _exp - _condition_1
+				local elapsed = _exp - _condition_2
 				el.troubleshootLabel.Visible = elapsed > 8
 				UI.startPulseAnimation()
 			end
@@ -1102,11 +1135,9 @@ local function checkForUpdates()
 			if _condition ~= "" and _condition then
 				local latestVersion = data.version
 				if Utils.compareVersions(State.CURRENT_VERSION, latestVersion) < 0 then
-					local ui = UI.getElements()
-					ui.updateBannerText.Text = `v{latestVersion} available - github.com/chrrxs/robloxstudio-mcp`
-					ui.updateBanner.Visible = true
-					ui.contentFrame.Position = UDim2.new(0, 8, 0, 92)
-					ui.contentFrame.Size = UDim2.new(1, -16, 1, -100)
+					if not hasVersionMismatch then
+						UI.showBanner("update", `v{latestVersion} available - github.com/chrrxs/robloxstudio-mcp`)
+					end
 				end
 			end
 		end
@@ -1256,9 +1287,9 @@ local function computeBridgeStamp()
 	for i = 1, #combined do
 		h = (h * 33 + (string.byte(combined, i))) % 2147483647
 	end
-	-- "2.14.0" is replaced with the package version at package time
+	-- "2.15.1" is replaced with the package version at package time
 	-- (scripts/build-plugin.mjs injectVersion), so a release bump also restamps.
-	return `{tostring(h)}-2.14.0`
+	return `{tostring(h)}-2.15.1`
 end
 local BRIDGE_STAMP = computeBridgeStamp()
 local function setSource(scriptInst, source)
@@ -5070,6 +5101,255 @@ return {
           </Properties>
         </Item>
         <Item class="ModuleScript" referent="16">
+          <Properties>
+            <string name="Name">SceneAnalysisHandlers</string>
+            <string name="Source"><![CDATA[-- Compiled with roblox-ts v3.0.0
+local MODE_CONFIGS = {
+	instance_composition = {
+		method = "GetInstanceCompositionAsync",
+		query = function(service)
+			return service:GetInstanceCompositionAsync()
+		end,
+	},
+	script_memory = {
+		method = "GetScriptMemoryAsync",
+		query = function(service)
+			return service:GetScriptMemoryAsync()
+		end,
+	},
+	unparented_instances = {
+		method = "GetUnparentedInstancesAsync",
+		query = function(service)
+			return service:GetUnparentedInstancesAsync()
+		end,
+	},
+	triangle_composition = {
+		method = "GetTriangleCompositionAsync",
+		query = function(service)
+			return service:GetTriangleCompositionAsync()
+		end,
+		sortByTriangles = true,
+	},
+	animation_memory = {
+		method = "GetAnimationMemoryAsync",
+		query = function(service)
+			return service:GetAnimationMemoryAsync()
+		end,
+	},
+	audio_memory = {
+		method = "GetAudioMemoryAsync",
+		query = function(service)
+			return service:GetAudioMemoryAsync()
+		end,
+	},
+}
+local ALL_MODES = { "instance_composition", "script_memory", "unparented_instances", "triangle_composition", "animation_memory", "audio_memory" }
+local function betaDisabledError()
+	return {
+		error = "scene_analysis_not_enabled",
+		message = "SceneAnalysisService is not enabled. Enable Scene Analysis in Studio Beta Features and restart Studio.",
+		betaFeatureRequired = true,
+	}
+end
+local function isBetaDisabledError(value)
+	local _value = value
+	local _condition = type(_value) == "string"
+	if _condition then
+		_condition = (string.find(value, "SceneAnalysisService is not enabled", 1, true)) ~= nil
+	end
+	return _condition
+end
+local function getSceneAnalysisService()
+	local provider = game
+	local ok, service = pcall(function()
+		return provider:GetService("SceneAnalysisService")
+	end)
+	if not ok or not service then
+		return {
+			error = "scene_analysis_unavailable",
+			message = `SceneAnalysisService is unavailable: {tostring(service)}`,
+		}
+	end
+	return service
+end
+local function normalizeMode(mode)
+	if mode == nil or mode == "all" then
+		return "all"
+	end
+	local _mode = mode
+	local _condition = not (type(_mode) == "string")
+	if not _condition then
+		_condition = MODE_CONFIGS[mode] == nil
+	end
+	if _condition then
+		return {
+			error = "invalid_mode",
+			message = `mode must be one of: all, {table.concat(ALL_MODES, ", ")}`,
+		}
+	end
+	return mode
+end
+local function normalizeTopN(topN)
+	local _topN = topN
+	if not (type(_topN) == "number") then
+		return 10
+	end
+	return math.clamp(math.floor(topN), 1, 100)
+end
+local function countLeaves(node)
+	local children = node.Children
+	if children and #children > 0 then
+		local total = 0
+		for _, child in children do
+			total += countLeaves(child)
+		end
+		return total
+	end
+	return 1
+end
+local function flattenLeaves(node, out)
+	local children = node.Children
+	if children and #children > 0 then
+		for _, child in children do
+			flattenLeaves(child, out)
+		end
+		return nil
+	end
+	local _out = out
+	local _node = node
+	table.insert(_out, _node)
+end
+local function compactEntry(node)
+	local entry = {
+		name = node.Name,
+	}
+	if node.Size ~= nil then
+		entry.size = node.Size
+	end
+	if node.Sizes ~= nil then
+		entry.sizes = node.Sizes
+	end
+	if node.AssetId ~= nil then
+		entry.asset_id = node.AssetId
+	end
+	return entry
+end
+local function compactRoot(node, leafCount)
+	local children = node.Children
+	local root = {
+		name = node.Name,
+		child_count = if children then #children else 0,
+		leaf_count = leafCount,
+	}
+	if node.Size ~= nil then
+		root.size = node.Size
+	end
+	if node.Sizes ~= nil then
+		root.sizes = node.Sizes
+	end
+	return root
+end
+local function metric(node, sortByTriangles)
+	if sortByTriangles then
+		local sizes = node.Sizes
+		local triangles = if sizes then sizes.Triangles else nil
+		local _condition = triangles
+		if _condition == nil then
+			_condition = 0
+		end
+		return _condition
+	end
+	local _condition = node.Size
+	if _condition == nil then
+		_condition = 0
+	end
+	return _condition
+end
+local function summarizeMode(mode, config, service, topN, raw)
+	local started = os.clock()
+	local ok, result = pcall(function()
+		return config.query(service)
+	end)
+	local elapsedMs = math.floor((os.clock() - started) * 1000)
+	if not ok then
+		if isBetaDisabledError(result) then
+			return betaDisabledError()
+		end
+		return {
+			error = "scene_analysis_query_failed",
+			mode = mode,
+			method = config.method,
+			message = tostring(result),
+		}
+	end
+	local tree = result
+	local leaves = {}
+	flattenLeaves(tree, leaves)
+	table.sort(leaves, function(a, b)
+		return metric(a, config.sortByTriangles == true) > metric(b, config.sortByTriangles == true)
+	end)
+	local top = {}
+	do
+		local i = 0
+		local _shouldIncrement = false
+		while true do
+			if _shouldIncrement then
+				i += 1
+			else
+				_shouldIncrement = true
+			end
+			if not (i < math.min(topN, #leaves)) then
+				break
+			end
+			local _arg0 = compactEntry(leaves[i + 1])
+			table.insert(top, _arg0)
+		end
+	end
+	local body = {
+		mode = mode,
+		method = config.method,
+		elapsed_ms = elapsedMs,
+		root = compactRoot(tree, #leaves),
+		top = top,
+	}
+	if raw then
+		body.tree = tree
+	end
+	return body
+end
+local function getSceneAnalysis(requestData)
+	local mode = normalizeMode(requestData.mode)
+	if not (type(mode) == "string") then
+		return mode
+	end
+	local serviceOrError = getSceneAnalysisService()
+	local _value = serviceOrError.IsA
+	if not (_value ~= 0 and _value == _value and _value ~= "" and _value) then
+		return serviceOrError
+	end
+	local service = serviceOrError
+	local topN = normalizeTopN(requestData.topN)
+	local raw = requestData.raw == true
+	if mode ~= "all" then
+		return summarizeMode(mode, MODE_CONFIGS[mode], service, topN, raw)
+	end
+	local body = {}
+	for _, m in ALL_MODES do
+		local result = summarizeMode(m, MODE_CONFIGS[m], service, topN, raw)
+		if result.error == "scene_analysis_not_enabled" then
+			return result
+		end
+		body[m] = result
+	end
+	return body
+end
+return {
+	getSceneAnalysis = getSceneAnalysis,
+}
+]]></string>
+          </Properties>
+        </Item>
+        <Item class="ModuleScript" referent="17">
           <Properties>
             <string name="Name">ScriptHandlers</string>
             <string name="Source"><![CDATA[-- Compiled with roblox-ts v3.0.0
@@ -5765,7 +6045,7 @@ return {
 ]]></string>
           </Properties>
         </Item>
-        <Item class="ModuleScript" referent="17">
+        <Item class="ModuleScript" referent="18">
           <Properties>
             <string name="Name">SerializationHandlers</string>
             <string name="Source"><![CDATA[-- Compiled with roblox-ts v3.0.0
@@ -5951,7 +6231,7 @@ return {
 ]]></string>
           </Properties>
         </Item>
-        <Item class="ModuleScript" referent="18">
+        <Item class="ModuleScript" referent="19">
           <Properties>
             <string name="Name">TestHandlers</string>
             <string name="Source"><![CDATA[-- Compiled with roblox-ts v3.0.0
@@ -6584,7 +6864,7 @@ return {
           </Properties>
         </Item>
       </Item>
-      <Item class="ModuleScript" referent="19">
+      <Item class="ModuleScript" referent="20">
         <Properties>
           <string name="Name">LuauExec</string>
           <string name="Source"><![CDATA[-- Compiled with roblox-ts v3.0.0
@@ -6902,7 +7182,7 @@ return {
 ]]></string>
         </Properties>
       </Item>
-      <Item class="ModuleScript" referent="20">
+      <Item class="ModuleScript" referent="21">
         <Properties>
           <string name="Name">Recording</string>
           <string name="Source"><![CDATA[-- Compiled with roblox-ts v3.0.0
@@ -6932,7 +7212,7 @@ return {
 ]]></string>
         </Properties>
       </Item>
-      <Item class="ModuleScript" referent="21">
+      <Item class="ModuleScript" referent="22">
         <Properties>
           <string name="Name">RenderMonitor</string>
           <string name="Source"><![CDATA[-- Compiled with roblox-ts v3.0.0
@@ -7000,7 +7280,7 @@ return {
 ]]></string>
         </Properties>
       </Item>
-      <Item class="ModuleScript" referent="22">
+      <Item class="ModuleScript" referent="23">
         <Properties>
           <string name="Name">RuntimeLogBuffer</string>
           <string name="Source"><![CDATA[-- Compiled with roblox-ts v3.0.0
@@ -7181,11 +7461,12 @@ return {
 ]]></string>
         </Properties>
       </Item>
-      <Item class="ModuleScript" referent="23">
+      <Item class="ModuleScript" referent="24">
         <Properties>
           <string name="Name">State</string>
           <string name="Source"><![CDATA[-- Compiled with roblox-ts v3.0.0
-local CURRENT_VERSION = "2.14.0"
+local CURRENT_VERSION = "2.15.1"
+local PLUGIN_VARIANT = "inspector"
 local MAX_CONNECTIONS = 5
 local BASE_PORT = 58741
 local activeTabIndex = 0
@@ -7263,6 +7544,7 @@ local function getConnections()
 end
 return {
 	CURRENT_VERSION = CURRENT_VERSION,
+	PLUGIN_VARIANT = PLUGIN_VARIANT,
 	MAX_CONNECTIONS = MAX_CONNECTIONS,
 	BASE_PORT = BASE_PORT,
 	connections = connections,
@@ -7277,7 +7559,7 @@ return {
 ]]></string>
         </Properties>
       </Item>
-      <Item class="ModuleScript" referent="24">
+      <Item class="ModuleScript" referent="25">
         <Properties>
           <string name="Name">StopPlayMonitor</string>
           <string name="Source"><![CDATA[-- Compiled with roblox-ts v3.0.0
@@ -7422,7 +7704,7 @@ return {
 ]]></string>
         </Properties>
       </Item>
-      <Item class="ModuleScript" referent="25">
+      <Item class="ModuleScript" referent="26">
         <Properties>
           <string name="Name">UI</string>
           <string name="Source"><![CDATA[-- Compiled with roblox-ts v3.0.0
@@ -7435,6 +7717,7 @@ local buttonHover = false
 local toolbarButton
 local toolbarIcons
 local lastToolbarIcon
+local activeBannerKind
 local updateToolbarIcon
 local function setToolbarButton(btn, icons)
 	toolbarButton = btn
@@ -7465,6 +7748,23 @@ local TWEEN_QUICK = TweenInfo.new(0.15, Enum.EasingStyle.Quad, Enum.EasingDirect
 local function tweenProp(instance, props)
 	TweenService:Create(instance, TWEEN_QUICK, props):Play()
 end
+local function showBanner(kind, text)
+	activeBannerKind = kind
+	elements.updateBannerText.Text = text
+	elements.updateBanner.Visible = true
+	elements.contentFrame.Position = UDim2.new(0, 8, 0, 92)
+	elements.contentFrame.Size = UDim2.new(1, -16, 1, -100)
+end
+local function hideBanner(kind)
+	if kind ~= nil and activeBannerKind ~= kind then
+		return nil
+	end
+	activeBannerKind = nil
+	elements.updateBanner.Visible = false
+	elements.updateBannerText.Text = ""
+	elements.contentFrame.Position = UDim2.new(0, 8, 0, 66)
+	elements.contentFrame.Size = UDim2.new(1, -16, 1, -74)
+end
 local C = {
 	bg = Color3.fromRGB(14, 14, 14),
 	card = Color3.fromRGB(22, 22, 22),
@@ -8166,6 +8466,8 @@ return {
 	startPulseAnimation = startPulseAnimation,
 	setToolbarButton = setToolbarButton,
 	updateToolbarIcon = updateToolbarIcon,
+	showBanner = showBanner,
+	hideBanner = hideBanner,
 	getElements = function()
 		return elements
 	end,
@@ -8173,7 +8475,7 @@ return {
 ]]></string>
         </Properties>
       </Item>
-      <Item class="ModuleScript" referent="26">
+      <Item class="ModuleScript" referent="27">
         <Properties>
           <string name="Name">Utils</string>
           <string name="Source"><![CDATA[-- Compiled with roblox-ts v3.0.0
@@ -8703,11 +9005,11 @@ return {
         </Properties>
       </Item>
     </Item>
-    <Item class="Folder" referent="30">
+    <Item class="Folder" referent="31">
       <Properties>
         <string name="Name">include</string>
       </Properties>
-      <Item class="ModuleScript" referent="27">
+      <Item class="ModuleScript" referent="28">
         <Properties>
           <string name="Name">Promise</string>
           <string name="Source"><![CDATA[--[[
@@ -10781,7 +11083,7 @@ return Promise
 ]]></string>
         </Properties>
       </Item>
-      <Item class="ModuleScript" referent="28">
+      <Item class="ModuleScript" referent="29">
         <Properties>
           <string name="Name">RuntimeLib</string>
           <string name="Source"><![CDATA[local Promise = require(script.Parent.Promise)
@@ -11048,15 +11350,15 @@ return TS
         </Properties>
       </Item>
     </Item>
-    <Item class="Folder" referent="31">
+    <Item class="Folder" referent="32">
       <Properties>
         <string name="Name">node_modules</string>
       </Properties>
-      <Item class="Folder" referent="32">
+      <Item class="Folder" referent="33">
         <Properties>
           <string name="Name">@rbxts</string>
         </Properties>
-      <Item class="ModuleScript" referent="29">
+      <Item class="ModuleScript" referent="30">
         <Properties>
           <string name="Name">services</string>
           <string name="Source"><![CDATA[return setmetatable({}, {