vendor/vich/uploader-bundle/src/Metadata/Driver/AnnotationDriver.php line 86

  1. <?php
  3. namespace Vich\UploaderBundle\Metadata\Driver;
  5. use Doctrine\Common\Annotations\Reader as AnnotationReader;
  6. use Doctrine\Persistence\ManagerRegistry;
  7. use Metadata\ClassMetadata as JMSClassMetadata;
  8. use Metadata\Driver\AdvancedDriverInterface;
  9. use Vich\UploaderBundle\Mapping\Annotation\Uploadable;
  10. use Vich\UploaderBundle\Mapping\Annotation\UploadableField;
  11. use Vich\UploaderBundle\Metadata\ClassMetadata;
  13. /**
  14.  * @author Kévin Gomez <contact@kevingomez.fr>
  15.  * @author Konstantin Myakshin <koc-dp@yandex.ru>
  16.  */
  17. class AnnotationDriver implements AdvancedDriverInterface
  18. {
  19.     /**
  20.      * @deprecated
  21.      */
  22.     public const UPLOADABLE_ANNOTATION = Uploadable::class;
  24.     /**
  25.      * @deprecated
  26.      */
  27.     public const UPLOADABLE_FIELD_ANNOTATION = UploadableField::class;
  29.     /** @var AnnotationReader|AttributeReader */
  30.     protected $reader;
  32.     /** @var ManagerRegistry[] */
  33.     private $managerRegistryList;
  35.     public function __construct(AnnotationReader $reader, array $managerRegistryList)
  36.     {
  37.         $this->reader = $reader;
  38.         $this->managerRegistryList = $managerRegistryList;
  39.     }
  41.     public function loadMetadataForClass(\ReflectionClass $class): ?JMSClassMetadata
  42.     {
  43.         if (!$this->isUploadable($class)) {
  44.             return null;
  45.         }
  47.         $classMetadata = new ClassMetadata($class->name);
  48.         $classMetadata->fileResources[] = $class->getFileName();
  50.         $classes = [];
  51.         do {
  52.             $classes[] = $class;
  53.             $class = $class->getParentClass();
  54.         } while (false !== $class);
  55.         $classes = \array_reverse($classes, false);
  56.         $properties = [];
  57.         foreach ($classes as $class) {
  58.             $properties = \array_merge($properties, $class->getProperties());
  59.         }
  61.         foreach ($properties as $property) {
  62.             $uploadableField = $this->reader->getPropertyAnnotation($property, UploadableField::class);
  63.             if (null === $uploadableField) {
  64.                 continue;
  65.             }
  66.             /* @var $uploadableField UploadableField */
  67.             // TODO: try automatically determinate target fields if embeddable used
  69.             $fieldMetadata = [
  70.                 'mapping' => $uploadableField->getMapping(),
  71.                 'propertyName' => $property->getName(),
  72.                 'fileNameProperty' => $uploadableField->getFileNameProperty(),
  73.                 'size' => $uploadableField->getSize(),
  74.                 'mimeType' => $uploadableField->getMimeType(),
  75.                 'originalName' => $uploadableField->getOriginalName(),
  76.                 'dimensions' => $uploadableField->getDimensions(),
  77.             ];
  79.             // TODO: store UploadableField object instead of array
  80.             $classMetadata->fields[$property->getName()] = $fieldMetadata;
  81.         }
  83.         return $classMetadata;
  84.     }
  86.     public function getAllClassNames(): array
  87.     {
  88.         $classes = [];
  89.         $metadata = [];
  91.         foreach ($this->managerRegistryList as $managerRegisty) {
  92.             $managers = $managerRegisty->getManagers();
  93.             foreach ($managers as $manager) {
  94.                 $metadata[] = $manager->getMetadataFactory()->getAllMetadata();
  95.             }
  96.         }
  98.         $metadata = \array_merge(...$metadata);
  100.         /** @var \Doctrine\Persistence\Mapping\ClassMetadata $classMeta */
  101.         foreach ($metadata as $classMeta) {
  102.             if ($this->isUploadable(new \ReflectionClass($classMeta->getName()))) {
  103.                 $classes[] = $classMeta->getName();
  104.             }
  105.         }
  107.         return $classes;
  108.     }
  110.     protected function isUploadable(\ReflectionClass $class): bool
  111.     {
  112.         return null !== $this->reader->getClassAnnotation($class, Uploadable::class);
  113.     }
  114. }