@nbakka/mcp-appium 2.0.44 → 2.0.46

package/lib/server.js

@@ -474,280 +474,117 @@ const createMcpServer = () => {
       return [...new Set(figmaLinks)];
     }
 
-    tool(
-      "mobile_export_figma_images",
-      "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"),
-        nodeId: zod_1.z.string().optional().describe("Specific node ID from URL (e.g., 13223-1724)")
-      },
-      async ({ figmaUrl, nodeId }) => {
-        try {
-          // Read Figma credentials from desktop/figma.json file
-          const figmaConfigPath = path.join(os.homedir(), 'Desktop', 'figma.json');
-
-          let figmaConfig;
-          try {
-            const configContent = await fs.readFile(figmaConfigPath, 'utf-8');
-            figmaConfig = JSON.parse(configContent);
-          } catch (error) {
-            throw new Error(`Failed to read Figma config from ${figmaConfigPath}: ${error.message}`);
-          }
-
-          // Extract API token from config
-          const { token: figmaToken } = figmaConfig;
-
-          if (!figmaToken) {
-            throw new Error('Figma API token not found in figma.json file. Please ensure the file contains "token" field.');
-          }
-
-          // 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');
-
-          // Create export directory if it doesn't exist
-          await fs.mkdir(exportPath, { recursive: true });
-
-          // Clear existing PNG files in the directory
-          try {
-            const existingFiles = await fs.readdir(exportPath);
-            const pngFiles = existingFiles.filter(file => file.endsWith('.png'));
-            for (const file of pngFiles) {
-              await fs.unlink(path.join(exportPath, file));
-            }
-          } catch (cleanupError) {
-            // Continue if cleanup fails
-          }
-
-          // Get file information
-          const fileResponse = await axios.get(
-            `https://api.figma.com/v1/files/${fileId}`,
-            {
-              headers: {
-                'X-Figma-Token': figmaToken
-              }
-            }
-          );
-
-          const fileData = fileResponse.data;
-
-          let framesToExport = [];
-
-          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);
-          }
-
-          if (framesToExport.length === 0) {
-            return `No main frames found in Figma file.`;
-          }
-
-          // Limit to reasonable number to avoid API issues
-          const limitedFrameIds = framesToExport.slice(0, 10);
-
-          let successCount = 0;
-          const batchSize = 5; // Smaller batches for main frames
-
-          // 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);
-
-            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}
-    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) {
-          if (error.response && error.response.status === 403) {
-            return `Error: Access denied. Please check your Figma API token and file permissions.`;
-          } else if (error.response && error.response.status === 404) {
-            return `Error: Figma file not found. Please check the URL and ensure the file is accessible.`;
-          } else {
-            return `Error exporting from Figma: ${error.message}`;
-          }
-        }
+tool(
+  "mobile_export_figma_images",
+  "Export Figma file as PDF",
+  {
+    figmaUrl: zod_1.z.string().describe("The Figma file URL to export as PDF")
+  },
+  async ({ figmaUrl }) => {
+    try {
+      // Read Figma credentials from desktop/figma.json file
+      const figmaConfigPath = path.join(os.homedir(), 'Desktop', 'figma.json');
+
+      let figmaConfig;
+      try {
+        const configContent = await fs.readFile(figmaConfigPath, 'utf-8');
+        figmaConfig = JSON.parse(configContent);
+      } catch (error) {
+        throw new Error(`Failed to read Figma config from ${figmaConfigPath}: ${error.message}`);
       }
-    );
 
-    // Helper function to extract file ID from Figma URL
-    function extractFileIdFromUrl(url) {
-      const patterns = [
-        /figma\.com\/file\/([a-zA-Z0-9]+)/,
-        /figma\.com\/design\/([a-zA-Z0-9]+)/,
-        /figma\.com\/proto\/([a-zA-Z0-9]+)/
-      ];
-
-      for (const pattern of patterns) {
-        const match = url.match(pattern);
-        if (match) {
-          return match[1];
-        }
-      }
-      return null;
-    }
+      // Extract API token from config
+      const { token: figmaToken } = figmaConfig;
 
-    // Helper function to extract node ID from Figma URL
-    function extractNodeIdFromUrl(url) {
-      const match = url.match(/node-id=([^&]+)/);
-      if (match) {
-        return match[1];
+      if (!figmaToken) {
+        throw new Error('Figma API token not found in figma.json file. Please ensure the file contains "token" field.');
       }
-      return null;
-    }
-
-    // Helper function to find a specific node by ID
-    function findNodeById(document, nodeId) {
-      let foundNode = null;
 
-      function traverse(node) {
-        if (node.id === nodeId) {
-          foundNode = node;
-          return;
-        }
-
-        if (node.children && Array.isArray(node.children)) {
-          node.children.forEach(traverse);
-        }
-      }
-
-      if (document.children && Array.isArray(document.children)) {
-        document.children.forEach(page => {
-          if (page.children && Array.isArray(page.children)) {
-            page.children.forEach(traverse);
-          }
-        });
+      // Extract file ID from Figma URL
+      const fileId = extractFileIdFromUrl(figmaUrl);
+      if (!fileId) {
+        throw new Error('Invalid Figma URL. Unable to extract file ID.');
       }
 
-      return foundNode;
-    }
+      // Set up export directory
+      const exportPath = path.join(os.homedir(), 'Desktop', 'figma');
 
-    // 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
-          }
-        }
+      // Create export directory if it doesn't exist
+      await fs.mkdir(exportPath, { recursive: true });
 
-        // Continue traversing children for non-main-frame nodes
-        if (node.children && Array.isArray(node.children)) {
-          node.children.forEach(child => traversePage(child, depth + 1));
+      // Clear existing PDF files in the directory
+      try {
+        const existingFiles = await fs.readdir(exportPath);
+        const pdfFiles = existingFiles.filter(file => file.endsWith('.pdf'));
+        for (const file of pdfFiles) {
+          await fs.unlink(path.join(exportPath, file));
         }
+      } catch (cleanupError) {
+        // Continue if cleanup fails
       }
 
-      // 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));
+      // Request PDF export of entire file
+      const exportResponse = await axios.get(
+        `https://api.figma.com/v1/images/${fileId}`,
+        {
+          headers: {
+            'X-Figma-Token': figmaToken
+          },
+          params: {
+            format: 'pdf'
           }
-        });
-      }
-
-      return mainFrames;
-    }
+        }
+      );
 
-    // Helper function to count child frames (to detect component libraries)
-    function countChildFrames(node) {
-      let count = 0;
+      const pdfUrl = exportResponse.data.images[fileId];
 
-      function traverse(n) {
-        if (n.type === 'FRAME') {
-          count++;
-        }
-        if (n.children && Array.isArray(n.children)) {
-          n.children.forEach(traverse);
-        }
+      if (!pdfUrl) {
+        throw new Error('Failed to get PDF export URL from Figma');
       }
 
-      if (node.children && Array.isArray(node.children)) {
-        node.children.forEach(traverse);
+      // Download PDF
+      const pdfResponse = await axios.get(pdfUrl, {
+        responseType: 'arraybuffer'
+      });
+
+      // Save PDF to file with timestamp
+      const timestamp = new Date().toISOString().replace(/[:.]/g, '-').slice(0, -5);
+      const pdfPath = path.join(exportPath, `figma-export-${timestamp}.pdf`);
+      await fs.writeFile(pdfPath, pdfResponse.data);
+
+      return `Figma PDF Export Complete!
+Export Path: ${pdfPath}
+File: figma-export-${timestamp}.pdf`;
+
+    } catch (error) {
+      if (error.response && error.response.status === 403) {
+        return `Error: Access denied. Please check your Figma API token and file permissions.`;
+      } else if (error.response && error.response.status === 404) {
+        return `Error: Figma file not found. Please check the URL and ensure the file is accessible.`;
+      } else {
+        return `Error exporting from Figma: ${error.message}`;
       }
-
-      return count;
     }
+  }
+);
+
+// Helper function to extract file ID from Figma URL
+function extractFileIdFromUrl(url) {
+  const patterns = [
+    /figma\.com\/file\/([a-zA-Z0-9]+)/,
+    /figma\.com\/design\/([a-zA-Z0-9]+)/,
+    /figma\.com\/proto\/([a-zA-Z0-9]+)/
+  ];
+
+  for (const pattern of patterns) {
+    const match = url.match(pattern);
+    if (match) {
+      return match[1];
+    }
+  }
+  return null;
+}
+
 
     return server;
 };

package/package.json

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