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

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.             /** @psalm-suppress InvalidArgument */
  71.             if (false === $this->isResource($parentMetadata)) {
  72.                 continue;
  73.             }
  75.             if ($parentMetadata->isMappedSuperclass) {
  76.                 foreach ($parentMetadata->getAssociationMappings() as $key => $value) {
  77.                     if ($this->isRelation($value['type']) && !isset($metadata->associationMappings[$key])) {
  78.                         $metadata->associationMappings[$key] = $value;
  79.                     }
  80.                 }
  81.             }
  82.         }
  83.     }
  85.     private function unsetAssociationMappings(ClassMetadataInfo $metadata): void
  86.     {
  87.         /** @psalm-suppress InvalidArgument */
  88.         if (false === $this->isResource($metadata)) {
  89.             return;
  90.         }
  92.         foreach ($metadata->getAssociationMappings() as $key => $value) {
  93.             if ($this->isRelation($value['type'])) {
  94.                 unset($metadata->associationMappings[$key]);
  95.             }
  96.         }
  97.     }
  99.     private function isRelation(int $type): bool
  100.     {
  101.         return in_array(
  102.             $type,
  103.             [
  104.                 ClassMetadataInfo::MANY_TO_MANY,
  105.                 ClassMetadataInfo::ONE_TO_MANY,
  106.                 ClassMetadataInfo::ONE_TO_ONE,
  107.             ],
  108.             true
  109.         );
  110.     }
  111. }