@locofy/mcp 1.0.2 → 1.0.4

package/build/tools/pullComponents.js

@@ -34,17 +34,16 @@ export async function runPullComponentsTool(args) {
             // Check for code changes and mark changed files
             await checkForCodeChanges(result.data);
             // clean the directory structure
-            const cleanedResult = cleanDirectoryStructure(result.data);
-            const resultJSON = cleanedResult;
+            const cleanedResult = processDirectoryStructure(result.data);
             // write resultJSON to a file
             if (process.env.DEBUG_MODE === 'true') {
-                fs.writeFileSync('result.json', JSON.stringify(resultJSON.children[0], null, 2));
+                fs.writeFileSync('result.json', JSON.stringify(cleanedResult, null, 2));
             }
             return {
                 content: [
                     {
                         type: 'text',
-                        text: JSON.stringify(resultJSON.children[0], null, 2)
+                        text: JSON.stringify(cleanedResult, null, 2)
                     }
                 ]
             };
@@ -77,9 +76,12 @@ export async function runPullComponentsTool(args) {
     }
 }
 async function fetchDirectoryStructure(componentNames, projectID, personalAccessToken) {
-    // TODO: change to PROD url
     const encodedNames = componentNames.map(name => encodeURIComponent(name)).join(',');
-    const url = 'https://codegen-api.locofy.dev/mcp/generators/' + projectID + '?name=' + encodedNames;
+    let baseURL = 'https://codegen-api.locofy.ai/mcp/generators/';
+    if (process.env.IS_DEV === 'true') {
+        baseURL = 'https://codegen-api.locofy.dev/mcp/generators/';
+    }
+    const url = baseURL + projectID + '?name=' + encodedNames;
     const headers = {
         'Accept': '*/*',
         'Content-Type': 'application/json',
@@ -414,7 +416,7 @@ function normalizeForComparison(code) {
  * @param directoryStructure The directory structure to clean
  * @returns The cleaned directory structure
  */
-function cleanDirectoryStructure(directoryStructure) {
+function processDirectoryStructure(directoryStructure) {
     const propertiesToRemove = [
         'files',
         'component',
@@ -427,6 +429,40 @@ function cleanDirectoryStructure(directoryStructure) {
         'storybookFileName',
         'isAutoSyncNode'
     ];
+    const componentsToUpdate = [];
+    const componentsUnchanged = [];
+    function collectChangeStatus(node) {
+        // Check if this is a component node with a name and change status
+        if (node.name && node.codeChanged !== undefined) {
+            if (node.codeChanged) {
+                componentsToUpdate.push(node.name);
+            }
+            else {
+                componentsUnchanged.push(node.name);
+            }
+        }
+        // Check dependencies
+        if (node.exportData && node.exportData.dependencies) {
+            node.exportData.dependencies.forEach((dep) => {
+                if (dep.compName && dep.codeChanged !== undefined) {
+                    if (dep.codeChanged) {
+                        componentsToUpdate.push(dep.compName);
+                    }
+                    else {
+                        componentsUnchanged.push(dep.compName);
+                    }
+                }
+                // Check nested dependencies
+                if (dep.dependencies && Array.isArray(dep.dependencies)) {
+                    dep.dependencies.forEach((nestedDep) => collectChangeStatus(nestedDep));
+                }
+            });
+        }
+        // Process children recursively
+        if (node.children && Array.isArray(node.children)) {
+            node.children.forEach((child) => collectChangeStatus(child));
+        }
+    }
     function cleanNode(node) {
         // Remove unwanted properties from the current node
         propertiesToRemove.forEach(prop => {
@@ -449,7 +485,16 @@ function cleanDirectoryStructure(directoryStructure) {
             node.children.forEach((child) => cleanNode(child));
         }
     }
-    // Start cleaning from the root
+    // First collect all change statuses
+    collectChangeStatus(directoryStructure);
+    // Then clean the structure
     cleanNode(directoryStructure);
-    return directoryStructure;
+    // Get the first child (the actual component data we want)
+    const result = directoryStructure.children && directoryStructure.children.length > 0
+        ? directoryStructure.children[0]
+        : directoryStructure;
+    // Add the component lists to the component data
+    result.componentsToUpdate = [...new Set(componentsToUpdate)];
+    result.componentsUnchanged = [...new Set(componentsUnchanged)];
+    return result;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@locofy/mcp",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "Locofy MCP Server with Cursor",
   "keywords": [
     "figma",