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

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\Common\Persistence\Mapping\Driver\MappingDriver;
  17. use Doctrine\ORM\Configuration;
  18. use Doctrine\ORM\Event\LoadClassMetadataEventArgs;
  19. use Doctrine\ORM\Events;
  20. use Doctrine\ORM\Mapping\ClassMetadata;
  21. use Doctrine\ORM\Mapping\ClassMetadataInfo;
  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.         $metadataDriver = $configuration->getMetadataDriverImpl();
  52.         Assert::isInstanceOf($metadataDriver, MappingDriver::class);
  54.         foreach (class_parents($class) as $parent) {
  55.             if (false === in_array($parent, $metadataDriver->getAllClassNames(), true)) {
  56.                 continue;
  57.             }
  59.             $parentMetadata = new ClassMetadata(
  60.                 $parent,
  61.                 $configuration->getNamingStrategy()
  62.             );
  64.             // Wakeup Reflection
  65.             $parentMetadata->wakeupReflection($this->getReflectionService());
  67.             // Load Metadata
  68.             $metadataDriver->loadMetadataForClass($parent, $parentMetadata);
  70.             if (false === $this->isResource($parentMetadata)) {
  71.                 continue;
  72.             }
  74.             if ($parentMetadata->isMappedSuperclass) {
  75.                 foreach ($parentMetadata->getAssociationMappings() as $key => $value) {
  76.                     if ($this->isRelation($value['type']) && !isset($metadata->associationMappings[$key])) {
  77.                         $metadata->associationMappings[$key] = $value;
  78.                     }
  79.                 }
  80.             }
  81.         }
  82.     }
  84.     private function unsetAssociationMappings(ClassMetadataInfo $metadata): void
  85.     {
  86.         if (false === $this->isResource($metadata)) {
  87.             return;
  88.         }
  90.         foreach ($metadata->getAssociationMappings() as $key => $value) {
  91.             if ($this->isRelation($value['type'])) {
  92.                 unset($metadata->associationMappings[$key]);
  93.             }
  94.         }
  95.     }
  97.     private function isRelation(int $type): bool
  98.     {
  99.         return in_array(
  100.             $type,
  101.             [
  102.                 ClassMetadataInfo::MANY_TO_MANY,
  103.                 ClassMetadataInfo::ONE_TO_MANY,
  104.                 ClassMetadataInfo::ONE_TO_ONE,
  105.             ],
  106.             true
  107.         );
  108.     }
  109. }