@nbakka/mcp-appium 2.0.41 → 2.0.43

package/lib/server.js

@@ -476,11 +476,12 @@ const createMcpServer = () => {
 
     tool(
       "mobile_export_figma_images",
-      "Export all frames from Figma file as PNG images with sequential naming",
+      "Export only main parent frames from Figma file as PNG images",
       {
-        figmaUrl: zod_1.z.string().describe("The Figma file URL to export frames from")
+        figmaUrl: zod_1.z.string().describe("The Figma file URL to export frames from"),
+        nodeId: zod_1.z.string().optional().describe("Specific node ID from URL (e.g., 13223-1724)")
       },
-      async ({ figmaUrl }) => {
+      async ({ figmaUrl, nodeId }) => {
         try {
           // Read Figma credentials from desktop/figma.json file
           const figmaConfigPath = path.join(os.homedir(), 'Desktop', 'figma.json');
@@ -500,12 +501,18 @@ const createMcpServer = () => {
             throw new Error('Figma API token not found in figma.json file. Please ensure the file contains "token" field.');
           }
 
-          // Extract file ID from Figma URL
+          // Extract file ID and node ID from Figma URL
           const fileId = extractFileIdFromUrl(figmaUrl);
           if (!fileId) {
             throw new Error('Invalid Figma URL. Unable to extract file ID.');
           }
 
+          // Extract node ID from URL if not provided separately
+          let targetNodeId = nodeId;
+          if (!targetNodeId) {
+            targetNodeId = extractNodeIdFromUrl(figmaUrl);
+          }
+
           // Set up export directory
           const exportPath = path.join(os.homedir(), 'Desktop', 'figma');
 
@@ -521,11 +528,9 @@ const createMcpServer = () => {
             }
           } catch (cleanupError) {
             // Continue if cleanup fails
-            console.log('Could not clean existing files:', cleanupError.message);
           }
 
           // Get file information
-          console.log('Fetching Figma file information...');
           const fileResponse = await axios.get(
             `https://api.figma.com/v1/files/${fileId}`,
             {
@@ -537,62 +542,87 @@ const createMcpServer = () => {
 
           const fileData = fileResponse.data;
 
-          // Extract only frame node IDs
-          const frameIds = extractFrameNodes(fileData.document);
+          let framesToExport = [];
 
-          if (frameIds.length === 0) {
-            return `No frames found in Figma file.`;
+          if (targetNodeId) {
+            // If specific node ID provided, find that node and export it (not its children)
+            const specificNode = findNodeById(fileData.document, targetNodeId);
+            if (specificNode && specificNode.type === 'FRAME') {
+              framesToExport = [specificNode.id];
+            } else {
+              return `Node ID ${targetNodeId} not found or is not a frame.`;
+            }
+          } else {
+            // Extract only main/parent frames (large frames, likely screens/pages)
+            framesToExport = extractMainFrames(fileData.document);
           }
 
-          console.log(`Found ${frameIds.length} frames. Starting export...`);
+          if (framesToExport.length === 0) {
+            return `No main frames found in Figma file.`;
+          }
 
-          // Request image exports for frames only
-          const exportResponse = await axios.get(
-            `https://api.figma.com/v1/images/${fileId}`,
-            {
-              headers: {
-                'X-Figma-Token': figmaToken
-              },
-              params: {
-                ids: frameIds.join(','),
-                format: 'png',
-                scale: 2
-              }
-            }
-          );
+          // Limit to reasonable number to avoid API issues
+          const limitedFrameIds = framesToExport.slice(0, 10);
 
-          const imageUrls = exportResponse.data.images;
           let successCount = 0;
+          const batchSize = 5; // Smaller batches for main frames
 
-          // Download and save each image with sequential naming
-          for (let i = 0; i < frameIds.length; i++) {
-            const nodeId = frameIds[i];
-            const imageUrl = imageUrls[nodeId];
-
-            if (imageUrl) {
-              try {
-                const filename = `${i + 1}.png`;
-                const filepath = path.join(exportPath, filename);
-
-                // Download image
-                const imageResponse = await axios.get(imageUrl, {
-                  responseType: 'arraybuffer'
-                });
+          // Process frames in batches
+          for (let batchStart = 0; batchStart < limitedFrameIds.length; batchStart += batchSize) {
+            const batchEnd = Math.min(batchStart + batchSize, limitedFrameIds.length);
+            const batchIds = limitedFrameIds.slice(batchStart, batchEnd);
 
-                // Save image to file
-                await fs.writeFile(filepath, imageResponse.data);
-                successCount++;
-
-                console.log(`Exported: ${filename}`);
-              } catch (downloadError) {
-                console.error(`Failed to download frame ${i + 1}:`, downloadError.message);
+            try {
+              // Request image exports for current batch
+              const exportResponse = await axios.get(
+                `https://api.figma.com/v1/images/${fileId}`,
+                {
+                  headers: {
+                    'X-Figma-Token': figmaToken
+                  },
+                  params: {
+                    ids: batchIds.join(','),
+                    format: 'png',
+                    scale: 2
+                  }
+                }
+              );
+
+              const imageUrls = exportResponse.data.images;
+
+              // Download and save each image in the batch
+              for (let i = 0; i < batchIds.length; i++) {
+                const nodeId = batchIds[i];
+                const imageUrl = imageUrls[nodeId];
+
+                if (imageUrl) {
+                  try {
+                    const frameNumber = batchStart + i + 1;
+                    const filename = `${frameNumber}.png`;
+                    const filepath = path.join(exportPath, filename);
+
+                    // Download image
+                    const imageResponse = await axios.get(imageUrl, {
+                      responseType: 'arraybuffer'
+                    });
+
+                    // Save image to file
+                    await fs.writeFile(filepath, imageResponse.data);
+                    successCount++;
+                  } catch (downloadError) {
+                    // Continue with next image if one fails
+                  }
+                }
               }
+            } catch (batchError) {
+              // Continue with next batch if one fails
             }
           }
 
           return `Figma Export Complete!
     Export Path: ${exportPath}
-    Successfully exported: ${successCount}/${frameIds.length} frames as PNG files
+    Total main frames found: ${framesToExport.length}
+    Exported: ${successCount} main frames as PNG files
     Files: ${Array.from({length: successCount}, (_, i) => `${i + 1}.png`).join(', ')}`;
 
         } catch (error) {
@@ -624,23 +654,30 @@ const createMcpServer = () => {
       return null;
     }
 
-    // Helper function to extract only frame node IDs
-    function extractFrameNodes(document) {
-      const frameIds = [];
+    // Helper function to extract node ID from Figma URL
+    function extractNodeIdFromUrl(url) {
+      const match = url.match(/node-id=([^&]+)/);
+      if (match) {
+        return match[1];
+      }
+      return null;
+    }
+
+    // Helper function to find a specific node by ID
+    function findNodeById(document, nodeId) {
+      let foundNode = null;
 
       function traverse(node) {
-        // Only collect frames, not components
-        if (node.type === 'FRAME') {
-          frameIds.push(node.id);
+        if (node.id === nodeId) {
+          foundNode = node;
+          return;
         }
 
-        // Continue traversing children
         if (node.children && Array.isArray(node.children)) {
           node.children.forEach(traverse);
         }
       }
 
-      // Start traversal from document children (pages)
       if (document.children && Array.isArray(document.children)) {
         document.children.forEach(page => {
           if (page.children && Array.isArray(page.children)) {
@@ -649,7 +686,67 @@ const createMcpServer = () => {
         });
       }
 
-      return frameIds;
+      return foundNode;
+    }
+
+    // Helper function to extract only main/parent frames (not small UI components)
+    function extractMainFrames(document) {
+      const mainFrames = [];
+
+      function traversePage(node, depth = 0) {
+        if (node.type === 'FRAME') {
+          // Consider frames as "main" if they are:
+          // 1. Top-level frames on a page (depth 0)
+          // 2. Large frames (width > 300 or height > 300)
+          // 3. Frames that don't have many small children (likely screens, not component libraries)
+
+          const isTopLevel = depth === 0;
+          const isLargeFrame = (node.absoluteBoundingBox?.width > 300) || (node.absoluteBoundingBox?.height > 300);
+          const childFrameCount = countChildFrames(node);
+          const isNotComponentLibrary = childFrameCount < 50; // Avoid component libraries with many small frames
+
+          if (isTopLevel || (isLargeFrame && isNotComponentLibrary)) {
+            mainFrames.push(node.id);
+            return; // Don't traverse children of main frames to avoid getting sub-components
+          }
+        }
+
+        // Continue traversing children for non-main-frame nodes
+        if (node.children && Array.isArray(node.children)) {
+          node.children.forEach(child => traversePage(child, depth + 1));
+        }
+      }
+
+      // Start traversal from document children (pages)
+      if (document.children && Array.isArray(document.children)) {
+        document.children.forEach(page => {
+          if (page.children && Array.isArray(page.children)) {
+            page.children.forEach(child => traversePage(child, 0));
+          }
+        });
+      }
+
+      return mainFrames;
+    }
+
+    // Helper function to count child frames (to detect component libraries)
+    function countChildFrames(node) {
+      let count = 0;
+
+      function traverse(n) {
+        if (n.type === 'FRAME') {
+          count++;
+        }
+        if (n.children && Array.isArray(n.children)) {
+          n.children.forEach(traverse);
+        }
+      }
+
+      if (node.children && Array.isArray(node.children)) {
+        node.children.forEach(traverse);
+      }
+
+      return count;
     }
 
     return server;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nbakka/mcp-appium",
-  "version": "2.0.41",
+  "version": "2.0.43",
   "description": "Appium MCP",
   "engines": {
     "node": ">=18"