@createlex/figma-swiftui-mcp 1.2.5 → 1.2.8

package/companion/bridge-server.cjs

@@ -13,6 +13,7 @@ const {
   setSavedProjectPath,
   writeSwiftUIScreen,
 } = require('./xcode-writer.cjs');
+const { buildGenerationPrompt } = require('./local-llm-generator.cjs');
 
 const DEFAULT_PORT = 7765;
 const DEFAULT_HOST = 'localhost';
@@ -114,7 +115,25 @@ function startBridgeServer(options = {}) {
   app.use(cors({ origin: '*' }));
   app.use(express.json({ limit: '50mb' }));
 
+  // Auth state exposed to the plugin UI via /ping
+  let authRequired = false;
+  let authRequiredReason = null;
+
+  function setAuthRequired(reason) {
+    authRequired = true;
+    authRequiredReason = reason || 'Token expired';
+    logger.warn(`[figma-swiftui-bridge] Auth required: ${authRequiredReason}`);
+  }
+
   app.get('/ping', (req, res) => {
+    if (authRequired) {
+      return res.status(401).json({
+        ok: false,
+        authRequired: true,
+        authRequiredReason,
+        loginCommand: 'npx @createlex/figma-swiftui-mcp login',
+      });
+    }
     res.json({
       ok: true,
       projectPath: projectPath || null,
@@ -123,6 +142,13 @@ function startBridgeServer(options = {}) {
     });
   });
 
+  // Graceful shutdown — called by the plugin UI "Disconnect MCP" to stop the runtime
+  app.post('/shutdown', (req, res) => {
+    res.json({ ok: true, message: 'Shutting down bridge server…' });
+    logger.info('[figma-swiftui-bridge] Shutdown requested by plugin UI');
+    setTimeout(() => process.exit(0), 400);
+  });
+
   app.get('/bridge/info', (req, res) => {
     res.json({
       ok: true,
@@ -278,21 +304,35 @@ function startBridgeServer(options = {}) {
 
     const data = response.data || {};
 
-    // Attach explicit AI workflow instructions so any agent that fetches this URL
-    // knows exactly how to generate and write correct SwiftUI code with real images.
+    // Build the full generation prompt so AI tools without MCP access
+    // can still generate proper native SwiftUI with individual elements.
+    let generationPrompt = null;
+    try {
+      generationPrompt = buildGenerationPrompt(data);
+    } catch (_) {
+      // Non-fatal — prompt is a bonus, not required
+    }
+
     const assetCount = data.assetExportPlan?.candidates?.length ?? 0;
     const aiWorkflow = {
-      step1: 'You have the Figma design context. Do NOT generate SwiftUI code from scratch — that produces grey placeholder rectangles instead of real images.',
-      step2: 'Call the MCP tool `write_selection_to_xcode`. It generates SwiftUI with correct Image("name") asset references AND writes the PNG files to Assets.xcassets automatically.',
-      step3: 'After write_selection_to_xcode returns, ONLY refine the code it produced: add GeometryReader for responsiveness, wire interactions. DO NOT rewrite Image("name") calls or replace them with Rectangle()/shapes.',
-      step4: 'If the caller has no MCP access, use the structName and assetExportPlan below to write SwiftUI that references Image("name") for every asset candidate listed.',
-      mcpTool: 'write_selection_to_xcode',
-      mcpServer: 'figma-swiftui',
+      IF_YOU_HAVE_MCP_ACCESS: {
+        step1: 'Call the MCP tool `write_selection_to_xcode` (server: figma-swiftui). It generates SwiftUI with correct Image("name") asset references AND writes the PNG files to Assets.xcassets automatically.',
+        step2: 'After write_selection_to_xcode returns, ONLY refine the code it produced: add GeometryReader for responsiveness, wire @State interactions. DO NOT rewrite Image("name") calls or replace them with Rectangle()/shapes.',
+        mcpTool: 'write_selection_to_xcode',
+        mcpServer: 'figma-swiftui',
+      },
+      IF_NO_MCP_ACCESS: {
+        instruction: 'Use _generationPrompt.systemPrompt as the system message and _generationPrompt.userMessage as the user message to generate native SwiftUI code. Output individual SwiftUI views — Text(), VStack, HStack, ZStack, Image("name") — NOT a single composite Image().',
+        rule1: 'Every asset in assetExportPlan.candidates MUST become Image("assetName").resizable() — never Rectangle() or Color.',
+        rule2: 'Text nodes become Text("content") with .font(.system(size: X, weight: .bold)) — NEVER custom font names.',
+        rule3: 'Output a full SwiftUI View struct with individual elements positioned using the node geometry in the design context.',
+        rule4: 'End with #Preview { StructName() }.',
+      },
       assetCandidateCount: assetCount,
-      warning: 'Photos, 3D renders, illustrations and icons are pre-exported as PNG assets. Always use Image("name") — never reconstruct them with SwiftUI shapes or colors.',
+      warning: 'DO NOT generate a single Image() wrapping the whole frame. Generate individual SwiftUI elements for each node.',
     };
 
-    res.json({ ok: true, ...data, _aiWorkflow: aiWorkflow });
+    res.json({ ok: true, ...data, _generationPrompt: generationPrompt, _aiWorkflow: aiWorkflow });
   });
 
   function sendBridgeMessage(ws, payload) {
@@ -667,6 +707,7 @@ function startBridgeServer(options = {}) {
                 host,
                 alreadyRunning: false,
                 getProjectPath: () => projectPath,
+                setAuthRequired,
                 getBridgeInfo: () => ({
                   protocolVersion: BRIDGE_PROTOCOL_VERSION,
                   pluginConnected: !!pluginBridgeClient,
@@ -712,6 +753,7 @@ function startBridgeServer(options = {}) {
         host,
         alreadyRunning: false,
         getProjectPath: () => projectPath,
+        setAuthRequired,
         getBridgeInfo: () => ({
           protocolVersion: BRIDGE_PROTOCOL_VERSION,
           pluginConnected: !!pluginBridgeClient,

package/companion/mcp-server.mjs

@@ -765,11 +765,11 @@ server.registerTool('write_generated_swiftui_to_xcode', {
 });
 
 server.registerTool('write_selection_to_xcode', {
-  description: 'Generate SwiftUI from the connected Figma selection, write it into the configured Xcode project, and return refinement hints. After writing, the AI assistant SHOULD automatically apply the returned refinementInstructions to make the generated code adaptive and interactive — replacing raster placeholders with native SwiftUI controls, adding GeometryReader for adaptive layout, and wiring basic interactions.',
+  description: 'Generate SwiftUI from the connected Figma selection and write it into the configured Xcode project. THIS IS THE CORRECT TOOL TO CALL — it exports real PNG assets to Assets.xcassets and generates Image("name") references automatically. Use generationMode="fidelity" for designs with blend modes, 3D renders, complex shadows, or photographic images — fidelity rasterizes the whole frame as one pixel-perfect PNG instead of trying to reconstruct it with shapes.',
   inputSchema: {
     nodeIds: z.array(z.string()).optional().describe('Optional list of Figma node ids. If omitted, uses the current selection'),
     includeOverflow: z.boolean().default(false).describe('Ignore Figma clipping when generating layout'),
-    generationMode: z.enum(['editable', 'fidelity']).default('editable').describe('Editable keeps more native SwiftUI structure; fidelity rasterizes more complex layouts'),
+    generationMode: z.enum(['editable', 'fidelity']).default('editable').describe('Use fidelity for designs with blend modes, 3D renders, complex shadows, or photographic content — rasterizes the whole frame as one pixel-perfect PNG. Use editable for UI screens with native controls (buttons, lists, forms).'),
     projectPath: z.string().optional().describe('Optional Xcode source folder override'),
   },
 }, async ({ nodeIds, includeOverflow, generationMode, projectPath }) => {

package/companion/server.js

@@ -30,12 +30,13 @@ async function main() {
       const validation = await validateRuntimeSession(authState);
       if (!validation.valid) {
         console.error(`[figma-swiftui-bridge] Authorization lost: ${validation.error}`);
-        if (bridgeRuntime && typeof bridgeRuntime.close === 'function' && !bridgeRuntime.alreadyRunning) {
-          await bridgeRuntime.close().catch((error) => {
-            console.error('[figma-swiftui-bridge] Failed to close bridge cleanly:', error.message);
-          });
+        // Signal the bridge to return authRequired from /ping so the plugin UI
+        // can show a "Login required" panel instead of just "runtime off".
+        if (bridgeRuntime && typeof bridgeRuntime.setAuthRequired === 'function') {
+          bridgeRuntime.setAuthRequired(validation.error || 'Token expired. Run: npx @createlex/figma-swiftui-mcp login');
+        } else {
+          process.exit(1);
         }
-        process.exit(1);
         return;
       }
 
@@ -44,13 +45,13 @@ async function main() {
         console.log('[figma-swiftui-bridge] Refreshed CreateLex MCP authorization');
       }
     } catch (error) {
-      console.error(`[figma-swiftui-bridge] Authorization revalidation failed: ${error instanceof Error ? error.message : 'unknown_error'}`);
-      if (bridgeRuntime && typeof bridgeRuntime.close === 'function' && !bridgeRuntime.alreadyRunning) {
-        await bridgeRuntime.close().catch((closeError) => {
-          console.error('[figma-swiftui-bridge] Failed to close bridge cleanly:', closeError.message);
-        });
+      const reason = error instanceof Error ? error.message : 'unknown_error';
+      console.error(`[figma-swiftui-bridge] Authorization revalidation failed: ${reason}`);
+      if (bridgeRuntime && typeof bridgeRuntime.setAuthRequired === 'function') {
+        bridgeRuntime.setAuthRequired(`Session revalidation failed: ${reason}`);
+      } else {
+        process.exit(1);
       }
-      process.exit(1);
     }
   }, AUTH_REVALIDATION_INTERVAL_MS);
 

package/package.json

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