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

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