ai-first-cli 1.1.2 → 1.1.5

package/src/core/semanticContexts.ts

@@ -28,7 +28,11 @@ export interface SemanticContexts {
 // ============================================================
 
 // 1. Candidate roots - scan inside these directories
-const CANDIDATE_ROOTS = ["src", "app", "packages", "services", "modules", "features"];
+const CANDIDATE_ROOTS = [
+  "src", "app", "packages", "services", "modules", "features",
+  // MVC patterns - for projects without src/ prefix
+  "controllers", "routes", "handlers", "views", "pages"
+];
 
 // 2. Ignore folders - these are technical, not business features
 const IGNORED_FOLDERS = new Set([
@@ -39,7 +43,6 @@ const IGNORED_FOLDERS = new Set([
   "constants",
   "config",
   "dto",
-  "models",
   "common",
   "shared"
 ]);
@@ -54,7 +57,11 @@ const ENTRYPOINT_PATTERNS = [
 ];
 
 // 4. Flow-specific patterns
-const FLOW_ENTRY_PATTERNS = ["controller", "route", "handler", "command"];
+const FLOW_ENTRY_PATTERNS = [
+  "controller", "route", "handler", "command",
+  // Frontend patterns
+  "page", "screen", "view", "component"
+];
 
 // 5. Flow exclusion patterns
 const FLOW_EXCLUDE = new Set([
@@ -83,7 +90,9 @@ const MAX_FLOW_FILES = 30;
 
 // Supported source file extensions
 const SOURCE_EXTENSIONS = new Set([
-  ".ts", ".tsx", ".js", ".jsx", ".py", ".java", ".kt", ".go", ".rs", ".rb", ".php", ".cs", ".vue", ".svelte"
+  ".ts", ".tsx", ".js", ".jsx", ".py", ".java", ".kt", ".go", ".rs", ".rb", ".php", ".cs", ".vue", ".svelte",
+  // Salesforce/Apex
+  ".cls", ".trigger", ".apex", ".object"
 ]);
 
 // ============================================================
@@ -96,13 +105,20 @@ function isSourceFile(filePath: string): boolean {
 }
 
 function isEntrypoint(filePath: string): boolean {
+  const lower = filePath.toLowerCase();
+  // Check both filename and directory path
   const basename = path.basename(filePath).toLowerCase();
-  return ENTRYPOINT_PATTERNS.some(pattern => basename.includes(pattern));
+  return ENTRYPOINT_PATTERNS.some(pattern => 
+    basename.includes(pattern) || lower.includes("/" + pattern + "s/")
+  );
 }
 
 function isFlowEntrypoint(filePath: string): boolean {
+  const lower = filePath.toLowerCase();
   const basename = path.basename(filePath).toLowerCase();
-  return FLOW_ENTRY_PATTERNS.some(pattern => basename.includes(pattern));
+  return FLOW_ENTRY_PATTERNS.some(pattern => 
+    basename.includes(pattern) || lower.includes("/" + pattern + "s/")
+  );
 }
 
 function isIgnoredFolder(folderName: string): boolean {
@@ -164,7 +180,20 @@ function findFeatureCandidates(files: string[]): Map<string, string[]> {
     
     if (rootIdx === -1) continue;
     
-    // Check depth 1: src/auth
+    // Check depth 0: controllers/ itself is a feature (when file is directly in root)
+    const depth0FeatureIdx = rootIdx;
+    if (depth0FeatureIdx < parts.length - 1) {
+      const featureName0 = parts[depth0FeatureIdx];
+      if (!isIgnoredFolder(featureName0) && !featureName0.includes(".")) {
+        const featurePath0 = parts.slice(0, depth0FeatureIdx + 1).join("/");
+        if (!candidates.has(featurePath0)) {
+          candidates.set(featurePath0, []);
+        }
+        candidates.get(featurePath0)!.push(file);
+      }
+    }
+    
+    // Check depth 1 and 2: src/auth, src/modules/auth
     for (let depth = 1; depth <= 2; depth++) {
       const featureIdx = rootIdx + depth;
       
@@ -249,36 +278,29 @@ export function generateFeatures(
     // Ignore errors, return empty
   }
 
-  // Get all files from modules
-  const allFiles = Object.values(modules).flatMap(m => m.files || []);
-  
-  // Find candidate features
-  const candidates = findFeatureCandidates(allFiles);
-
-  // Filter and validate candidates
-  for (const [featurePath, files] of candidates) {
+  // Use modules directly from modules.json
+  for (const [moduleName, mod] of Object.entries(modules)) {
+    const files = mod.files || [];
+    
     // Filter to source files only
     const sourceFiles = files.filter(isSourceFile);
     
-    // Must have at least 3 source files
-    if (sourceFiles.length < 3) continue;
+    // Must have at least 2 source files (relaxed from 3 for smaller projects)
+    if (sourceFiles.length < 2) continue;
     
     // Must have at least one entrypoint
     const entrypoints = sourceFiles.filter(isEntrypoint);
     if (entrypoints.length === 0) continue;
     
-    // Extract feature name from path
-    const featureName = featurePath.split("/").pop() || featurePath;
-    
-    // Get dependencies
-    const dependencies = getFeatureDependencies(featurePath, modules);
+    // Extract feature name from path - get the last segment
+    const featureName = mod.path.split("/").pop() || moduleName;
     
     features.push({
       name: featureName,
-      path: featurePath,
+      path: mod.path,
       files: sourceFiles.slice(0, 50),
       entrypoints: entrypoints.slice(0, 10),
-      dependencies
+      dependencies: []
     });
   }
 
@@ -372,9 +394,9 @@ function flowsFromFolders(files: string[]): Flow[] {
   for (const file of files) {
     if (!isSourceFile(file)) continue;
     
-    const baseName = path.basename(file).replace(/\.(ts|js|tsx|jsx)$/, "");
-    // Extract feature: authController -> auth, userService -> user
-    const feature = baseName.replace(/(Controller|Service|Repository|Handler|Route|Model|Entity)$/i, "");
+    const baseName = path.basename(file).replace(/\.[^.]+$/, "");
+    // Extract feature: authController -> auth, userService -> user, LoginPage -> login, Dashboard -> dashboard
+    const feature = baseName.replace(/(Controller|Service|Repository|Handler|Route|Model|Entity|Command|Page|Screen|View|Component)$/i, "");
     const key = feature.toLowerCase();
     
     if (!byFeature.has(key)) {
@@ -384,10 +406,13 @@ function flowsFromFolders(files: string[]): Flow[] {
   }
   
   for (const [feature, featureFiles] of byFeature) {
-    if (featureFiles.length < 3) continue;
+    // Relaxed: allow flows with 2+ files OR files that match entrypoint patterns
+    const hasEntrypoint = featureFiles.some(f => isFlowEntrypoint(f));
+    if (featureFiles.length < 2 && !hasEntrypoint) continue;
     
     const layers = new Set(featureFiles.map(getLayer).filter(l => l !== "unknown"));
-    if (layers.size < 2) continue;
+    // Relaxed: allow single layer if files match entrypoint patterns
+    if (layers.size < 2 && !hasEntrypoint) continue;
     
     const sorted = [...featureFiles].sort((a, b) => 
       getLayerPriority(a) - getLayerPriority(b)

package/src/core/symbolGraph.ts

@@ -145,7 +145,7 @@ function buildSymbolRelationships(
   // 1. Process imports relationships
   for (const dep of depsAnalysis.dependencies) {
     const sourceSymbols = symbols.filter(s => s.file === dep.source);
-    let targetSymbol = findTargetSymbol(dep.target, symbolMap);
+    let targetSymbol = findTargetSymbol(dep.target, symbolMap, symbols);
 
     if (targetSymbol && sourceSymbols.length > 0) {
       for (const source of sourceSymbols) {
@@ -185,7 +185,7 @@ function buildSymbolRelationships(
 /**
  * Find target symbol from import path
  */
-function findTargetSymbol(targetPath: string, symbolMap: Map<string, Symbol>): Symbol | undefined {
+function findTargetSymbol(targetPath: string, symbolMap: Map<string, Symbol>, allSymbols?: Symbol[]): Symbol | undefined {
   // Try exact match
   let target = symbolMap.get(targetPath);
   
@@ -208,6 +208,20 @@ function findTargetSymbol(targetPath: string, symbolMap: Map<string, Symbol>): S
     }
   }
 
+  // Try to find any symbol in the target file
+  if (!target && allSymbols) {
+    const targetFile = targetPath.replace(/\.[^.]+$/, '');
+    const fileSymbols = allSymbols.filter(s => {
+      const symFile = s.file.replace(/\.[^.]+$/, '');
+      return symFile === targetFile || symFile.endsWith('/' + targetFile) || symFile.endsWith('/' + targetPath.replace(/\.[^.]+$/, ''));
+    });
+    
+    if (fileSymbols.length > 0) {
+      // Return the first exported symbol, or just the first one
+      target = fileSymbols.find(s => s.export) || fileSymbols[0];
+    }
+  }
+
   return target;
 }
 

package/src/utils/fileUtils.ts

@@ -51,6 +51,11 @@ export const DEFAULT_INCLUDE_EXTENSIONS = [
   ".yaml",
   ".yml",
   ".md",
+  // Salesforce
+  ".cls",
+  ".trigger",
+  ".apex",
+  ".object",
 ];
 
 /**

package/test-projects/express-api/ai/ai_context.md

@@ -0,0 +1,112 @@
+# AI Context
+
+> This file provides a comprehensive overview of the repository for AI coding assistants.
+
+---
+
+## Quick Overview
+
+- **Pattern**: MVC (Model-View-Controller)
+- **Languages**: JavaScript, JSON
+- **Frameworks**: Express.js
+- **Total Files**: 8
+
+---
+
+## Table of Contents
+
+1. [Tech Stack](#tech-stack)
+2. [Architecture](#architecture)
+3. [Key Entrypoints](#key-entrypoints)
+4. [Code Conventions](#code-conventions)
+5. [AI Rules](#ai-rules)
+6. [Repository Map](#repository-map)
+
+---
+
+## Tech Stack
+
+**Languages**: JavaScript, JSON
+
+**Frameworks**: Express.js
+
+**Libraries**: Bcrypt, JWT
+
+**Package Managers**: npm
+
+---
+
+## Architecture
+
+## Architectural Pattern
+**Primary**: MVC (Model-View-Controller)
+**Secondary**: Layered Architecture, Microservices
+
+## Layers Identified
+- Application
+- Domain
+
+## Key Modules
+| Module | Responsibility |
+|--------|----------------|
+| `controllers` | API endpoints and request handling |
+| `index.js` | Contains 0 files |
+| `middleware` | Request/response middleware |
+| `models` | Data models and entities |
+| `package.json` | Contains 0 files |
+| `services` | Business logic and use cases |
+
+
+---
+
+## Key Entrypoints
+
+### Library
+- `index.js` - Main entry point
+
+### Server
+- `package.json#scripts.start` - start: node index.js
+- `package.json#scripts.dev` - dev: nodemon index.js
+
+---
+
+## Code Conventions
+
+## Naming Conventions
+- **Files**: camelCase
+
+## Project Structure
+
+## Code Style
+- **Semicolons**: Required
+
+
+---
+
+## AI Rules
+
+- Language: Use JavaScript, JSON
+- Frameworks: Express.js
+- Architecture: Follow MVC (Model-View-Controller) pattern
+- File naming: camelCase
+- Main entry: package.json#scripts.start
+
+---
+
+## Repository Map
+
+See repo_map.md for the full structure.
+
+---
+
+## Notes for AI Assistants
+
+1. Follow the established naming conventions (see conventions.md)
+2. Use the detected frameworks and libraries
+3. Target the correct entrypoints for modifications
+4. Maintain the detected architecture patterns
+5. Follow AI rules in ai_rules.md
+
+---
+
+*Generated by ai-first*

package/test-projects/express-api/ai/ai_rules.md

@@ -0,0 +1,50 @@
+# AI Rules
+
+> Guidelines for AI assistants working on this project
+
+---
+
+## Guidelines
+
+- Language: Use JavaScript, JSON
+- Frameworks: Express.js
+- Architecture: Follow MVC (Model-View-Controller) pattern
+- File naming: camelCase
+- Main entry: package.json#scripts.start
+- Dev command: npm run dev
+
+## Patterns
+
+- Layer structure: application → domain
+
+## Constraints
+
+- Always use semicolons
+
+---
+
+## Quick Reference
+
+| Category | Value |
+|----------|-------|
+| Language | JavaScript, JSON |
+| Framework | Express.js |
+| Architecture | MVC (Model-View-Controller) |
+| Naming | camelCase |
+| Indentation | unknown |
+
+---
+
+## Best Practices
+
+1. Follow the established naming conventions
+2. Keep functions small and focused
+3. Write tests for new features
+4. Use descriptive variable names
+5. Keep the architecture consistent
+6. Document complex logic
+7. Review code before committing
+
+---
+
+*Generated by ai-first*

package/test-projects/express-api/ai/architecture.md

@@ -0,0 +1,62 @@
+# Architecture
+
+## Architectural Pattern
+**Primary**: MVC (Model-View-Controller)
+**Secondary**: Layered Architecture, Microservices
+
+## Layers Identified
+- Application
+- Domain
+
+## Key Modules
+| Module | Responsibility |
+|--------|----------------|
+| `controllers` | API endpoints and request handling |
+| `index.js` | Contains 0 files |
+| `middleware` | Request/response middleware |
+| `models` | Data models and entities |
+| `package.json` | Contains 0 files |
+| `services` | Business logic and use cases |
+
+
+## Module Details
+
+### controllers
+- **Path**: `controllers`
+- **Responsibility**: API endpoints and request handling
+- **Depends on**: service, model
+
+### index.js
+- **Path**: `index.js`
+- **Responsibility**: Contains 0 files
+
+
+### middleware
+- **Path**: `middleware`
+- **Responsibility**: Request/response middleware
+
+
+### models
+- **Path**: `models`
+- **Responsibility**: Data models and entities
+
+
+### package.json
+- **Path**: `package.json`
+- **Responsibility**: Contains 0 files
+
+
+### services
+- **Path**: `services`
+- **Responsibility**: Business logic and use cases
+- **Depends on**: model, repository
+
+
+## Recommendations
+
+- Keep the architecture consistent as the project grows
+- Follow the established layer structure
+- Use dependency injection to manage module relationships
+
+---
+*Generated by ai-first*

package/test-projects/express-api/ai/cache.json

@@ -0,0 +1,125 @@
+{
+  "files": {
+    "ai/ai_context.md": {
+      "hash": "286f0083010c4a66f3906b020b9cc7b4",
+      "mtime": 1773796492348.4023
+    },
+    "ai/ai_rules.md": {
+      "hash": "fa1a632e71f5b7055dee7d1368600ce0",
+      "mtime": 1773796492348.4023
+    },
+    "ai/architecture.md": {
+      "hash": "5e9f157447aac07a89cf6fe7bffcdad4",
+      "mtime": 1773796492342.4023
+    },
+    "ai/context/features/controllers.json": {
+      "hash": "65a9f94a405e51bd65bd7a5378dbb164",
+      "mtime": 1773796492350.4023
+    },
+    "ai/context/features/services.json": {
+      "hash": "113525da53ad952071cca194ba355d48",
+      "mtime": 1773796492350.4023
+    },
+    "ai/context/flows/auth.json": {
+      "hash": "a6200259a521a38e08dc6396d870a633",
+      "mtime": 1773796492350.4023
+    },
+    "ai/context/flows/user.json": {
+      "hash": "af049ebfc3b8ce808e6091ac96a651ee",
+      "mtime": 1773796492350.4023
+    },
+    "ai/conventions.md": {
+      "hash": "fa8f117204dbd53d6b787e6ec1791dbf",
+      "mtime": 1773796492344.4023
+    },
+    "ai/dependencies.json": {
+      "hash": "b59a369568a90136bf6d54a022aae411",
+      "mtime": 1773796492346.4023
+    },
+    "ai/entrypoints.md": {
+      "hash": "17b75097fff8baf7e8218553431dad64",
+      "mtime": 1773796492343.4023
+    },
+    "ai/files.json": {
+      "hash": "4041be286c1f3bef227566eb3131babc",
+      "mtime": 1773796517397.7932
+    },
+    "ai/graph/module-graph.json": {
+      "hash": "43383456483e635b66ff8960f3b39f03",
+      "mtime": 1773796517391.793
+    },
+    "ai/modules.json": {
+      "hash": "52396bb2eb3346493e373bf55805ef90",
+      "mtime": 1773796517382.793
+    },
+    "ai/project.json": {
+      "hash": "08ba7541bf16fe9e746aa0ed3ef4a7b3",
+      "mtime": 1773796492350.4023
+    },
+    "ai/repo-map.json": {
+      "hash": "bc79826d46b669d42b546057efb73255",
+      "mtime": 1773796517383.793
+    },
+    "ai/repo_map.json": {
+      "hash": "474f2af3cc084780fcdc989f865cb19d",
+      "mtime": 1773796492340.4023
+    },
+    "ai/repo_map.md": {
+      "hash": "35963e46f602438bc7b1318c630c9616",
+      "mtime": 1773796492340.4023
+    },
+    "ai/schema.json": {
+      "hash": "09deb83ac436baaa3ee8ba47c1b6876e",
+      "mtime": 1773796492350.4023
+    },
+    "ai/summary.md": {
+      "hash": "792c7fa8536cdf57b7506a5b4390e1fd",
+      "mtime": 1773796492341.4023
+    },
+    "ai/symbols.json": {
+      "hash": "55f09cedb48590dcdc4c0a4506026d52",
+      "mtime": 1773796492346.4023
+    },
+    "ai/tech_stack.md": {
+      "hash": "bce98c307ddce9864d341031c914ba8a",
+      "mtime": 1773796492343.4023
+    },
+    "ai/tools.json": {
+      "hash": "b1356e52c74e9b3b6a0d397b790a4968",
+      "mtime": 1773796492350.4023
+    },
+    "controllers/authController.js": {
+      "hash": "7354721b063c069c013d70e04f4bee0f",
+      "mtime": 1773776832721.0393
+    },
+    "controllers/userController.js": {
+      "hash": "e43bb45b1c75b7e4bcd072c60a70d848",
+      "mtime": 1773776845003.1538
+    },
+    "index.js": {
+      "hash": "b866780bf160f2bfb2ac6baba6a1f6b1",
+      "mtime": 1773776791058.662
+    },
+    "middleware/authMiddleware.js": {
+      "hash": "bd1176bd90f83f93b3f66d07758267e5",
+      "mtime": 1773776926397.9497
+    },
+    "models/userRepository.js": {
+      "hash": "fdf8dfbb70b15c82f0d2b13e51a640f6",
+      "mtime": 1773776885672.544
+    },
+    "package.json": {
+      "hash": "c64180130f77b5472a2850aafa61c7e9",
+      "mtime": 1773776784437.6038
+    },
+    "services/authService.js": {
+      "hash": "cacf5bd493a669c8ef7b95bef67cf7e5",
+      "mtime": 1773776853984.2388
+    },
+    "services/userService.js": {
+      "hash": "cb3b6f655977d154edee4038d6105d10",
+      "mtime": 1773776865182.3455
+    }
+  },
+  "lastIndexed": "2026-03-18T01:15:17.399Z"
+}

package/test-projects/express-api/ai/context/features/controllers.json

@@ -0,0 +1,13 @@
+{
+  "name": "controllers",
+  "path": "controllers",
+  "files": [
+    "controllers/authController.js",
+    "controllers/userController.js"
+  ],
+  "entrypoints": [
+    "controllers/authController.js",
+    "controllers/userController.js"
+  ],
+  "dependencies": []
+}

package/test-projects/express-api/ai/context/features/services.json

@@ -0,0 +1,13 @@
+{
+  "name": "services",
+  "path": "services",
+  "files": [
+    "services/authService.js",
+    "services/userService.js"
+  ],
+  "entrypoints": [
+    "services/authService.js",
+    "services/userService.js"
+  ],
+  "dependencies": []
+}

package/test-projects/express-api/ai/context/flows/auth.json

@@ -0,0 +1,12 @@
+{
+  "name": "auth",
+  "entrypoint": "controllers/authController.js",
+  "files": [
+    "controllers/authController.js",
+    "services/authService.js"
+  ],
+  "depth": 1,
+  "layers": [
+    "service"
+  ]
+}

package/test-projects/express-api/ai/context/flows/user.json

@@ -0,0 +1,13 @@
+{
+  "name": "user",
+  "entrypoint": "controllers/userController.js",
+  "files": [
+    "controllers/userController.js",
+    "models/userRepository.js",
+    "services/userService.js"
+  ],
+  "depth": 1,
+  "layers": [
+    "service"
+  ]
+}

package/test-projects/express-api/ai/conventions.md

@@ -0,0 +1,51 @@
+# Code Conventions
+
+## Naming Conventions
+- **Files**: camelCase
+
+## Project Structure
+
+## Code Style
+- **Semicolons**: Required
+
+
+## Naming Conventions
+
+| Element | Convention |
+|---------|------------|
+| Files | camelCase |
+| Components | unknown |
+| Test Files | unknown |
+| Directories | unknown |
+
+## Project Structure
+
+| Directory | Location |
+|-----------|----------|
+| Source | unknown |
+| Tests | unknown |
+| Config | unknown |
+
+## Code Style
+
+| Style Element | Value |
+|---------------|-------|
+| Indentation | unknown |
+| Semicolons | Required |
+
+## Testing
+
+| Aspect | Value |
+|--------|-------|
+| Framework | unknown |
+| Test Pattern | unknown |
+
+## Git
+
+| Aspect | Value |
+|--------|-------|
+| Branch Strategy | Trunk-based (main/master) |
+| Commit Format | unknown |
+
+---
+*Generated by ai-first*