Multiple Connections: Correctly inject code generators

Technically we do use tags and interface implementations to do this automatically

Author: homersimpsons

DependencyInjection/TdbmCompilerPass.php

@@ -0,0 +1,41 @@
+<?php
+
+declare(strict_types=1);
+
+namespace TheCodingMachine\TDBM\Bundle\DependencyInjection;
+
+use Symfony\Component\DependencyInjection\Compiler\CompilerPassInterface;
+use Symfony\Component\DependencyInjection\ContainerBuilder;
+use Symfony\Component\DependencyInjection\Reference;
+
+class TdbmCompilerPass implements CompilerPassInterface
+{
+
+    /**
+     * @inheritDoc
+     */
+    public function process(ContainerBuilder $container)
+    {
+        $generatorListeners = $this->taggedServicesToReferences($container->findTaggedServiceIds(TdbmExtension::TAG_GENERATOR_LISTENER));
+        $codeGeneratorListeners = $this->taggedServicesToReferences($container->findTaggedServiceIds(TdbmExtension::TAG_CODE_GENERATOR_LISTENER));
+
+        $tdbmConfigurations = array_keys($container->findTaggedServiceIds(TdbmExtension::TAG_TDBM_CONFIGURATION));
+
+        foreach ($tdbmConfigurations as $tdbmConfiguration) {
+            $configuration = $container->getDefinition($tdbmConfiguration);
+            $configuration->setArgument('$generatorListeners', $generatorListeners);
+            $configuration->setArgument('$codeGeneratorListeners', $codeGeneratorListeners);
+        }
+    }
+
+    /**
+     * @param array<string, array> $taggedServices Keys are services ids, this is the output of `ContainerBuilder::findTaggedServiceIds`
+     * @return array<Reference>
+     */
+    private function taggedServicesToReferences(array $taggedServices): array
+    {
+        return array_map(static function (string $serviceId) {
+            return new Reference($serviceId);
+        }, array_keys($taggedServices));
+    }
+}

DependencyInjection/TdbmExtension.php

@@ -25,7 +25,9 @@
 use TheCodingMachine\TDBM\Schema\LockFileSchemaManager;
 use TheCodingMachine\TDBM\TDBMService;
 use TheCodingMachine\TDBM\Utils\Annotation\AnnotationParser;
+use TheCodingMachine\TDBM\Utils\CodeGeneratorListenerInterface;
 use TheCodingMachine\TDBM\Utils\DefaultNamingStrategy;
+use TheCodingMachine\TDBM\Utils\GeneratorListenerInterface;
 use TheCodingMachine\TDBM\Utils\NamingStrategyInterface;
 use TheCodingMachine\TDBM\SchemaLockFileDumper;
 use TheCodingMachine\TDBM\Utils\RootProjectLocator;
@@ -39,6 +41,10 @@
 
 class TdbmExtension extends Extension
 {
+    public const TAG_GENERATOR_LISTENER = 'tdbm.generatorListener';
+    public const TAG_CODE_GENERATOR_LISTENER = 'tdbm.codeGeneratorListener';
+    public const TAG_TDBM_CONFIGURATION = 'tdbm.configuration';
+
     private const DEFAULT_CONFIGURATION_ID = TDBMConfiguration::class;
     private const DEFAULT_NAMING_STRATEGY_ID = DefaultNamingStrategy::class;
 
@@ -51,6 +57,9 @@ class TdbmExtension extends Extension
      */
     public function load(array $configs, ContainerBuilder $container): void
     {
+        $container->registerForAutoconfiguration(GeneratorListenerInterface::class)->addTag(self::TAG_GENERATOR_LISTENER);
+        $container->registerForAutoconfiguration(CodeGeneratorListenerInterface::class)->addTag(self::TAG_CODE_GENERATOR_LISTENER);
+
         $configuration = new Configuration();
         $processedConfig = $this->processConfiguration($configuration, $configs);
 
@@ -138,8 +147,8 @@ private function getConfigurationDefinition(ConnectionConfiguration $config, str
         $configuration->setArgument(1, $config->getDaoNamespace());
         $configuration->setArgument('$connection', new Reference($config->getConnection()));
         $configuration->setArgument('$namingStrategy', new Reference($namingStrategyServiceId));
-        $configuration->setArgument('$codeGeneratorListeners', [new Reference(SymfonyCodeGeneratorListener::class)]);
         $configuration->setArgument('$cache', new Reference('tdbm.cache'));
+        $configuration->addTag(self::TAG_TDBM_CONFIGURATION);
 
         // Let's name the tdbm lock file after the name of the DBAL connection.
 

TdbmBundle.php

@@ -1,12 +1,20 @@
 <?php
 
+declare(strict_types=1);
 
 namespace TheCodingMachine\TDBM\Bundle;
 
-
+use Symfony\Component\DependencyInjection\Compiler\PassConfig;
+use Symfony\Component\DependencyInjection\ContainerBuilder;
 use Symfony\Component\HttpKernel\Bundle\Bundle;
+use TheCodingMachine\TDBM\Bundle\DependencyInjection\TdbmCompilerPass;
 
 class TdbmBundle extends Bundle
 {
+    public function build(ContainerBuilder $container)
+    {
+        parent::build($container);
 
-}
+        $container->addCompilerPass(new TdbmCompilerPass(), PassConfig::TYPE_BEFORE_OPTIMIZATION, 1);
+    }
+}