dominus-cli 0.1.0

package/plugin/src/init.server.lua

@@ -0,0 +1,315 @@
+--[[
+	Dominus - Roblox Studio Plugin
+	AI-powered development agent bridge via WebSocket
+
+	This plugin connects to the Dominus CLI running on your machine,
+	enabling real-time bidirectional communication for AI-assisted development.
+]]
+
+local toolbar = plugin:CreateToolbar("Dominus")
+local connectButton = toolbar:CreateButton(
+	"Connect",
+	"Connect to Dominus CLI agent",
+	"rbxassetid://12974519994"
+)
+
+local WsClient = require(script.WsClient)
+local Protocol = require(script.Protocol)
+local Explorer = require(script.Explorer)
+local Executor = require(script.Executor)
+local Watcher = require(script.Watcher)
+local Properties = require(script.Properties)
+local Reflection = require(script.Reflection)
+local TestRunner = require(script.TestRunner)
+local UIBuilder = require(script.UIBuilder)
+
+local BASE_PORT = 18088
+local MAX_PORT_SCAN = 10
+local activePort = BASE_PORT
+local connected = false
+local ws = nil
+
+local function log(msg)
+	print("[Dominus]", msg)
+end
+
+local function handleMessage(rawMessage)
+	local ok, msg = pcall(Protocol.decode, rawMessage)
+	if not ok then
+		warn("[Dominus] Failed to decode message:", msg)
+		return
+	end
+
+	local msgType = msg.type
+	local payload = msg.payload
+	local msgId = msg.id
+
+	-- Route message to appropriate handler
+	if msgType == "cli:get:explorer" then
+		local tree = Explorer.getTree(payload.rootPath, payload.maxDepth or 3)
+		WsClient.send(Protocol.encode({
+			id = msgId,
+			type = "studio:response",
+			payload = { tree = tree },
+			ts = os.clock(),
+		}))
+
+	elseif msgType == "cli:get:script" then
+		local result = Explorer.getScriptSource(payload.path)
+		WsClient.send(Protocol.encode({
+			id = msgId,
+			type = "studio:response",
+			payload = result,
+			ts = os.clock(),
+		}))
+
+	elseif msgType == "cli:set:script" then
+		local result = Explorer.setScriptSource(payload.path, payload.source)
+		WsClient.send(Protocol.encode({
+			id = msgId,
+			type = "studio:response",
+			payload = result,
+			ts = os.clock(),
+		}))
+
+	elseif msgType == "cli:run" then
+		local result = Executor.run(payload.code)
+		WsClient.send(Protocol.encode({
+			id = msgId,
+			type = "studio:response",
+			payload = result,
+			ts = os.clock(),
+		}))
+
+	elseif msgType == "cli:get:properties" then
+		local result = Properties.get(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({
+			id = msgId,
+			type = "studio:response",
+			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({
+			id = msgId,
+			type = "studio:response",
+			payload = result,
+			ts = os.clock(),
+		}))
+
+	elseif msgType == "cli:delete" then
+		local result = Explorer.deleteInstance(payload.path)
+		WsClient.send(Protocol.encode({
+			id = msgId,
+			type = "studio:response",
+			payload = result,
+			ts = os.clock(),
+		}))
+
+	elseif msgType == "cli:get:selection" then
+		local Selection = game:GetService("Selection")
+		local selected = Selection:Get()
+		local paths = {}
+		local classNames = {}
+		for _, obj in selected do
+			table.insert(paths, Explorer.getPath(obj))
+			table.insert(classNames, obj.ClassName)
+		end
+		WsClient.send(Protocol.encode({
+			id = msgId,
+			type = "studio:response",
+			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({
+			id = msgId,
+			type = "studio:response",
+			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({
+			id = msgId,
+			type = "studio:response",
+			payload = result,
+			ts = os.clock(),
+		}))
+
+	elseif msgType == "cli:run:tests" then
+		local result = TestRunner.run(payload.testName)
+		WsClient.send(Protocol.encode({
+			id = msgId,
+			type = "studio:response",
+			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()
+			local result = TestRunner.executeRunMode(payload.args)
+			WsClient.send(Protocol.encode({
+				id = msgId,
+				type = "studio:response",
+				payload = result,
+				ts = os.clock(),
+			}))
+		end)
+
+	elseif msgType == "cli:test:play" then
+		-- ExecutePlayModeAsync yields until EndTest is called, so run in a thread
+		task.spawn(function()
+			local result = TestRunner.executePlayMode(payload.args)
+			WsClient.send(Protocol.encode({
+				id = msgId,
+				type = "studio:response",
+				payload = result,
+				ts = os.clock(),
+			}))
+		end)
+
+	elseif msgType == "cli:test:end" then
+		local result = TestRunner.endTest(payload.value)
+		WsClient.send(Protocol.encode({
+			id = msgId,
+			type = "studio:response",
+			payload = result,
+			ts = os.clock(),
+		}))
+
+	elseif msgType == "cli:build:ui" then
+		local result = UIBuilder.build(payload)
+		WsClient.send(Protocol.encode({
+			id = msgId,
+			type = "studio:response",
+			payload = result,
+			ts = os.clock(),
+		}))
+
+	elseif msgType == "cli:get:reflection" then
+		local result = Reflection.getClassInfo(payload.className)
+		WsClient.send(Protocol.encode({
+			id = msgId,
+			type = "studio:response",
+			payload = result,
+			ts = os.clock(),
+		}))
+
+	else
+		warn("[Dominus] Unknown message type:", msgType)
+	end
+end
+
+local function tryConnectToPort(port)
+	log("Trying ws://localhost:" .. port .. "...")
+
+	local success = WsClient.connect("ws://localhost:" .. port, {
+		onOpen = function()
+			connected = true
+			activePort = port
+			connectButton:SetActive(true)
+			log("Connected to Dominus on port " .. port .. "!")
+
+			local studioVersion = game:GetService("RobloxPluginGuiService") and "Studio" or "Unknown"
+			local placeName = "Unknown"
+			pcall(function()
+				placeName = game:GetService("MarketplaceService"):GetProductInfo(game.PlaceId).Name
+			end)
+
+			WsClient.send(Protocol.encode({
+				id = Protocol.generateId(),
+				type = "studio:connected",
+				payload = {
+					studioVersion = studioVersion,
+					placeId = game.PlaceId,
+					placeName = placeName,
+				},
+				ts = os.clock(),
+			}))
+
+			Watcher.start(function(entry)
+				if connected then
+					WsClient.send(Protocol.encode({
+						id = Protocol.generateId(),
+						type = "studio:output",
+						payload = entry,
+						ts = os.clock(),
+					}))
+				end
+			end)
+		end,
+
+		onMessage = function(message)
+			task.spawn(handleMessage, message)
+		end,
+
+		onClose = function()
+			connected = false
+			connectButton:SetActive(false)
+			Watcher.stop()
+			log("Disconnected from Dominus CLI")
+		end,
+
+		onError = function(err)
+			warn("[Dominus] WebSocket error on port " .. port .. ":", err)
+		end,
+	})
+
+	return success
+end
+
+local function connect()
+	if connected then
+		log("Already connected, disconnecting first...")
+		WsClient.disconnect()
+		connected = false
+	end
+
+	log("Scanning for Dominus server (ports " .. BASE_PORT .. "-" .. (BASE_PORT + MAX_PORT_SCAN - 1) .. ")...")
+
+	for i = 0, MAX_PORT_SCAN - 1 do
+		local port = BASE_PORT + i
+		local success = tryConnectToPort(port)
+		if success then
+			return
+		end
+	end
+
+	warn("[Dominus] No Dominus server found. Is `dominus` or `dominus mcp` running?")
+end
+
+connectButton.Click:Connect(function()
+	if connected then
+		WsClient.disconnect()
+		connected = false
+		connectButton:SetActive(false)
+	else
+		connect()
+	end
+end)
+
+-- Auto-connect on plugin load
+task.delay(1, function()
+	connect()
+end)
+
+plugin.Unloading:Connect(function()
+	Watcher.stop()
+	WsClient.disconnect()
+end)