vendor/mgilet/notification-bundle/NotifiableDiscovery.php line 76

Open in your IDE?
  1. <?php
  3. namespace Mgilet\NotificationBundle;
  5. use Doctrine\Common\Annotations\Reader;
  6. use Doctrine\Common\Util\ClassUtils;
  7. use Doctrine\ORM\EntityManager;
  8. use Metadata\ClassMetadata;
  10. class NotifiableDiscovery
  11. {
  12.     /**
  13.      * @var Reader
  14.      */
  15.     private $annotationReader;
  17.     /**
  18.      * @var EntityManager
  19.      */
  20.     private $em;
  22.     /**
  23.      * @var array
  24.      */
  25.     private $notifiables = [];
  28.     /**
  29.      * WorkerDiscovery constructor.
  30.      *
  31.      * @param EntityManager $em
  32.      * @param Reader        $annotationReader
  33.      *
  34.      * @throws \InvalidArgumentException
  35.      */
  36.     public function __construct(EntityManager $em, Reader $annotationReader)
  37.     {
  38.         $this->annotationReader = $annotationReader;
  39.         $this->em = $em;
  40.         $this->discoverNotifiables();
  41.     }
  43.     /**
  44.      * Returns all the workers
  45.      * @throws \InvalidArgumentException
  46.      */
  47.     public function getNotifiables()
  48.     {
  49.         return $this->notifiables;
  50.     }
  52.     /**
  53.      * @param NotifiableInterface $notifiable
  54.      *
  55.      * @return string|null
  56.      */
  57.     public function getNotifiableName(NotifiableInterface $notifiable)
  58.     {
  59.         // fixes the case when the notifiable is a proxy
  60.         $class = ClassUtils::getRealClass(get_class($notifiable));
  61.         $annotation = $this->annotationReader->getClassAnnotation(new \ReflectionClass($class), 'Mgilet\NotificationBundle\Annotation\Notifiable');
  62.         if ($annotation) {
  63.             return $annotation->getName();
  64.         }
  66.         return null;
  67.     }
  69.     /**
  70.      * Discovers workers
  71.      * @throws \InvalidArgumentException
  72.      */
  73.     private function discoverNotifiables()
  74.     {
  75.         /** @var ClassMetadata[] $entities */
  76.         $entities = $this->em->getMetadataFactory()->getAllMetadata();
  77.         foreach ($entities as $entity) {
  78.             $class = $entity->name;
  79.             $annotation = $this->annotationReader->getClassAnnotation(new \ReflectionClass($class), 'Mgilet\NotificationBundle\Annotation\Notifiable');
  80.             if ($annotation) {
  81.                 $this->notifiables[$annotation->getName()] = [
  82.                     'class' => $entity->name,
  83.                     'annotation' => $annotation,
  84.                     'identifiers' => $entity->getIdentifier()
  85.                 ];
  86.             }
  87.         }
  88.     }
  89. }