@locofy/mcp 1.0.6 → 1.0.8

package/build/helpers/helpers.js

@@ -0,0 +1,31 @@
+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;
+}

package/build/tools/pullComponents.js

@@ -2,6 +2,7 @@ import { z } from 'zod';
 import * as fs from 'fs';
 import * as path from 'path';
 import axios from 'axios';
+import { getProjectID } from '../helpers/helpers.js';
 /**
  * PullComponents tool
  *   - Returns a hardcoded representation of a directory structure
@@ -19,10 +20,13 @@ export async function runPullComponentsTool(args) {
     // clean the component names, remove extensions
     componentNames = componentNames.map(name => name.replace(/\.[^/.]+$/, ''));
     const workspacePath = decodeURIComponent(args.workspacePath);
-    const projectID = process.env.PROJECT_ID;
+    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 (!projectID || !personalAccessToken) {
-        throw new Error('PROJECT_ID or PERSONAL_ACCESS_TOKEN is not set');
+    if (!personalAccessToken) {
+        throw new Error('PERSONAL_ACCESS_TOKEN is not set');
     }
     try {
         const result = await fetchDirectoryStructure(componentNames, projectID, personalAccessToken);
@@ -77,10 +81,11 @@ export async function runPullComponentsTool(args) {
 }
 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/';
-    }
+    // let baseURL = 'https://codegen-api.locofy.ai/mcp/generators/';
+    // if (process.env.IS_DEV === 'true') {
+    //   baseURL = 'https://codegen-api.locofy.dev/mcp/generators/';
+    // }
+    const baseURL = 'https://codegen-api.locofy.dev/mcp/generators/';
     const url = baseURL + projectID + '?name=' + encodedNames;
     const headers = {
         'Accept': '*/*',

package/build/tools/pullFiles.js

@@ -2,6 +2,7 @@ import { z } from 'zod';
 import * as fs from 'fs';
 import * as path from 'path';
 import axios from 'axios';
+import { getProjectID } from '../helpers/helpers.js';
 /**
  * PullFiles tool
  *   - Returns a hardcoded representation of a directory structure
@@ -19,10 +20,13 @@ export async function runPullFilesTool(args) {
     // clean the file names, remove extensions
     fileNames = fileNames.map(name => name.replace(/\.[^/.]+$/, ''));
     const workspacePath = decodeURIComponent(args.workspacePath);
-    const projectID = process.env.PROJECT_ID;
+    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 (!projectID || !personalAccessToken) {
-        throw new Error('PROJECT_ID or PERSONAL_ACCESS_TOKEN is not set');
+    if (!personalAccessToken) {
+        throw new Error('PERSONAL_ACCESS_TOKEN is not set');
     }
     try {
         const result = await fetchDirectoryStructure(fileNames, projectID, personalAccessToken);
@@ -77,10 +81,11 @@ export async function runPullFilesTool(args) {
 }
 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/';
-    }
+    // let baseURL = 'https://codegen-api.locofy.ai/mcp/generators/';
+    // if (process.env.IS_DEV === 'true') {
+    //   baseURL = 'https://codegen-api.locofy.dev/mcp/generators/';
+    // }
+    const baseURL = 'https://codegen-api.locofy.dev/mcp/generators/';
     const url = baseURL + projectID + '?name=' + encodedNames;
     const headers = {
         'Accept': '*/*',

package/package.json

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