c4-gen 1.5.1 → 1.5.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "c4-gen",
-  "version": "1.5.1",
+  "version": "1.5.2",
   "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/markdown.js

@@ -693,59 +693,71 @@ ${seq.mermaidCode || generateSequenceMermaid(seq)}
 
 ## 2. Class Diagrams & Details
 
-${code.classes?.map((cls, index) => `
+${code.classes?.map((cls, index) => {
+  if (!cls || !cls.name) return '';
+  return `
 ### ${index + 1}. ${cls.name}
 
 | Property | Value |
 |----------|-------|
-| **Type** | ${cls.type} |
-| **File** | \`${cls.file}\` |
+| **Type** | ${cls.type || 'class'} |
+| **File** | \`${cls.file || 'Not specified'}\` |
 | **Extends** | ${cls.extends || 'None'} |
 | **Implements** | ${cls.implements?.join(', ') || 'None'} |
 
-**Description:** ${cls.description}
+**Description:** ${cls.description || 'No description'}
 
 #### Design Patterns:
 ${cls.designPatterns?.map(p => `- ${p}`).join('\n') || '- None'}
 
 #### Properties:
 
-${cls.properties?.map(prop => `
-- **\`${prop.name}\`** (${prop.type})
+${cls.properties?.map(prop => {
+  if (!prop || !prop.name) return '';
+  return `
+- **\`${prop.name}\`** (${prop.type || 'any'})
   - Visibility: ${prop.visibility || 'public'}
   - Required: ${prop.isRequired ? 'Yes' : 'No'}
   ${prop.defaultValue ? `- Default: \`${prop.defaultValue}\`` : ''}
-  - Description: ${prop.description}
-`).join('\n') || '- No properties'}
+  - Description: ${prop.description || 'N/A'}
+`;
+}).filter(p => p).join('\n') || '- No properties'}
 
 #### Methods:
 
-${cls.methods?.map(method => `
-##### \`${method.name}(${method.parameters?.map(p => `${p.name}: ${p.type}`).join(', ') || ''})\`
+${cls.methods?.map(method => {
+  if (!method || !method.name) return '';
+  return `
+##### \`${method.name}(${method.parameters?.map(p => `${p.name || 'param'}: ${p.type || 'any'}`).join(', ') || ''})\`
 
 - **Visibility:** ${method.visibility || 'public'}
 - **Static:** ${method.isStatic ? 'Yes' : 'No'}
 - **Async:** ${method.isAsync ? 'Yes' : 'No'}
-- **Returns:** ${method.returnType}
-- **Description:** ${method.description}
+- **Returns:** ${method.returnType || 'void'}
+- **Description:** ${method.description || 'N/A'}
 
 ${method.parameters?.length > 0 ? `**Parameters:**
-${method.parameters.map(p => `- \`${p.name}\` (${p.type}) ${p.isOptional ? '[Optional]' : '[Required]'}: ${p.description}`).join('\n')}
+${method.parameters.map(p => `- \`${p.name || 'param'}\` (${p.type || 'any'}) ${p.isOptional ? '[Optional]' : '[Required]'}: ${p.description || 'N/A'}`).join('\n')}
 ` : ''}
 
 ${method.exceptions?.length > 0 ? `**Exceptions:** ${method.exceptions.join(', ')}` : ''}
-`).join('\n') || '- No methods'}
+`;
+}).filter(m => m).join('\n') || '- No methods'}
 
 #### Relationships:
 
-${cls.relationships?.map(rel => `
+${cls.relationships?.map(rel => {
+  if (!rel || !rel.type || !rel.to) return '';
+  return `
 - **${rel.type}** → **${rel.to}**
-  - Description: ${rel.description}
+  - Description: ${rel.description || 'N/A'}
   - Cardinality: ${rel.cardinality || 'Not specified'}
-`).join('\n') || '- No relationships'}
+`;
+}).filter(r => r).join('\n') || '- No relationships'}
 
 ---
-`).join('\n') || 'No classes documented'}
+`;
+}).filter(c => c).join('\n') || 'No classes documented'}
 
 ---
 
@@ -791,8 +803,13 @@ ${generateClassMermaid(code)}
 }
 
 function generateSequenceMermaid(seq) {
+  if (!seq || !seq.steps || !Array.isArray(seq.steps)) {
+    return 'sequenceDiagram\n    Note: No steps defined\n';
+  }
+  
   let mermaid = 'sequenceDiagram\n';
   seq.steps?.forEach(step => {
+    if (!step || !step.from || !step.to || !step.action) return;
     mermaid += `    ${step.from}->>+${step.to}: ${step.action}\n`;
     if (step.responseType) {
       mermaid += `    ${step.to}-->>-${step.from}: ${step.responseType}\n`;
@@ -804,26 +821,32 @@ function generateSequenceMermaid(seq) {
 function generateClassMermaid(code) {
   const classes = code.classes || [];
   return classes.map(cls => {
+    // Safe check for class name
+    if (!cls || !cls.name) return '';
+    
     const className = cls.name.replace(/[^a-zA-Z0-9]/g, '_');
     let mermaid = `    class ${className} {\n`;
     
     // Properties
     cls.properties?.forEach(prop => {
+      if (!prop || !prop.name || !prop.type) return;
       const visibility = prop.visibility === 'private' ? '-' : prop.visibility === 'protected' ? '#' : '+';
       mermaid += `        ${visibility}${prop.type} ${prop.name}\n`;
     });
     
     // Methods
     cls.methods?.forEach(method => {
+      if (!method || !method.name) return;
       const visibility = method.visibility === 'private' ? '-' : method.visibility === 'protected' ? '#' : '+';
       const params = method.parameters?.map(p => `${p.name}: ${p.type}`).join(', ') || '';
-      mermaid += `        ${visibility}${method.name}(${params}) ${method.returnType}\n`;
+      mermaid += `        ${visibility}${method.name}(${params}) ${method.returnType || 'void'}\n`;
     });
     
     mermaid += '    }\n';
     
     // Relationships
     cls.relationships?.forEach(rel => {
+      if (!rel || !rel.to || !rel.type) return;
       const target = rel.to.replace(/[^a-zA-Z0-9]/g, '_');
       if (rel.type === 'extends') {
         mermaid += `    ${target} <|-- ${className}\n`;
@@ -835,7 +858,7 @@ function generateClassMermaid(code) {
     });
     
     return mermaid;
-  }).join('\n');
+  }).filter(m => m).join('\n'); // Filter out empty strings
 }
 
 function generateDatabaseSchemaDoc(schema) {