From 0eb42eab820093be15317e7b70004785475be5d6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?K=C3=A9vin=20Dunglas?= <dunglas@gmail.com>
Date: Sun, 31 Mar 2019 21:06:18 +0200
Subject: [PATCH] Add support for validator automapping

---
 DependencyInjection/DoctrineExtension.php | 18 ++++++++++++++++++
 Tests/ContainerTest.php                   |  7 +++++++
 2 files changed, 25 insertions(+)

diff --git a/DependencyInjection/DoctrineExtension.php b/DependencyInjection/DoctrineExtension.php
index 8c54f7bef..6a887de6a 100644
--- a/DependencyInjection/DoctrineExtension.php
+++ b/DependencyInjection/DoctrineExtension.php
@@ -12,6 +12,7 @@
 use Symfony\Bridge\Doctrine\DependencyInjection\AbstractDoctrineExtension;
 use Symfony\Bridge\Doctrine\Messenger\DoctrineTransactionMiddleware;
 use Symfony\Bridge\Doctrine\PropertyInfo\DoctrineExtractor;
+use Symfony\Bridge\Doctrine\Validator\DoctrineLoader;
 use Symfony\Component\Config\FileLocator;
 use Symfony\Component\DependencyInjection\Alias;
 use Symfony\Component\DependencyInjection\ChildDefinition;
@@ -21,6 +22,7 @@
 use Symfony\Component\DependencyInjection\Loader\XmlFileLoader;
 use Symfony\Component\DependencyInjection\Reference;
 use Symfony\Component\Messenger\MessageBusInterface;
+use Symfony\Component\Validator\Mapping\Loader\LoaderInterface;
 
 /**
  * DoctrineExtension is an extension for the Doctrine DBAL and ORM library.
@@ -349,6 +351,7 @@ protected function ormLoad(array $config, ContainerBuilder $container)
             $this->loadOrmEntityManager($entityManager, $container);
 
             $this->loadPropertyInfoExtractor($name, $container);
+            $this->loadValidatorLoader($name, $container);
         }
 
         if ($config['resolve_target_entities']) {
@@ -761,6 +764,21 @@ private function loadPropertyInfoExtractor($entityManagerName, ContainerBuilder
         $propertyExtractorDefinition->addTag('property_info.type_extractor', ['priority' => -999]);
     }
 
+    /**
+     * Loads a validator loader for each defined entity manager.
+     */
+    private function loadValidatorLoader(string $entityManagerName, ContainerBuilder $container) : void
+    {
+        if (! interface_exists(LoaderInterface::class) || ! class_exists(DoctrineLoader::class)) {
+            return;
+        }
+
+        $validatorLoaderDefinition = $container->register(sprintf('doctrine.orm.%s_entity_manager.validator_loader', $entityManagerName), DoctrineLoader::class);
+        $validatorLoaderDefinition->addArgument(new Reference(sprintf('doctrine.orm.%s_entity_manager', $entityManagerName)));
+
+        $validatorLoaderDefinition->addTag('validator.auto_mapper', ['priority' => -100]);
+    }
+
     /**
      * @param array  $objectManager
      * @param string $cacheName
diff --git a/Tests/ContainerTest.php b/Tests/ContainerTest.php
index 240ac815b..db669f687 100644
--- a/Tests/ContainerTest.php
+++ b/Tests/ContainerTest.php
@@ -19,6 +19,7 @@
 use Symfony\Bridge\Doctrine\Logger\DbalLogger;
 use Symfony\Bridge\Doctrine\PropertyInfo\DoctrineExtractor;
 use Symfony\Bridge\Doctrine\Validator\Constraints\UniqueEntityValidator;
+use Symfony\Bridge\Doctrine\Validator\DoctrineLoader;
 use Symfony\Component\PropertyInfo\PropertyInitializableExtractorInterface;
 
 class ContainerTest extends TestCase
@@ -68,5 +69,11 @@ public function testContainer()
             $this->assertInstanceOf(ClassMetadataFactory::class, $container->get('doctrine.orm.default_entity_manager.metadata_factory'));
         }
         $this->assertInstanceOf(DoctrineExtractor::class, $container->get('doctrine.orm.default_entity_manager.property_info_extractor'));
+
+        if (class_exists(DoctrineLoader::class)) {
+            $this->assertInstanceOf(DoctrineLoader::class, $container->get('doctrine.orm.default_entity_manager.validator_loader'));
+        } else {
+            $this->assertFalse($container->has('doctrine.orm.default_entity_manager.validator_loader'));
+        }
     }
 }