c4-gen 1.5.2 → 1.6.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "c4-gen",
-  "version": "1.5.2",
+  "version": "1.6.0",
   "description": "Công cụ tự động sinh tài liệu kiến trúc C4 Model từ source code sử dụng AI",
   "main": "index.js",
   "type": "module",

package/src/generator.js

@@ -122,10 +122,11 @@ export async function generateC4Documentation({ sourcePath, outputPath, apiKey,
 }
 
 async function generateContextDiagram(model, codeStructure) {
-  // Chia nhỏ data để tránh vượt quá context limit
-  const filesList = codeStructure.files.slice(0, 30).map(f => f.path).join('\n');
+  // Tăng context để phân tích chi tiết hơn
+  const filesList = codeStructure.files.slice(0, 50).map(f => f.path).join('\n'); // Tăng từ 30 lên 50
   const structureSummary = summarizeStructure(codeStructure.structure);
-  const sampleCode = truncateText(codeStructure.sampleCode, 3000);
+  const sampleCode = truncateText(codeStructure.sampleCode, 5000); // Tăng từ 3000 lên 5000
+  const filesByFolder = groupFilesByFolder(codeStructure.files);
   
   const prompt = `
 Phân tích source code sau và tạo C4 LEVEL 1: CONTEXT DIAGRAM (System Overview):
@@ -134,7 +135,10 @@ Phân tích source code sau và tạo C4 LEVEL 1: CONTEXT DIAGRAM (System Overvi
 - Total Files: ${codeStructure.files.length}
 - Project Structure: ${structureSummary}
 
-📁 KEY FILES (30 first files):
+📁 FILES BY FOLDER:
+${filesByFolder}
+
+📁 KEY FILES (50 first files):
 ${filesList}
 
 💻 SAMPLE CODE:
@@ -144,89 +148,127 @@ ${sampleCode}
 ${formatDependencies(codeStructure.dependencies)}
 
 ---
-🎯 HƯỚNG DẪN PHÂN TÍCH:
-
-QUAN TRỌNG: Chỉ phân tích dựa trên CODE THỰC TẾ được cung cấp, KHÔNG tự nghĩ ra hoặc dùng ví dụ.
-
-1. BUSINESS CONTEXT: 
-   - Đọc package.json name/description
-   - Phân tích folder structure (ecommerce/cms/crm/pos/blog...)
-   - Xem code để hiểu domain nghiệp vụ
-
-2. STAKEHOLDERS: 
-   - Tìm trong code: roles, permissions, user types
-   - Xem routes/controllers để hiểu ai dùng hệ thống
-   
-3. EXTERNAL SYSTEMS:
-   - Phân tích DEPENDENCIES để tìm third-party services
-   - Tìm API keys, SDK imports (stripe, firebase, aws-sdk, nodemailer...)
-   - Xem imports để phát hiện integrations
+🎯 HƯỚNG DẪN PHÂN TÍCH CHI TIẾT:
+
+QUAN TRỌNG: Phải phân tích TOÀN DIỆN và CHI TIẾT, KHÔNG tự nghĩ ra hoặc dùng ví dụ.
+
+1. BUSINESS CONTEXT - PHÂN TÍCH SÂU:
+   - Đọc package.json name/description để hiểu domain
+   - Phân tích folder structure: pos/ → Point of Sale, ecommerce/ → E-commerce, etc.
+   - Scan routes/controllers để hiểu business features
+   - Identify domain entities từ models/schemas
+   - Phát hiện business flows từ service layer
+
+2. STAKEHOLDERS - TÌM TẤT CẢ:
+   - Scan ALL files chứa "role", "permission", "user", "admin"
+   - Đọc authentication/authorization code
+   - Tìm trong migrations/seeds cho user roles
+   - Check middleware cho role-based access
+   - List ĐẦY ĐỦ mọi loại người dùng hệ thống
    
-4. COMMUNICATION PROTOCOLS:
-   - Xem có express/fastify → HTTP/REST
-   - Có socket.io → WebSocket
-   - Có amqplib → AMQP
-   - Có grpc → gRPC
-
-5. TECHNOLOGIES:
-   - Frontend: Tìm react, vue, angular, next.js trong dependencies
-   - Backend: Xem runtime (node.js, python, java từ files)
-   - Database: Tìm pg, mysql2, mongodb, sequelize, prisma...
-   - Infrastructure: Tìm Dockerfile, docker-compose.yml, kubernetes yaml
-
-Hãy trả về JSON với format:
+3. EXTERNAL SYSTEMS - LIỆT KÊ TẤT CẢ:
+   - Scan TOÀN BỘ dependencies cho third-party services:
+     * Payment: stripe, paypal, vnpay, momo
+     * Email: nodemailer, sendgrid, mailgun
+     * SMS: twilio, nexmo
+     * Storage: aws-s3, cloudinary, firebase-storage
+     * Auth: passport, jwt, oauth
+     * Analytics: google-analytics, mixpanel
+     * Monitoring: sentry, newrelic
+     * Search: elasticsearch, algolia
+     * Cache: redis, memcached
+   - Tìm API integrations trong code (axios calls, fetch to external URLs)
+   - Check config files cho external service endpoints
+   - PHẢI liệt kê TẤT CẢ integrations tìm được
+
+4. USERS - PHÂN LOẠI CHI TIẾT:
+   - Scan routes để identify user journeys:
+     * /api/admin/* → Admin users
+     * /api/customer/* → Customers
+     * /api/vendor/* → Vendors
+     * /api/staff/* → Staff members
+   - Phân tích permissions/roles cho từng loại user
+   - List ĐẦY ĐỦ interactions của từng user type
+   - Identify access channels (Web App, Mobile App, API)
+
+5. COMMUNICATION PROTOCOLS:
+   - HTTP/REST: express, fastify, koa
+   - WebSocket: socket.io, ws
+   - GraphQL: apollo, graphql
+   - gRPC: @grpc/grpc-js
+   - Message Queue: amqplib (RabbitMQ), kafkajs (Kafka)
+   - Database protocols: postgres, mysql, mongodb
+
+6. TECHNOLOGIES - LIỆT KÊ ĐẦY ĐỦ:
+   - Frontend: React, Vue, Angular, Next.js, Nuxt, Svelte...
+   - Backend: Node.js, Python, Java, Go, .NET...
+   - Database: PostgreSQL, MySQL, MongoDB, Redis, Elasticsearch...
+   - Infrastructure: Docker, Kubernetes, AWS, GCP, Azure...
+
+Hãy trả về JSON với format CHI TIẾT:
 {
-  "systemName": "Tên từ package.json hoặc suy ra từ folder structure",
-  "description": "Mô tả dựa trên phân tích code và structure",
-  "businessContext": "Domain nghiệp vụ phát hiện từ code",
-  "purpose": "Mục đích chính suy ra từ features",
+  "systemName": "Tên CHÍNH XÁC từ package.json",
+  "description": "Mô tả ĐẦY ĐỦ dựa trên phân tích code (3-5 câu)",
+  "businessContext": "Domain nghiệp vụ CHI TIẾT (VD: Restaurant POS system with inventory management, order processing, and payment integration)",
+  "purpose": "Mục đích CỤTH (liệt kê 3-5 mục đích chính)",
   "stakeholders": [
     {
-      "role": "Vai trò TÌM ĐƯỢC trong code (Admin/User/...)",
-      "description": "Mô tả dựa trên permissions/roles trong code",
-      "concerns": "Concerns suy ra từ features họ dùng"
+      "role": "Vai trò CHÍNH XÁC từ code",
+      "description": "Mô tả CHI TIẾT responsibilities",
+      "concerns": "Concerns CỤ THỂ (security, performance, usability...)"
     }
   ],
   "users": [
     {
-      "name": "User type TÌM ĐƯỢC trong code",
-      "type": "Internal/External (dựa vào phân tích)",
-      "description": "Mô tả dựa trên routes/controllers",
+      "name": "User type TỪ CODE (Admin, Customer, Vendor, Staff...)",
+      "type": "Internal/External",
+      "description": "Mô tả ĐẦY ĐỦ role và responsibilities",
       "interactions": [
-        "Interactions TÌM ĐƯỢC từ API endpoints",
-        "VÍ DỤ: Nếu có /api/products → 'Quản lý sản phẩm'"
+        "LIST TẤT CẢ interactions từ routes (Manage products, Process orders, View reports...)"
       ],
-      "access": "Phát hiện từ code (Web nếu có React, Mobile nếu có React Native...)"
+      "access": "Access method CHI TIẾT (Web Dashboard, Mobile App, REST API...)"
     }
   ],
   "externalSystems": [
     {
-      "name": "Tên service TÌM ĐƯỢC trong dependencies (stripe, firebase, aws-sdk...)",
-      "category": "Category dựa vào tên package (stripe→Payment, nodemailer→Email...)",
-      "description": "Mô tả service",
-      "purpose": "Purpose suy từ cách dùng",
-      "protocol": "Protocol phát hiện từ code",
-      "direction": "Inbound/Outbound (phân tích từ code flow)",
-      "dataFlow": "Data type từ code"
+      "name": "Service name CHÍNH XÁC từ dependencies",
+      "category": "Category CỤ THỂ (Payment Gateway, Email Service, Cloud Storage...)",
+      "description": "Mô tả CHI TIẾT service và version",
+      "purpose": "Purpose CỤ THỂ (Process payments, Send notifications...)",
+      "protocol": "Protocol (HTTPS/REST, SMTP, S3...)",
+      "direction": "Inbound/Outbound/Bidirectional",
+      "dataFlow": "Data type CỤ THỂ (Payment data, Email content, File uploads...)"
     }
   ],
-  "architectureStyle": "Style phát hiện từ structure (1 app→Monolithic, nhiều services→Microservices)",
-  "deploymentModel": "Dựa vào Dockerfile/cloud config",
+  "architectureStyle": "Architecture CHI TIẾT (Monolithic with modular structure, Microservices...)",
+  "deploymentModel": "Deployment CỤ THỂ (Docker containers, Kubernetes, Serverless...)",
   "technologies": {
-    "frontend": ["Technologies TÌM ĐƯỢC trong package.json devDependencies"],
-    "backend": ["Technologies TÌM ĐƯỢC trong dependencies + file extensions"],
-    "database": ["Databases TÌM ĐƯỢC từ imports/dependencies"],
-    "infrastructure": ["Infrastructure TÌM ĐƯỢC từ Dockerfile, k8s files..."]
+    "frontend": ["TẤT CẢ frontend tech + versions"],
+    "backend": ["TẤT CẢ backend tech + versions"],
+    "database": ["TẤT CẢ databases + versions"],
+    "infrastructure": ["TẤT CẢ infrastructure tools"],
+    "devOps": ["CI/CD tools, monitoring..."],
+    "security": ["Auth methods, encryption..."]
   },
   "keyPrinciples": [
-    "Principles SUY RA từ code structure và patterns"
+    "LIST principles từ code (RESTful API, Microservices, DDD, CQRS...)"
+  ],
+  "businessFlows": [
+    "KEY business flows từ code (Order Processing, Inventory Management...)"
   ]
 }
 
+YÊU CẦU ĐẦY ĐỦ:
+✅ Stakeholders: PHẢI có ít nhất 3-5 stakeholder types
+✅ Users: PHẢI liệt kê TẤT CẢ user types tìm được
+✅ External Systems: PHẢI liệt kê TẤT CẢ dependencies/integrations
+✅ Technologies: PHẢI đầy đủ tất cả tech stack
+✅ Business Flows: PHẢI liệt kê các flows chính
+
 LƯU Ý: 
-- Nếu KHÔNG TÌM THẤY thông tin → để mảng rỗng [] hoặc "Not detected"
-- KHÔNG tự nghĩ ra data không có trong code
-- Nếu không chắc → ghi "Detected from code analysis" thay vì bịa
+- Nếu KHÔNG TÌM THẤY → để mảng rỗng [], KHÔNG bịa
+- PHẢI CHI TIẾT và ĐẦY ĐỦ
+- Ưu tiên QUALITY over QUANTITY nhưng KHÔNG bỏ sót
 `;
 
   checkPromptSize(prompt);
@@ -240,8 +282,9 @@ LƯU Ý:
 async function generateContainerDiagram(model, codeStructure) {
   // Tăng context để AI hiểu rõ hơn
   const filesByType = groupFilesByType(codeStructure.files);
+  const filesByFolder = groupFilesByFolder(codeStructure.files);
   const structureSummary = summarizeStructure(codeStructure.structure);
-  const sampleCode = truncateText(codeStructure.sampleCode, 5000); // Tăng từ 2000 lên 5000
+  const sampleCode = truncateText(codeStructure.sampleCode, 8000); // Tăng từ 5000 lên 8000
   
   const prompt = `
 Phân tích source code và tạo C4 LEVEL 2: CONTAINER DIAGRAM (System Components):
@@ -252,112 +295,165 @@ ${structureSummary}
 📁 FILES BY TYPE:
 ${filesByType}
 
-📦 DEPENDENCIES:
+� FILES BY FOLDER:
+${filesByFolder}
+
+�📦 DEPENDENCIES:
 ${formatDependencies(codeStructure.dependencies)}
 
 💻 SAMPLE CODE:
 ${sampleCode}
 
 ---
-🎯 HƯỚNG DẪN PHÂN TÍCH:
-
-QUAN TRỌNG: Chỉ dựa trên CODE và DEPENDENCIES thực tế, KHÔNG dùng ví dụ mẫu.
-
-1. IDENTIFY CONTAINERS:
-   - Frontend: Tìm React/Vue/Angular trong dependencies → Web Application container
-   - Backend: Tìm Express/Fastify/NestJS → API Server container
-   - Database: Tìm pg/mysql2/mongodb → Database container
-   - Cache: Tìm redis/ioredis → Cache container
-   - Message Queue: Tìm amqplib/kafkajs → MQ container
-   - Search: Tìm @elastic/elasticsearch → Search Engine container
+🎯 HƯỚNG DẪN PHÂN TÍCH CHI TIẾT:
+
+QUAN TRỌNG: Chỉ dựa trên CODE và DEPENDENCIES thực tế, PHẢI CHI TIẾT và ĐẦY ĐỦ.
+
+1. IDENTIFY ALL CONTAINERS - LIỆT KÊ TẤT CẢ:
+   - Frontend Containers:
+     * React/Vue/Angular → Web Application
+     * React Native → Mobile App
+     * Next.js → SSR Application
+   - Backend Containers:
+     * Express/Fastify/NestJS → API Server
+     * GraphQL Server → GraphQL API
+     * WebSocket Server → Realtime Service
+   - Data Containers:
+     * PostgreSQL/MySQL → Primary Database
+     * MongoDB → Document Store
+     * Redis → Cache Layer
+     * Elasticsearch → Search Engine
+   - Infrastructure Containers:
+     * Nginx → Reverse Proxy
+     * RabbitMQ/Kafka → Message Broker
+     * Worker processes → Background Jobs
+   
+2. TECHNOLOGY STACK - CHÍNH XÁC:
+   - Đọc package.json cho EXACT versions
+   - VD: "express": "^4.18.2" → "Express.js 4.18.2"
+   - List ALL major dependencies với versions
    
-2. TECHNOLOGY & VERSION:
-   - Đọc package.json để lấy version CHÍNH XÁC
-   - VÍ DỤ: "react": "^18.2.0" → ghi "React 18.2"
+3. CONTAINER SPECIFICATIONS:
+   - Ports: Scan code cho app.listen(), PORT, env variables
+   - Environment: Development, Staging, Production configs
+   - Scaling: Replicas từ docker-compose hoặc k8s
+   - Resources: Memory/CPU limits từ deployment configs
+   - Health checks: Tìm /health, /ping endpoints
    
-3. CONTAINER SPECS:
-   - Ports: Tìm trong code (app.listen(3000), PORT env...)
-   - Replicas: Mặc định 1, trừ khi thấy docker-compose scale
-   - Resources: Đọc từ Dockerfile/k8s manifests nếu có
+4. COMMUNICATION PATTERNS:
+   - Synchronous:
+     * HTTP/REST → express routes
+     * GraphQL → apollo-server
+     * gRPC → @grpc/grpc-js
+   - Asynchronous:
+     * Message Queue → amqplib, kafkajs
+     * WebSocket → socket.io, ws
+     * Event Bus → eventemitter2
+   - Caching:
+     * Redis → ioredis
+     * In-memory → node-cache
    
-4. PROTOCOLS:
-   - Có express → HTTP/REST
-   - Có socket.io → WebSocket
-   - Có import amqp → AMQP
-   - Có @grpc → gRPC
-
-5. DATA STORES:
-   - Sequelize/TypeORM/Prisma → Relational DB
-   - Mongoose → MongoDB (NoSQL)
-   - ioredis → Redis (Cache)
-   - @elastic → Elasticsearch (Search)
-
-Hãy trả về JSON với format:
+5. CONTAINER INTERACTIONS - CHI TIẾT:
+   - API Server ↔ Database: Query patterns, connection pooling
+   - API Server ↔ Cache: Cache strategies (read-through, write-through)
+   - API Server ↔ Message Queue: Pub/Sub patterns
+   - Frontend ↔ API Server: REST endpoints, WebSocket connections
+   - Workers ↔ Queue: Job processing patterns
+
+6. DATA STORES - LIỆT KÊ TẤT CẢ:
+   - Primary DB: Main data storage với schema
+   - Cache: Redis cho session, caching
+   - Search: Elasticsearch cho full-text search
+   - File Storage: S3, Cloudinary cho uploads
+   - Analytics: Time-series DB nếu có
+
+Hãy trả về JSON CHI TIẾT:
 {
-  "architectureStyle": "Phát hiện từ structure (1 app→Monolithic, nhiều services→Microservices)",
+  "architectureStyle": "Architecture CỤ THỂ (Monolithic, Microservices, Modular Monolith...)",
   "containers": [
     {
-      "name": "Tên TỰ ĐẶT dựa vào technology (React App, Node API, PostgreSQL DB...)",
-      "type": "Type dựa vào phát hiện (Web Application/API Server/Database/Cache...)",
-      "technology": "Technology + version TÌM ĐƯỢC (Express.js 4.18, PostgreSQL 14...)",
-      "description": "Mô tả dựa trên responsibilities",
+      "name": "Tên CHÍNH XÁC (API Server, Web Dashboard, Mobile App, PostgreSQL DB...)",
+      "type": "Type CỤ THỂ (Web Application/API Server/Database/Cache/Message Queue...)",
+      "technology": "Tech + version CHÍNH XÁC (Express.js 4.18.2, React 18.2.0, PostgreSQL 14.5...)",
+      "description": "Mô tả ĐẦY ĐỦ responsibilities và purpose",
       "responsibilities": [
-        "Responsibility SUY RA từ code (Nếu Frontend→UI rendering, routing...)",
-        "Nếu Backend→Business logic, API endpoints..."
+        "LIST TẤT CẢ responsibilities từ code (Handle API requests, Process payments, Manage inventory...)"
       ],
-      "port": "Port TÌM ĐƯỢC từ code hoặc 'Not specified'",
-      "replicas": "1 hoặc số TÌM ĐƯỢC trong docker-compose",
+      "port": "Port CHÍNH XÁC hoặc 'Not specified'",
+      "replicas": "Số instances (1, 3, auto-scale...)",
       "resources": {
-        "memory": "TÌM từ Dockerfile/k8s hoặc 'Not specified'",
-        "cpu": "TÌM từ Dockerfile/k8s hoặc 'Not specified'"
+        "memory": "Memory limit (512MB, 1GB, 2GB...)",
+        "cpu": "CPU limit (0.5, 1.0, 2.0...)"
       },
-      "storage": "TÌM từ volumes hoặc 'Not applicable'",
-      "protocols": ["Protocols PHÁT HIỆN từ code"],
-      "dependencies": ["Dependencies THỰC TẾ từ imports"],
+      "storage": "Storage type và size",
+      "protocols": ["TẤT CẢ protocols (HTTP/1.1, HTTP/2, WebSocket, gRPC...)"],
+      "dependencies": ["TẤT CẢ runtime dependencies"],
       "interactions": [
         {
-          "to": "Container đích THỰC SỰ connect tới",
-          "description": "Mô tả TỪ code",
-          "protocol": "Protocol PHÁT HIỆN",
-          "direction": "Direction dựa vào code flow",
-          "dataExchanged": "Data type TỪ code",
-          "isAsync": "true nếu có await/Promise, false nếu sync"
+          "to": "Target container CHÍNH XÁC",
+          "description": "Mô tả CHI TIẾT interaction",
+          "protocol": "Protocol CỤ THỂ",
+          "direction": "Inbound/Outbound/Bidirectional",
+          "dataExchanged": "Data type CHI TIẾT",
+          "isAsync": "true/false",
+          "frequency": "Request frequency (high/medium/low)"
         }
       ],
       "endpoints": [
-        "API endpoints TÌM ĐƯỢC từ routes/controllers"
-      ]
+        "TẤT CẢ endpoints/routes phục vụ"
+      ],
+      "healthCheck": "Health check endpoint",
+      "logging": "Logging strategy",
+      "monitoring": "Monitoring tools"
     }
   ],
   "dataStores": [
     {
-      "name": "Tên DB TỪ dependencies",
-      "type": "Type dựa vào package (pg→Relational, mongodb→NoSQL, redis→Cache)",
-      "technology": "Technology + version từ package.json",
-      "purpose": "Purpose suy từ code usage",
-      "dataModel": "Model TỪ code (Multi-tenant nếu thấy schema separation...)",
-      "port": "Port TỪ config hoặc default"
+      "name": "Store name CHÍNH XÁC",
+      "type": "Type CỤ THỂ (Relational/NoSQL/Cache/Search/File Storage...)",
+      "technology": "Tech + version CHÍNH XÁC",
+      "purpose": "Purpose CHI TIẾT",
+      "dataModel": "Data model description",
+      "port": "Port hoặc connection string",
+      "replication": "Replication strategy",
+      "backup": "Backup strategy"
     }
   ],
   "messagingLayer": {
-    "technology": "Technology TÌM ĐƯỢC (RabbitMQ từ amqplib, Kafka từ kafkajs...)",
-    "queues": ["Queue names TÌM ĐƯỢC trong code"],
-    "purpose": "Purpose SUY từ usage"
+    "technology": "Technology CHÍNH XÁC (RabbitMQ 3.11, Kafka 3.3...)",
+    "queues": ["TẤT CẢ queue names"],
+    "exchanges": ["Exchange names nếu có"],
+    "purpose": "Purpose CHI TIẾT",
+    "patterns": ["Pub/Sub, Work Queue, RPC..."]
   },
   "infrastructureComponents": [
     {
-      "name": "Component TÌM ĐƯỢC (Nginx từ nginx.conf, Docker từ Dockerfile...)",
-      "technology": "Technology từ config files",
-      "purpose": "Purpose từ config"
+      "name": "Component name (Nginx, API Gateway, Load Balancer...)",
+      "technology": "Technology + version",
+      "purpose": "Purpose CHI TIẾT",
+      "configuration": "Key configurations"
     }
-  ]
+  ],
+  "networkingAndSecurity": {
+    "loadBalancing": "Load balancing strategy",
+    "ssl": "SSL/TLS configuration",
+    "authentication": "Auth methods",
+    "rateLimit": "Rate limiting strategy"
+  }
 }
 
+YÊU CẦU ĐẦY ĐỦ:
+✅ Containers: PHẢI liệt kê TẤT CẢ containers (API, DB, Cache, Queue...)
+✅ Technology: PHẢI có version CHÍNH XÁC từ package.json
+✅ Interactions: PHẢI mô tả CHI TIẾT mọi connection
+✅ Endpoints: PHẢI list routes/APIs của container
+✅ Data Stores: PHẢI liệt kê TẤT CẢ storage systems
+
 LƯU Ý:
-- Nếu KHÔNG có Frontend dependencies → KHÔNG tạo Frontend container
-- Nếu KHÔNG có Database package → KHÔNG tạo DB container
-- Chỉ tạo containers CHO CÁI CÓ TRONG CODE
-- Nếu không tìm thấy → để "Not detected" hoặc mảng rỗng []
+- Nếu KHÔNG có Frontend → không tạo Frontend container
+- Nếu KHÔNG có Queue → messagingLayer null
+- CHỈ những gì CÓ TRONG CODE
+- PHẢI CHI TIẾT và ĐẦY ĐỦ
 `;
 
   checkPromptSize(prompt);
@@ -371,8 +467,9 @@ LƯU Ý:
 async function generateComponentDiagram(model, codeStructure) {
   // Tăng context để đọc nhiều files hơn
   const filesByFolder = groupFilesByFolder(codeStructure.files);
+  const filesByType = groupFilesByType(codeStructure.files);
   const structureSummary = summarizeStructure(codeStructure.structure);
-  const sampleCode = truncateText(codeStructure.sampleCode, 8000); // Tăng từ 2500 lên 8000
+  const sampleCode = truncateText(codeStructure.sampleCode, 10000); // Tăng từ 8000 lên 10000
   
   const prompt = `
 Phân tích source code và tạo C4 LEVEL 3: COMPONENT DIAGRAM (Internal Structure):
@@ -383,119 +480,186 @@ ${structureSummary}
 📁 FILES BY FOLDER (${codeStructure.files.length} total):
 ${filesByFolder}
 
-📦 DEPENDENCIES:
+� FILES BY TYPE:
+${filesByType}
+
+�📦 DEPENDENCIES:
 ${formatDependencies(codeStructure.dependencies)}
 
 💻 SAMPLE CODE:
 ${sampleCode}
 
 ---
-🎯 HƯỚNG DẪN PHÂN TÍCH:
-
-QUAN TRỌNG: Chỉ liệt kê components TÌM ĐƯỢC trong code, KHÔNG tự nghĩ ra.
-
-1. SCAN FILES ĐỂ TÌM COMPONENTS:
-   - Controllers: Tìm files chứa "Controller", "controller", "*Controller.js", "routes/*.js"
-   - Services: Tìm files chứa "Service", "service", "*Service.js"
-   - Repositories: Tìm files chứa "Repository", "Repo", "DAO", "models/*.js"
-   - Middleware: Tìm files trong "middleware/", "middlewares/"
-   - Utils: Tìm files trong "utils/", "helpers/", "lib/"
+🎯 HƯỚNG DẪN PHÂN TÍCH CHI TIẾT:
+
+QUAN TRỌNG: Chỉ liệt kê components TÌM ĐƯỢC trong code, PHẢI ĐẦY ĐỦ và CHI TIẾT.
+
+1. SCAN ALL FILES ĐỂ TÌM TẤT CẢ COMPONENTS:
+   - Controllers: 
+     * Scan ALL files matching: *Controller.js, *controller.ts, routes/*.js
+     * Extract class names và route handlers
+   - Services:
+     * Scan ALL files matching: *Service.js, *service.ts, services/*.js
+     * Identify business logic components
+   - Repositories/DAOs:
+     * Scan ALL files matching: *Repository.js, *Repo.js, *DAO.js, models/*.js
+     * Map database access layers
+   - Middleware:
+     * List ALL files in middleware/, middlewares/
+     * Identify auth, validation, logging middleware
+   - Utilities:
+     * List ALL files in utils/, helpers/, lib/
+     * Categorize by function
    
-2. PHÁT HIỆN ARCHITECTURE LAYERS:
-   - Presentation: Nếu có controllers/, routes/, views/
-   - Business: Nếu có services/, business/, domain/
-   - Data: Nếu có repositories/, models/, data-access/
-   - Infrastructure: Nếu có config/, infrastructure/, lib/
-
-3. API ENDPOINTS - SCAN ROUTES:
-   - Đọc files trong routes/ hoặc controllers/
-   - Tìm app.get(), router.post(), @Get(), @Post()...
-   - Lấy path VÀ method CHÍNH XÁC từ code
+2. IDENTIFY ARCHITECTURE LAYERS - CHI TIẾT:
+   - Presentation Layer:
+     * Routes, Controllers, API handlers
+     * Input validation, request mapping
+   - Business Logic Layer:
+     * Services, Use cases, Domain logic
+     * Business rules, calculations
+   - Data Access Layer:
+     * Repositories, ORMs, Query builders
+     * Database operations, caching
+   - Infrastructure Layer:
+     * Config, External integrations
+     * Logging, monitoring, error handling
+   - Cross-Cutting Concerns:
+     * Authentication, Authorization
+     * Logging, Caching, Error handling
+
+3. API ENDPOINTS - SCAN TẤT CẢ ROUTES:
+   - Đọc TOÀN BỘ route files
+   - Extract:
+     * Method: GET, POST, PUT, DELETE, PATCH
+     * Path: /api/users, /api/products/:id
+     * Handler: Controller method name
+     * Middleware: Auth, validation middleware
+   - PHẢI liệt kê ĐẦY ĐỦ TẤT CẢ endpoints
    
-4. COMPONENT DEPENDENCIES:
-   - Xem imports/require trong từng file
-   - Nếu ProductController import ProductService → dependency
+4. COMPONENT DEPENDENCIES - MAP TOÀN BỘ:
+   - Scan imports/require trong MỌI file
+   - Build dependency graph:
+     * Controller → Service
+     * Service → Repository
+     * Repository → Database
+   - Identify circular dependencies
    
-5. DESIGN PATTERNS:
-   - Có Controller + Service → MVC pattern
-   - Có Repository → Repository pattern
-   - Có inject/DI container → Dependency Injection
-
-Hãy trả về JSON:
+5. DESIGN PATTERNS - PHÁT HIỆN:
+   - MVC: Controller + Model + View/Routes
+   - Repository: Data access abstraction
+   - Service Layer: Business logic separation
+   - Dependency Injection: IoC containers
+   - Factory: Object creation patterns
+   - Singleton: Shared instances
+   - Observer: Event emitters
+   - Strategy: Pluggable algorithms
+
+6. MIDDLEWARE CHAIN - LIỆT KÊ:
+   - Global middleware
+   - Route-specific middleware
+   - Error handling middleware
+   - Execution order
+
+Hãy trả về JSON CHI TIẾT:
 {
   "architectureLayers": [
     {
-      "name": "Tên layer PHÁT HIỆN từ folder structure",
-      "purpose": "Purpose SUY RA từ files trong layer",
-      "components": ["Component names TÌM ĐƯỢC trong layer này"]
+      "name": "Layer name CỤ THỂ (Presentation, Business Logic, Data Access, Infrastructure)",
+      "purpose": "Purpose CHI TIẾT",
+      "components": ["TẤT CẢ component names trong layer"],
+      "responsibilities": ["Chi tiết responsibilities"],
+      "patterns": ["Design patterns used"]
     }
   ],
   "components": [
     {
-      "name": "Tên component TỪ FILENAME (VD: ProductController từ ProductController.js)",
-      "type": "Type DỰA VÀO file name và location (Controller/Service/Repository...)",
-      "layer": "Layer PHÁT HIỆN từ folder path",
-      "technology": "Technology TỪ imports (express, sequelize...)",
-      "description": "Mô tả DỰA VÀO code trong file",
+      "name": "Component name CHÍNH XÁC từ filename",
+      "type": "Type CỤ THỂ (Controller/Service/Repository/Middleware/Utility...)",
+      "layer": "Layer CỤ THỂ",
+      "technology": "Technology stack (Express, Sequelize, TypeORM...)",
+      "description": "Mô tả ĐẦY ĐỦ purpose và functionality",
       "responsibilities": [
-        "Responsibilities SUY RA từ methods trong file"
+        "LIST TẤT CẢ responsibilities từ methods"
       ],
       "files": [
-        "File paths CHÍNH XÁC từ scan"
+        "TẤT CẢ file paths liên quan"
       ],
       "apiEndpoints": [
         {
-          "method": "HTTP method TÌM ĐƯỢC (GET/POST/PUT/DELETE)",
-          "path": "Path CHÍNH XÁC từ route definition",
-          "description": "Description TỪ comments hoặc suy từ path",
-          "authentication": "Required nếu thấy auth middleware, None nếu không",
-          "authorization": "Authorization level TỪ middleware/decorators"
+          "method": "HTTP method CHÍNH XÁC",
+          "path": "Path CHÍNH XÁC từ route",
+          "description": "Description CHI TIẾT",
+          "handler": "Handler function name",
+          "authentication": "Auth requirement (Required/Optional/None)",
+          "authorization": "Role/Permission requirements",
+          "validation": "Input validation rules",
+          "requestBody": "Request body schema",
+          "responseType": "Response type"
         }
       ],
       "dependencies": [
         {
           "component": "Component name TỪ IMPORTS",
-          "purpose": "Purpose SUY từ cách dùng",
-          "type": "Type TỪ file path của dependency"
+          "purpose": "Purpose CHI TIẾT",
+          "type": "Dependency type (Import/Injection/Composition)"
         }
       ],
-      "designPatterns": ["Patterns PHÁT HIỆN từ code structure"],
       "methods": [
         {
           "name": "Method name TỪ CODE",
-          "purpose": "Purpose TỪ function name hoặc comments",
-          "parameters": ["Params TỪ function signature"],
-          "returnType": "Return type TỪ code hoặc TypeScript types"
+          "purpose": "Purpose CHI TIẾT",
+          "parameters": ["TẤT CẢ params với types"],
+          "returnType": "Return type CHÍNH XÁC",
+          "async": "true/false",
+          "visibility": "public/private/protected"
         }
-      ]
+      ],
+      "designPatterns": ["Patterns PHÁT HIỆN được"]
     }
   ],
   "middlewareChain": [
     {
       "name": "Middleware name TỪ FILES",
-      "order": "Order number NẾUPHÁT HIỆN được",
-      "purpose": "Purpose TỪ middleware code",
-      "appliesTo": "Routes THẤY middleware được apply"
+      "order": "Execution order number",
+      "purpose": "Purpose CHI TIẾT",
+      "appliesTo": "Routes áp dụng (Global/Specific routes)",
+      "type": "Middleware type (Auth/Validation/Logging/ErrorHandling...)"
     }
   ],
   "dataFlow": {
-    "requestFlow": "Flow PHÁT HIỆN từ code (Client → Middleware → Controller → Service...)",
-    "responseFlow": "Response flow TỪ code"
+    "requestFlow": "Flow CHI TIẾT (Client → Route → Middleware → Controller → Service → Repository → Database)",
+    "responseFlow": "Response flow (Database → Repository → Service → Controller → Response)",
+    "errorFlow": "Error handling flow"
   },
   "crossCuttingConcerns": [
     {
-      "name": "Concern TÌM ĐƯỢC (Logging/Caching/ErrorHandling...)",
-      "implementation": "Implementation type TỪ code",
-      "components": ["Components THỰC SỰ implement concern này"]
+      "name": "Concern name (Logging/Caching/ErrorHandling/Security/Monitoring...)",
+      "implementation": "Implementation CHI TIẾT",
+      "components": ["TẤT CẢ components liên quan"],
+      "strategy": "Strategy used"
     }
-  ]
+  ],
+  "securityMeasures": {
+    "authentication": "Auth methods CHI TIẾT",
+    "authorization": "Authorization strategy",
+    "inputValidation": "Validation approach",
+    "encryption": "Encryption methods"
+  }
 }
 
+YÊU CẦU ĐẦY ĐỦ:
+✅ Components: PHẢI liệt kê TẤT CẢ components từ files scan
+✅ API Endpoints: PHẢI có TẤT CẢ routes (50+, 100+, 200+...)
+✅ Dependencies: PHẢI map TẤT CẢ imports/dependencies
+✅ Middleware: PHẢI liệt kê TẤT CẢ middleware và order
+✅ Layers: PHẢI phân loại components vào layers
+
 LƯU Ý:
-- Nếu KHÔNG TÌM THẤY controllers → components array sẽ rỗng hoặc chỉ có những cái TÌM ĐƯỢC
-- Nếu KHÔNG có routes → apiEndpoints sẽ rỗng []
+- Nếu KHÔNG TÌM THẤY controllers → components array rỗng cho layer đó
+- Nếu KHÔNG có routes → apiEndpoints rỗng []
 - CHỈ liệt kê những gì CÓ TRONG CODE
-- Nếu không chắc → ghi "Detected from code" thay vì bịa
+- PHẢI CHI TIẾT và ĐẦY ĐỦ
 `;
 
   const result = await model.generateContent(prompt);