Fix declaring a repository when it is a service entity repository

Sylius · Nov 29, 2022 · a6c4c59 · a6c4c59
1 parent aa1fdd6
commit a6c4c59
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/Bundle/DependencyInjection/Driver/Doctrine/DoctrineORMDriver.php b/src/Bundle/DependencyInjection/Driver/Doctrine/DoctrineORMDriver.php
@@ -13,7 +13,9 @@
 
 namespace Sylius\Bundle\ResourceBundle\DependencyInjection\Driver\Doctrine;
 
+use App\Repository\BookRepository;
 use Doctrine\Bundle\DoctrineBundle\DependencyInjection\Compiler\ServiceRepositoryCompilerPass;
+use Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository;
 use Doctrine\Common\Persistence\ObjectManager as DeprecatedObjectManager;
 use Doctrine\ORM\EntityManagerInterface;
 use Doctrine\ORM\Mapping\ClassMetadata;
@@ -56,7 +58,6 @@ protected function addRepository(ContainerBuilder $container, MetadataInterface
         $managerReference = new Reference($metadata->getServiceId('manager'));
         $definition = new Definition($repositoryClass);
         $definition->setPublic(true);
-
         if ($repositoryClass === EntityRepository::class) {
             /** @var string $entityClass */
             $entityClass = $metadata->getClass('model');
@@ -68,7 +69,12 @@ protected function addRepository(ContainerBuilder $container, MetadataInterface
 
             $genericEntities[] = $entityClass;
         } else {
-            $definition->setArguments([$managerReference, $this->getClassMetadataDefinition($metadata)]);
+            if (is_a($repositoryClass, ServiceEntityRepository::class, true)) {
+                $definition->setArguments([new Reference('doctrine')]);
+                $container->setDefinition($serviceId, $definition);
+            } else  {
+                $definition->setArguments([$managerReference, $this->getClassMetadataDefinition($metadata)]);
+            }
 
             $container->setDefinition($serviceId, $definition);