feat: Add trait rendering mode option to GenerateCommand

Author: tasuku43

src/Console/Command/GenerateCommand.php

@@ -14,6 +14,7 @@
 use Tasuku43\MermaidClassDiagram\ClassDiagramRenderer\ClassDiagramBuilder;
 use Tasuku43\MermaidClassDiagram\ClassDiagramRenderer\Node\NodeParser;
 use Tasuku43\MermaidClassDiagram\ClassDiagramRenderer\RenderOptions\RenderOptions;
+use Tasuku43\MermaidClassDiagram\ClassDiagramRenderer\TraitRenderMode;
 
 class GenerateCommand extends Command
 {
@@ -32,6 +33,12 @@ protected function configure()
             InputOption::VALUE_REQUIRED,
             'Comma-separated relationship types to exclude: dependency,composition,inheritance,realization'
         );
+        $this->addOption(
+            'trait-mode',
+            null,
+            InputOption::VALUE_REQUIRED,
+            'Trait rendering mode: with-traits | flatten (default: flatten)'
+        );
     }
 
     protected function execute(InputInterface $input, OutputInterface $output): int
@@ -56,6 +63,24 @@ private function buildRenderOptions(InputInterface $input): RenderOptions
     {
         $options = RenderOptions::default();
 
+        // Trait render mode
+        $traitMode = (string)($input->getOption('trait-mode') ?? '');
+        switch (strtolower($traitMode)) {
+            case 'with-traits':
+            case 'with_traits':
+            case 'with':
+                $options->traitRenderMode = TraitRenderMode::WithTraits;
+                break;
+            case 'flatten':
+            case 'flat':
+            case '':
+                // default (Flatten)
+                break;
+            default:
+                // ignore unknown; keep default
+                break;
+        }
+
         $exclude = (string)($input->getOption('exclude-relationships') ?? '');
         if ($exclude === '') {
             return $options;