@locofy/mcp 1.1.0 → 1.1.2

package/build/api.js

@@ -1,50 +1 @@
-import express from 'express';
-import { z } from 'zod';
-import { PullComponentsToolSchema, runPullComponentsTool } from './tools/pullComponents.js';
-import { PullFilesToolSchema, runPullFilesTool } from './tools/pullFiles.js';
-// Only start the API server if DEBUG_MODE is set
-if (process.env.DEBUG_MODE === 'true') {
-    const app = express();
-    const port = process.env.PORT || 3000;
-    // Middleware to parse JSON bodies
-    app.use(express.json());
-    // API endpoint for pullComponents
-    app.post('/api/pullComponents', async (req, res) => {
-        try {
-            const validatedArgs = PullComponentsToolSchema.parse(req.body);
-            const result = await runPullComponentsTool(validatedArgs);
-            res.json(result);
-        }
-        catch (error) {
-            if (error instanceof z.ZodError) {
-                res.status(400).json({ error: 'Invalid request body', details: error.errors });
-            }
-            else {
-                console.error('Error processing request:', error);
-                res.status(500).json({ error: 'Internal server error', message: error instanceof Error ? error.message : 'Unknown error' });
-            }
-        }
-    });
-    // API endpoint for pullFiles
-    app.post('/api/pullFiles', async (req, res) => {
-        try {
-            const validatedArgs = PullFilesToolSchema.parse(req.body);
-            const result = await runPullFilesTool(validatedArgs);
-            res.json(result);
-        }
-        catch (error) {
-            if (error instanceof z.ZodError) {
-                res.status(400).json({ error: 'Invalid request body', details: error.errors });
-            }
-            else {
-                console.error('Error processing request:', error);
-                res.status(500).json({ error: 'Internal server error', message: error instanceof Error ? error.message : 'Unknown error' });
-            }
-        }
-    });
-    // Start the server
-    app.listen(port, () => {
-        console.log(`Debug API server running on http://localhost:${port}`);
-    });
-}
-export default {};
+import express from"express";import{z}from"zod";import{PullComponentsToolSchema,runPullComponentsTool}from"./tools/pullComponents.js";import{PullFilesToolSchema,runPullFilesTool}from"./tools/pullFiles.js";if("true"===process.env.DEBUG_MODE){const app=express(),port=process.env.PORT||3e3;app.use(express.json()),app.post("/api/pullComponents",(async(req,res)=>{try{const validatedArgs=PullComponentsToolSchema.parse(req.body),result=await runPullComponentsTool(validatedArgs);res.json(result)}catch(error){error instanceof z.ZodError?res.status(400).json({error:"Invalid request body",details:error.errors}):res.status(500).json({error:"Internal server error",message:error instanceof Error?error.message:"Unknown error"})}})),app.post("/api/pullFiles",(async(req,res)=>{try{const validatedArgs=PullFilesToolSchema.parse(req.body),result=await runPullFilesTool(validatedArgs);res.json(result)}catch(error){error instanceof z.ZodError?res.status(400).json({error:"Invalid request body",details:error.errors}):res.status(500).json({error:"Internal server error",message:error instanceof Error?error.message:"Unknown error"})}})),app.listen(port,(()=>{}))}export default{};

package/build/constants.js

@@ -1,7 +1 @@
-const isDev = process.env.IS_DEV === 'true';
-export const MIXPANEL_PROJECT_TOKEN = isDev
-    ? '77d8775b23ede44f6f58a6a0c677e881'
-    : '6ba4b43a52dcbc80ee759b59f2dc8c5f';
-export const MIXPANEL_PROXY_DOMAIN = isDev
-    ? 'https://event-tracking.locofy.dev/events/mixpanel'
-    : 'https://events.locofy.ai/events/mixpanel';
+const isDev="true"===process.env.IS_DEV;export const MIXPANEL_PROJECT_TOKEN=isDev?"77d8775b23ede44f6f58a6a0c677e881":"6ba4b43a52dcbc80ee759b59f2dc8c5f";export const MIXPANEL_PROXY_DOMAIN=isDev?"https://event-tracking.locofy.dev/events/mixpanel":"https://events.locofy.ai/events/mixpanel";

package/build/helpers/helpers.js

@@ -1,31 +1 @@
-import path from 'path';
-import fs from 'fs';
-/**
- * Get the PROJECT_ID
- * If PROJECT_ID is not found in the workspacePath/.locofy/config.json, return the environment variable value
- * @param workspacePath
- * @returns The PROJECT_ID value
- */
-export async function getProjectID(workspacePath) {
-    const projectIDKey = 'PROJECT_ID';
-    const configPath = path.join(workspacePath, '.locofy/config.json');
-    if (fs.existsSync(configPath)) {
-        try {
-            const configData = fs.readFileSync(configPath, 'utf8');
-            const config = JSON.parse(configData);
-            if (config[projectIDKey]) {
-                console.log(`found ${projectIDKey} in config file, returning value ${config[projectIDKey]}`);
-                return config[projectIDKey];
-            }
-            if (config[projectIDKey.toLowerCase()]) {
-                console.log(`found ${projectIDKey.toLowerCase()} in config file, returning value ${config[projectIDKey.toLowerCase()]}`);
-                return config[projectIDKey.toLowerCase()];
-            }
-        }
-        catch (error) {
-            console.error('Error reading config file:', error);
-        }
-    }
-    console.log(`PROJECT_ID was not found in config file, returning environment variable value ${process.env.PROJECT_ID}`);
-    return process.env.PROJECT_ID;
-}
+import path from"path";import fs from"fs";export async function getProjectID(workspacePath){const configPath=path.join(workspacePath,".locofy/config.json");if(fs.existsSync(configPath))try{const configData=fs.readFileSync(configPath,"utf8"),config=JSON.parse(configData);if(config.PROJECT_ID)return config.PROJECT_ID;if(config["PROJECT_ID".toLowerCase()])return config["PROJECT_ID".toLowerCase()]}catch(error){}return process.env.PROJECT_ID}

package/build/index.js

@@ -1,89 +1,2 @@
 #!/usr/bin/env node
-import { Server } from '@modelcontextprotocol/sdk/server/index.js';
-import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
-import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
-import { pullComponentsToolName, pullComponentsToolDescription, PullComponentsToolSchema, runPullComponentsTool } from './tools/pullComponents.js';
-import { pullFilesToolName, pullFilesToolDescription, PullFilesToolSchema, runPullFilesTool } from './tools/pullFiles.js';
-import './api.js';
-const server = new Server({
-    name: 'locofy-mcp',
-    version: '2.0.1',
-}, {
-    capabilities: {
-        tools: {},
-    },
-});
-server.setRequestHandler(ListToolsRequestSchema, async () => {
-    return {
-        tools: [
-            {
-                name: pullComponentsToolName,
-                description: pullComponentsToolDescription,
-                inputSchema: {
-                    type: 'object',
-                    properties: {
-                        componentNames: {
-                            type: 'array',
-                            items: {
-                                type: 'string',
-                            },
-                            description: 'The list of component or screen names to be retrieved.'
-                        },
-                        workspacePath: {
-                            type: 'string',
-                            description: 'The full path to the workspace.'
-                        },
-                    },
-                    required: ['componentNames', 'workspacePath'],
-                },
-            },
-            {
-                name: pullFilesToolName,
-                description: pullFilesToolDescription,
-                inputSchema: {
-                    type: 'object',
-                    properties: {
-                        fileNames: {
-                            type: 'array',
-                            items: {
-                                type: 'string',
-                            },
-                            description: 'The list of specific file names to be retrieved, without resolving their component dependencies.'
-                        },
-                        workspacePath: {
-                            type: 'string',
-                            description: 'The full path to the workspace.'
-                        },
-                    },
-                }
-            }
-        ],
-    };
-});
-server.setRequestHandler(CallToolRequestSchema, async (request) => {
-    const { name, arguments: args } = request.params;
-    switch (name) {
-        case pullComponentsToolName: {
-            const validated = PullComponentsToolSchema.parse(args);
-            return await runPullComponentsTool(validated);
-        }
-        case pullFilesToolName: {
-            const validated = PullFilesToolSchema.parse(args);
-            return await runPullFilesTool(validated);
-        }
-        default:
-            throw new Error(`Unknown tool: ${name}`);
-    }
-});
-async function main() {
-    const transport = new StdioServerTransport();
-    await server.connect(transport);
-    console.log('Cursor Tools MCP Server running on stdio');
-    if (process.env.DEBUG_MODE === 'true') {
-        console.log('Debug mode active - REST API available for Postman testing');
-    }
-}
-main().catch((error) => {
-    console.error('Fatal error:', error);
-    process.exit(1);
-});
+import{Server}from"@modelcontextprotocol/sdk/server/index.js";import{StdioServerTransport}from"@modelcontextprotocol/sdk/server/stdio.js";import{CallToolRequestSchema,ListToolsRequestSchema}from"@modelcontextprotocol/sdk/types.js";import{pullComponentsToolName,pullComponentsToolDescription,PullComponentsToolSchema,runPullComponentsTool}from"./tools/pullComponents.js";import{pullFilesToolName,pullFilesToolDescription,PullFilesToolSchema,runPullFilesTool}from"./tools/pullFiles.js";import"./api.js";const server=new Server({name:"locofy-mcp",version:"2.0.1"},{capabilities:{tools:{}}});async function main(){const transport=new StdioServerTransport;await server.connect(transport),process.env.DEBUG_MODE}server.setRequestHandler(ListToolsRequestSchema,(async()=>({tools:[{name:pullComponentsToolName,description:pullComponentsToolDescription,inputSchema:{type:"object",properties:{componentNames:{type:"array",items:{type:"string"},description:"The list of component or screen names to be retrieved."},workspacePath:{type:"string",description:"The full path to the workspace."}},required:["componentNames","workspacePath"]}},{name:pullFilesToolName,description:pullFilesToolDescription,inputSchema:{type:"object",properties:{fileNames:{type:"array",items:{type:"string"},description:"The list of specific file names to be retrieved, without resolving their component dependencies."},workspacePath:{type:"string",description:"The full path to the workspace."}}}}]}))),server.setRequestHandler(CallToolRequestSchema,(async request=>{const{name:name,arguments:args}=request.params;switch(name){case pullComponentsToolName:{const validated=PullComponentsToolSchema.parse(args);return await runPullComponentsTool(validated)}case pullFilesToolName:{const validated=PullFilesToolSchema.parse(args);return await runPullFilesTool(validated)}default:throw new Error(`Unknown tool: ${name}`)}})),main().catch((error=>{process.exit(1)}));

package/build/tools/pullComponents.js

@@ -1,507 +1 @@
-import { z } from 'zod';
-import * as fs from 'fs';
-import * as path from 'path';
-import axios from 'axios';
-import { getProjectID } from '../helpers/helpers.js';
-import { track } from '../utils/analytics.js';
-/**
- * PullComponents tool
- *   - Returns a hardcoded representation of a directory structure
- *   - Contains nested folders, files, and file contents
- */
-export const pullComponentsToolName = 'getLatestComponentAndDependencyCode';
-export const pullComponentsToolDescription = `Cursor IDE leverages this MCP tool to scan directory structures with nested folders, files, and their contents. It then compares this information with the current code in the IDE to intelligently update components and their dependencies. If codeChanged is true, the tool will prompt the user to accept the changes before updating the component and its dependencies. If codeChanged is false, the tool will not update the component and its dependencies. The tool ensures that users are informed of the components retrieved and their relationships, providing a clear understanding of the updates being made
-  `;
-export const PullComponentsToolSchema = z.object({
-    componentNames: z.array(z.string()).describe('The list of component or screen names to be retrieved.').max(10).min(1),
-    workspacePath: z.string().describe('The full path to the workspace.'),
-});
-export async function runPullComponentsTool(args) {
-    let componentNames = args.componentNames;
-    // clean the component names, remove extensions
-    componentNames = componentNames.map(name => name.replace(/\.[^/.]+$/, ''));
-    const workspacePath = decodeURIComponent(args.workspacePath);
-    const projectID = await getProjectID(workspacePath);
-    if (!projectID || projectID.length === 0) {
-        throw new Error('PROJECT_ID is not set');
-    }
-    const personalAccessToken = process.env.PERSONAL_ACCESS_TOKEN;
-    if (!personalAccessToken) {
-        throw new Error('PERSONAL_ACCESS_TOKEN is not set');
-    }
-    try {
-        const result = await fetchDirectoryStructure(componentNames, projectID, personalAccessToken);
-        if (result.success) {
-            const userID = result.user_id;
-            // Prepend workspacePath to all filePath attributes
-            prepareFilePaths(result.data, workspacePath);
-            // Download and save any files referenced in the response
-            await downloadAndSaveAssets(result.data);
-            // Check for code changes and mark changed files
-            await checkForCodeChanges(result.data);
-            // clean the directory structure
-            const cleanedResult = processDirectoryStructure(result.data);
-            track('mcp_execute_tool_e', {
-                tool_name: pullComponentsToolName,
-                project_id: projectID,
-                user_id: userID,
-                source: 'cursor', // TODO: update it when we release vscode extension
-            });
-            return {
-                content: [
-                    {
-                        type: 'text',
-                        text: JSON.stringify(cleanedResult, null, 2)
-                    }
-                ]
-            };
-        }
-        else {
-            // Return the error message from fetchDirectoryStructure
-            return {
-                content: [
-                    {
-                        type: 'text',
-                        statusCode: result.statusCode || 'unknown',
-                        text: JSON.stringify({
-                            error: result.error,
-                        }, null, 2)
-                    }
-                ]
-            };
-        }
-    }
-    catch (error) {
-        console.error('Error fetching directory structure:', error);
-        return {
-            content: [
-                {
-                    type: 'text',
-                    text: JSON.stringify({ error: 'Failed to fetch directory structure ' + error }, null, 2)
-                }
-            ]
-        };
-    }
-}
-async function fetchDirectoryStructure(componentNames, projectID, personalAccessToken) {
-    const encodedNames = componentNames.map(name => encodeURIComponent(name)).join(',');
-    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',
-        'Authorization': 'Bearer ' + personalAccessToken
-    };
-    const data = {};
-    try {
-        const response = await axios.post(url, data, { headers, });
-        if (response.status !== 200) {
-            // Instead of throwing, return an error object
-            return {
-                success: false,
-                error: `API returned status code ${response.status}`,
-                statusCode: response.status
-            };
-        }
-        if (!response.data || !response.data.directoryTree) {
-            return {
-                success: false,
-                error: 'API response is missing directoryTree property',
-                statusCode: response.status
-            };
-        }
-        return {
-            success: true,
-            data: response.data.directoryTree,
-            user_id: response.data.user_id
-        };
-    }
-    catch (error) {
-        if (axios.isAxiosError(error)) {
-            if (error.response) {
-                const statusCode = error.response.status;
-                let errorMessage = '';
-                if (statusCode === 401) {
-                    errorMessage = 'Authentication failed: Invalid or expired token. Please check your mcp.json configuration in the .cursor folder to ensure your PERSONAL_ACCESS_TOKEN is correct. You can visit the Locofy project settings page to check or regenerate your token if needed.';
-                }
-                else if (statusCode === 500) {
-                    errorMessage = 'Server error: The API service is experiencing issues';
-                }
-                else {
-                    errorMessage = `API request failed with status ${statusCode}: ${error.message}`;
-                }
-                return {
-                    success: false,
-                    error: errorMessage,
-                    statusCode: statusCode
-                };
-            }
-            else if (error.request) {
-                return {
-                    success: false,
-                    error: 'Network error: No response received from API',
-                    statusCode: 'network_error'
-                };
-            }
-        }
-        console.error('API request failed:', error);
-        return {
-            success: false,
-            error: `Failed to fetch directory structure: ${error instanceof Error ? error.message : 'Unknown error'}`,
-            statusCode: 'unknown_error'
-        };
-    }
-}
-/**
- * Traverses the directory structure and prepends workspacePath to all filePath attributes
- * @param directoryStructure The directory structure from the API response
- * @param workspacePath The workspace path to prepend to file paths
- */
-function prepareFilePaths(directoryStructure, workspacePath) {
-    const processNode = (node) => {
-        if (node.filePath) {
-            if (!path.isAbsolute(node.filePath)) {
-                node.filePath = path.join(workspacePath, node.filePath);
-            }
-        }
-        // Process all file paths in exportData
-        if (node.exportData) {
-            // Handle exportData.filePath
-            if (node.exportData.filePath) {
-                if (!path.isAbsolute(node.exportData.filePath)) {
-                    node.exportData.filePath = path.join(workspacePath, node.exportData.filePath);
-                }
-            }
-            // Handle exportData.files array
-            if (node.exportData.files && Array.isArray(node.exportData.files)) {
-                for (const file of node.exportData.files) {
-                    if (file.filePath && !path.isAbsolute(file.filePath)) {
-                        file.filePath = path.join(workspacePath, file.filePath);
-                    }
-                }
-            }
-            // Handle dependencies array
-            if (node.exportData.dependencies && Array.isArray(node.exportData.dependencies)) {
-                for (const dependency of node.exportData.dependencies) {
-                    // Process dependency filePath
-                    if (dependency.filePath && !path.isAbsolute(dependency.filePath)) {
-                        dependency.filePath = path.join(workspacePath, dependency.filePath);
-                    }
-                    // Process dependency files array
-                    if (dependency.files && Array.isArray(dependency.files)) {
-                        for (const file of dependency.files) {
-                            if (file.filePath && !path.isAbsolute(file.filePath)) {
-                                file.filePath = path.join(workspacePath, file.filePath);
-                            }
-                        }
-                    }
-                    // Recursively process any nested dependencies if they exist
-                    if (dependency.dependencies && Array.isArray(dependency.dependencies)) {
-                        for (const nestedDep of dependency.dependencies) {
-                            processNode(nestedDep); // Recursively process nested dependencies
-                        }
-                    }
-                }
-            }
-        }
-        // Process content.filePath if it exists
-        if (node.content && node.content.filePath) {
-            if (!path.isAbsolute(node.content.filePath)) {
-                node.content.filePath = path.join(workspacePath, node.content.filePath);
-            }
-        }
-        // Process children recursively
-        if (node.children && Array.isArray(node.children)) {
-            for (const child of node.children) {
-                processNode(child);
-            }
-        }
-    };
-    // Start processing from the root
-    processNode(directoryStructure);
-}
-/**
- * Downloads and saves asset files from the directory structure
- * @param directoryStructure The directory structure from the API response
- */
-async function downloadAndSaveAssets(directoryStructure) {
-    // Process each file in the directory structure
-    const processNode = async (node) => {
-        /* eslint-enable */
-        if (node.type === 'file' && node.exportData && node.exportData.files) {
-            // For each file in the files array
-            for (const file of node.exportData.files) {
-                if (file.url && file.fileName) {
-                    const filePath = file.filePath;
-                    // Create the directory structure if it doesn't exist
-                    const dirPath = path.dirname(filePath);
-                    if (!fs.existsSync(dirPath)) {
-                        fs.mkdirSync(dirPath, { recursive: true });
-                        console.log(`Created directory: ${dirPath}`);
-                    }
-                    // Only download if file doesn't exist
-                    if (!fs.existsSync(filePath)) {
-                        try {
-                            console.log(`Downloading ${file.fileName} from ${file.url}`);
-                            const response = await axios.get(file.url, { responseType: 'arraybuffer' });
-                            fs.writeFileSync(filePath, Buffer.from(response.data));
-                            console.log(`Saved ${file.fileName} to ${filePath}`);
-                        }
-                        catch (error) {
-                            console.error(`Error downloading ${file.fileName}:`, error);
-                        }
-                    }
-                    else {
-                        console.log(`File ${filePath} already exists, skipping download`);
-                    }
-                }
-            }
-        }
-        // Process children recursively
-        if (node.children && Array.isArray(node.children)) {
-            for (const child of node.children) {
-                await processNode(child);
-            }
-        }
-    };
-    // Start processing from the root
-    await processNode(directoryStructure);
-}
-/**
- * Checks if the current code on disk differs from the code received from the API
- * @param directoryStructure The directory structure from the API response
- */
-async function checkForCodeChanges(directoryStructure) {
-    // Process a single file and check if its content has changed
-    function checkFile(filePath, content) {
-        if (!fs.existsSync(filePath)) {
-            console.log(`New file: ${filePath}`);
-            return true;
-        }
-        try {
-            const fileContent = fs.readFileSync(filePath, 'utf8');
-            const codeChanged = hasCodeChanged(fileContent, content);
-            if (codeChanged) {
-                console.log(`File changed: ${filePath}`);
-            }
-            return codeChanged;
-        }
-        catch (error) {
-            console.error(`Error reading file ${filePath}:`, error);
-            return true; // Consider file as changed if we can't read it
-        }
-    }
-    // Check CSS module file if it exists
-    function checkCssFile(basePath, cssFileName, cssContent) {
-        if (!cssFileName || !cssContent)
-            return false;
-        const cssFilePath = path.join(path.dirname(basePath), cssFileName);
-        if (fs.existsSync(cssFilePath)) {
-            const cssFileContent = fs.readFileSync(cssFilePath, 'utf8');
-            const cssChanged = hasCodeChanged(cssFileContent, cssContent);
-            if (cssChanged) {
-                console.log(`CSS module changed: ${cssFilePath}`);
-                return true;
-            }
-            else {
-                return false;
-            }
-        }
-        return true; // CSS file doesn't exist, consider as new/changed
-    }
-    // Update node with change status and instructions
-    function updateNodeStatus(node, fileName, hasChanged, isCss = false) {
-        if (!hasChanged) {
-            const message = `This file ${fileName} has no changes, no need to update`;
-            if (isCss) {
-                node.cssContent = message;
-                node.instructionCSSForIDE = message;
-            }
-            else {
-                node.instructionForIDE = message;
-                node.content = message;
-            }
-        }
-        else if (!fs.existsSync(node.filePath)) {
-            node.instructionForIDE = 'This file does not exist, please create it';
-        }
-        return hasChanged;
-    }
-    // The main recursive function to process nodes
-    function processNode(node) {
-        let hasChanges = false;
-        // Remove experimental fields first
-        if (node.exportData) {
-            // Process main component file
-            if (node.filePath) {
-                const nodeContent = node.name.includes('.css') ?
-                    node.exportData.cssContent :
-                    node.exportData.content;
-                const fileChanged = checkFile(node.filePath, nodeContent);
-                node.codeChanged = fileChanged;
-                hasChanges = hasChanges || fileChanged;
-                updateNodeStatus(node, node.name, fileChanged);
-                // Check associated CSS module
-                if (node.exportData.cssFileName && node.exportData.cssContent) {
-                    const cssChanged = checkCssFile(node.filePath, node.exportData.cssFileName, node.exportData.cssContent);
-                    hasChanges = hasChanges || cssChanged;
-                    updateNodeStatus(node.exportData, node.exportData.cssFileName, cssChanged, true);
-                }
-            }
-            // Process dependencies
-            if (node.exportData.dependencies && Array.isArray(node.exportData.dependencies)) {
-                for (const dependency of node.exportData.dependencies) {
-                    // Process dependency file
-                    if (dependency.filePath) {
-                        const isCSS = dependency.name && dependency.name.includes('.css');
-                        const depContent = isCSS ? dependency.cssContent : dependency.content;
-                        const fileChanged = checkFile(dependency.filePath, depContent);
-                        dependency.codeChanged = fileChanged;
-                        hasChanges = hasChanges || fileChanged;
-                        updateNodeStatus(dependency, dependency.compName, fileChanged);
-                        // Check associated CSS module
-                        if (dependency.cssFileName && dependency.cssContent) {
-                            const cssChanged = checkCssFile(dependency.filePath, dependency.cssFileName, dependency.cssContent);
-                            hasChanges = hasChanges || cssChanged;
-                            updateNodeStatus(dependency, dependency.cssFileName, cssChanged, true);
-                        }
-                    }
-                    // Process nested dependencies recursively
-                    if (dependency.dependencies && Array.isArray(dependency.dependencies)) {
-                        for (const nestedDep of dependency.dependencies) {
-                            const nestedChanged = processNode(nestedDep);
-                            hasChanges = hasChanges || nestedChanged;
-                        }
-                    }
-                }
-            }
-        }
-        // Process children recursively
-        if (node.children && Array.isArray(node.children)) {
-            for (const child of node.children) {
-                const childChanged = processNode(child);
-                hasChanges = hasChanges || childChanged;
-            }
-        }
-        return hasChanges;
-    }
-    // Start processing from the root
-    return processNode(directoryStructure);
-}
-/**
- * Compares two code snippets to determine if they are functionally different
- * @param fileContent The code currently in the file
- * @param apiCode The code provided by the API
- * @returns True if the code has changed, false otherwise
- */
-function hasCodeChanged(fileContent, apiCode) {
-    if (!fileContent || !apiCode) {
-        return true;
-    }
-    // Normalize both code snippets
-    const normalizedFile = normalizeForComparison(fileContent);
-    const normalizedApi = normalizeForComparison(apiCode);
-    // Simple equality check on normalized versions
-    return normalizedFile !== normalizedApi;
-}
-/**
- * Normalizes code for comparison by removing irrelevant differences
- * @param code The code to normalize
- * @returns Normalized code string
- */
-function normalizeForComparison(code) {
-    return code
-        .replace(/\s+/g, '') // Remove all whitespace (spaces, tabs, newlines)
-        .replace(/[;,'"]/g, '') // Remove semicolons, commas, and quotes
-        .replace(/\/\/.*$/gm, '') // Remove single-line comments
-        .replace(/\/\*[\s\S]*?\*\//g, '') // Remove multi-line comments
-        .toLowerCase(); // Convert to lowercase for case-insensitive comparison
-}
-/**
- * Cleans the directory structure by removing unwanted properties from all nodes
- * @param directoryStructure The directory structure to clean
- * @returns The cleaned directory structure
- */
-function processDirectoryStructure(directoryStructure) {
-    const propertiesToRemove = [
-        'files',
-        'component',
-        'jsLC',
-        'assetCount',
-        'cssLC',
-        'componentCount',
-        'pageMetaData',
-        'isHomePage',
-        '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 => {
-            if (node[prop] !== undefined)
-                delete node[prop];
-        });
-        // Clean exportData if it exists
-        if (node.exportData) {
-            propertiesToRemove.forEach(prop => {
-                if (node.exportData[prop] !== undefined)
-                    delete node.exportData[prop];
-            });
-            // Process dependencies in exportData
-            if (node.exportData.dependencies && Array.isArray(node.exportData.dependencies)) {
-                node.exportData.dependencies.forEach((dep) => cleanNode(dep));
-            }
-        }
-        // Process children recursively
-        if (node.children && Array.isArray(node.children)) {
-            node.children.forEach((child) => cleanNode(child));
-        }
-    }
-    // First collect all change statuses
-    collectChangeStatus(directoryStructure);
-    // Then clean the structure
-    cleanNode(directoryStructure);
-    // Add the component lists to the component data
-    return {
-        directoryStructure: directoryStructure.children,
-        componentsToUpdate: [...new Set(componentsToUpdate)],
-        componentsUnchanged: [...new Set(componentsUnchanged)]
-    };
-}
+import{z}from"zod";import*as fs from"fs";import*as path from"path";import axios from"axios";import{getProjectID}from"../helpers/helpers.js";import{track}from"../utils/analytics.js";export const pullComponentsToolName="getLatestComponentAndDependencyCode";export const pullComponentsToolDescription="Cursor IDE leverages this MCP tool to scan directory structures with nested folders, files, and their contents. It then compares this information with the current code in the IDE to intelligently update components and their dependencies. If codeChanged is true, the tool will prompt the user to accept the changes before updating the component and its dependencies. If codeChanged is false, the tool will not update the component and its dependencies. The tool ensures that users are informed of the components retrieved and their relationships, providing a clear understanding of the updates being made\n  ";export const PullComponentsToolSchema=z.object({componentNames:z.array(z.string()).describe("The list of component or screen names to be retrieved.").max(10).min(1),workspacePath:z.string().describe("The full path to the workspace.")});export async function runPullComponentsTool(args){let componentNames=args.componentNames;componentNames=componentNames.map((name=>name.replace(/\.[^/.]+$/,"")));const workspacePath=decodeURIComponent(args.workspacePath),projectID=await getProjectID(workspacePath);if(!projectID||0===projectID.length)throw new Error("PROJECT_ID is not set");const personalAccessToken=process.env.PERSONAL_ACCESS_TOKEN;if(!personalAccessToken)throw new Error("PERSONAL_ACCESS_TOKEN is not set");try{const result=await fetchDirectoryStructure(componentNames,projectID,personalAccessToken);if(result.success){const userID=result.user_id;prepareFilePaths(result.data,workspacePath),await downloadAndSaveAssets(result.data),await checkForCodeChanges(result.data);const cleanedResult=processDirectoryStructure(result.data),source=(process.env.IDE_SOURCE||"other").toLowerCase();return track("mcp_execute_tool_e",{tool_name:pullComponentsToolName,project_id:projectID,user_id:userID,source:source}),{content:[{type:"text",text:JSON.stringify(cleanedResult,null,2)}]}}return{content:[{type:"text",statusCode:result.statusCode||"unknown",text:JSON.stringify({error:result.error},null,2)}]}}catch(error){return{content:[{type:"text",text:JSON.stringify({error:"Failed to fetch directory structure "+error},null,2)}]}}}async function fetchDirectoryStructure(componentNames,projectID,personalAccessToken){const encodedNames=componentNames.map((name=>encodeURIComponent(name))).join(",");let baseURL="https://codegen-api.locofy.ai/mcp/generators/";"true"===process.env.IS_DEV&&(baseURL="https://codegen-api.locofy.dev/mcp/generators/");const url=baseURL+projectID+"?name="+encodedNames,headers={Accept:"*/*","Content-Type":"application/json",Authorization:"Bearer "+personalAccessToken},data={};try{const response=await axios.post(url,data,{headers:headers});return 200!==response.status?{success:!1,error:`API returned status code ${response.status}`,statusCode:response.status}:response.data&&response.data.directoryTree?{success:!0,data:response.data.directoryTree,user_id:response.data.user_id}:{success:!1,error:"API response is missing directoryTree property",statusCode:response.status}}catch(error){if(axios.isAxiosError(error)){if(error.response){const statusCode=error.response.status;let errorMessage="";return errorMessage=401===statusCode?"Authentication failed: Invalid or expired token. Please check your mcp.json configuration in the .cursor folder to ensure your PERSONAL_ACCESS_TOKEN is correct. You can visit the Locofy project settings page to check or regenerate your token if needed.":500===statusCode?"Server error: The API service is experiencing issues":`API request failed with status ${statusCode}: ${error.message}`,{success:!1,error:errorMessage,statusCode:statusCode}}if(error.request)return{success:!1,error:"Network error: No response received from API",statusCode:"network_error"}}return{success:!1,error:`Failed to fetch directory structure: ${error instanceof Error?error.message:"Unknown error"}`,statusCode:"unknown_error"}}}function prepareFilePaths(directoryStructure,workspacePath){const processNode=node=>{if(node.filePath&&(path.isAbsolute(node.filePath)||(node.filePath=path.join(workspacePath,node.filePath))),node.exportData){if(node.exportData.filePath&&(path.isAbsolute(node.exportData.filePath)||(node.exportData.filePath=path.join(workspacePath,node.exportData.filePath))),node.exportData.files&&Array.isArray(node.exportData.files))for(const file of node.exportData.files)file.filePath&&!path.isAbsolute(file.filePath)&&(file.filePath=path.join(workspacePath,file.filePath));if(node.exportData.dependencies&&Array.isArray(node.exportData.dependencies))for(const dependency of node.exportData.dependencies){if(dependency.filePath&&!path.isAbsolute(dependency.filePath)&&(dependency.filePath=path.join(workspacePath,dependency.filePath)),dependency.files&&Array.isArray(dependency.files))for(const file of dependency.files)file.filePath&&!path.isAbsolute(file.filePath)&&(file.filePath=path.join(workspacePath,file.filePath));if(dependency.dependencies&&Array.isArray(dependency.dependencies))for(const nestedDep of dependency.dependencies)processNode(nestedDep)}}if(node.content&&node.content.filePath&&(path.isAbsolute(node.content.filePath)||(node.content.filePath=path.join(workspacePath,node.content.filePath))),node.children&&Array.isArray(node.children))for(const child of node.children)processNode(child)};processNode(directoryStructure)}async function downloadAndSaveAssets(directoryStructure){const processNode=async node=>{if("file"===node.type&&node.exportData&&node.exportData.files)for(const file of node.exportData.files)if(file.url&&file.fileName){const filePath=file.filePath;if(!filePath)continue;const dirPath=path.dirname(filePath);if(fs.existsSync(dirPath)||fs.mkdirSync(dirPath,{recursive:!0}),!fs.existsSync(filePath))try{const response=await axios.get(file.url,{responseType:"arraybuffer"});fs.writeFileSync(filePath,Buffer.from(response.data))}catch(error){}}if(node.children&&Array.isArray(node.children))for(const child of node.children)await processNode(child)};await processNode(directoryStructure)}async function checkForCodeChanges(directoryStructure){function checkFile(filePath,content){if(!fs.existsSync(filePath))return!0;try{const codeChanged=hasCodeChanged(fs.readFileSync(filePath,"utf8"),content);return codeChanged}catch(error){return!0}}function checkCssFile(basePath,cssFileName,cssContent){if(!cssFileName||!cssContent)return!1;const cssFilePath=path.join(path.dirname(basePath),cssFileName);if(fs.existsSync(cssFilePath)){return!!hasCodeChanged(fs.readFileSync(cssFilePath,"utf8"),cssContent)}return!0}function updateNodeStatus(node,fileName,hasChanged,isCss=!1){if(hasChanged)fs.existsSync(node.filePath)||(node.instructionForIDE="This file does not exist, please create it");else{const message=`This file ${fileName} has no changes, no need to update`;isCss?(node.cssContent=message,node.instructionCSSForIDE=message):(node.instructionForIDE=message,node.content=message)}return hasChanged}return function processNode(node){let hasChanges=!1;if(node.exportData){if(node.filePath){const nodeContent=node.name.includes(".css")?node.exportData.cssContent:node.exportData.content,fileChanged=checkFile(node.filePath,nodeContent);if(node.codeChanged=fileChanged,hasChanges=hasChanges||fileChanged,updateNodeStatus(node,node.name,fileChanged),node.exportData.cssFileName&&node.exportData.cssContent){const cssChanged=checkCssFile(node.filePath,node.exportData.cssFileName,node.exportData.cssContent);hasChanges=hasChanges||cssChanged,updateNodeStatus(node.exportData,node.exportData.cssFileName,cssChanged,!0)}}if(node.exportData.dependencies&&Array.isArray(node.exportData.dependencies))for(const dependency of node.exportData.dependencies){if(dependency.filePath){const depContent=dependency.name&&dependency.name.includes(".css")?dependency.cssContent:dependency.content,fileChanged=checkFile(dependency.filePath,depContent);if(dependency.codeChanged=fileChanged,hasChanges=hasChanges||fileChanged,updateNodeStatus(dependency,dependency.compName,fileChanged),dependency.cssFileName&&dependency.cssContent){const cssChanged=checkCssFile(dependency.filePath,dependency.cssFileName,dependency.cssContent);hasChanges=hasChanges||cssChanged,updateNodeStatus(dependency,dependency.cssFileName,cssChanged,!0)}}if(dependency.dependencies&&Array.isArray(dependency.dependencies))for(const nestedDep of dependency.dependencies){const nestedChanged=processNode(nestedDep);hasChanges=hasChanges||nestedChanged}}}if(node.children&&Array.isArray(node.children))for(const child of node.children){const childChanged=processNode(child);hasChanges=hasChanges||childChanged}return hasChanges}(directoryStructure)}function hasCodeChanged(fileContent,apiCode){if(!fileContent||!apiCode)return!0;return normalizeForComparison(fileContent)!==normalizeForComparison(apiCode)}function normalizeForComparison(code){return code.replace(/\s+/g,"").replace(/[;,'"]/g,"").replace(/\/\/.*$/gm,"").replace(/\/\*[\s\S]*?\*\//g,"").toLowerCase()}function processDirectoryStructure(directoryStructure){const propertiesToRemove=["files","component","jsLC","assetCount","cssLC","componentCount","pageMetaData","isHomePage","storybookFileName","isAutoSyncNode"],componentsToUpdate=[],componentsUnchanged=[];return function collectChangeStatus(node){node.name&&void 0!==node.codeChanged&&(node.codeChanged?componentsToUpdate.push(node.name):componentsUnchanged.push(node.name)),node.exportData&&node.exportData.dependencies&&node.exportData.dependencies.forEach((dep=>{dep.compName&&void 0!==dep.codeChanged&&(dep.codeChanged?componentsToUpdate.push(dep.compName):componentsUnchanged.push(dep.compName)),dep.dependencies&&Array.isArray(dep.dependencies)&&dep.dependencies.forEach((nestedDep=>collectChangeStatus(nestedDep)))})),node.children&&Array.isArray(node.children)&&node.children.forEach((child=>collectChangeStatus(child)))}(directoryStructure),function cleanNode(node){propertiesToRemove.forEach((prop=>{void 0!==node[prop]&&delete node[prop]})),node.exportData&&(propertiesToRemove.forEach((prop=>{void 0!==node.exportData[prop]&&delete node.exportData[prop]})),node.exportData.dependencies&&Array.isArray(node.exportData.dependencies)&&node.exportData.dependencies.forEach((dep=>cleanNode(dep)))),node.children&&Array.isArray(node.children)&&node.children.forEach((child=>cleanNode(child)))}(directoryStructure),{directoryStructure:directoryStructure.children,componentsToUpdate:[...new Set(componentsToUpdate)],componentsUnchanged:[...new Set(componentsUnchanged)]}}

package/build/tools/pullFiles.js

@@ -1,505 +1 @@
-import { z } from 'zod';
-import * as fs from 'fs';
-import * as path from 'path';
-import axios from 'axios';
-import { getProjectID } from '../helpers/helpers.js';
-import { track } from '../utils/analytics.js';
-/**
- * PullFiles tool
- *   - Returns a hardcoded representation of a directory structure
- *   - Contains nested folders, files, and file contents
- */
-export const pullFilesToolName = 'getLatestFileCode';
-export const pullFilesToolDescription = `Cursor IDE uses this tool to retrieve individual file contents without component dependency management. Unlike the component tool that handles components and their dependencies as a unit, this tool focuses on specific files by name, retrieving only their content without dependency resolution. Use this tool when you need the raw content of specific files rather than complete components with their dependency trees. If codeChanged is true, the tool will prompt the user to accept the changes before updating the file. If codeChanged is false, the tool will not update the files.
-  `;
-export const PullFilesToolSchema = z.object({
-    fileNames: z.array(z.string()).describe('The list of specific file names to be retrieved, without resolving their component dependencies.').max(10).min(1),
-    workspacePath: z.string().describe('The full path to the workspace.'),
-});
-export async function runPullFilesTool(args) {
-    let fileNames = args.fileNames;
-    // clean the file names, remove extensions
-    fileNames = fileNames.map(name => name.replace(/\.[^/.]+$/, ''));
-    const workspacePath = decodeURIComponent(args.workspacePath);
-    const projectID = await getProjectID(workspacePath);
-    if (!projectID || projectID.length === 0) {
-        throw new Error('PROJECT_ID is not set');
-    }
-    const personalAccessToken = process.env.PERSONAL_ACCESS_TOKEN;
-    if (!personalAccessToken) {
-        throw new Error('PERSONAL_ACCESS_TOKEN is not set');
-    }
-    try {
-        const result = await fetchDirectoryStructure(fileNames, projectID, personalAccessToken);
-        if (result.success) {
-            const userID = result.user_id;
-            // Prepend workspacePath to all filePath attributes
-            prepareFilePaths(result.data, workspacePath);
-            // Download and save any files referenced in the response
-            await downloadAndSaveAssets(result.data);
-            // Check for code changes and mark changed files
-            await checkForCodeChanges(result.data);
-            // clean the directory structure
-            const cleanedResult = processDirectoryStructure(result.data);
-            track('mcp_execute_tool_e', {
-                tool_name: pullFilesToolName,
-                project_id: projectID,
-                user_id: userID,
-                source: 'cursor', // TODO: update it when we release vscode extension
-            });
-            return {
-                content: [
-                    {
-                        type: 'text',
-                        text: JSON.stringify(cleanedResult, null, 2)
-                    }
-                ]
-            };
-        }
-        else {
-            // Return the error message from fetchDirectoryStructure
-            return {
-                content: [
-                    {
-                        type: 'text',
-                        statusCode: result.statusCode || 'unknown',
-                        text: JSON.stringify({
-                            error: result.error,
-                        }, null, 2)
-                    }
-                ]
-            };
-        }
-    }
-    catch (error) {
-        console.error('Error fetching directory structure:', error);
-        return {
-            content: [
-                {
-                    type: 'text',
-                    text: JSON.stringify({ error: 'Failed to fetch directory structure ' + error }, null, 2)
-                }
-            ]
-        };
-    }
-}
-async function fetchDirectoryStructure(fileNames, projectID, personalAccessToken) {
-    const encodedNames = fileNames.map(name => encodeURIComponent(name)).join(',');
-    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',
-        'Authorization': 'Bearer ' + personalAccessToken
-    };
-    const data = {};
-    try {
-        const response = await axios.post(url, data, { headers, });
-        if (response.status !== 200) {
-            // Instead of throwing, return an error object
-            return {
-                success: false,
-                error: `API returned status code ${response.status}`,
-                statusCode: response.status
-            };
-        }
-        if (!response.data || !response.data.directoryTree) {
-            return {
-                success: false,
-                error: 'API response is missing directoryTree property',
-                statusCode: response.status
-            };
-        }
-        return {
-            success: true,
-            data: response.data.directoryTree,
-            user_id: response.data.user_id
-        };
-    }
-    catch (error) {
-        if (axios.isAxiosError(error)) {
-            if (error.response) {
-                const statusCode = error.response.status;
-                let errorMessage = '';
-                if (statusCode === 401) {
-                    errorMessage = 'Authentication failed: Invalid or expired token. Please check your mcp.json configuration in the .cursor folder to ensure your PERSONAL_ACCESS_TOKEN is correct. You can visit the Locofy project settings page to check or regenerate your token if needed.';
-                }
-                else if (statusCode === 500) {
-                    errorMessage = 'Server error: The API service is experiencing issues';
-                }
-                else {
-                    errorMessage = `API request failed with status ${statusCode}: ${error.message}`;
-                }
-                return {
-                    success: false,
-                    error: errorMessage,
-                    statusCode: statusCode
-                };
-            }
-            else if (error.request) {
-                return {
-                    success: false,
-                    error: 'Network error: No response received from API',
-                    statusCode: 'network_error'
-                };
-            }
-        }
-        console.error('API request failed:', error);
-        return {
-            success: false,
-            error: `Failed to fetch directory structure: ${error instanceof Error ? error.message : 'Unknown error'}`,
-            statusCode: 'unknown_error'
-        };
-    }
-}
-/**
- * Traverses the directory structure and prepends workspacePath to all filePath attributes
- * @param directoryStructure The directory structure from the API response
- * @param workspacePath The workspace path to prepend to file paths
- */
-function prepareFilePaths(directoryStructure, workspacePath) {
-    const processNode = (node) => {
-        if (node.filePath) {
-            if (!path.isAbsolute(node.filePath)) {
-                node.filePath = path.join(workspacePath, node.filePath);
-            }
-        }
-        // Process all file paths in exportData
-        if (node.exportData) {
-            // Handle exportData.filePath
-            if (node.exportData.filePath) {
-                if (!path.isAbsolute(node.exportData.filePath)) {
-                    node.exportData.filePath = path.join(workspacePath, node.exportData.filePath);
-                }
-            }
-            // Handle exportData.files array
-            if (node.exportData.files && Array.isArray(node.exportData.files)) {
-                for (const file of node.exportData.files) {
-                    if (file.filePath && !path.isAbsolute(file.filePath)) {
-                        file.filePath = path.join(workspacePath, file.filePath);
-                    }
-                }
-            }
-            // Handle dependencies array
-            if (node.exportData.dependencies && Array.isArray(node.exportData.dependencies)) {
-                for (const dependency of node.exportData.dependencies) {
-                    // Process dependency filePath
-                    if (dependency.filePath && !path.isAbsolute(dependency.filePath)) {
-                        dependency.filePath = path.join(workspacePath, dependency.filePath);
-                    }
-                    // Process dependency files array
-                    if (dependency.files && Array.isArray(dependency.files)) {
-                        for (const file of dependency.files) {
-                            if (file.filePath && !path.isAbsolute(file.filePath)) {
-                                file.filePath = path.join(workspacePath, file.filePath);
-                            }
-                        }
-                    }
-                    // Recursively process any nested dependencies if they exist
-                    if (dependency.dependencies && Array.isArray(dependency.dependencies)) {
-                        for (const nestedDep of dependency.dependencies) {
-                            processNode(nestedDep); // Recursively process nested dependencies
-                        }
-                    }
-                }
-            }
-        }
-        // Process content.filePath if it exists
-        if (node.content && node.content.filePath) {
-            if (!path.isAbsolute(node.content.filePath)) {
-                node.content.filePath = path.join(workspacePath, node.content.filePath);
-            }
-        }
-        // Process children recursively
-        if (node.children && Array.isArray(node.children)) {
-            for (const child of node.children) {
-                processNode(child);
-            }
-        }
-    };
-    // Start processing from the root
-    processNode(directoryStructure);
-}
-/**
- * Downloads and saves asset files from the directory structure
- * @param directoryStructure The directory structure from the API response
- */
-async function downloadAndSaveAssets(directoryStructure) {
-    // Process each file in the directory structure
-    const processNode = async (node) => {
-        /* eslint-enable */
-        if (node.type === 'file' && node.exportData && node.exportData.files) {
-            // For each file in the files array
-            for (const file of node.exportData.files) {
-                if (file.url && file.fileName) {
-                    const filePath = file.filePath;
-                    // Create the directory structure if it doesn't exist
-                    const dirPath = path.dirname(filePath);
-                    if (!fs.existsSync(dirPath)) {
-                        fs.mkdirSync(dirPath, { recursive: true });
-                        console.log(`Created directory: ${dirPath}`);
-                    }
-                    // Only download if file doesn't exist
-                    if (!fs.existsSync(filePath)) {
-                        try {
-                            console.log(`Downloading ${file.fileName} from ${file.url}`);
-                            const response = await axios.get(file.url, { responseType: 'arraybuffer' });
-                            fs.writeFileSync(filePath, Buffer.from(response.data));
-                            console.log(`Saved ${file.fileName} to ${filePath}`);
-                        }
-                        catch (error) {
-                            console.error(`Error downloading ${file.fileName}:`, error);
-                        }
-                    }
-                    else {
-                        console.log(`File ${filePath} already exists, skipping download`);
-                    }
-                }
-            }
-        }
-        // Process children recursively
-        if (node.children && Array.isArray(node.children)) {
-            for (const child of node.children) {
-                await processNode(child);
-            }
-        }
-    };
-    // Start processing from the root
-    await processNode(directoryStructure);
-}
-/**
- * Checks if the current code on disk differs from the code received from the API
- * @param directoryStructure The directory structure from the API response
- */
-async function checkForCodeChanges(directoryStructure) {
-    // Process a single file and check if its content has changed
-    function checkFile(filePath, content) {
-        if (!fs.existsSync(filePath)) {
-            console.log(`New file: ${filePath}`);
-            return true;
-        }
-        try {
-            const fileContent = fs.readFileSync(filePath, 'utf8');
-            const codeChanged = hasCodeChanged(fileContent, content);
-            if (codeChanged) {
-                console.log(`File changed: ${filePath}`);
-            }
-            return codeChanged;
-        }
-        catch (error) {
-            console.error(`Error reading file ${filePath}:`, error);
-            return true; // Consider file as changed if we can't read it
-        }
-    }
-    // Check CSS module file if it exists
-    function checkCssFile(basePath, cssFileName, cssContent) {
-        if (!cssFileName || !cssContent)
-            return false;
-        const cssFilePath = path.join(path.dirname(basePath), cssFileName);
-        if (fs.existsSync(cssFilePath)) {
-            const cssFileContent = fs.readFileSync(cssFilePath, 'utf8');
-            const cssChanged = hasCodeChanged(cssFileContent, cssContent);
-            if (cssChanged) {
-                console.log(`CSS module changed: ${cssFilePath}`);
-                return true;
-            }
-            else {
-                return false;
-            }
-        }
-        return true; // CSS file doesn't exist, consider as new/changed
-    }
-    // Update node with change status and instructions
-    function updateNodeStatus(node, fileName, hasChanged, isCss = false) {
-        if (!hasChanged) {
-            const message = `This file ${fileName} has no changes, no need to update`;
-            if (isCss) {
-                node.cssContent = message;
-                node.instructionCSSForIDE = message;
-            }
-            else {
-                node.instructionForIDE = message;
-                node.content = message;
-            }
-        }
-        else if (!fs.existsSync(node.filePath)) {
-            node.instructionForIDE = 'This file does not exist, please create it';
-        }
-        return hasChanged;
-    }
-    // The main recursive function to process nodes
-    function processNode(node) {
-        let hasChanges = false;
-        // Remove experimental fields first
-        if (node.exportData) {
-            if (node.filePath) {
-                const nodeContent = node.name.includes('.css') ?
-                    node.exportData.cssContent :
-                    node.exportData.content;
-                const fileChanged = checkFile(node.filePath, nodeContent);
-                node.codeChanged = fileChanged;
-                hasChanges = hasChanges || fileChanged;
-                updateNodeStatus(node, node.name, fileChanged);
-                // Check associated CSS module
-                if (node.exportData.cssFileName && node.exportData.cssContent) {
-                    const cssChanged = checkCssFile(node.filePath, node.exportData.cssFileName, node.exportData.cssContent);
-                    hasChanges = hasChanges || cssChanged;
-                    updateNodeStatus(node.exportData, node.exportData.cssFileName, cssChanged, true);
-                }
-            }
-            // Process dependencies
-            if (node.exportData.dependencies && Array.isArray(node.exportData.dependencies)) {
-                for (const dependency of node.exportData.dependencies) {
-                    // Process dependency file
-                    if (dependency.filePath) {
-                        const isCSS = dependency.name && dependency.name.includes('.css');
-                        const depContent = isCSS ? dependency.cssContent : dependency.content;
-                        const fileChanged = checkFile(dependency.filePath, depContent);
-                        dependency.codeChanged = fileChanged;
-                        hasChanges = hasChanges || fileChanged;
-                        updateNodeStatus(dependency, dependency.compName, fileChanged);
-                        // Check associated CSS module
-                        if (dependency.cssFileName && dependency.cssContent) {
-                            const cssChanged = checkCssFile(dependency.filePath, dependency.cssFileName, dependency.cssContent);
-                            hasChanges = hasChanges || cssChanged;
-                            updateNodeStatus(dependency, dependency.cssFileName, cssChanged, true);
-                        }
-                    }
-                    // Process nested dependencies recursively
-                    if (dependency.dependencies && Array.isArray(dependency.dependencies)) {
-                        for (const nestedDep of dependency.dependencies) {
-                            const nestedChanged = processNode(nestedDep);
-                            hasChanges = hasChanges || nestedChanged;
-                        }
-                    }
-                }
-            }
-        }
-        // Process children recursively
-        if (node.children && Array.isArray(node.children)) {
-            for (const child of node.children) {
-                const childChanged = processNode(child);
-                hasChanges = hasChanges || childChanged;
-            }
-        }
-        return hasChanges;
-    }
-    // Start processing from the root
-    return processNode(directoryStructure);
-}
-/**
- * Compares two code snippets to determine if they are functionally different
- * @param fileContent The code currently in the file
- * @param apiCode The code provided by the API
- * @returns True if the code has changed, false otherwise
- */
-function hasCodeChanged(fileContent, apiCode) {
-    if (!fileContent || !apiCode) {
-        return true;
-    }
-    // Normalize both code snippets
-    const normalizedFile = normalizeForComparison(fileContent);
-    const normalizedApi = normalizeForComparison(apiCode);
-    // Simple equality check on normalized versions
-    return normalizedFile !== normalizedApi;
-}
-/**
- * Normalizes code for comparison by removing irrelevant differences
- * @param code The code to normalize
- * @returns Normalized code string
- */
-function normalizeForComparison(code) {
-    return code
-        .replace(/\s+/g, '') // Remove all whitespace (spaces, tabs, newlines)
-        .replace(/[;,'"]/g, '') // Remove semicolons, commas, and quotes
-        .replace(/\/\/.*$/gm, '') // Remove single-line comments
-        .replace(/\/\*[\s\S]*?\*\//g, '') // Remove multi-line comments
-        .toLowerCase(); // Convert to lowercase for case-insensitive comparison
-}
-/**
- * Cleans the directory structure by removing unwanted properties from all nodes
- * @param directoryStructure The directory structure to clean
- * @returns The cleaned directory structure
- */
-function processDirectoryStructure(directoryStructure) {
-    const propertiesToRemove = [
-        'files',
-        'component',
-        'jsLC',
-        'assetCount',
-        'cssLC',
-        'componentCount',
-        'pageMetaData',
-        'isHomePage',
-        'storybookFileName',
-        'isAutoSyncNode'
-    ];
-    const filesToUpdate = [];
-    const filesUnchanged = [];
-    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) {
-                filesToUpdate.push(node.name);
-            }
-            else {
-                filesUnchanged.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) {
-                        filesToUpdate.push(dep.compName);
-                    }
-                    else {
-                        filesUnchanged.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 => {
-            if (node[prop] !== undefined)
-                delete node[prop];
-        });
-        // Clean exportData if it exists
-        if (node.exportData) {
-            propertiesToRemove.forEach(prop => {
-                if (node.exportData[prop] !== undefined)
-                    delete node.exportData[prop];
-            });
-            // Process dependencies in exportData
-            if (node.exportData.dependencies && Array.isArray(node.exportData.dependencies)) {
-                node.exportData.dependencies.forEach((dep) => cleanNode(dep));
-            }
-        }
-        // Process children recursively
-        if (node.children && Array.isArray(node.children)) {
-            node.children.forEach((child) => cleanNode(child));
-        }
-    }
-    // First collect all change statuses
-    collectChangeStatus(directoryStructure);
-    // Then clean the structure
-    cleanNode(directoryStructure);
-    return {
-        directoryStructure: directoryStructure.children,
-        filesToUpdate: [...new Set(filesToUpdate)],
-        filesUnchanged: [...new Set(filesUnchanged)]
-    };
-}
+import{z}from"zod";import*as fs from"fs";import*as path from"path";import axios from"axios";import{getProjectID}from"../helpers/helpers.js";import{track}from"../utils/analytics.js";export const pullFilesToolName="getLatestFileCode";export const pullFilesToolDescription="Cursor IDE uses this tool to retrieve individual file contents without component dependency management. Unlike the component tool that handles components and their dependencies as a unit, this tool focuses on specific files by name, retrieving only their content without dependency resolution. Use this tool when you need the raw content of specific files rather than complete components with their dependency trees. If codeChanged is true, the tool will prompt the user to accept the changes before updating the file. If codeChanged is false, the tool will not update the files.\n  ";export const PullFilesToolSchema=z.object({fileNames:z.array(z.string()).describe("The list of specific file names to be retrieved, without resolving their component dependencies.").max(10).min(1),workspacePath:z.string().describe("The full path to the workspace.")});export async function runPullFilesTool(args){let fileNames=args.fileNames;fileNames=fileNames.map((name=>name.replace(/\.[^/.]+$/,"")));const workspacePath=decodeURIComponent(args.workspacePath),projectID=await getProjectID(workspacePath);if(!projectID||0===projectID.length)throw new Error("PROJECT_ID is not set");const personalAccessToken=process.env.PERSONAL_ACCESS_TOKEN;if(!personalAccessToken)throw new Error("PERSONAL_ACCESS_TOKEN is not set");try{const result=await fetchDirectoryStructure(fileNames,projectID,personalAccessToken);if(result.success){const userID=result.user_id;prepareFilePaths(result.data,workspacePath),await downloadAndSaveAssets(result.data),await checkForCodeChanges(result.data);const cleanedResult=processDirectoryStructure(result.data),source=(process.env.IDE_SOURCE||"other").toLowerCase();return track("mcp_execute_tool_e",{tool_name:pullFilesToolName,project_id:projectID,user_id:userID,source:source}),{content:[{type:"text",text:JSON.stringify(cleanedResult,null,2)}]}}return{content:[{type:"text",statusCode:result.statusCode||"unknown",text:JSON.stringify({error:result.error},null,2)}]}}catch(error){return{content:[{type:"text",text:JSON.stringify({error:"Failed to fetch directory structure "+error},null,2)}]}}}async function fetchDirectoryStructure(fileNames,projectID,personalAccessToken){const encodedNames=fileNames.map((name=>encodeURIComponent(name))).join(",");let baseURL="https://codegen-api.locofy.ai/mcp/generators/";"true"===process.env.IS_DEV&&(baseURL="https://codegen-api.locofy.dev/mcp/generators/");const url=baseURL+projectID+"?name="+encodedNames,headers={Accept:"*/*","Content-Type":"application/json",Authorization:"Bearer "+personalAccessToken},data={};try{const response=await axios.post(url,data,{headers:headers});return 200!==response.status?{success:!1,error:`API returned status code ${response.status}`,statusCode:response.status}:response.data&&response.data.directoryTree?{success:!0,data:response.data.directoryTree,user_id:response.data.user_id}:{success:!1,error:"API response is missing directoryTree property",statusCode:response.status}}catch(error){if(axios.isAxiosError(error)){if(error.response){const statusCode=error.response.status;let errorMessage="";return errorMessage=401===statusCode?"Authentication failed: Invalid or expired token. Please check your mcp.json configuration in the .cursor folder to ensure your PERSONAL_ACCESS_TOKEN is correct. You can visit the Locofy project settings page to check or regenerate your token if needed.":500===statusCode?"Server error: The API service is experiencing issues":`API request failed with status ${statusCode}: ${error.message}`,{success:!1,error:errorMessage,statusCode:statusCode}}if(error.request)return{success:!1,error:"Network error: No response received from API",statusCode:"network_error"}}return{success:!1,error:`Failed to fetch directory structure: ${error instanceof Error?error.message:"Unknown error"}`,statusCode:"unknown_error"}}}function prepareFilePaths(directoryStructure,workspacePath){const processNode=node=>{if(node.filePath&&(path.isAbsolute(node.filePath)||(node.filePath=path.join(workspacePath,node.filePath))),node.exportData){if(node.exportData.filePath&&(path.isAbsolute(node.exportData.filePath)||(node.exportData.filePath=path.join(workspacePath,node.exportData.filePath))),node.exportData.files&&Array.isArray(node.exportData.files))for(const file of node.exportData.files)file.filePath&&!path.isAbsolute(file.filePath)&&(file.filePath=path.join(workspacePath,file.filePath));if(node.exportData.dependencies&&Array.isArray(node.exportData.dependencies))for(const dependency of node.exportData.dependencies){if(dependency.filePath&&!path.isAbsolute(dependency.filePath)&&(dependency.filePath=path.join(workspacePath,dependency.filePath)),dependency.files&&Array.isArray(dependency.files))for(const file of dependency.files)file.filePath&&!path.isAbsolute(file.filePath)&&(file.filePath=path.join(workspacePath,file.filePath));if(dependency.dependencies&&Array.isArray(dependency.dependencies))for(const nestedDep of dependency.dependencies)processNode(nestedDep)}}if(node.content&&node.content.filePath&&(path.isAbsolute(node.content.filePath)||(node.content.filePath=path.join(workspacePath,node.content.filePath))),node.children&&Array.isArray(node.children))for(const child of node.children)processNode(child)};processNode(directoryStructure)}async function downloadAndSaveAssets(directoryStructure){const processNode=async node=>{if("file"===node.type&&node.exportData&&node.exportData.files)for(const file of node.exportData.files)if(file.url&&file.fileName){const filePath=file.filePath;if(!filePath)continue;const dirPath=path.dirname(filePath);if(fs.existsSync(dirPath)||fs.mkdirSync(dirPath,{recursive:!0}),!fs.existsSync(filePath))try{const response=await axios.get(file.url,{responseType:"arraybuffer"});fs.writeFileSync(filePath,Buffer.from(response.data))}catch(error){}}if(node.children&&Array.isArray(node.children))for(const child of node.children)await processNode(child)};await processNode(directoryStructure)}async function checkForCodeChanges(directoryStructure){function checkFile(filePath,content){if(!fs.existsSync(filePath))return!0;try{const codeChanged=hasCodeChanged(fs.readFileSync(filePath,"utf8"),content);return codeChanged}catch(error){return!0}}function checkCssFile(basePath,cssFileName,cssContent){if(!cssFileName||!cssContent)return!1;const cssFilePath=path.join(path.dirname(basePath),cssFileName);if(fs.existsSync(cssFilePath)){return!!hasCodeChanged(fs.readFileSync(cssFilePath,"utf8"),cssContent)}return!0}function updateNodeStatus(node,fileName,hasChanged,isCss=!1){if(hasChanged)fs.existsSync(node.filePath)||(node.instructionForIDE="This file does not exist, please create it");else{const message=`This file ${fileName} has no changes, no need to update`;isCss?(node.cssContent=message,node.instructionCSSForIDE=message):(node.instructionForIDE=message,node.content=message)}return hasChanged}return function processNode(node){let hasChanges=!1;if(node.exportData){if(node.filePath){const nodeContent=node.name.includes(".css")?node.exportData.cssContent:node.exportData.content,fileChanged=checkFile(node.filePath,nodeContent);if(node.codeChanged=fileChanged,hasChanges=hasChanges||fileChanged,updateNodeStatus(node,node.name,fileChanged),node.exportData.cssFileName&&node.exportData.cssContent){const cssChanged=checkCssFile(node.filePath,node.exportData.cssFileName,node.exportData.cssContent);hasChanges=hasChanges||cssChanged,updateNodeStatus(node.exportData,node.exportData.cssFileName,cssChanged,!0)}}if(node.exportData.dependencies&&Array.isArray(node.exportData.dependencies))for(const dependency of node.exportData.dependencies){if(dependency.filePath){const depContent=dependency.name&&dependency.name.includes(".css")?dependency.cssContent:dependency.content,fileChanged=checkFile(dependency.filePath,depContent);if(dependency.codeChanged=fileChanged,hasChanges=hasChanges||fileChanged,updateNodeStatus(dependency,dependency.compName,fileChanged),dependency.cssFileName&&dependency.cssContent){const cssChanged=checkCssFile(dependency.filePath,dependency.cssFileName,dependency.cssContent);hasChanges=hasChanges||cssChanged,updateNodeStatus(dependency,dependency.cssFileName,cssChanged,!0)}}if(dependency.dependencies&&Array.isArray(dependency.dependencies))for(const nestedDep of dependency.dependencies){const nestedChanged=processNode(nestedDep);hasChanges=hasChanges||nestedChanged}}}if(node.children&&Array.isArray(node.children))for(const child of node.children){const childChanged=processNode(child);hasChanges=hasChanges||childChanged}return hasChanges}(directoryStructure)}function hasCodeChanged(fileContent,apiCode){if(!fileContent||!apiCode)return!0;return normalizeForComparison(fileContent)!==normalizeForComparison(apiCode)}function normalizeForComparison(code){return code.replace(/\s+/g,"").replace(/[;,'"]/g,"").replace(/\/\/.*$/gm,"").replace(/\/\*[\s\S]*?\*\//g,"").toLowerCase()}function processDirectoryStructure(directoryStructure){const propertiesToRemove=["files","component","jsLC","assetCount","cssLC","componentCount","pageMetaData","isHomePage","storybookFileName","isAutoSyncNode"],filesToUpdate=[],filesUnchanged=[];return function collectChangeStatus(node){node.name&&void 0!==node.codeChanged&&(node.codeChanged?filesToUpdate.push(node.name):filesUnchanged.push(node.name)),node.exportData&&node.exportData.dependencies&&node.exportData.dependencies.forEach((dep=>{dep.compName&&void 0!==dep.codeChanged&&(dep.codeChanged?filesToUpdate.push(dep.compName):filesUnchanged.push(dep.compName)),dep.dependencies&&Array.isArray(dep.dependencies)&&dep.dependencies.forEach((nestedDep=>collectChangeStatus(nestedDep)))})),node.children&&Array.isArray(node.children)&&node.children.forEach((child=>collectChangeStatus(child)))}(directoryStructure),function cleanNode(node){propertiesToRemove.forEach((prop=>{void 0!==node[prop]&&delete node[prop]})),node.exportData&&(propertiesToRemove.forEach((prop=>{void 0!==node.exportData[prop]&&delete node.exportData[prop]})),node.exportData.dependencies&&Array.isArray(node.exportData.dependencies)&&node.exportData.dependencies.forEach((dep=>cleanNode(dep)))),node.children&&Array.isArray(node.children)&&node.children.forEach((child=>cleanNode(child)))}(directoryStructure),{directoryStructure:directoryStructure.children,filesToUpdate:[...new Set(filesToUpdate)],filesUnchanged:[...new Set(filesUnchanged)]}}

package/build/utils/analytics.js

@@ -1,47 +1 @@
-import { MIXPANEL_PROJECT_TOKEN, MIXPANEL_PROXY_DOMAIN } from '../constants.js';
-export function track(event, props = {}) {
-    if (process.env.NODE_ENV === 'test' || process.env.IS_TEST === 'true') {
-        console.log('Analytics disabled: NODE_ENV is test or IS_TEST is true');
-        return;
-    }
-    if (!MIXPANEL_PROJECT_TOKEN || !MIXPANEL_PROXY_DOMAIN) {
-        console.log('Analytics disabled: MIXPANEL_PROJECT_TOKEN or MIXPANEL_PROXY_DOMAIN not set');
-        return;
-    }
-    try {
-        // Create the Mixpanel-compatible event object
-        const eventObject = {
-            event: event,
-            properties: {
-                ...props,
-                token: MIXPANEL_PROJECT_TOKEN,
-                time: Math.floor(Date.now() / 1000),
-                distinct_id: props.user_id || 'anonymous',
-            }
-        };
-        const encodedData = Buffer.from(JSON.stringify([eventObject])).toString('base64');
-        console.log(`Analytics: tracking event '${event}'`);
-        fetch(MIXPANEL_PROXY_DOMAIN + '/track/?data=' + encodedData, {
-            method: 'GET',
-            headers: {
-                'Accept': 'text/plain'
-            }
-        })
-            .then(response => {
-            if (response.ok) { // Status code in the range 200-299
-                if (process.env.IS_DEV === 'true') {
-                    console.log(`Analytics: successfully sent event '${event}' with status ${response.status}`);
-                }
-            }
-            else {
-                console.error(`Error sending event '${event}' to Mixpanel:`, response.statusText);
-            }
-        })
-            .catch(error => {
-            console.error(`Error sending event '${event}' to Mixpanel:`, error);
-        });
-    }
-    catch (error) {
-        console.error(`Error preparing event '${event}' for tracking:`, error);
-    }
-}
+import{MIXPANEL_PROJECT_TOKEN,MIXPANEL_PROXY_DOMAIN}from"../constants.js";export function track(event,props={}){if("test"!==process.env.NODE_ENV&&"true"!==process.env.IS_TEST&&MIXPANEL_PROJECT_TOKEN&&MIXPANEL_PROXY_DOMAIN)try{const eventObject={event:event,properties:{...props,token:MIXPANEL_PROJECT_TOKEN,time:Math.floor(Date.now()/1e3),distinct_id:props.user_id||"anonymous"}},encodedData=Buffer.from(JSON.stringify([eventObject])).toString("base64");fetch(MIXPANEL_PROXY_DOMAIN+"/track/?data="+encodedData,{method:"GET",headers:{Accept:"text/plain"}}).then((response=>{response.ok&&process.env.IS_DEV})).catch((error=>{}))}catch(error){}}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@locofy/mcp",
-  "version": "1.1.0",
+  "version": "1.1.2",
   "description": "Locofy MCP Server with Cursor",
   "keywords": [
     "figma",
@@ -18,10 +18,11 @@
   "main": "./src/index.ts",
   "type": "module",
   "scripts": {
-    "build": "rm -rf build && tsc && node -e \"const fs=require('fs');const f='build/index.js';fs.writeFileSync(f,'#!/usr/bin/env node\\n'+fs.readFileSync(f));fs.chmodSync(f,'755')\"",
+    "build": "rm -rf build && tsc && node -e \"const fs=require('fs');const f='build/index.js';fs.writeFileSync(f,'#!/usr/bin/env node\\n'+fs.readFileSync(f));fs.chmodSync(f,'755')\" && find build -name '*.js' -exec npx terser {} --compress drop_console=true --output {} \\;",
     "link": "npm run build && npm link",
     "start": "node build/index.js",
-    "lint": "eslint . --ext .ts --fix"
+    "lint": "eslint . --ext .ts --fix --ignore-pattern \"**/__tests__/**\" --ignore-pattern \"**/*.test.ts\"",
+    "test": "NODE_ENV=test jest --coverage --coverageReporters=text-summary --detectOpenHandles --forceExit"
   },
   "bin": {
     "locofy-mcp": "./build/index.js"
@@ -36,12 +37,17 @@
     "zod": "^3.24.1"
   },
   "devDependencies": {
+    "@jest/globals": "^29.7.0",
     "@types/express": "^5.0.1",
+    "@types/jest": "^29.5.14",
     "@types/node": "^22.13.0",
     "@typescript-eslint/eslint-plugin": "^6.7.0",
     "@typescript-eslint/parser": "^6.7.0",
     "eslint": "^8.49.0",
     "eslint-config-prettier": "^9.0.0",
+    "terser": "^5.39.0",
+    "jest": "^29.7.0",
+    "ts-jest": "^29.3.0",
     "typescript": "^5.7.3"
   }
 }