carto-md 1.0.14 → 1.0.15

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "carto-md",
-  "version": "1.0.14",
+  "version": "1.0.15",
   "description": "The context layer for AI-native development.",
   "bin": {
     "carto": "src/cli/index.js"

package/src/detector/files.js

@@ -1,10 +1,10 @@
 const fs = require('fs');
 const path = require('path');
 
-const MAX_FILES_TOTAL = 50;
-const ROUTE_BUDGET = 20;
+const MAX_FILES_TOTAL = 80;
+const BASE_ROUTE_BUDGET = 20;
 const MODEL_BUDGET = 10;
-const UTILITY_BUDGET = 20;
+const BASE_UTILITY_BUDGET = 20;
 
 const PYTHON_IGNORE = new Set(['__pycache__', '.venv', 'venv', 'migrations', 'node_modules', '.git', '.carto']);
 const JS_IGNORE = new Set(['node_modules', '.git', 'dist', 'build', '.carto', '.next', '.turbo', 'coverage', 'out', '.cache', 'generated', '__generated__', 'storybook-static', 'public', 'static']);
@@ -92,17 +92,19 @@ function smartSelect(allFiles, htmlFiles, projectRoot) {
     }
   }
 
-  // Sort routes by score (entry points first)
+  // Dynamic route budget — expand if many route files, compensate from utility budget
+  const routeBudget = routeCandidates.length > BASE_ROUTE_BUDGET
+    ? Math.min(routeCandidates.length, 40)
+    : BASE_ROUTE_BUDGET;
+  const utilityBudget = Math.max(10, MAX_FILES_TOTAL - routeBudget - MODEL_BUDGET);
+
   routeCandidates.sort((a, b) => scoreRoute(b) - scoreRoute(a));
-  const selectedRoutes = routeCandidates.slice(0, ROUTE_BUDGET);
+  const selectedRoutes = routeCandidates.slice(0, routeBudget);
 
-  // Sort models by score (schema files first)
   modelCandidates.sort((a, b) => scoreModel(b) - scoreModel(a));
   const selectedModels = modelCandidates.slice(0, MODEL_BUDGET);
 
-  // For utilities: scan all files for import statements, count how many times each is imported
   const importCounts = countImportReferences(allFiles, projectRoot);
-  // Rank other files by how often they're imported
   otherFiles.sort((a, b) => {
     const relA = path.relative(projectRoot, a);
     const relB = path.relative(projectRoot, b);
@@ -113,7 +115,7 @@ function smartSelect(allFiles, htmlFiles, projectRoot) {
   const remainingBudget = MAX_FILES_TOTAL - alreadySelected.size;
   const selectedUtilities = otherFiles
     .filter(f => !alreadySelected.has(f))
-    .slice(0, Math.min(UTILITY_BUDGET, remainingBudget));
+    .slice(0, Math.min(utilityBudget, remainingBudget));
 
   const allSelected = [...new Set([...selectedRoutes, ...selectedModels, ...selectedUtilities])];
 

package/src/extractors/imports.js

@@ -199,9 +199,14 @@ function buildImportGraph(fileContents, projectRoot) {
   const graph = {};
 
   for (const { filePath, content } of fileContents) {
+    const relPath = path.relative(projectRoot, filePath);
+    const base = path.basename(filePath);
+
+    // Skip generated files — they produce massive noisy edges
+    if (base.includes('.generated.') || relPath.includes('__generated__')) continue;
+
     const deps = extractImports(content, filePath, projectRoot);
     if (deps.length > 0) {
-      const relPath = path.relative(projectRoot, filePath);
       graph[relPath] = deps;
     }
   }

package/src/extractors/languages/prisma.js

@@ -27,18 +27,13 @@ module.exports = {
 
 function extractPrismaModels(content) {
   const models = [];
-  const modelPattern = /^model\s+(\w+)\s*\{([^}]+)\}/gm;
+  const blocks = extractModelBlocks(content);
 
-  let match;
-  while ((match = modelPattern.exec(content)) !== null) {
-    const className = match[1];
-    const body = match[2];
+  for (const { name, body } of blocks) {
     const fields = [];
-
     const lines = body.split('\n');
     for (const line of lines) {
       const trimmed = line.trim();
-      // Skip empty lines, comments, and @@directives
       if (!trimmed || trimmed.startsWith('//') || trimmed.startsWith('@@')) continue;
 
       const fieldMatch = trimmed.match(/^(\w+)\s+(\w+[\[\]?]*)/);
@@ -46,8 +41,7 @@ function extractPrismaModels(content) {
         fields.push({ name: fieldMatch[1], type: fieldMatch[2] });
       }
     }
-
-    models.push({ className, fields });
+    models.push({ className: name, fields });
   }
 
   return models;
@@ -55,21 +49,43 @@ function extractPrismaModels(content) {
 
 function extractPrismaDBTables(content) {
   const tables = [];
-  const modelPattern = /^model\s+(\w+)\s*\{([^}]+)\}/gm;
+  const blocks = extractModelBlocks(content);
+
+  for (const { name, body } of blocks) {
+    const mapMatch = body.match(/@@map\s*\(\s*['"]([^'"]+)['"]\s*\)/);
+    const tableName = mapMatch ? mapMatch[1] : toSnakeCase(name);
+    tables.push({ tableName, modelName: name });
+  }
 
+  return tables;
+}
+
+/**
+ * Brace-counting approach to extract model blocks.
+ * Handles } inside /// comments and @zod annotations.
+ */
+function extractModelBlocks(content) {
+  const blocks = [];
+  const modelStart = /^model\s+(\w+)\s*\{/gm;
   let match;
-  while ((match = modelPattern.exec(content)) !== null) {
-    const modelName = match[1];
-    const body = match[2];
 
-    // Check for @@map('table_name')
-    const mapMatch = body.match(/@@map\s*\(\s*['"]([^'"]+)['"]\s*\)/);
-    const tableName = mapMatch ? mapMatch[1] : toSnakeCase(modelName);
+  while ((match = modelStart.exec(content)) !== null) {
+    const name = match[1];
+    const startIdx = match.index + match[0].length;
+    let depth = 1;
+    let i = startIdx;
 
-    tables.push({ tableName, modelName });
+    while (i < content.length && depth > 0) {
+      if (content[i] === '{') depth++;
+      else if (content[i] === '}') depth--;
+      if (depth > 0) i++;
+    }
+
+    const body = content.substring(startIdx, i);
+    blocks.push({ name, body });
   }
 
-  return tables;
+  return blocks;
 }
 
 function toSnakeCase(str) {