construct-shader-graph-mcp 0.3.0 → 0.5.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "construct-shader-graph-mcp",
-  "version": "0.3.0",
+  "version": "0.5.0",
   "description": "Standalone MCP server for Construct Shader Graph",
   "type": "module",
   "files": [

package/src/server.mjs

@@ -22,6 +22,7 @@ let localServer = null;
 let bridge = null;
 let controlServer = null;
 let isPrimaryInstance = false;
+let promotionInFlight = null;
 
 function log(message, ...args) {
   console.error(`[construct-shader-graph-mcp] ${message}`, ...args);
@@ -227,7 +228,9 @@ function createToolDefinitions() {
         description:
           "Choose which connected shader graph tab future MCP calls should target.",
         inputSchema: {
-          sessionId: z.string().describe("Session id returned by list_projects."),
+          sessionId: z
+            .string()
+            .describe("Session id returned by list_projects."),
         },
         outputSchema: {
           sessionId: z.string(),
@@ -291,7 +294,9 @@ function createToolDefinitions() {
             .describe("Optional session id; defaults to the selected project."),
           method: z
             .string()
-            .describe("Manifest method path, for example nodes.create or shader.getInfo."),
+            .describe(
+              "Manifest method path, for example nodes.create or shader.getInfo.",
+            ),
           args: z
             .array(z.any())
             .optional()
@@ -397,12 +402,15 @@ function registerPrompts(server) {
     "inspect-graph",
     {
       title: "Inspect Graph",
-      description: "Prompt for safely inspecting the current graph before any edits.",
+      description:
+        "Prompt for safely inspecting the current graph before any edits.",
       argsSchema: z.object({
         focus: z
           .string()
           .optional()
-          .describe("Optional area to inspect, like uniforms, preview, or node types."),
+          .describe(
+            "Optional area to inspect, like uniforms, preview, or node types.",
+          ),
       }),
     },
     ({ focus }) => ({
@@ -447,7 +455,10 @@ function registerPrompts(server) {
       description:
         "Prompt for debugging generated code or preview issues in a shader graph project.",
       argsSchema: z.object({
-        issue: z.string().optional().describe("Optional description of the observed preview issue."),
+        issue: z
+          .string()
+          .optional()
+          .describe("Optional description of the observed preview issue."),
       }),
     },
     ({ issue }) => ({
@@ -480,7 +491,6 @@ function createLocalServer() {
 }
 
 async function startPrimaryBackend() {
-  isPrimaryInstance = true;
   localServer = createLocalServer();
 
   bridge = new WebSocketServer({ noServer: true });
@@ -541,7 +551,9 @@ async function startPrimaryBackend() {
           selectedSessionId = sessionId;
         }
 
-        log(`registered ${sessionId} (${session.project?.name || "Untitled Shader"})`);
+        log(
+          `registered ${sessionId} (${session.project?.name || "Untitled Shader"})`,
+        );
         sendWsJson(socket, {
           type: "registered",
           sessionId,
@@ -631,9 +643,15 @@ async function startPrimaryBackend() {
         return;
       }
 
-      const tool = createToolDefinitions().find((entry) => entry.name === request.tool);
+      const tool = createToolDefinitions().find(
+        (entry) => entry.name === request.tool,
+      );
       if (!tool) {
-        sendJson(socket, { id: request.id, ok: false, error: `Unknown tool '${request.tool}'` });
+        sendJson(socket, {
+          id: request.id,
+          ok: false,
+          error: `Unknown tool '${request.tool}'`,
+        });
         return;
       }
 
@@ -650,14 +668,25 @@ async function startPrimaryBackend() {
     });
   });
 
-  await new Promise((resolve, reject) => {
-    controlServer.once("error", reject);
-    controlServer.listen(CONTROL_PORT, "127.0.0.1", () => {
-      controlServer.off("error", reject);
-      resolve();
+  try {
+    await new Promise((resolve, reject) => {
+      controlServer.once("error", reject);
+      controlServer.listen(CONTROL_PORT, "127.0.0.1", () => {
+        controlServer.off("error", reject);
+        resolve();
+      });
     });
-  });
+  } catch (err) {
+    httpServer.close();
+    bridge.close();
+    throw err;
+  }
+
+  // Permanent error handlers to prevent unhandled error crashes
+  httpServer.on("error", (err) => log("bridge http error:", err.message));
+  controlServer.on("error", (err) => log("control server error:", err.message));
 
+  isPrimaryInstance = true;
   log(`control listening on tcp://127.0.0.1:${CONTROL_PORT}`);
 }
 
@@ -678,9 +707,12 @@ function createProxyServer() {
   return server;
 }
 
-function callPrimaryTool(tool, input) {
+function callPrimaryToolDirect(tool, input) {
   return new Promise((resolve, reject) => {
-    const socket = net.createConnection({ host: "127.0.0.1", port: CONTROL_PORT });
+    const socket = net.createConnection({
+      host: "127.0.0.1",
+      port: CONTROL_PORT,
+    });
     const requestId = `rpc_${Date.now()}_${Math.random().toString(16).slice(2)}`;
     const rl = readline.createInterface({ input: socket });
 
@@ -720,6 +752,66 @@ function callPrimaryTool(tool, input) {
   });
 }
 
+async function tryPromoteToPrimary() {
+  if (isPrimaryInstance) return true;
+
+  // Deduplicate concurrent promotion attempts
+  if (promotionInFlight) return promotionInFlight;
+
+  promotionInFlight = (async () => {
+    try {
+      await startPrimaryBackend();
+      log("promoted to primary instance");
+      return true;
+    } catch (err) {
+      if (err?.code === "EADDRINUSE") {
+        log("promotion failed: another primary appeared");
+      } else {
+        log("promotion failed:", err.message);
+      }
+      // Reset partial state
+      isPrimaryInstance = false;
+      localServer = null;
+      bridge = null;
+      controlServer = null;
+      return false;
+    } finally {
+      promotionInFlight = null;
+    }
+  })();
+
+  return promotionInFlight;
+}
+
+function executeToolLocally(toolName, input) {
+  const tool = createToolDefinitions().find((t) => t.name === toolName);
+  if (!tool) throw new Error(`Unknown tool '${toolName}'`);
+  return tool.handler(input || {});
+}
+
+async function callPrimaryTool(toolName, input) {
+  // If we're already promoted, go local
+  if (isPrimaryInstance) {
+    return executeToolLocally(toolName, input);
+  }
+
+  try {
+    return await callPrimaryToolDirect(toolName, input);
+  } catch (error) {
+    // If connection refused, the primary is gone — try to take over
+    if (error?.code === "ECONNREFUSED") {
+      log("primary unreachable, attempting promotion...");
+      const promoted = await tryPromoteToPrimary();
+      if (promoted) {
+        return executeToolLocally(toolName, input);
+      }
+      // Another primary appeared while we promoted — retry via proxy
+      return callPrimaryToolDirect(toolName, input);
+    }
+    throw error;
+  }
+}
+
 async function ensureBackend() {
   try {
     await startPrimaryBackend();
@@ -728,6 +820,12 @@ async function ensureBackend() {
       throw error;
     }
 
+    // Clean up any partially created resources
+    isPrimaryInstance = false;
+    localServer = null;
+    bridge = null;
+    controlServer = null;
+
     log(`bridge already running on ${BRIDGE_PORT}; starting follower proxy`);
   }
 }