dominus-cli 0.1.0 → 0.2.1

package/dist/index.js

@@ -1,7 +1,7 @@
 #!/usr/bin/env node
-import fs4, { existsSync, readFileSync, unlinkSync } from 'fs';
-import path3, { join } from 'path';
-import os2, { tmpdir } from 'os';
+import fs5, { existsSync, readFileSync, unlinkSync } from 'fs';
+import path4, { join } from 'path';
+import os3, { tmpdir } from 'os';
 import { nanoid } from 'nanoid';
 import { WebSocket, WebSocketServer } from 'ws';
 import { EventEmitter } from 'events';
@@ -18,6 +18,7 @@ import { execSync } from 'child_process';
 import { randomBytes } from 'crypto';
 import Spinner from 'ink-spinner';
 import OpenAI from 'openai';
+import https from 'https';
 
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
@@ -55,14 +56,14 @@ function resolveProviderSettings(config) {
   };
 }
 function ensureDir() {
-  if (!fs4.existsSync(DOMINUS_DIR)) {
-    fs4.mkdirSync(DOMINUS_DIR, { recursive: true });
+  if (!fs5.existsSync(DOMINUS_DIR)) {
+    fs5.mkdirSync(DOMINUS_DIR, { recursive: true });
   }
 }
 function readJson(filePath, defaults) {
   try {
-    if (fs4.existsSync(filePath)) {
-      const raw = fs4.readFileSync(filePath, "utf-8");
+    if (fs5.existsSync(filePath)) {
+      const raw = fs5.readFileSync(filePath, "utf-8");
       return { ...defaults, ...JSON.parse(raw) };
     }
   } catch {
@@ -71,7 +72,7 @@ function readJson(filePath, defaults) {
 }
 function writeJson(filePath, data) {
   ensureDir();
-  fs4.writeFileSync(filePath, JSON.stringify(data, null, 2), "utf-8");
+  fs5.writeFileSync(filePath, JSON.stringify(data, null, 2), "utf-8");
 }
 function loadConfig() {
   ensureDir();
@@ -123,10 +124,10 @@ function createConfigAccess() {
 var DOMINUS_DIR, CONFIG_PATH, DB_PATH, RULES_PATH, PROVIDERS, DEFAULTS, DEFAULT_RULES;
 var init_config = __esm({
   "src/config/config.ts"() {
-    DOMINUS_DIR = path3.join(os2.homedir(), ".dominus");
-    CONFIG_PATH = path3.join(DOMINUS_DIR, "config.json");
-    DB_PATH = path3.join(DOMINUS_DIR, "dominus.db");
-    RULES_PATH = path3.join(DOMINUS_DIR, "rules.json");
+    DOMINUS_DIR = path4.join(os3.homedir(), ".dominus");
+    CONFIG_PATH = path4.join(DOMINUS_DIR, "config.json");
+    DB_PATH = path4.join(DOMINUS_DIR, "dominus.db");
+    RULES_PATH = path4.join(DOMINUS_DIR, "rules.json");
     PROVIDERS = {
       openai: {
         name: "OpenAI",
@@ -459,6 +460,166 @@ var init_ws_server = __esm({
     };
   }
 });
+async function isPortInUse(port) {
+  return new Promise((resolve) => {
+    const ws = new WebSocket(`ws://localhost:${port}`);
+    const timer = setTimeout(() => {
+      ws.close();
+      resolve(false);
+    }, 1e3);
+    ws.on("open", () => {
+      clearTimeout(timer);
+      ws.close();
+      resolve(true);
+    });
+    ws.on("error", () => {
+      clearTimeout(timer);
+      resolve(false);
+    });
+  });
+}
+var DominusClient;
+var init_ws_client = __esm({
+  "src/transport/ws-client.ts"() {
+    init_protocol();
+    DominusClient = class extends EventEmitter {
+      ws = null;
+      port;
+      studioConnected = false;
+      connectionInfo = null;
+      pendingRequests = /* @__PURE__ */ new Map();
+      constructor(port = 18088) {
+        super();
+        this.port = port;
+      }
+      connect() {
+        return new Promise((resolve, reject) => {
+          const url = `ws://localhost:${this.port}`;
+          const ws = new WebSocket(url);
+          ws.on("open", () => {
+            this.ws = ws;
+            ws.send(serializeMessage(createMessage("controller:connect", {})));
+            this.emit("server:started", { port: this.port, relay: true });
+            resolve();
+          });
+          ws.on("message", (raw) => {
+            try {
+              const msg = parseMessage(raw.toString());
+              this.handleMessage(msg);
+            } catch (err) {
+              this.emit("server:error", err);
+            }
+          });
+          ws.on("close", () => {
+            this.ws = null;
+            this.studioConnected = false;
+            this.emit("studio:disconnected");
+            for (const [id, pending] of this.pendingRequests) {
+              clearTimeout(pending.timer);
+              pending.reject(new Error("Relay connection lost"));
+              this.pendingRequests.delete(id);
+            }
+          });
+          ws.on("error", (err) => {
+            if (!this.ws) reject(err);
+            this.emit("server:error", err);
+          });
+        });
+      }
+      handleMessage(msg) {
+        if (msg.type === "controller:welcome") {
+          const p = msg.payload;
+          this.studioConnected = p.studioConnected;
+          if (this.studioConnected) {
+            this.emit("studio:connected", {});
+          }
+          return;
+        }
+        if (msg.type === "studio:connected") {
+          this.studioConnected = true;
+          const payload = msg.payload;
+          this.connectionInfo = {
+            connectedAt: Date.now(),
+            studioVersion: payload.studioVersion,
+            placeId: payload.placeId,
+            placeName: payload.placeName
+          };
+          this.emit("studio:connected", payload);
+          return;
+        }
+        if (msg.type === "studio:disconnected") {
+          this.studioConnected = false;
+          this.connectionInfo = null;
+          this.emit("studio:disconnected");
+          return;
+        }
+        if (msg.type === StudioMsg.RESPONSE) {
+          const pending = this.pendingRequests.get(msg.id);
+          if (pending) {
+            clearTimeout(pending.timer);
+            pending.resolve(msg);
+            this.pendingRequests.delete(msg.id);
+          }
+          return;
+        }
+        this.emit(msg.type, msg.payload);
+        this.emit("message", msg);
+      }
+      sendRequest(type, payload, timeoutMs = 1e4) {
+        return new Promise((resolve, reject) => {
+          if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
+            reject(new Error("Not connected to Dominus server"));
+            return;
+          }
+          if (!this.studioConnected) {
+            reject(new Error("Studio not connected"));
+            return;
+          }
+          const msg = createMessage(type, payload);
+          const timer = setTimeout(() => {
+            this.pendingRequests.delete(msg.id);
+            reject(new Error(`Request timed out: ${type}`));
+          }, timeoutMs);
+          this.pendingRequests.set(msg.id, {
+            resolve,
+            reject,
+            timer
+          });
+          this.ws.send(serializeMessage(msg));
+        });
+      }
+      sendNotification(type, payload) {
+        if (!this.ws || this.ws.readyState !== WebSocket.OPEN) return;
+        const msg = createMessage(type, payload);
+        this.ws.send(serializeMessage(msg));
+      }
+      isConnected() {
+        return this.studioConnected;
+      }
+      getConnectionInfo() {
+        if (!this.connectionInfo) return null;
+        return { ...this.connectionInfo, ws: this.ws };
+      }
+      getPort() {
+        return this.port;
+      }
+      stop() {
+        return new Promise((resolve) => {
+          for (const [, pending] of this.pendingRequests) {
+            clearTimeout(pending.timer);
+            pending.reject(new Error("Client shutting down"));
+          }
+          this.pendingRequests.clear();
+          if (this.ws) {
+            this.ws.close();
+            this.ws = null;
+          }
+          resolve();
+        });
+      }
+    };
+  }
+});
 
 // src/memory/store.ts
 var store_exports = {};
@@ -481,14 +642,14 @@ __export(store_exports, {
 function persist() {
   if (!db) return;
   const data = db.export();
-  fs4.writeFileSync(dbPath, Buffer.from(data));
+  fs5.writeFileSync(dbPath, Buffer.from(data));
 }
 async function initMemoryStore() {
   if (db) return;
   dbPath = getDbPath();
   const SQL = await initSqlJs();
-  if (fs4.existsSync(dbPath)) {
-    const buffer = fs4.readFileSync(dbPath);
+  if (fs5.existsSync(dbPath)) {
+    const buffer = fs5.readFileSync(dbPath);
     db = new SQL.Database(buffer);
   } else {
     db = new SQL.Database();
@@ -779,10 +940,10 @@ var init_read_script = __esm({
         if (!ctx.isStudioConnected()) {
           return { success: false, error: "Studio is not connected" };
         }
-        const path4 = params.path;
-        const response = await ctx.sendToStudio(CliMsg.GET_SCRIPT, { path: path4 });
+        const path5 = params.path;
+        const response = await ctx.sendToStudio(CliMsg.GET_SCRIPT, { path: path5 });
         if (!response.payload) {
-          return { success: false, error: `Script not found: ${path4}` };
+          return { success: false, error: `Script not found: ${path5}` };
         }
         return {
           success: true,
@@ -828,14 +989,14 @@ var init_edit_script = __esm({
         if (!ctx.isStudioConnected()) {
           return { success: false, error: "Studio is not connected" };
         }
-        const path4 = params.path;
+        const path5 = params.path;
         const source = params.source;
-        const response = await ctx.sendToStudio(CliMsg.SET_SCRIPT, { path: path4, source });
+        const response = await ctx.sendToStudio(CliMsg.SET_SCRIPT, { path: path5, source });
         const payload = response.payload;
         if (!payload.success) {
           return { success: false, error: payload.error ?? "Failed to edit script" };
         }
-        return { success: true, data: { path: path4, linesWritten: source.split("\n").length } };
+        return { success: true, data: { path: path5, linesWritten: source.split("\n").length } };
       }
     };
   }
@@ -1521,8 +1682,8 @@ __export(clone_instance_exports, {
   tool: () => tool18
 });
 function buildCloneCode(params) {
-  const path4 = params.path;
-  const parts = path4.split(".");
+  const path5 = params.path;
+  const parts = path5.split(".");
   let resolve = "game";
   for (const p of parts) {
     resolve += `["${p}"]`;
@@ -1663,8 +1824,8 @@ __export(build_multiple_exports, {
 });
 function generateBatchCode(parts, groupName, groupParent) {
   const lines = [];
-  function resolveParent(path4) {
-    const segs = path4.split(".");
+  function resolveParent(path5) {
+    const segs = path5.split(".");
     let r = "game";
     for (const s of segs) r += `["${s}"]`;
     return r;
@@ -1928,11 +2089,11 @@ var init_upload_asset = __esm({
         const assetType = params.assetType;
         const name = params.name;
         const description = params.description ?? "";
-        if (!fs4.existsSync(filePath)) {
+        if (!fs5.existsSync(filePath)) {
           return { success: false, error: `File not found: ${filePath}` };
         }
-        const fileBuffer = fs4.readFileSync(filePath);
-        const fileName = path3.basename(filePath);
+        const fileBuffer = fs5.readFileSync(filePath);
+        const fileName = path4.basename(filePath);
         const typeMapping = {
           Image: "Decal",
           Audio: "Audio",
@@ -1994,19 +2155,27 @@ __export(server_exports, {
 async function startMcpServer() {
   const config = loadConfig();
   await initMemoryStore();
-  wsServer = new DominusServer(config.port);
-  await wsServer.start();
+  const portTaken = await isPortInUse(config.port);
+  if (portTaken) {
+    const client = new DominusClient(config.port);
+    await client.connect();
+    bridge = client;
+  } else {
+    const server = new DominusServer(config.port);
+    await server.start();
+    bridge = server;
+  }
   const mcp = new McpServer({
     name: "dominus",
-    version: "0.1.0"
+    version: "0.2.1"
   });
   mcp.tool(
     "read_script",
     "Read the source code of a script in Roblox Studio",
     { path: z.string().describe('Full instance path, e.g. "ServerScriptService.GameManager"') },
-    async ({ path: path4 }) => {
+    async ({ path: path5 }) => {
       assertConnected();
-      const res = await wsServer.sendRequest(CliMsg.GET_SCRIPT, { path: path4 });
+      const res = await bridge.sendRequest(CliMsg.GET_SCRIPT, { path: path5 });
       return { content: [{ type: "text", text: JSON.stringify(res.payload, null, 2) }] };
     }
   );
@@ -2017,9 +2186,9 @@ async function startMcpServer() {
       path: z.string().describe("Full instance path"),
       source: z.string().describe("New complete source code")
     },
-    async ({ path: path4, source }) => {
+    async ({ path: path5, source }) => {
       assertConnected();
-      const res = await wsServer.sendRequest(CliMsg.SET_SCRIPT, { path: path4, source });
+      const res = await bridge.sendRequest(CliMsg.SET_SCRIPT, { path: path5, source });
       return { content: [{ type: "text", text: JSON.stringify(res.payload) }] };
     }
   );
@@ -2029,7 +2198,7 @@ async function startMcpServer() {
     { code: z.string().describe("Luau code to execute") },
     async ({ code }) => {
       assertConnected();
-      const res = await wsServer.sendRequest(CliMsg.RUN_CODE, { code });
+      const res = await bridge.sendRequest(CliMsg.RUN_CODE, { code });
       return { content: [{ type: "text", text: JSON.stringify(res.payload, null, 2) }] };
     }
   );
@@ -2042,7 +2211,7 @@ async function startMcpServer() {
     },
     async ({ rootPath, maxDepth }) => {
       assertConnected();
-      const res = await wsServer.sendRequest(CliMsg.GET_EXPLORER, {
+      const res = await bridge.sendRequest(CliMsg.GET_EXPLORER, {
         rootPath: rootPath ?? null,
         maxDepth
       });
@@ -2053,9 +2222,9 @@ async function startMcpServer() {
     "get_properties",
     "Get all properties of an instance in Roblox Studio",
     { path: z.string().describe("Full instance path") },
-    async ({ path: path4 }) => {
+    async ({ path: path5 }) => {
       assertConnected();
-      const res = await wsServer.sendRequest(CliMsg.GET_PROPERTIES, { path: path4 });
+      const res = await bridge.sendRequest(CliMsg.GET_PROPERTIES, { path: path5 });
       return { content: [{ type: "text", text: JSON.stringify(res.payload, null, 2) }] };
     }
   );
@@ -2066,9 +2235,9 @@ async function startMcpServer() {
       path: z.string().describe("Full instance path"),
       properties: z.record(z.unknown()).describe("Key-value pairs of properties to set")
     },
-    async ({ path: path4, properties }) => {
+    async ({ path: path5, properties }) => {
       assertConnected();
-      const res = await wsServer.sendRequest(CliMsg.SET_PROPERTIES, { path: path4, properties });
+      const res = await bridge.sendRequest(CliMsg.SET_PROPERTIES, { path: path5, properties });
       return { content: [{ type: "text", text: JSON.stringify(res.payload) }] };
     }
   );
@@ -2083,7 +2252,7 @@ async function startMcpServer() {
     },
     async ({ className, parentPath, name, properties }) => {
       assertConnected();
-      const res = await wsServer.sendRequest(CliMsg.INSERT_INSTANCE, {
+      const res = await bridge.sendRequest(CliMsg.INSERT_INSTANCE, {
         className,
         parentPath,
         name,
@@ -2096,9 +2265,9 @@ async function startMcpServer() {
     "delete_instance",
     "Delete an instance from the Roblox Studio DataModel",
     { path: z.string().describe("Full instance path to delete") },
-    async ({ path: path4 }) => {
+    async ({ path: path5 }) => {
       assertConnected();
-      const res = await wsServer.sendRequest(CliMsg.DELETE_INSTANCE, { path: path4 });
+      const res = await bridge.sendRequest(CliMsg.DELETE_INSTANCE, { path: path5 });
       return { content: [{ type: "text", text: JSON.stringify(res.payload) }] };
     }
   );
@@ -2108,7 +2277,7 @@ async function startMcpServer() {
     {},
     async () => {
       assertConnected();
-      const res = await wsServer.sendRequest(CliMsg.GET_SELECTION, {});
+      const res = await bridge.sendRequest(CliMsg.GET_SELECTION, {});
       return { content: [{ type: "text", text: JSON.stringify(res.payload, null, 2) }] };
     }
   );
@@ -2121,7 +2290,7 @@ async function startMcpServer() {
     },
     async ({ query, maxResults }) => {
       assertConnected();
-      const res = await wsServer.sendRequest(CliMsg.SEARCH_SCRIPTS, { query, maxResults });
+      const res = await bridge.sendRequest(CliMsg.SEARCH_SCRIPTS, { query, maxResults });
       return { content: [{ type: "text", text: JSON.stringify(res.payload, null, 2) }] };
     }
   );
@@ -2134,7 +2303,7 @@ async function startMcpServer() {
     },
     async ({ limit, level }) => {
       assertConnected();
-      const res = await wsServer.sendRequest(CliMsg.GET_OUTPUT, {
+      const res = await bridge.sendRequest(CliMsg.GET_OUTPUT, {
         limit,
         level: level ?? null
       });
@@ -2147,7 +2316,7 @@ async function startMcpServer() {
     { className: z.string().describe('Roblox class name, e.g. "Frame", "TextLabel", "Part", "MeshPart"') },
     async ({ className }) => {
       assertConnected();
-      const res = await wsServer.sendRequest(CliMsg.GET_REFLECTION, { className });
+      const res = await bridge.sendRequest(CliMsg.GET_REFLECTION, { className });
       return { content: [{ type: "text", text: JSON.stringify(res.payload, null, 2) }] };
     }
   );
@@ -2162,7 +2331,7 @@ async function startMcpServer() {
     },
     async ({ parent, tree }) => {
       assertConnected();
-      const res = await wsServer.sendRequest(
+      const res = await bridge.sendRequest(
         CliMsg.BUILD_UI,
         { parent: parent ?? "StarterGui", tree }
       );
@@ -2175,7 +2344,7 @@ async function startMcpServer() {
     { testName: z.string().optional().describe("Specific test name, or omit to run all") },
     async ({ testName }) => {
       assertConnected();
-      const res = await wsServer.sendRequest(CliMsg.RUN_TESTS, {
+      const res = await bridge.sendRequest(CliMsg.RUN_TESTS, {
         testName: testName ?? null
       });
       return { content: [{ type: "text", text: JSON.stringify(res.payload, null, 2) }] };
@@ -2187,7 +2356,7 @@ async function startMcpServer() {
     { args: z.unknown().optional().describe("Arguments passed to the test session. Server scripts get this via StudioTestService:GetTestArgs()") },
     async ({ args }) => {
       assertConnected();
-      const res = await wsServer.sendRequest(CliMsg.EXECUTE_RUN_TEST, { args: args ?? null }, 12e4);
+      const res = await bridge.sendRequest(CliMsg.EXECUTE_RUN_TEST, { args: args ?? null }, 12e4);
       return { content: [{ type: "text", text: JSON.stringify(res.payload, null, 2) }] };
     }
   );
@@ -2197,7 +2366,7 @@ async function startMcpServer() {
     { args: z.unknown().optional().describe("Arguments passed to the test session. Server scripts get this via StudioTestService:GetTestArgs()") },
     async ({ args }) => {
       assertConnected();
-      const res = await wsServer.sendRequest(CliMsg.EXECUTE_PLAY_TEST, { args: args ?? null }, 12e4);
+      const res = await bridge.sendRequest(CliMsg.EXECUTE_PLAY_TEST, { args: args ?? null }, 12e4);
       return { content: [{ type: "text", text: JSON.stringify(res.payload, null, 2) }] };
     }
   );
@@ -2236,7 +2405,7 @@ async function startMcpServer() {
       if (params.canCollide !== void 0) properties.CanCollide = params.canCollide;
       if (params.shape && params.shape !== "Block") properties.Shape = params.shape;
       const className = params.shape === "Wedge" ? "WedgePart" : params.shape === "CornerWedge" ? "CornerWedgePart" : "Part";
-      const res = await wsServer.sendRequest(CliMsg.INSERT_INSTANCE, {
+      const res = await bridge.sendRequest(CliMsg.INSERT_INSTANCE, {
         className,
         parentPath: params.parent ?? "Workspace",
         name: params.name,
@@ -2275,7 +2444,7 @@ async function startMcpServer() {
         lines.push(`end`);
       }
       lines.push(`print("Cloned: " .. clone:GetFullName())`);
-      const res = await wsServer.sendRequest(CliMsg.RUN_CODE, { code: lines.join("\n") });
+      const res = await bridge.sendRequest(CliMsg.RUN_CODE, { code: lines.join("\n") });
       return { content: [{ type: "text", text: JSON.stringify(res.payload, null, 2) }] };
     }
   );
@@ -2295,8 +2464,8 @@ async function startMcpServer() {
       const parentParts = parent.split(".");
       let parentResolve = "game";
       for (const p of parentParts) parentResolve += `["${p}"]`;
-      const resolves = params.paths.map((path4) => {
-        const parts = path4.split(".");
+      const resolves = params.paths.map((path5) => {
+        const parts = path5.split(".");
         let r = "game";
         for (const part of parts) r += `["${part}"]`;
         return r;
@@ -2308,7 +2477,7 @@ async function startMcpServer() {
         ...resolves.map((r) => `${r}.Parent = group`),
         `print("Grouped ${params.paths.length} instances into: " .. group:GetFullName())`
       ].join("\n");
-      const res = await wsServer.sendRequest(CliMsg.RUN_CODE, { code });
+      const res = await bridge.sendRequest(CliMsg.RUN_CODE, { code });
       return { content: [{ type: "text", text: JSON.stringify(res.payload, null, 2) }] };
     }
   );
@@ -2336,8 +2505,8 @@ async function startMcpServer() {
       const parts = params.parts;
       const groupName = params.groupName;
       const groupParent = params.groupParent ?? "Workspace";
-      function resolveP(path4) {
-        const segs = path4.split(".");
+      function resolveP(path5) {
+        const segs = path5.split(".");
         let r = "game";
         for (const s of segs) r += `["${s}"]`;
         return r;
@@ -2367,7 +2536,7 @@ async function startMcpServer() {
         lines.push(`${v}.Parent = ${groupName ? "_group" : resolveP(p.parent ?? "Workspace")}`);
       }
       lines.push(`print("Built ${parts.length} parts${groupName ? ` in ${groupName}` : ""}")`);
-      const res = await wsServer.sendRequest(CliMsg.RUN_CODE, { code: lines.join("\n") });
+      const res = await bridge.sendRequest(CliMsg.RUN_CODE, { code: lines.join("\n") });
       return { content: [{ type: "text", text: JSON.stringify(res.payload, null, 2) }] };
     }
   );
@@ -2403,16 +2572,16 @@ async function startMcpServer() {
     "studio://status",
     "studio://status",
     async () => {
-      const info = wsServer.getConnectionInfo();
+      const info = bridge.getConnectionInfo();
       return {
         contents: [{
           uri: "studio://status",
           text: JSON.stringify({
-            connected: wsServer.isConnected(),
+            connected: bridge.isConnected(),
             placeName: info?.placeName,
             placeId: info?.placeId,
             studioVersion: info?.studioVersion,
-            port: wsServer.getPort()
+            port: bridge.getPort()
           }, null, 2),
           mimeType: "application/json"
         }]
@@ -2423,16 +2592,17 @@ async function startMcpServer() {
   await mcp.connect(transport);
 }
 function assertConnected() {
-  if (!wsServer.isConnected()) {
+  if (!bridge.isConnected()) {
     throw new Error(
       "Roblox Studio is not connected. Open Studio and click the Dominus plugin Connect button."
     );
   }
 }
-var wsServer;
+var bridge;
 var init_server = __esm({
   "src/mcp/server.ts"() {
     init_ws_server();
+    init_ws_client();
     init_protocol();
     init_store();
     init_config();
@@ -3475,163 +3645,7 @@ var App = ({ server, agent: initialAgent, version }) => {
 
 // src/index.tsx
 init_ws_server();
-
-// src/transport/ws-client.ts
-init_protocol();
-var DominusClient = class extends EventEmitter {
-  ws = null;
-  port;
-  studioConnected = false;
-  connectionInfo = null;
-  pendingRequests = /* @__PURE__ */ new Map();
-  constructor(port = 18088) {
-    super();
-    this.port = port;
-  }
-  connect() {
-    return new Promise((resolve, reject) => {
-      const url = `ws://localhost:${this.port}`;
-      const ws = new WebSocket(url);
-      ws.on("open", () => {
-        this.ws = ws;
-        ws.send(serializeMessage(createMessage("controller:connect", {})));
-        this.emit("server:started", { port: this.port, relay: true });
-        resolve();
-      });
-      ws.on("message", (raw) => {
-        try {
-          const msg = parseMessage(raw.toString());
-          this.handleMessage(msg);
-        } catch (err) {
-          this.emit("server:error", err);
-        }
-      });
-      ws.on("close", () => {
-        this.ws = null;
-        this.studioConnected = false;
-        this.emit("studio:disconnected");
-        for (const [id, pending] of this.pendingRequests) {
-          clearTimeout(pending.timer);
-          pending.reject(new Error("Relay connection lost"));
-          this.pendingRequests.delete(id);
-        }
-      });
-      ws.on("error", (err) => {
-        if (!this.ws) reject(err);
-        this.emit("server:error", err);
-      });
-    });
-  }
-  handleMessage(msg) {
-    if (msg.type === "controller:welcome") {
-      const p = msg.payload;
-      this.studioConnected = p.studioConnected;
-      if (this.studioConnected) {
-        this.emit("studio:connected", {});
-      }
-      return;
-    }
-    if (msg.type === "studio:connected") {
-      this.studioConnected = true;
-      const payload = msg.payload;
-      this.connectionInfo = {
-        connectedAt: Date.now(),
-        studioVersion: payload.studioVersion,
-        placeId: payload.placeId,
-        placeName: payload.placeName
-      };
-      this.emit("studio:connected", payload);
-      return;
-    }
-    if (msg.type === "studio:disconnected") {
-      this.studioConnected = false;
-      this.connectionInfo = null;
-      this.emit("studio:disconnected");
-      return;
-    }
-    if (msg.type === StudioMsg.RESPONSE) {
-      const pending = this.pendingRequests.get(msg.id);
-      if (pending) {
-        clearTimeout(pending.timer);
-        pending.resolve(msg);
-        this.pendingRequests.delete(msg.id);
-      }
-      return;
-    }
-    this.emit(msg.type, msg.payload);
-    this.emit("message", msg);
-  }
-  sendRequest(type, payload, timeoutMs = 1e4) {
-    return new Promise((resolve, reject) => {
-      if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
-        reject(new Error("Not connected to Dominus server"));
-        return;
-      }
-      if (!this.studioConnected) {
-        reject(new Error("Studio not connected"));
-        return;
-      }
-      const msg = createMessage(type, payload);
-      const timer = setTimeout(() => {
-        this.pendingRequests.delete(msg.id);
-        reject(new Error(`Request timed out: ${type}`));
-      }, timeoutMs);
-      this.pendingRequests.set(msg.id, {
-        resolve,
-        reject,
-        timer
-      });
-      this.ws.send(serializeMessage(msg));
-    });
-  }
-  sendNotification(type, payload) {
-    if (!this.ws || this.ws.readyState !== WebSocket.OPEN) return;
-    const msg = createMessage(type, payload);
-    this.ws.send(serializeMessage(msg));
-  }
-  isConnected() {
-    return this.studioConnected;
-  }
-  getConnectionInfo() {
-    if (!this.connectionInfo) return null;
-    return { ...this.connectionInfo, ws: this.ws };
-  }
-  getPort() {
-    return this.port;
-  }
-  stop() {
-    return new Promise((resolve) => {
-      for (const [, pending] of this.pendingRequests) {
-        clearTimeout(pending.timer);
-        pending.reject(new Error("Client shutting down"));
-      }
-      this.pendingRequests.clear();
-      if (this.ws) {
-        this.ws.close();
-        this.ws = null;
-      }
-      resolve();
-    });
-  }
-};
-async function isPortInUse(port) {
-  return new Promise((resolve) => {
-    const ws = new WebSocket(`ws://localhost:${port}`);
-    const timer = setTimeout(() => {
-      ws.close();
-      resolve(false);
-    }, 1e3);
-    ws.on("open", () => {
-      clearTimeout(timer);
-      ws.close();
-      resolve(true);
-    });
-    ws.on("error", () => {
-      clearTimeout(timer);
-      resolve(false);
-    });
-  });
-}
+init_ws_client();
 
 // src/ai/provider.ts
 init_config();
@@ -4293,7 +4307,98 @@ function createDefaultRegistry() {
 // src/index.tsx
 init_store();
 init_config();
-var VERSION = "0.1.0";
+var CACHE_DIR = path4.join(os3.homedir(), ".dominus");
+var CACHE_FILE = path4.join(CACHE_DIR, "update-check.json");
+var CHECK_INTERVAL_MS = 4 * 60 * 60 * 1e3;
+var NPM_PACKAGE = "dominus-cli";
+function readCache() {
+  try {
+    if (fs5.existsSync(CACHE_FILE)) {
+      return JSON.parse(fs5.readFileSync(CACHE_FILE, "utf-8"));
+    }
+  } catch {
+  }
+  return { lastCheck: 0, latestVersion: null };
+}
+function writeCache(cache) {
+  try {
+    if (!fs5.existsSync(CACHE_DIR)) fs5.mkdirSync(CACHE_DIR, { recursive: true });
+    fs5.writeFileSync(CACHE_FILE, JSON.stringify(cache), "utf-8");
+  } catch {
+  }
+}
+function fetchLatestVersion() {
+  return new Promise((resolve) => {
+    const req = https.get(
+      `https://registry.npmjs.org/${NPM_PACKAGE}/latest`,
+      { headers: { Accept: "application/json" }, timeout: 5e3 },
+      (res) => {
+        if (res.statusCode !== 200) {
+          resolve(null);
+          res.resume();
+          return;
+        }
+        let body = "";
+        res.on("data", (c) => body += c);
+        res.on("end", () => {
+          try {
+            resolve(JSON.parse(body).version ?? null);
+          } catch {
+            resolve(null);
+          }
+        });
+      }
+    );
+    req.on("error", () => resolve(null));
+    req.on("timeout", () => {
+      req.destroy();
+      resolve(null);
+    });
+  });
+}
+function compareVersions(current, latest) {
+  const a = current.split(".").map(Number);
+  const b = latest.split(".").map(Number);
+  for (let i = 0; i < 3; i++) {
+    if ((b[i] ?? 0) > (a[i] ?? 0)) return 1;
+    if ((b[i] ?? 0) < (a[i] ?? 0)) return -1;
+  }
+  return 0;
+}
+async function checkForUpdate(currentVersion) {
+  try {
+    const cache = readCache();
+    const now = Date.now();
+    if (now - cache.lastCheck < CHECK_INTERVAL_MS && cache.latestVersion) {
+      if (compareVersions(currentVersion, cache.latestVersion) > 0) {
+        return { updateAvailable: true, currentVersion, latestVersion: cache.latestVersion };
+      }
+      return null;
+    }
+    const latest = await fetchLatestVersion();
+    writeCache({ lastCheck: now, latestVersion: latest });
+    if (latest && compareVersions(currentVersion, latest) > 0) {
+      return { updateAvailable: true, currentVersion, latestVersion: latest };
+    }
+  } catch {
+  }
+  return null;
+}
+function formatUpdateMessage(info) {
+  const line1 = `Update available: ${info.currentVersion} \u2192 ${info.latestVersion}`;
+  const line2 = `Run: npm i -g ${NPM_PACKAGE}`;
+  const width = Math.max(line1.length, line2.length) + 4;
+  const pad = (s) => `  \u2502  ${s}${" ".repeat(width - s.length - 4)}  \u2502`;
+  return [
+    "",
+    `  \u256D${"\u2500".repeat(width)}\u256E`,
+    pad(line1),
+    pad(line2),
+    `  \u2570${"\u2500".repeat(width)}\u256F`,
+    ""
+  ].join("\n");
+}
+var VERSION = "0.2.1";
 var program = new Command();
 program.name("dominus").description("AI-powered autonomous agent for Roblox Studio development").version(VERSION);
 program.command("start", { isDefault: true }).description("Launch the Dominus TUI agent").option("-p, --port <port>", "WebSocket server port", "18088").action(async (opts) => {
@@ -4313,6 +4418,10 @@ program.command("start", { isDefault: true }).description("Launch the Dominus TU
       setConfigValue("userName", firstName);
     }
   }
+  checkForUpdate(VERSION).then((info) => {
+    if (info) process.stderr.write(formatUpdateMessage(info));
+  }).catch(() => {
+  });
   await launchApp(port);
 });
 var configCmd = program.command("config").description("Manage Dominus configuration");
@@ -4403,24 +4512,24 @@ rulesCmd.command("list").description("List all rules").action(() => {
 });
 program.command("install-plugin").description("Build & install the Dominus plugin into Roblox Studio").action(async () => {
   const { execSync: execSync2 } = await import('child_process');
-  const packageRoot = path3.resolve(import.meta.dirname ?? __dirname, "..");
-  const pluginDir = path3.join(packageRoot, "plugin");
-  const pluginProject = path3.join(pluginDir, "default.project.json");
-  const studioPlugins = path3.join(
-    os2.homedir(),
+  const packageRoot = path4.resolve(import.meta.dirname ?? __dirname, "..");
+  const pluginDir = path4.join(packageRoot, "plugin");
+  const pluginProject = path4.join(pluginDir, "default.project.json");
+  const studioPlugins = path4.join(
+    os3.homedir(),
     "AppData",
     "Local",
     "Roblox",
     "Plugins"
   );
-  const destFile = path3.join(studioPlugins, "Dominus.rbxm");
-  if (!fs4.existsSync(pluginProject)) {
+  const destFile = path4.join(studioPlugins, "Dominus.rbxm");
+  if (!fs5.existsSync(pluginProject)) {
     console.log("  \u2716 Plugin source not found.");
     console.log(`  Expected at: ${pluginDir}`);
     return;
   }
-  if (!fs4.existsSync(studioPlugins)) {
-    fs4.mkdirSync(studioPlugins, { recursive: true });
+  if (!fs5.existsSync(studioPlugins)) {
+    fs5.mkdirSync(studioPlugins, { recursive: true });
   }
   let rojoAvailable = false;
   try {
@@ -4447,49 +4556,49 @@ program.command("mcp").description("Start the Dominus MCP server (for Cursor, Cl
   await startMcpServer2();
 });
 function getMcpTargets() {
-  const home = os2.homedir();
-  const appData = process.env.APPDATA || path3.join(home, "AppData", "Roaming");
-  const localAppData = process.env.LOCALAPPDATA || path3.join(home, "AppData", "Local");
+  const home = os3.homedir();
+  const appData = process.env.APPDATA || path4.join(home, "AppData", "Roaming");
+  const localAppData = process.env.LOCALAPPDATA || path4.join(home, "AppData", "Local");
   const cwd = process.cwd();
   return [
     {
       name: "Cursor (project)",
-      configPath: path3.join(cwd, ".cursor", "mcp.json"),
+      configPath: path4.join(cwd, ".cursor", "mcp.json"),
       keyPath: ["mcpServers"]
     },
     {
       name: "Cursor (global)",
-      configPath: path3.join(home, ".cursor", "mcp.json"),
+      configPath: path4.join(home, ".cursor", "mcp.json"),
       keyPath: ["mcpServers"]
     },
     {
       name: "Claude Desktop",
-      configPath: path3.join(appData, "Claude", "claude_desktop_config.json"),
+      configPath: path4.join(appData, "Claude", "claude_desktop_config.json"),
       keyPath: ["mcpServers"]
     },
     {
       name: "Claude Code",
-      configPath: path3.join(home, ".claude.json"),
+      configPath: path4.join(home, ".claude.json"),
       keyPath: ["mcpServers"]
     },
     {
       name: "Windsurf (project)",
-      configPath: path3.join(cwd, ".windsurf", "mcp.json"),
+      configPath: path4.join(cwd, ".windsurf", "mcp.json"),
       keyPath: ["mcpServers"]
     },
     {
       name: "Windsurf (global)",
-      configPath: path3.join(home, ".codeium", "windsurf", "mcp_config.json"),
+      configPath: path4.join(home, ".codeium", "windsurf", "mcp_config.json"),
       keyPath: ["mcpServers"]
     },
     {
       name: "VS Code (project)",
-      configPath: path3.join(cwd, ".vscode", "mcp.json"),
+      configPath: path4.join(cwd, ".vscode", "mcp.json"),
       keyPath: ["servers"]
     },
     {
       name: "VS Code (global)",
-      configPath: path3.join(localAppData, "Code", "User", "settings.json"),
+      configPath: path4.join(localAppData, "Code", "User", "settings.json"),
       keyPath: ["mcp", "servers"]
     }
   ];
@@ -4501,9 +4610,9 @@ var dominusEntry = {
 };
 function injectMcpConfig(target) {
   let config = {};
-  if (fs4.existsSync(target.configPath)) {
+  if (fs5.existsSync(target.configPath)) {
     try {
-      config = JSON.parse(fs4.readFileSync(target.configPath, "utf-8"));
+      config = JSON.parse(fs5.readFileSync(target.configPath, "utf-8"));
     } catch {
       config = {};
     }
@@ -4519,12 +4628,12 @@ function injectMcpConfig(target) {
     return { status: "already" };
   }
   obj["dominus"] = { ...dominusEntry };
-  const dir = path3.dirname(target.configPath);
-  if (!fs4.existsSync(dir)) {
-    fs4.mkdirSync(dir, { recursive: true });
+  const dir = path4.dirname(target.configPath);
+  if (!fs5.existsSync(dir)) {
+    fs5.mkdirSync(dir, { recursive: true });
   }
-  fs4.writeFileSync(target.configPath, JSON.stringify(config, null, 2), "utf-8");
-  return { status: fs4.existsSync(target.configPath) ? "installed" : "created" };
+  fs5.writeFileSync(target.configPath, JSON.stringify(config, null, 2), "utf-8");
+  return { status: fs5.existsSync(target.configPath) ? "installed" : "created" };
 }
 program.command("mcp-install").description("Auto-install Dominus MCP into Cursor, VS Code, Claude Desktop, Windsurf, etc.").option("-a, --all", "Install to all detected editors").action(async (opts) => {
   const targets = getMcpTargets();
@@ -4546,7 +4655,7 @@ program.command("mcp-install").description("Auto-install Dominus MCP into Cursor
   }
   const detected = targets.map((t) => ({
     target: t,
-    exists: fs4.existsSync(path3.dirname(t.configPath))
+    exists: fs5.existsSync(path4.dirname(t.configPath))
   }));
   const toInstall = detected.filter((d) => d.exists);
   if (toInstall.length === 0) {
@@ -4619,23 +4728,23 @@ program.command("mcp-setup").description("Print MCP configuration for manual set
   console.log("");
 });
 function buildPluginXml(pluginDir, destFile) {
-  const srcDir = path3.join(pluginDir, "src");
-  if (!fs4.existsSync(srcDir)) {
+  const srcDir = path4.join(pluginDir, "src");
+  if (!fs5.existsSync(srcDir)) {
     console.log(`  \u2716 Plugin source directory not found: ${srcDir}`);
     return;
   }
-  const files = fs4.readdirSync(srcDir).filter((f) => f.endsWith(".lua"));
+  const files = fs5.readdirSync(srcDir).filter((f) => f.endsWith(".lua"));
   const mainFile = files.find((f) => f === "init.server.lua");
   const moduleFiles = files.filter((f) => f !== "init.server.lua");
   if (!mainFile) {
     console.log("  \u2716 init.server.lua not found in plugin source");
     return;
   }
-  const mainSource = fs4.readFileSync(path3.join(srcDir, mainFile), "utf-8");
+  const mainSource = fs5.readFileSync(path4.join(srcDir, mainFile), "utf-8");
   let moduleItems = "";
   for (const mf of moduleFiles) {
     const modName = mf.replace(".lua", "");
-    const modSource = fs4.readFileSync(path3.join(srcDir, mf), "utf-8");
+    const modSource = fs5.readFileSync(path4.join(srcDir, mf), "utf-8");
     moduleItems += `
       <Item class="ModuleScript" referent="${modName}">
         <Properties>
@@ -4653,7 +4762,7 @@ function buildPluginXml(pluginDir, destFile) {
     </Properties>${moduleItems}
   </Item>
 </roblox>`;
-  fs4.writeFileSync(destFile, rbxmx, "utf-8");
+  fs5.writeFileSync(destFile, rbxmx, "utf-8");
   console.log(`  \u2714 Plugin installed to: ${destFile}`);
   console.log("  Restart Roblox Studio to load the plugin.");
 }