From 3fff1564a5f55b058c73d50ac40f91c2798cae16 Mon Sep 17 00:00:00 2001
From: Nicolas Grekas <nicolas.grekas@gmail.com>
Date: Tue, 23 Mar 2021 11:49:11 +0100
Subject: [PATCH] Fix dealing with Definition objects in
 SetDoctrineAnnotatedPrefixesPass

---
 .../SetDoctrineAnnotatedPrefixesPass.php      | 36 +++++++++++++++----
 1 file changed, 29 insertions(+), 7 deletions(-)

diff --git a/src/DependencyInjection/CompilerPass/SetDoctrineAnnotatedPrefixesPass.php b/src/DependencyInjection/CompilerPass/SetDoctrineAnnotatedPrefixesPass.php
index ff2f2bba5..f51fc944f 100644
--- a/src/DependencyInjection/CompilerPass/SetDoctrineAnnotatedPrefixesPass.php
+++ b/src/DependencyInjection/CompilerPass/SetDoctrineAnnotatedPrefixesPass.php
@@ -11,8 +11,11 @@
 
 namespace Symfony\Bundle\MakerBundle\DependencyInjection\CompilerPass;
 
+use Doctrine\ORM\Mapping\Driver\AnnotationDriver;
+use Doctrine\Persistence\Mapping\Driver\AnnotationDriver as AbstractAnnotationDriver;
 use Symfony\Component\DependencyInjection\Compiler\CompilerPassInterface;
 use Symfony\Component\DependencyInjection\ContainerBuilder;
+use Symfony\Component\DependencyInjection\Definition;
 use Symfony\Component\DependencyInjection\Reference;
 
 class SetDoctrineAnnotatedPrefixesPass implements CompilerPassInterface
@@ -35,16 +38,35 @@ public function process(ContainerBuilder $container)
             }
 
             $managerName = $m[1];
+            $methodCalls = $metadataDriverImpl->getMethodCalls();
 
-            foreach ($metadataDriverImpl->getMethodCalls() as [$method, $arguments]) {
-                if ('addDriver' === $method) {
-                    $isAnnotated = 'doctrine.orm.'.$managerName.'_annotation_metadata_driver' === (string) $arguments[0];
-                    $annotatedPrefixes[$managerName][] = [
-                        $arguments[1],
-                        $isAnnotated ? new Reference($arguments[0]) : null,
-                    ];
+            foreach ($methodCalls as $i => [$method, $arguments]) {
+                if ('addDriver' !== $method) {
+                    continue;
                 }
+
+                if ($arguments[0] instanceof Definition) {
+                    $class = $arguments[0]->getClass();
+                    $namespace = substr($class, 0, strrpos($class, '\\'));
+
+                    if ('Doctrine\ORM\Mapping\Driver' === $namespace ? AnnotationDriver::class !== $class : !is_subclass_of($class, AbstractAnnotationDriver::class)) {
+                        continue;
+                    }
+
+                    $id = sprintf('.%d_annotation_metadata_driver~%s', $i, ContainerBuilder::hash($arguments));
+                    $container->setDefinition($id, $arguments[0]);
+                    $arguments[0] = new Reference($id);
+                    $methodCalls[$i] = $arguments;
+                }
+
+                $isAnnotated = false !== strpos($arguments[0], '_annotation_metadata_driver');
+                $annotatedPrefixes[$managerName][] = [
+                    $arguments[1],
+                    $isAnnotated ? new Reference($arguments[0]) : null,
+                ];
             }
+
+            $metadataDriverImpl->setMethodCalls($methodCalls);
         }
 
         if (null !== $annotatedPrefixes) {