feat: Add command options and config to exclude tables and views

composer.json

@@ -23,7 +23,7 @@
     "php": "^8.0",
     "doctrine/doctrine-bundle": "^2.5",
     "doctrine/orm": "^2.9 || ^3.0",
-    "jawira/db-draw": "^1.2",
+    "jawira/db-draw": "^1.6",
     "jawira/plantuml-client": "^1.0",
     "symfony/console": "^6.0 || ^7.0",
     "symfony/filesystem": "^6.0 || ^7.0",

src/Command/DoctrineDiagramCommand.php

@@ -47,7 +47,8 @@ protected function configure(): void
       ->addOption(Config::SIZE, null, InputOption::VALUE_REQUIRED, sprintf('Diagram size (<info>%s</info>, <info>%s</info> or <info>%s</info>).', Size::MINI, Size::MIDI, Size::MAXI))
       ->addOption(Config::SERVER, null, InputOption::VALUE_REQUIRED, 'PlantUML server URL, only used to convert puml diagrams to svg and png.')
       ->addOption(Config::CONNECTION, null, InputOption::VALUE_REQUIRED, 'Doctrine connection to use.')
-      ->addOption(Config::THEME, null, InputOption::VALUE_REQUIRED, 'Change diagram colors and style.');
+      ->addOption(Config::THEME, null, InputOption::VALUE_REQUIRED, 'Change diagram colors and style.')
+      ->addOption(Config::EXCLUDE, null, InputOption::VALUE_REQUIRED, 'Comma separated list of tables to exclude from diagram.');
   }
 
   protected function execute(InputInterface $input, OutputInterface $output): int
@@ -61,8 +62,11 @@ protected function execute(InputInterface $input, OutputInterface $output): int
     $server         = $this->stringOrNullOption($input, Config::SERVER);
     $filename       = $this->stringOrNullOption($input, Config::FILENAME);
     $theme          = $this->stringOrNullOption($input, Config::THEME);
+    $exclude        = $this->stringOrNullOption($input, Config::EXCLUDE);
 
-    $puml     = $this->doctrineDiagram->generatePuml($connectionName, $size, $theme);
+    $excludeArray = is_string($exclude) ? explode(',', $exclude) : null;
+
+    $puml     = $this->doctrineDiagram->generatePuml($connectionName, $size, $theme, $excludeArray);
     $content  = $this->doctrineDiagram->convertWithServer($puml, $format, $server);
     $fullName = $this->doctrineDiagram->dumpDiagram($content, $filename, $format);
 

src/Constants/Config.php

@@ -10,4 +10,5 @@ class Config
   public const SERVER = 'server';
   public const CONNECTION = 'connection';
   public const THEME = 'theme';
+  public const EXCLUDE = 'exclude';
 }

src/DependencyInjection/Configuration.php

@@ -18,8 +18,8 @@ class Configuration implements ConfigurationInterface
   public function getConfigTreeBuilder(): TreeBuilder
   {
     $buildTree = (new TreeBuilder('doctrine_diagram'));
-    /** @var ParentNodeDefinitionInterface $rootNode */
-    $rootNode = $buildTree->getRootNode();
+    $rootNode  = $buildTree->getRootNode();
+    assert($rootNode instanceof ParentNodeDefinitionInterface);
     $rootNode->children()
       ->enumNode(Config::SIZE)
       ->values([Size::MINI, Size::MIDI, Size::MAXI])
@@ -41,6 +41,9 @@ public function getConfigTreeBuilder(): TreeBuilder
       ->scalarNode(Config::CONNECTION)
       ->defaultValue(null)
       ->end()
+      ->arrayNode(Config::EXCLUDE)
+      ->scalarPrototype()->end()
+      ->end()
       ->end();
 
     return $buildTree;

src/DependencyInjection/DoctrineDiagramExtension.php

@@ -22,6 +22,7 @@ public function load(array $configs, ContainerBuilder $container): void
     $container->setParameter('doctrine_diagram.' . Config::SERVER, $config[Config::SERVER]);
     $container->setParameter('doctrine_diagram.' . Config::THEME, $config[Config::THEME]);
     $container->setParameter('doctrine_diagram.' . Config::CONNECTION, $config[Config::CONNECTION]);
+    $container->setParameter('doctrine_diagram.' . Config::EXCLUDE, $config[Config::EXCLUDE]);
 
     $loader = new PhpFileLoader($container, new FileLocator(__DIR__ . '/../Resources/config'));
     $loader->load('services.php');

src/Resources/config/services.php

@@ -20,7 +20,8 @@
     ->arg('$format', param('doctrine_diagram.' . Config::FORMAT))
     ->arg('$server', param('doctrine_diagram.' . Config::SERVER))
     ->arg('$theme', param('doctrine_diagram.' . Config::THEME))
-    ->arg('$connection', param('doctrine_diagram.' . Config::CONNECTION));
+    ->arg('$connection', param('doctrine_diagram.' . Config::CONNECTION))
+    ->arg('$exclude', param('doctrine_diagram.' . Config::EXCLUDE));
 
   $services->set('jawira.doctrine_diagram.command', DoctrineDiagramCommand::class)
     ->arg('$doctrineDiagram', service('jawira.doctrine_diagram.service'))

src/Service/DoctrineDiagram.php

@@ -2,6 +2,7 @@
 
 namespace Jawira\DoctrineDiagramBundle\Service;
 
+use Doctrine\DBAL\Connection;
 use Doctrine\Persistence\ConnectionRegistry;
 use Jawira\DbDraw\DbDraw;
 use Jawira\DoctrineDiagramBundle\Constants\Format;
@@ -26,7 +27,10 @@ public function __construct(
     private string             $theme,
     /** This value comes from doctrine_diagram.yaml */
     private ?string            $connection,
-  ) {
+    /** This value comes from doctrine_diagram.yaml */
+    private array              $exclude,
+  )
+  {
   }
 
   /**
@@ -37,19 +41,20 @@ public function __construct(
    *
    * @param ?string $connectionName Doctrine connection name,this value comes from console.
    */
-  public function generatePuml(?string $connectionName = null, ?string $size = null, ?string $theme = null): string
+  public function generatePuml(?string $connectionName = null, ?string $size = null, ?string $theme = null, ?array $exclude = null): string
   {
     // Fallback values from doctrine_diagram.yaml
     $connectionName ??= $this->connection;
     $size           ??= $this->size;
     $theme          ??= $this->theme;
+    $exclude        ??= $this->exclude;
 
     // Generate puml diagram
     $connection = $this->doctrine->getConnection($connectionName);
-    /** @var \Doctrine\DBAL\Connection $connection */
+    assert($connection instanceof Connection);
     $dbDraw = new DbDraw($connection);
 
-    return $dbDraw->generatePuml($size, $theme);
+    return $dbDraw->generatePuml($size, $theme, $exclude);
   }
 
   /**