sonance-brand-mcp 1.3.31 → 1.3.32

package/dist/assets/api/sonance-vision-apply/route.ts

@@ -212,8 +212,8 @@ export async function POST(request: Request) {
       // Generate a unique session ID
       const newSessionId = randomUUID().slice(0, 8);
 
-      // Gather page context
-      const pageContext = gatherPageContext(pageRoute || "/", projectRoot);
+      // Gather page context (with keyword search based on user prompt)
+      const pageContext = gatherPageContext(pageRoute || "/", projectRoot, userPrompt);
 
       // Build user message with vision
       const messageContent: Anthropic.MessageCreateParams["messages"][0]["content"] = [];
@@ -843,11 +843,12 @@ function discoverLayoutFiles(pageFile: string | null, projectRoot: string): stri
 
 /**
  * Gather context about the current page for AI analysis
- * Uses recursive import resolution to build complete component graph
+ * Uses recursive import resolution AND keyword search to build complete component graph
  */
 function gatherPageContext(
   pageRoute: string,
-  projectRoot: string
+  projectRoot: string,
+  userPrompt?: string
 ): {
   pageFile: string | null;
   pageContent: string;
@@ -886,6 +887,24 @@ function gatherPageContext(
     }
   }
 
+  // KEYWORD SEARCH: Find additional files based on user prompt keywords
+  // This catches files that aren't in the import chain but contain relevant UI text
+  if (userPrompt) {
+    const keywords = extractKeywordsFromPrompt(userPrompt);
+    if (keywords.length > 0) {
+      const searchResults = searchFilesForKeywords(keywords, projectRoot, 15);
+      
+      for (const result of searchResults) {
+        // Only add if not already in our list
+        if (!visited.has(result.path)) {
+          visited.add(result.path);
+          componentSources.push({ path: result.path, content: result.content });
+          debugLog("[apply] Added file from keyword search", { path: result.path, matchCount: result.matchCount });
+        }
+      }
+    }
+  }
+
   let globalsCSS = "";
   const globalsCSSPatterns = [
     "src/app/globals.css",
@@ -1087,19 +1106,142 @@ function findDynamicRoute(cleanRoute: string, projectRoot: string): string | nul
 }
 
 function extractImports(content: string): string[] {
-  // Match ALL @/ imports (not just @/components) and relative imports
-  const importRegex = /import\s+.*?\s+from\s+["'](@\/[^"']+|\.\.?\/[^"']+)["']/g;
+  // Match both import AND export statements for @/ and relative paths
+  // This catches barrel files (index.ts) that use: export { Foo } from './Foo'
+  const importExportRegex = /(?:import|export)\s+.*?\s+from\s+["'](@\/[^"']+|\.\.?\/[^"']+)["']/g;
   const imports: string[] = [];
   let match;
 
-  while ((match = importRegex.exec(content)) !== null) {
+  while ((match = importExportRegex.exec(content)) !== null) {
     imports.push(match[1]);
   }
 
-  debugLog("[apply] Extracted imports", { count: imports.length, imports: imports.slice(0, 10) });
+  debugLog("[apply] Extracted imports/exports", { count: imports.length, imports: imports.slice(0, 10) });
   return imports;
 }
 
+/**
+ * Extract meaningful keywords from user prompt for file searching
+ * Looks for quoted strings, capitalized phrases, and specific patterns
+ */
+function extractKeywordsFromPrompt(prompt: string): string[] {
+  const keywords: string[] = [];
+  
+  // 1. Extract quoted strings (e.g., "View Assets", 'Edit Process')
+  const quotedRegex = /["']([^"']+)["']/g;
+  let match;
+  while ((match = quotedRegex.exec(prompt)) !== null) {
+    if (match[1].length > 2 && match[1].length < 50) {
+      keywords.push(match[1]);
+    }
+  }
+  
+  // 2. Extract capitalized phrases (e.g., "View Assets", "Quick Actions")
+  // Matches 2+ capitalized words together
+  const capitalizedRegex = /\b([A-Z][a-z]+(?:\s+[A-Z][a-z]+)+)\b/g;
+  while ((match = capitalizedRegex.exec(prompt)) !== null) {
+    if (!keywords.includes(match[1])) {
+      keywords.push(match[1]);
+    }
+  }
+  
+  // 3. Extract single capitalized words that might be component names
+  const singleCapRegex = /\b([A-Z][a-z]{3,})\b/g;
+  while ((match = singleCapRegex.exec(prompt)) !== null) {
+    // Skip common words
+    const skipWords = ["The", "This", "That", "Make", "Change", "Edit", "Update", "Add", "Remove", "Delete"];
+    if (!skipWords.includes(match[1]) && !keywords.includes(match[1])) {
+      keywords.push(match[1]);
+    }
+  }
+  
+  debugLog("[apply] Extracted keywords from prompt", { prompt: prompt.substring(0, 100), keywords });
+  return keywords;
+}
+
+/**
+ * Recursively search files in a directory for keyword matches
+ * Returns file paths that contain any of the keywords
+ */
+function searchFilesForKeywords(
+  keywords: string[],
+  projectRoot: string,
+  maxFiles: number = 10
+): { path: string; content: string; matchCount: number }[] {
+  if (keywords.length === 0) return [];
+  
+  const results: { path: string; content: string; matchCount: number }[] = [];
+  const searchDirs = ["components", "src/components", "app", "src/app", "pages", "src/pages"];
+  const extensions = [".tsx", ".jsx", ".ts", ".js"];
+  const visited = new Set<string>();
+  
+  function searchDir(dirPath: string, depth: number = 0) {
+    if (depth > 5 || results.length >= maxFiles) return;
+    
+    const fullDirPath = path.join(projectRoot, dirPath);
+    if (!fs.existsSync(fullDirPath) || visited.has(fullDirPath)) return;
+    visited.add(fullDirPath);
+    
+    try {
+      const entries = fs.readdirSync(fullDirPath, { withFileTypes: true });
+      
+      for (const entry of entries) {
+        if (results.length >= maxFiles) break;
+        
+        const entryPath = path.join(dirPath, entry.name);
+        const fullEntryPath = path.join(projectRoot, entryPath);
+        
+        if (entry.isDirectory()) {
+          // Skip node_modules, .git, etc.
+          if (!entry.name.startsWith(".") && entry.name !== "node_modules") {
+            searchDir(entryPath, depth + 1);
+          }
+        } else if (entry.isFile() && extensions.some(ext => entry.name.endsWith(ext))) {
+          try {
+            const content = fs.readFileSync(fullEntryPath, "utf-8");
+            
+            // Count keyword matches
+            let matchCount = 0;
+            for (const keyword of keywords) {
+              // Case-insensitive search
+              const regex = new RegExp(keyword.replace(/[.*+?^${}()|[\]\\]/g, "\\$&"), "gi");
+              const matches = content.match(regex);
+              if (matches) {
+                matchCount += matches.length;
+              }
+            }
+            
+            if (matchCount > 0) {
+              results.push({ path: entryPath, content, matchCount });
+            }
+          } catch {
+            // Skip files that can't be read
+          }
+        }
+      }
+    } catch {
+      // Skip directories that can't be read
+    }
+  }
+  
+  // Search each directory
+  for (const dir of searchDirs) {
+    searchDir(dir);
+    if (results.length >= maxFiles) break;
+  }
+  
+  // Sort by match count (most matches first)
+  results.sort((a, b) => b.matchCount - a.matchCount);
+  
+  debugLog("[apply] Keyword search results", { 
+    keywords, 
+    filesFound: results.length,
+    topMatches: results.slice(0, 5).map(r => ({ path: r.path, matchCount: r.matchCount }))
+  });
+  
+  return results.slice(0, maxFiles);
+}
+
 // Cache for tsconfig path aliases
 let cachedPathAliases: Map<string, string> | null = null;
 let cachedProjectRoot: string | null = null;

package/dist/assets/api/sonance-vision-edit/route.ts

@@ -222,8 +222,8 @@ export async function POST(request: Request) {
         );
       }
 
-      // Gather page context (including focused element files via project-wide search)
-      const pageContext = gatherPageContext(pageRoute, projectRoot, focusedElements);
+      // Gather page context (including focused element files and keyword search)
+      const pageContext = gatherPageContext(pageRoute, projectRoot, focusedElements, userPrompt);
 
       // Build user message with vision
       const messageContent: Anthropic.MessageCreateParams["messages"][0]["content"] = [];
@@ -711,12 +711,13 @@ function discoverLayoutFiles(pageFile: string | null, projectRoot: string): stri
 
 /**
  * Gather context about the current page for AI analysis
- * Uses recursive import resolution to build complete component graph
+ * Uses recursive import resolution AND keyword search to build complete component graph
  */
 function gatherPageContext(
   pageRoute: string,
   projectRoot: string,
-  focusedElements?: VisionFocusedElement[]
+  focusedElements?: VisionFocusedElement[],
+  userPrompt?: string
 ): {
   pageFile: string | null;
   pageContent: string;
@@ -768,6 +769,24 @@ function gatherPageContext(
     }
   }
 
+  // KEYWORD SEARCH: Find additional files based on user prompt keywords
+  // This catches files that aren't in the import chain but contain relevant UI text
+  if (userPrompt) {
+    const keywords = extractKeywordsFromPrompt(userPrompt);
+    if (keywords.length > 0) {
+      const searchResults = searchFilesForKeywords(keywords, projectRoot, 15);
+      
+      for (const result of searchResults) {
+        // Only add if not already in our list
+        if (!visited.has(result.path)) {
+          visited.add(result.path);
+          componentSources.push({ path: result.path, content: result.content });
+          debugLog("[edit] Added file from keyword search", { path: result.path, matchCount: result.matchCount });
+        }
+      }
+    }
+  }
+
   // Read globals.css - check multiple possible locations
   let globalsCSS = "";
   const globalsCSSPatterns = [
@@ -978,19 +997,142 @@ function findDynamicRoute(cleanRoute: string, projectRoot: string): string | nul
  * Extract import paths from file content
  */
 function extractImports(content: string): string[] {
-  // Match ALL @/ imports (not just @/components) and relative imports
-  const importRegex = /import\s+.*?\s+from\s+["'](@\/[^"']+|\.\.?\/[^"']+)["']/g;
+  // Match both import AND export statements for @/ and relative paths
+  // This catches barrel files (index.ts) that use: export { Foo } from './Foo'
+  const importExportRegex = /(?:import|export)\s+.*?\s+from\s+["'](@\/[^"']+|\.\.?\/[^"']+)["']/g;
   const imports: string[] = [];
   let match;
 
-  while ((match = importRegex.exec(content)) !== null) {
+  while ((match = importExportRegex.exec(content)) !== null) {
     imports.push(match[1]);
   }
 
-  debugLog("[edit] Extracted imports", { count: imports.length, imports: imports.slice(0, 10) });
+  debugLog("[edit] Extracted imports/exports", { count: imports.length, imports: imports.slice(0, 10) });
   return imports;
 }
 
+/**
+ * Extract meaningful keywords from user prompt for file searching
+ * Looks for quoted strings, capitalized phrases, and specific patterns
+ */
+function extractKeywordsFromPrompt(prompt: string): string[] {
+  const keywords: string[] = [];
+  
+  // 1. Extract quoted strings (e.g., "View Assets", 'Edit Process')
+  const quotedRegex = /["']([^"']+)["']/g;
+  let match;
+  while ((match = quotedRegex.exec(prompt)) !== null) {
+    if (match[1].length > 2 && match[1].length < 50) {
+      keywords.push(match[1]);
+    }
+  }
+  
+  // 2. Extract capitalized phrases (e.g., "View Assets", "Quick Actions")
+  // Matches 2+ capitalized words together
+  const capitalizedRegex = /\b([A-Z][a-z]+(?:\s+[A-Z][a-z]+)+)\b/g;
+  while ((match = capitalizedRegex.exec(prompt)) !== null) {
+    if (!keywords.includes(match[1])) {
+      keywords.push(match[1]);
+    }
+  }
+  
+  // 3. Extract single capitalized words that might be component names
+  const singleCapRegex = /\b([A-Z][a-z]{3,})\b/g;
+  while ((match = singleCapRegex.exec(prompt)) !== null) {
+    // Skip common words
+    const skipWords = ["The", "This", "That", "Make", "Change", "Edit", "Update", "Add", "Remove", "Delete"];
+    if (!skipWords.includes(match[1]) && !keywords.includes(match[1])) {
+      keywords.push(match[1]);
+    }
+  }
+  
+  debugLog("[edit] Extracted keywords from prompt", { prompt: prompt.substring(0, 100), keywords });
+  return keywords;
+}
+
+/**
+ * Recursively search files in a directory for keyword matches
+ * Returns file paths that contain any of the keywords
+ */
+function searchFilesForKeywords(
+  keywords: string[],
+  projectRoot: string,
+  maxFiles: number = 10
+): { path: string; content: string; matchCount: number }[] {
+  if (keywords.length === 0) return [];
+  
+  const results: { path: string; content: string; matchCount: number }[] = [];
+  const searchDirs = ["components", "src/components", "app", "src/app", "pages", "src/pages"];
+  const extensions = [".tsx", ".jsx", ".ts", ".js"];
+  const visited = new Set<string>();
+  
+  function searchDir(dirPath: string, depth: number = 0) {
+    if (depth > 5 || results.length >= maxFiles) return;
+    
+    const fullDirPath = path.join(projectRoot, dirPath);
+    if (!fs.existsSync(fullDirPath) || visited.has(fullDirPath)) return;
+    visited.add(fullDirPath);
+    
+    try {
+      const entries = fs.readdirSync(fullDirPath, { withFileTypes: true });
+      
+      for (const entry of entries) {
+        if (results.length >= maxFiles) break;
+        
+        const entryPath = path.join(dirPath, entry.name);
+        const fullEntryPath = path.join(projectRoot, entryPath);
+        
+        if (entry.isDirectory()) {
+          // Skip node_modules, .git, etc.
+          if (!entry.name.startsWith(".") && entry.name !== "node_modules") {
+            searchDir(entryPath, depth + 1);
+          }
+        } else if (entry.isFile() && extensions.some(ext => entry.name.endsWith(ext))) {
+          try {
+            const content = fs.readFileSync(fullEntryPath, "utf-8");
+            
+            // Count keyword matches
+            let matchCount = 0;
+            for (const keyword of keywords) {
+              // Case-insensitive search
+              const regex = new RegExp(keyword.replace(/[.*+?^${}()|[\]\\]/g, "\\$&"), "gi");
+              const matches = content.match(regex);
+              if (matches) {
+                matchCount += matches.length;
+              }
+            }
+            
+            if (matchCount > 0) {
+              results.push({ path: entryPath, content, matchCount });
+            }
+          } catch {
+            // Skip files that can't be read
+          }
+        }
+      }
+    } catch {
+      // Skip directories that can't be read
+    }
+  }
+  
+  // Search each directory
+  for (const dir of searchDirs) {
+    searchDir(dir);
+    if (results.length >= maxFiles) break;
+  }
+  
+  // Sort by match count (most matches first)
+  results.sort((a, b) => b.matchCount - a.matchCount);
+  
+  debugLog("[edit] Keyword search results", { 
+    keywords, 
+    filesFound: results.length,
+    topMatches: results.slice(0, 5).map(r => ({ path: r.path, matchCount: r.matchCount }))
+  });
+  
+  return results.slice(0, maxFiles);
+}
+
 // Cache for tsconfig path aliases
 let cachedPathAliases: Map<string, string> | null = null;
 let cachedProjectRoot: string | null = null;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sonance-brand-mcp",
-  "version": "1.3.31",
+  "version": "1.3.32",
   "description": "MCP Server for Sonance Brand Guidelines and Component Library - gives Claude instant access to brand colors, typography, and UI components.",
   "main": "dist/index.js",
   "type": "module",