scaffold-doc-cli 1.0.1 → 1.0.3

package/.github/workflows/ai-doc-sync.yml

@@ -8,6 +8,7 @@ on:
 
 permissions:
   contents: write
+  pull-requests: write
 
 jobs:
   doc-completion:
@@ -30,8 +31,16 @@ jobs:
           GEMINI_API_KEY: ${{ secrets.GEMINI_API_KEY }}
         run: node .github/scripts/doc-sync.mjs
 
-      - name: Commit Changes
-        uses: stefanzweifel/git-auto-commit-action@v5
+      - name: Create Pull Request
+        uses: peter-evans/create-pull-request@v5
         with:
-          commit_message: "docs: auto-generated documentation updates"
-          file_pattern: .docs/** mkdocs.yml
+          token: ${{ secrets.GITHUB_TOKEN }}
+          commit-message: "docs: auto-generated documentation updates"
+          title: "docs: AI-generated documentation updates"
+          body: |
+            Automatic documentation updates generated by AI based on recent code changes.
+            
+            Please review and merge if accurate.
+          branch: ai-docs/${{ github.ref_name }}
+          base: ${{ github.head_ref || github.ref_name }}
+          delete-branch: true

package/index.js

@@ -8,53 +8,61 @@ import { fileURLToPath } from 'url';
 
 const __dirname = path.dirname(fileURLToPath(import.meta.url));
 
+// Handle version check
+const args = process.argv.slice(2);
+if (args.includes('-v') || args.includes('--version')) {
+  const packageJson = await fs.readJson(path.join(__dirname, 'package.json'));
+  console.log(packageJson.version);
+  process.exit(0);
+}
+
 async function init() {
-    console.log(chalk.green.bold('Welcome to the Motin Documentation Initializer! 🚀'));
-
-    const answers = await inquirer.prompt([
-        {
-            type: 'input',
-            name: 'projectName',
-            message: 'What is the name of your project?',
-            default: path.basename(process.cwd()),
-        },
-        {
-            type: 'list',
-            name: 'stack',
-            message: 'Which technical stack are you using?',
-            choices: [
-                new inquirer.Separator('--- JavaScript/TypeScript ---'),
-                'Nuxt', 'Next.js', 'Vue', 'React', 'Angular', 'Svelte', 'Node (Express/NestJS)',
-                new inquirer.Separator('--- PHP ---'),
-                'Laravel', 'Symfony', 'Generic PHP',
-                new inquirer.Separator('--- Python ---'),
-                'Django', 'FastAPI', 'Flask', 'Generic Python',
-                new inquirer.Separator('--- Other ---'),
-                'Java (Spring)', 'Ruby (Rails)', 'Go', 'Rust', 'Other'
-            ],
-        },
-        {
-            type: 'confirm',
-            name: 'includeWorkflow',
-            message: 'Do you want to include a GitHub Workflow for AI doc completion?',
-            default: true,
-        },
-    ]);
-
-    const docsDir = path.join(process.cwd(), '.docs');
-    const referenceDir = path.join(docsDir, 'reference');
-
-    try {
-        const relativeDocsDir = path.relative(process.cwd(), docsDir);
-        console.log(chalk.blue(`\nCreating documentation directory at ./${relativeDocsDir}...`));
-        await fs.ensureDir(docsDir);
-        await fs.ensureDir(referenceDir);
-
-        // 1. Create Standard Files
-        // ------------------------
-
-        // index.md
-        const indexContent = `<!-- 
+  console.log(chalk.green.bold('Welcome to the Motin Documentation Initializer! 🚀'));
+
+  const answers = await inquirer.prompt([
+    {
+      type: 'input',
+      name: 'projectName',
+      message: 'What is the name of your project?',
+      default: path.basename(process.cwd()),
+    },
+    {
+      type: 'list',
+      name: 'stack',
+      message: 'Which technical stack are you using?',
+      choices: [
+        new inquirer.Separator('--- JavaScript/TypeScript ---'),
+        'Nuxt', 'Next.js', 'Vue', 'React', 'Angular', 'Svelte', 'Node (Express/NestJS)',
+        new inquirer.Separator('--- PHP ---'),
+        'Laravel', 'Symfony', 'Generic PHP',
+        new inquirer.Separator('--- Python ---'),
+        'Django', 'FastAPI', 'Flask', 'Generic Python',
+        new inquirer.Separator('--- Other ---'),
+        'Java (Spring)', 'Ruby (Rails)', 'Go', 'Rust', 'Other'
+      ],
+    },
+    {
+      type: 'confirm',
+      name: 'includeWorkflow',
+      message: 'Do you want to include a GitHub Workflow for AI doc completion?',
+      default: true,
+    },
+  ]);
+
+  const docsDir = path.join(process.cwd(), '.docs');
+  const referenceDir = path.join(docsDir, 'reference');
+
+  try {
+    const relativeDocsDir = path.relative(process.cwd(), docsDir);
+    console.log(chalk.blue(`\nCreating documentation directory at ./${relativeDocsDir}...`));
+    await fs.ensureDir(docsDir);
+    await fs.ensureDir(referenceDir);
+
+    // 1. Create Standard Files
+    // ------------------------
+
+    // index.md
+    const indexContent = `<!-- 
   AI_DOC_START
   CONTEXT: This is the entry point for the project documentation.
   INSTRUCTION: Analyze the project structure and update the "Overview" section.
@@ -70,10 +78,10 @@ Documentation initialized.
 - [Getting Started](./getting-started.md)
 - [Architecture](./architecture.md)
 `;
-        await fs.writeFile(path.join(docsDir, 'index.md'), indexContent);
+    await fs.writeFile(path.join(docsDir, 'index.md'), indexContent);
 
-        // getting-started.md
-        const gettingStartedContent = `# Getting Started
+    // getting-started.md
+    const gettingStartedContent = `# Getting Started
 
 <!-- 
   INSTRUCTION: Document the installation and setup steps.
@@ -87,10 +95,10 @@ Documentation initialized.
 <!-- AI_CONTENT_START -->
 <!-- AI_CONTENT_END -->
 `;
-        await fs.writeFile(path.join(docsDir, 'getting-started.md'), gettingStartedContent);
+    await fs.writeFile(path.join(docsDir, 'getting-started.md'), gettingStartedContent);
 
-        // architecture.md
-        const architectureContent = `# Architecture
+    // architecture.md
+    const architectureContent = `# Architecture
 
 <!-- 
   INSTRUCTION: Document the high-level system architecture, data flow, and key design decisions.
@@ -100,31 +108,31 @@ Documentation initialized.
 <!-- AI_CONTENT_START -->
 <!-- AI_CONTENT_END -->
 `;
-        await fs.writeFile(path.join(docsDir, 'architecture.md'), architectureContent);
+    await fs.writeFile(path.join(docsDir, 'architecture.md'), architectureContent);
 
-        // 2. Create Stack Specific Files (Granular)
-        // ----------------------------------------
-        const stackConfig = getStackConfig(answers.stack);
+    // 2. Create Stack Specific Files (Granular)
+    // ----------------------------------------
+    const stackConfig = getStackConfig(answers.stack);
 
-        // Generate each reference file defined in the config
-        for (const [filename, contentGenerator] of Object.entries(stackConfig.files)) {
-            await fs.writeFile(path.join(referenceDir, filename), contentGenerator());
-        }
+    // Generate each reference file defined in the config
+    for (const [filename, contentGenerator] of Object.entries(stackConfig.files)) {
+      await fs.writeFile(path.join(referenceDir, filename), contentGenerator());
+    }
 
-        console.log(chalk.green('✔ Standard documentation files created'));
-        console.log(chalk.green(`✔ ${Object.keys(stackConfig.files).length} reference doc(s) created`));
+    console.log(chalk.green('✔ Standard documentation files created'));
+    console.log(chalk.green(`✔ ${Object.keys(stackConfig.files).length} reference doc(s) created`));
 
-        // 3. Create MkDocs Config
-        // -----------------------
+    // 3. Create MkDocs Config
+    // -----------------------
 
-        // Construct navigation for Reference section
-        const referenceNav = Object.keys(stackConfig.files).map(filename => {
-            // Convert 'controllers.md' -> 'Controllers'
-            const title = filename.replace('.md', '').charAt(0).toUpperCase() + filename.replace('.md', '').slice(1);
-            return `    - ${title}: reference/${filename}`;
-        }).join('\n');
+    // Construct navigation for Reference section
+    const referenceNav = Object.keys(stackConfig.files).map(filename => {
+      // Convert 'controllers.md' -> 'Controllers'
+      const title = filename.replace('.md', '').charAt(0).toUpperCase() + filename.replace('.md', '').slice(1);
+      return `    - ${title}: reference/${filename}`;
+    }).join('\n');
 
-        const mkdocsContent = `site_name: ${answers.projectName} Documentation
+    const mkdocsContent = `site_name: ${answers.projectName} Documentation
 docs_dir: .docs
 theme: readthedocs
 
@@ -135,19 +143,19 @@ nav:
   - Reference:
 ${referenceNav}
 `;
-        await fs.writeFile(path.join(process.cwd(), 'mkdocs.yml'), mkdocsContent);
-        console.log(chalk.green('✔ mkdocs.yml created'));
+    await fs.writeFile(path.join(process.cwd(), 'mkdocs.yml'), mkdocsContent);
+    console.log(chalk.green('✔ mkdocs.yml created'));
 
-        // 4. Create GitHub Workflow & AI Script
-        // -------------------------
-        if (answers.includeWorkflow) {
-            const workflowDir = path.join(process.cwd(), '.github', 'workflows');
-            const scriptsDir = path.join(process.cwd(), '.github', 'scripts');
-            await fs.ensureDir(workflowDir);
-            await fs.ensureDir(scriptsDir);
+    // 4. Create GitHub Workflow & AI Script
+    // -------------------------
+    if (answers.includeWorkflow) {
+      const workflowDir = path.join(process.cwd(), '.github', 'workflows');
+      const scriptsDir = path.join(process.cwd(), '.github', 'scripts');
+      await fs.ensureDir(workflowDir);
+      await fs.ensureDir(scriptsDir);
 
-            // Workflow File
-            const workflowContent = `name: AI Doc Sync
+      // Workflow File
+      const workflowContent = `name: AI Doc Sync
 
 on:
   push:
@@ -157,6 +165,7 @@ on:
 
 permissions:
   contents: write
+  pull-requests: write
 
 jobs:
   doc-completion:
@@ -179,17 +188,25 @@ jobs:
           GEMINI_API_KEY: \${{ secrets.GEMINI_API_KEY }}
         run: node .github/scripts/doc-sync.mjs
 
-      - name: Commit Changes
-        uses: stefanzweifel/git-auto-commit-action@v5
+      - name: Create Pull Request
+        uses: peter-evans/create-pull-request@v5
         with:
-          commit_message: "docs: auto-generated documentation updates"
-          file_pattern: .docs/** mkdocs.yml
+          token: \${{ secrets.GITHUB_TOKEN }}
+          commit-message: "docs: auto-generated documentation updates"
+          title: "docs: AI-generated documentation updates"
+          body: |
+            Automatic documentation updates generated by AI based on recent code changes.
+            
+            Please review and merge if accurate.
+          branch: ai-docs/\${{ github.ref_name }}
+          base: \${{ github.head_ref || github.ref_name }}
+          delete-branch: true
 `;
-            await fs.writeFile(path.join(workflowDir, 'ai-doc-sync.yml'), workflowContent);
-            console.log(chalk.green('✔ .github/workflows/ai-doc-sync.yml created'));
+      await fs.writeFile(path.join(workflowDir, 'ai-doc-sync.yml'), workflowContent);
+      console.log(chalk.green('✔ .github/workflows/ai-doc-sync.yml created'));
 
-            // AI Sync Script
-            const scriptContent = `import { GoogleGenAI } from "@google/genai";
+      // AI Sync Script
+      const scriptContent = `import { GoogleGenAI } from "@google/genai";
 import fs from 'fs';
 import path from 'path';
 import { execSync } from 'child_process';
@@ -271,112 +288,112 @@ async function main() {
 }
 
 main();`;
-            await fs.writeFile(path.join(scriptsDir, 'doc-sync.mjs'), scriptContent);
-            console.log(chalk.green('✔ .github/scripts/doc-sync.mjs created'));
-        }
-
-        console.log(chalk.green.bold('\nDocumentation scaffolding complete! 🎉'));
-        console.log(chalk.yellow('Next steps:'));
-        console.log(`1. Review the generated files in ./${relativeDocsDir}`);
-        console.log('2. Run "mkdocs serve" to preview your documentation');
-        console.log('3. Commit the changes');
-
-    } catch (err) {
-        console.error(chalk.red('Error creating documentation files:'), err);
+      await fs.writeFile(path.join(scriptsDir, 'doc-sync.mjs'), scriptContent);
+      console.log(chalk.green('✔ .github/scripts/doc-sync.mjs created'));
     }
+
+    console.log(chalk.green.bold('\nDocumentation scaffolding complete! 🎉'));
+    console.log(chalk.yellow('Next steps:'));
+    console.log(`1. Review the generated files in ./${relativeDocsDir}`);
+    console.log('2. Run "mkdocs serve" to preview your documentation');
+    console.log('3. Commit the changes');
+
+  } catch (err) {
+    console.error(chalk.red('Error creating documentation files:'), err);
+  }
 }
 
 // Config Factory
 function getStackConfig(stack) {
-    const commonHeader = (type) => `<!-- 
+  const commonHeader = (type) => `<!-- 
   AI_DOC_START
   STACK: ${stack}
   TYPE: ${type}
 -->\n`;
 
-    // --- PHP: Symfony ---
-    if (stack.includes('Symfony')) {
-        return {
-            files: {
-                'controllers.md': () => `${commonHeader('Controllers')}# Controllers\n\n<!-- INSTRUCTION: Document all Symfony Controllers found in /src/Controller -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-                'entities.md': () => `${commonHeader('Entities')}# Entities\n\n<!-- INSTRUCTION: Document all Doctrine Entities found in /src/Entity -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-                'services.md': () => `${commonHeader('Services')}# Services\n\n<!-- INSTRUCTION: Document core application services found in /src/Service -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-            }
-        };
-    }
-
-    // --- PHP: Laravel ---
-    if (stack.includes('Laravel')) {
-        return {
-            files: {
-                'controllers.md': () => `${commonHeader('Controllers')}# Controllers\n\n<!-- INSTRUCTION: Document Http Controllers found in /app/Http/Controllers -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-                'models.md': () => `${commonHeader('Models')}# Models\n\n<!-- INSTRUCTION: Document Eloquent Models found in /app/Models -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-                'routes.md': () => `${commonHeader('Routes')}# Routes\n\n<!-- INSTRUCTION: Document application routes found in /routes -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-            }
-        };
-    }
+  // --- PHP: Symfony ---
+  if (stack.includes('Symfony')) {
+    return {
+      files: {
+        'controllers.md': () => `${commonHeader('Controllers')}# Controllers\n\n<!-- INSTRUCTION: Document all Symfony Controllers found in /src/Controller -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+        'entities.md': () => `${commonHeader('Entities')}# Entities\n\n<!-- INSTRUCTION: Document all Doctrine Entities found in /src/Entity -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+        'services.md': () => `${commonHeader('Services')}# Services\n\n<!-- INSTRUCTION: Document core application services found in /src/Service -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+      }
+    };
+  }
 
-    // --- JS: Nuxt ---
-    if (stack.includes('Nuxt')) {
-        return {
-            files: {
-                'components.md': () => `${commonHeader('Components')}# Components\n\n<!-- INSTRUCTION: Document Vue components in /components -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-                'composables.md': () => `${commonHeader('Composables')}# Composables\n\n<!-- INSTRUCTION: Document auto-imported composables in /composables -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-                'server-routes.md': () => `${commonHeader('Server Routes')}# Server Routes\n\n<!-- INSTRUCTION: Document Nitro server routes in /server/api -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-            }
-        };
-    }
+  // --- PHP: Laravel ---
+  if (stack.includes('Laravel')) {
+    return {
+      files: {
+        'controllers.md': () => `${commonHeader('Controllers')}# Controllers\n\n<!-- INSTRUCTION: Document Http Controllers found in /app/Http/Controllers -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+        'models.md': () => `${commonHeader('Models')}# Models\n\n<!-- INSTRUCTION: Document Eloquent Models found in /app/Models -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+        'routes.md': () => `${commonHeader('Routes')}# Routes\n\n<!-- INSTRUCTION: Document application routes found in /routes -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+      }
+    };
+  }
 
-    // --- JS: General (React, Next, Vue) ---
-    if (stack.includes('React') || stack.includes('Next') || stack.includes('Vue') || stack.includes('Angular') || stack.includes('Svelte')) {
-        return {
-            files: {
-                'components.md': () => `${commonHeader('Components')}# Components\n\n<!-- INSTRUCTION: Document UI components -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-                'hooks.md': () => `${commonHeader('Hooks')}# Hooks\n\n<!-- INSTRUCTION: Document custom hooks/composables -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-                'api.md': () => `${commonHeader('API')}# API Integration\n\n<!-- INSTRUCTION: Document API clients or fetch wrappers -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-            }
-        };
-    }
+  // --- JS: Nuxt ---
+  if (stack.includes('Nuxt')) {
+    return {
+      files: {
+        'components.md': () => `${commonHeader('Components')}# Components\n\n<!-- INSTRUCTION: Document Vue components in /components -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+        'composables.md': () => `${commonHeader('Composables')}# Composables\n\n<!-- INSTRUCTION: Document auto-imported composables in /composables -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+        'server-routes.md': () => `${commonHeader('Server Routes')}# Server Routes\n\n<!-- INSTRUCTION: Document Nitro server routes in /server/api -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+      }
+    };
+  }
 
-    // --- Python: Django ---
-    if (stack.includes('Django')) {
-        return {
-            files: {
-                'models.md': () => `${commonHeader('Models')}# Models\n\n<!-- INSTRUCTION: Document Django Models -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-                'views.md': () => `${commonHeader('Views')}# Views\n\n<!-- INSTRUCTION: Document Django Views/ViewSets -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-                'admin.md': () => `${commonHeader('Admin')}# Admin Customization\n\n<!-- INSTRUCTION: Document Admin classes -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-            }
-        };
-    }
+  // --- JS: General (React, Next, Vue) ---
+  if (stack.includes('React') || stack.includes('Next') || stack.includes('Vue') || stack.includes('Angular') || stack.includes('Svelte')) {
+    return {
+      files: {
+        'components.md': () => `${commonHeader('Components')}# Components\n\n<!-- INSTRUCTION: Document UI components -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+        'hooks.md': () => `${commonHeader('Hooks')}# Hooks\n\n<!-- INSTRUCTION: Document custom hooks/composables -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+        'api.md': () => `${commonHeader('API')}# API Integration\n\n<!-- INSTRUCTION: Document API clients or fetch wrappers -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+      }
+    };
+  }
 
-    // --- Python: FastAPI ---
-    if (stack.includes('FastAPI')) {
-        return {
-            files: {
-                'endpoints.md': () => `${commonHeader('Endpoints')}# Endpoints\n\n<!-- INSTRUCTION: Document FastAPI routes and decorators -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-                'models.md': () => `${commonHeader('Models')}# Pydantic Models\n\n<!-- INSTRUCTION: Document Pydantic models -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-            }
-        };
-    }
+  // --- Python: Django ---
+  if (stack.includes('Django')) {
+    return {
+      files: {
+        'models.md': () => `${commonHeader('Models')}# Models\n\n<!-- INSTRUCTION: Document Django Models -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+        'views.md': () => `${commonHeader('Views')}# Views\n\n<!-- INSTRUCTION: Document Django Views/ViewSets -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+        'admin.md': () => `${commonHeader('Admin')}# Admin Customization\n\n<!-- INSTRUCTION: Document Admin classes -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+      }
+    };
+  }
 
-    // --- Java/Spring ---
-    if (stack.includes('Spring') || stack.includes('Java')) {
-        return {
-            files: {
-                'controllers.md': () => `${commonHeader('Controllers')}# Controllers\n\n<!-- INSTRUCTION: Document REST Controllers -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-                'services.md': () => `${commonHeader('Services')}# Services\n\n<!-- INSTRUCTION: Document Service beans -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-                'domain.md': () => `${commonHeader('Domain')}# Domain Models\n\n<!-- INSTRUCTION: Document Entity classes -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-            }
-        };
-    }
+  // --- Python: FastAPI ---
+  if (stack.includes('FastAPI')) {
+    return {
+      files: {
+        'endpoints.md': () => `${commonHeader('Endpoints')}# Endpoints\n\n<!-- INSTRUCTION: Document FastAPI routes and decorators -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+        'models.md': () => `${commonHeader('Models')}# Pydantic Models\n\n<!-- INSTRUCTION: Document Pydantic models -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+      }
+    };
+  }
 
-    // --- Default Fallback ---
+  // --- Java/Spring ---
+  if (stack.includes('Spring') || stack.includes('Java')) {
     return {
-        files: {
-            'api.md': () => `${commonHeader('API')}# API Reference\n\n<!-- INSTRUCTION: Document public API endpoints -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-            'modules.md': () => `${commonHeader('Modules')}# Core Modules\n\n<!-- INSTRUCTION: Document core logical modules -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
-        }
+      files: {
+        'controllers.md': () => `${commonHeader('Controllers')}# Controllers\n\n<!-- INSTRUCTION: Document REST Controllers -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+        'services.md': () => `${commonHeader('Services')}# Services\n\n<!-- INSTRUCTION: Document Service beans -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+        'domain.md': () => `${commonHeader('Domain')}# Domain Models\n\n<!-- INSTRUCTION: Document Entity classes -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+      }
     };
+  }
+
+  // --- Default Fallback ---
+  return {
+    files: {
+      'api.md': () => `${commonHeader('API')}# API Reference\n\n<!-- INSTRUCTION: Document public API endpoints -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+      'modules.md': () => `${commonHeader('Modules')}# Core Modules\n\n<!-- INSTRUCTION: Document core logical modules -->\n\n## List\n<!-- AI_CONTENT_START -->\n<!-- AI_CONTENT_END -->`,
+    }
+  };
 }
 
 init();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "scaffold-doc-cli",
-  "version": "1.0.1",
+  "version": "1.0.3",
   "description": "CLI tool to scaffold standardized documentation structure for projects.",
   "main": "index.js",
   "type": "module",