dominus-cli 0.2.1 → 0.5.0

package/plugin/src/init.server.lua

@@ -7,11 +7,7 @@
 ]]
 
 local toolbar = plugin:CreateToolbar("Dominus")
-local connectButton = toolbar:CreateButton(
-	"Connect",
-	"Connect to Dominus CLI agent",
-	"rbxassetid://12974519994"
-)
+local connectButton = toolbar:CreateButton("Connect", "Connect to Dominus CLI agent", "rbxassetid://12974519994")
 
 local WsClient = require(script.WsClient)
 local Protocol = require(script.Protocol)
@@ -53,7 +49,6 @@ local function handleMessage(rawMessage)
 			payload = { tree = tree },
 			ts = os.clock(),
 		}))
-
 	elseif msgType == "cli:get:script" then
 		local result = Explorer.getScriptSource(payload.path)
 		WsClient.send(Protocol.encode({
@@ -62,7 +57,6 @@ local function handleMessage(rawMessage)
 			payload = result,
 			ts = os.clock(),
 		}))
-
 	elseif msgType == "cli:set:script" then
 		local result = Explorer.setScriptSource(payload.path, payload.source)
 		WsClient.send(Protocol.encode({
@@ -71,7 +65,6 @@ local function handleMessage(rawMessage)
 			payload = result,
 			ts = os.clock(),
 		}))
-
 	elseif msgType == "cli:run" then
 		local result = Executor.run(payload.code)
 		WsClient.send(Protocol.encode({
@@ -80,7 +73,6 @@ local function handleMessage(rawMessage)
 			payload = result,
 			ts = os.clock(),
 		}))
-
 	elseif msgType == "cli:get:properties" then
 		local result = Properties.get(payload.path)
 		WsClient.send(Protocol.encode({
@@ -89,7 +81,14 @@ local function handleMessage(rawMessage)
 			payload = result,
 			ts = os.clock(),
 		}))
-
+	elseif msgType == "cli:get:descendants:properties" then
+		local result = Properties.getDescendants(payload.path)
+		WsClient.send(Protocol.encode({
+			id = msgId,
+			type = "studio:response",
+			payload = result,
+			ts = os.clock(),
+		}))
 	elseif msgType == "cli:set:properties" then
 		local result = Properties.set(payload.path, payload.properties)
 		WsClient.send(Protocol.encode({
@@ -98,7 +97,6 @@ local function handleMessage(rawMessage)
 			payload = result,
 			ts = os.clock(),
 		}))
-
 	elseif msgType == "cli:insert" then
 		local result = Explorer.insertInstance(payload.className, payload.parentPath, payload.name, payload.properties)
 		WsClient.send(Protocol.encode({
@@ -107,7 +105,6 @@ local function handleMessage(rawMessage)
 			payload = result,
 			ts = os.clock(),
 		}))
-
 	elseif msgType == "cli:delete" then
 		local result = Explorer.deleteInstance(payload.path)
 		WsClient.send(Protocol.encode({
@@ -116,7 +113,6 @@ local function handleMessage(rawMessage)
 			payload = result,
 			ts = os.clock(),
 		}))
-
 	elseif msgType == "cli:get:selection" then
 		local Selection = game:GetService("Selection")
 		local selected = Selection:Get()
@@ -132,7 +128,6 @@ local function handleMessage(rawMessage)
 			payload = { paths = paths, classNames = classNames },
 			ts = os.clock(),
 		}))
-
 	elseif msgType == "cli:search:scripts" then
 		local result = Explorer.searchScripts(payload.query, payload.maxResults or 20)
 		WsClient.send(Protocol.encode({
@@ -141,7 +136,6 @@ local function handleMessage(rawMessage)
 			payload = result,
 			ts = os.clock(),
 		}))
-
 	elseif msgType == "cli:get:output" then
 		local result = Watcher.getRecentOutput(payload.limit or 50, payload.level)
 		WsClient.send(Protocol.encode({
@@ -150,7 +144,6 @@ local function handleMessage(rawMessage)
 			payload = result,
 			ts = os.clock(),
 		}))
-
 	elseif msgType == "cli:run:tests" then
 		local result = TestRunner.run(payload.testName)
 		WsClient.send(Protocol.encode({
@@ -159,7 +152,6 @@ local function handleMessage(rawMessage)
 			payload = result,
 			ts = os.clock(),
 		}))
-
 	elseif msgType == "cli:test:run" then
 		-- ExecuteRunModeAsync yields until EndTest is called, so run in a thread
 		task.spawn(function()
@@ -171,7 +163,6 @@ local function handleMessage(rawMessage)
 				ts = os.clock(),
 			}))
 		end)
-
 	elseif msgType == "cli:test:play" then
 		-- ExecutePlayModeAsync yields until EndTest is called, so run in a thread
 		task.spawn(function()
@@ -183,7 +174,6 @@ local function handleMessage(rawMessage)
 				ts = os.clock(),
 			}))
 		end)
-
 	elseif msgType == "cli:test:end" then
 		local result = TestRunner.endTest(payload.value)
 		WsClient.send(Protocol.encode({
@@ -192,7 +182,6 @@ local function handleMessage(rawMessage)
 			payload = result,
 			ts = os.clock(),
 		}))
-
 	elseif msgType == "cli:build:ui" then
 		local result = UIBuilder.build(payload)
 		WsClient.send(Protocol.encode({
@@ -201,7 +190,6 @@ local function handleMessage(rawMessage)
 			payload = result,
 			ts = os.clock(),
 		}))
-
 	elseif msgType == "cli:get:reflection" then
 		local result = Reflection.getClassInfo(payload.className)
 		WsClient.send(Protocol.encode({
@@ -210,7 +198,108 @@ local function handleMessage(rawMessage)
 			payload = result,
 			ts = os.clock(),
 		}))
-
+	elseif msgType == "cli:serialize:ui" then
+		local result = UIBuilder.serialize(payload)
+		WsClient.send(Protocol.encode({
+			id = msgId,
+			type = "studio:response",
+			payload = result,
+			ts = os.clock(),
+		}))
+	elseif msgType == "cli:move:instance" then
+		local instance = Explorer.resolveInstance(payload.path)
+		if not instance then
+			WsClient.send(Protocol.encode({
+				id = msgId,
+				type = "studio:response",
+				payload = { success = false, error = "Instance not found: " .. payload.path },
+				ts = os.clock(),
+			}))
+		else
+			local newParent = Explorer.resolveInstance(payload.newParent)
+			if not newParent then
+				local svcOk, svc = pcall(function()
+					return game:GetService(payload.newParent)
+				end)
+				if svcOk then
+					newParent = svc
+				end
+			end
+			if not newParent then
+				WsClient.send(Protocol.encode({
+					id = msgId,
+					type = "studio:response",
+					payload = { success = false, error = "New parent not found: " .. payload.newParent },
+					ts = os.clock(),
+				}))
+			else
+				local ChangeHistoryService = game:GetService("ChangeHistoryService")
+				local recording = ChangeHistoryService:TryBeginRecording("Dominus: Move " .. instance.Name)
+				local ok, err = pcall(function()
+					instance.Parent = newParent
+				end)
+				if recording then
+					if ok then
+						ChangeHistoryService:FinishRecording(recording, Enum.FinishRecordingOperation.Commit)
+					else
+						ChangeHistoryService:FinishRecording(recording, Enum.FinishRecordingOperation.Cancel)
+					end
+				end
+				if ok then
+					WsClient.send(Protocol.encode({
+						id = msgId,
+						type = "studio:response",
+						payload = { success = true, newPath = Explorer.getPath(instance) },
+						ts = os.clock(),
+					}))
+				else
+					WsClient.send(Protocol.encode({
+						id = msgId,
+						type = "studio:response",
+						payload = { success = false, error = "Move failed: " .. tostring(err) },
+						ts = os.clock(),
+					}))
+				end
+			end
+		end
+	elseif msgType == "cli:undo" then
+		local ChangeHistoryService = game:GetService("ChangeHistoryService")
+		local ok, err = pcall(function()
+			ChangeHistoryService:Undo()
+		end)
+		WsClient.send(Protocol.encode({
+			id = msgId,
+			type = "studio:response",
+			payload = ok and { success = true } or { success = false, error = tostring(err) },
+			ts = os.clock(),
+		}))
+	elseif msgType == "cli:redo" then
+		local ChangeHistoryService = game:GetService("ChangeHistoryService")
+		local ok, err = pcall(function()
+			ChangeHistoryService:Redo()
+		end)
+		WsClient.send(Protocol.encode({
+			id = msgId,
+			type = "studio:response",
+			payload = ok and { success = true } or { success = false, error = tostring(err) },
+			ts = os.clock(),
+		}))
+	elseif msgType == "cli:bulk:set:properties" then
+		local result = Properties.bulkSet(payload)
+		WsClient.send(Protocol.encode({
+			id = msgId,
+			type = "studio:response",
+			payload = result,
+			ts = os.clock(),
+		}))
+	elseif msgType == "cli:find:replace:scripts" then
+		local result = Explorer.findReplaceScripts(payload)
+		WsClient.send(Protocol.encode({
+			id = msgId,
+			type = "studio:response",
+			payload = result,
+			ts = os.clock(),
+		}))
 	else
 		warn("[Dominus] Unknown message type:", msgType)
 	end