vendor/sylius/resource-bundle/src/Bundle/EventListener/ORMMappedSuperClassSubscriber.php line 58

Open in your IDE?
  1. <?php
  3. /*
  4.  * This file is part of the Sylius package.
  5.  *
  6.  * (c) Paweł Jędrzejewski
  7.  *
  8.  * For the full copyright and license information, please view the LICENSE
  9.  * file that was distributed with this source code.
  10.  */
  12. declare(strict_types=1);
  14. namespace Sylius\Bundle\ResourceBundle\EventListener;
  16. use Doctrine\ORM\Configuration;
  17. use Doctrine\ORM\Event\LoadClassMetadataEventArgs;
  18. use Doctrine\ORM\Events;
  19. use Doctrine\ORM\Mapping\ClassMetadata;
  20. use Doctrine\ORM\Mapping\ClassMetadataInfo;
  21. use Doctrine\Persistence\Mapping\Driver\MappingDriver;
  22. use Webmozart\Assert\Assert;
  24. final class ORMMappedSuperClassSubscriber extends AbstractDoctrineSubscriber
  25. {
  26.     public function getSubscribedEvents(): array
  27.     {
  28.         return [
  29.             Events::loadClassMetadata,
  30.         ];
  31.     }
  33.     public function loadClassMetadata(LoadClassMetadataEventArgs $eventArgs): void
  34.     {
  35.         $metadata = $eventArgs->getClassMetadata();
  37.         if (!$metadata->isMappedSuperclass) {
  38.             $this->setAssociationMappings($metadata, $eventArgs->getEntityManager()->getConfiguration());
  39.         } else {
  40.             $this->unsetAssociationMappings($metadata);
  41.         }
  42.     }
  44.     private function setAssociationMappings(ClassMetadataInfo $metadata, Configuration $configuration): void
  45.     {
  46.         $class = $metadata->getName();
  47.         if (!class_exists($class)) {
  48.             return;
  49.         }
  51.         /** @psalm-suppress DeprecatedClass */
  52.         $metadataDriver = $configuration->getMetadataDriverImpl();
  53.         Assert::isInstanceOf($metadataDriver, MappingDriver::class);
  55.         $parents = class_parents($class) ?: [];
  57.         foreach ($parents as $parent) {
  58.             if (false === in_array($parent, $metadataDriver->getAllClassNames(), true)) {
  59.                 continue;
  60.             }
  62.             $parentMetadata = new ClassMetadata(
  63.                 $parent,
  64.                 $configuration->getNamingStrategy()
  65.             );
  67.             // Wakeup Reflection
  68.             /** @psalm-suppress ArgumentTypeCoercion */
  69.             $parentMetadata->wakeupReflection($this->getReflectionService());
  71.             // Load Metadata
  72.             $metadataDriver->loadMetadataForClass($parent, $parentMetadata);
  74.             /** @psalm-suppress InvalidArgument */
  75.             if (false === $this->isResource($parentMetadata)) {
  76.                 continue;
  77.             }
  79.             if ($parentMetadata->isMappedSuperclass) {
  80.                 foreach ($parentMetadata->getAssociationMappings() as $key => $value) {
  81.                     if ($this->isRelation($value['type']) && !isset($metadata->associationMappings[$key])) {
  82.                         $metadata->associationMappings[$key] = $value;
  83.                     }
  84.                 }
  85.             }
  86.         }
  87.     }
  89.     private function unsetAssociationMappings(ClassMetadataInfo $metadata): void
  90.     {
  91.         /** @psalm-suppress InvalidArgument */
  92.         if (false === $this->isResource($metadata)) {
  93.             return;
  94.         }
  96.         foreach ($metadata->getAssociationMappings() as $key => $value) {
  97.             if ($this->isRelation($value['type'])) {
  98.                 unset($metadata->associationMappings[$key]);
  99.             }
  100.         }
  101.     }
  103.     private function isRelation(int $type): bool
  104.     {
  105.         return in_array(
  106.             $type,
  107.             [
  108.                 ClassMetadataInfo::MANY_TO_MANY,
  109.                 ClassMetadataInfo::ONE_TO_MANY,
  110.                 ClassMetadataInfo::ONE_TO_ONE,
  111.             ],
  112.             true
  113.         );
  114.     }
  115. }