@createlex/figma-swiftui-mcp 1.0.8 → 1.0.9

package/companion/mcp-server.mjs

@@ -30,6 +30,8 @@ const BRIDGE_HTTP_URL = process.env.FIGMA_SWIFTUI_BRIDGE_HTTP_URL || 'http://loc
 const BRIDGE_WS_URL = process.env.FIGMA_SWIFTUI_BRIDGE_WS_URL || 'ws://localhost:7765/bridge';
 const REQUEST_TIMEOUT_MS = Number(process.env.FIGMA_SWIFTUI_BRIDGE_TIMEOUT_MS || 30000);
 const AUTH_REVALIDATION_INTERVAL_MS = Number(process.env.FIGMA_SWIFTUI_AUTH_REVALIDATION_MS || (10 * 60 * 1000));
+const RESPONSE_SIZE_CAP = Number(process.env.FIGMA_SWIFTUI_RESPONSE_SIZE_CAP || 102400); // 100 KB default
+const MAX_DIAGNOSTICS = 20;
 
 let bridgeSocket = null;
 let connectPromise = null;
@@ -666,7 +668,7 @@ server.registerTool('write_generated_swiftui_to_xcode', {
   }
 
   return jsonResult({
-    ...result,
+    ...stripImageData(result),
     structName: effectiveStructName,
   });
 });
@@ -739,16 +741,125 @@ server.registerTool('write_selection_to_xcode', {
     filePath: result.results?.swiftFile ?? null,
   });
 
-  return jsonResult({
-    ...result,
+  return jsonResult(buildCompactResponse({
+    result,
     structName: effectiveStructName,
     selection: generated.selection ?? null,
     diagnostics: generated.diagnostics ?? [],
     refinementInstructions,
-    ...(analysisHints ? { analysisHints } : {}),
-  });
+    analysisHints,
+  }));
 });
 
+// ---------------------------------------------------------------------------
+// Response-size mitigation helpers
+// ---------------------------------------------------------------------------
+
+/**
+ * Remove base64 / svg binary data from any image results.
+ * Files are already on disk — the model only needs names and paths.
+ */
+function stripImageData(obj) {
+  if (!obj || typeof obj !== 'object') return obj;
+
+  // Deep-clone to avoid mutating the original
+  const clone = JSON.parse(JSON.stringify(obj));
+
+  const walk = (node) => {
+    if (Array.isArray(node)) {
+      node.forEach(walk);
+      return;
+    }
+    if (node && typeof node === 'object') {
+      delete node.base64;
+      delete node.svg;
+      delete node.data;
+      for (const value of Object.values(node)) {
+        walk(value);
+      }
+    }
+  };
+
+  walk(clone);
+  return clone;
+}
+
+/**
+ * Cap the diagnostics array, prioritising rasterized nodes first.
+ */
+function capDiagnostics(diagnostics) {
+  if (!Array.isArray(diagnostics) || diagnostics.length <= MAX_DIAGNOSTICS) {
+    return diagnostics;
+  }
+
+  // Sort: rasterized nodes first (most useful for refinement)
+  const sorted = [...diagnostics].sort((a, b) => {
+    const aRaster = a.rasterized || a.reason?.toLowerCase().includes('raster') ? 1 : 0;
+    const bRaster = b.rasterized || b.reason?.toLowerCase().includes('raster') ? 1 : 0;
+    return bRaster - aRaster;
+  });
+
+  const kept = sorted.slice(0, MAX_DIAGNOSTICS);
+  kept.push({
+    _truncated: true,
+    totalCount: diagnostics.length,
+    shownCount: MAX_DIAGNOSTICS,
+    message: `${diagnostics.length - MAX_DIAGNOSTICS} additional diagnostic(s) omitted to reduce response size.`,
+  });
+  return kept;
+}
+
+/**
+ * Build a compact MCP response for write_selection_to_xcode.
+ *
+ * Strategy:
+ *  1. Strip all binary (base64/svg) data from image results.
+ *  2. Cap diagnostics to MAX_DIAGNOSTICS most relevant entries.
+ *  3. Estimate serialised size; if still over RESPONSE_SIZE_CAP,
+ *     compress analysis hints to summary-only.
+ */
+function buildCompactResponse({ result, structName, selection, diagnostics, refinementInstructions, analysisHints }) {
+  const compact = {
+    ...stripImageData(result),
+    structName,
+    selection,
+    diagnostics: capDiagnostics(diagnostics),
+    refinementInstructions,
+    ...(analysisHints ? { analysisHints } : {}),
+  };
+
+  // First size check
+  let serialised = JSON.stringify(compact);
+  if (serialised.length <= RESPONSE_SIZE_CAP) {
+    return compact;
+  }
+
+  // Compress: remove verbose analysis hints, keep only summary-level data
+  if (compact.analysisHints) {
+    const { generationHints, manualRefinementHints, ...rest } = compact.analysisHints;
+    compact.analysisHints = {
+      ...rest,
+      _compressed: true,
+      generationHintCount: generationHints?.interactiveElements?.length ?? 0,
+      manualRefinementHintCount: manualRefinementHints?.length ?? 0,
+    };
+  }
+
+  // Second size check — drop the full Swift code echo if still too large
+  serialised = JSON.stringify(compact);
+  if (serialised.length > RESPONSE_SIZE_CAP && compact.results?.swiftFile) {
+    // The code is already on disk at swiftFile; no need to echo it
+    if (compact.results) {
+      compact.results = { ...compact.results };
+      delete compact.results.code;
+    }
+    compact._responseCapped = true;
+    compact._responseNote = `Response was compressed to stay under ${Math.round(RESPONSE_SIZE_CAP / 1024)}KB. All files were written successfully to disk.`;
+  }
+
+  return compact;
+}
+
 function buildRefinementInstructions({ diagnostics, rasterNodes, analysisHints, structName, filePath }) {
   const instructions = [];
 

package/companion/package.json

@@ -1,6 +1,6 @@
 {
   "name": "figma-swiftui-companion",
-  "version": "1.0.0",
+  "version": "1.0.9",
   "description": "Local server that writes Figma-generated SwiftUI code and images into an Xcode project",
   "main": "server.js",
   "scripts": {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@createlex/figma-swiftui-mcp",
-  "version": "1.0.8",
+  "version": "1.0.9",
   "description": "CreateLex MCP runtime for Figma-to-SwiftUI generation and Xcode export",
   "bin": {
     "figma-swiftui-mcp": "bin/figma-swiftui-mcp.js"