vendor/sulu/sulu/src/Sulu/Bundle/PersistenceBundle/DependencyInjection/PersistenceExtensionTrait.php line 93

Open in your IDE?
  1. <?php
  3. /*
  4.  * This file is part of Sulu.
  5.  *
  6.  * (c) Sulu GmbH
  7.  *
  8.  * This source file is subject to the MIT license that is bundled
  9.  * with this source code in the file LICENSE.
  10.  */
  12. namespace Sulu\Bundle\PersistenceBundle\DependencyInjection;
  14. use Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepositoryInterface;
  15. use Symfony\Component\DependencyInjection\ContainerBuilder;
  16. use Symfony\Component\DependencyInjection\Definition;
  17. use Symfony\Component\DependencyInjection\Reference;
  19. /**
  20.  * Define repository services for each object (e.g. sulu.repository.[object name]) and
  21.  * map all object parameters to the container.
  22.  */
  23. trait PersistenceExtensionTrait
  24. {
  25.     /**
  26.      * @param array<string, array{model: class-string, repository?: class-string}> $objects
  27.      *
  28.      * @return void
  29.      */
  30.     protected function configurePersistence(array $objects, ContainerBuilder $container)
  31.     {
  32.         $this->defineRepositories($objects, $container);
  34.         $this->remapObjectParameters($objects, $container);
  36.         $configObjects = ['sulu' => $objects];
  38.         if ($container->hasParameter('sulu.persistence.objects')) {
  39.             /** @var mixed[] $existingConfigObjects */
  40.             $existingConfigObjects = $container->getParameter('sulu.persistence.objects');
  41.             $configObjects = \array_merge_recursive($configObjects, $existingConfigObjects);
  42.         }
  44.         $container->setParameter('sulu.persistence.objects', $configObjects);
  45.     }
  47.     /**
  48.      * @param array<string, array{model: class-string, repository?: class-string}> $objects
  49.      */
  50.     private function defineRepositories(array $objects, ContainerBuilder $container): void
  51.     {
  52.         foreach ($objects as $object => $services) {
  53.             if (\array_key_exists('model', $services)) {
  54.                 $repositoryDefinition = $this->getRepositoryDefinition($object, $services, $container);
  56.                 $container->setDefinition($this->getContainerKey('repository', $object), $repositoryDefinition)
  57.                     ->setPublic(true)
  58.                     ->setLazy(true);
  59.             }
  60.         }
  61.     }
  63.     /**
  64.      * Get the repository service definition.
  65.      *
  66.      * @param string $object
  67.      * @param array{model: class-string, repository?: class-string} $services
  68.      *
  69.      * @return Definition
  70.      */
  71.     private function getRepositoryDefinition($object, array $services, ContainerBuilder $container)
  72.     {
  73.         $repositoryKey = $this->getContainerKey('repository', $object, '.class');
  75.         // default repository
  76.         $repositoryClass = 'Sulu\Component\Persistence\Repository\ORM\EntityRepository';
  78.         if ($container->hasParameter($repositoryKey)) {
  79.             /** @var class-string $repositoryClass */
  80.             $repositoryClass = $container->getParameter($repositoryKey);
  81.         }
  83.         if (isset($services['repository'])) {
  84.             $repositoryClass = $services['repository'];
  85.         }
  87.         $definition = new Definition($repositoryClass);
  88.         $definition->setArguments([
  89.             new Reference($this->getEntityManagerServiceKey()),
  90.             $this->getClassMetadataDefinition($services['model']),
  91.         ]);
  93.         $repositoryReflectionClass = new \ReflectionClass($repositoryClass);
  94.         if (
  95.             $repositoryReflectionClass->hasMethod('setAccessControlQueryEnhancer')
  96.             && !$repositoryReflectionClass->implementsInterface(ServiceEntityRepositoryInterface::class)
  97.         ) {
  98.             $definition->addMethodCall(
  99.                 'setAccessControlQueryEnhancer',
  100.                 [new Reference('sulu_security.access_control_query_enhancer')]
  101.             );
  102.         }
  104.         return $definition;
  105.     }
  107.     /**
  108.      * @param class-string $model
  109.      *
  110.      * @return Definition
  111.      */
  112.     private function getClassMetadataDefinition($model)
  113.     {
  114.         $definition = new Definition('Doctrine\ORM\Mapping\ClassMetadata');
  115.         $definition
  116.             ->setFactory([
  117.                 new Reference($this->getEntityManagerServiceKey()),
  118.                 'getClassMetadata',
  119.             ])
  120.             ->setArguments([$model])
  121.             ->setPublic(false);
  123.         return $definition;
  124.     }
  126.     /**
  127.      * @param array<string, array{model: class-string, repository?: class-string}> $objects
  128.      */
  129.     private function remapObjectParameters(array $objects, ContainerBuilder $container): void
  130.     {
  131.         foreach ($objects as $object => $services) {
  132.             foreach ($services as $service => $class) {
  133.                 $container->setParameter(
  134.                     \sprintf(
  135.                         'sulu.%s.%s.class',
  136.                         $service,
  137.                         $object
  138.                     ),
  139.                     $class
  140.                 );
  141.             }
  142.         }
  143.     }
  145.     /**
  146.      * Get container key.
  147.      *
  148.      * @param string $key
  149.      * @param string $object
  150.      * @param string|null $suffix
  151.      *
  152.      * @return string
  153.      */
  154.     private function getContainerKey($key, $object, $suffix = null)
  155.     {
  156.         return \sprintf('sulu.%s.%s%s', $key, $object, $suffix);
  157.     }
  159.     /**
  160.      * Get the entity manager.
  161.      *
  162.      * @return string
  163.      */
  164.     private function getEntityManagerServiceKey()
  165.     {
  166.         return 'doctrine.orm.default_entity_manager';
  167.     }
  168. }