vendor/vich/uploader-bundle/src/Metadata/Driver/AttributeDriver.php line 93

Open in your IDE?
  1. <?php
  3. namespace Vich\UploaderBundle\Metadata\Driver;
  5. use Metadata\ClassMetadata as JMSClassMetadata;
  6. use Metadata\Driver\AdvancedDriverInterface;
  7. use Vich\UploaderBundle\Mapping\Annotation\Uploadable as UploadableAnnotation;
  8. use Vich\UploaderBundle\Mapping\Annotation\UploadableField as UploadableFieldAnnotation;
  9. use Vich\UploaderBundle\Mapping\Attribute\Uploadable;
  10. use Vich\UploaderBundle\Mapping\Attribute\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 AttributeDriver implements AdvancedDriverInterface
  18. {
  19.     /**
  20.      * @param \Doctrine\Persistence\ManagerRegistry[] $managerRegistryList
  21.      */
  22.     public function __construct(
  23.         protected readonly AttributeReader $reader,
  24.         private readonly array $managerRegistryList
  25.     ) {
  26.     }
  28.     public function loadMetadataForClass(\ReflectionClass $class): ?JMSClassMetadata
  29.     {
  30.         if (!$this->isUploadable($class)) {
  31.             return null;
  32.         }
  34.         $classMetadata = new ClassMetadata($class->name);
  35.         $classMetadata->fileResources[] = $class->getFileName();
  37.         $classes = [];
  38.         do {
  39.             $classes[] = $class;
  40.             $class = $class->getParentClass();
  41.         } while (false !== $class);
  42.         $classes = \array_reverse($classes);
  43.         $properties = [];
  44.         foreach ($classes as $cls) {
  45.             $properties = [...$properties, ...$cls->getProperties()];
  46.         }
  48.         foreach ($properties as $property) {
  49.             // Support both new Attribute\ and deprecated Annotation\ namespaces
  50.             $uploadableField = $this->reader->getPropertyAttribute($property, UploadableField::class);
  51.             if (null === $uploadableField) {
  52.                 // Fallback to deprecated Annotation namespace
  53.                 $uploadableField = $this->reader->getPropertyAttribute($property, UploadableFieldAnnotation::class);
  54.             }
  55.             if (!$uploadableField instanceof UploadableField && !$uploadableField instanceof UploadableFieldAnnotation) {
  56.                 continue;
  57.             }
  58.             // TODO: try automatically determinate target fields if embeddable used
  60.             $fieldMetadata = [
  61.                 'mapping' => $uploadableField->getMapping(),
  62.                 'propertyName' => $property->getName(),
  63.                 'fileNameProperty' => $uploadableField->getFileNameProperty(),
  64.                 'size' => $uploadableField->getSize(),
  65.                 'mimeType' => $uploadableField->getMimeType(),
  66.                 'originalName' => $uploadableField->getOriginalName(),
  67.                 'dimensions' => $uploadableField->getDimensions(),
  68.             ];
  70.             // TODO: store UploadableField object instead of array
  71.             $classMetadata->fields[$property->getName()] = $fieldMetadata;
  72.         }
  74.         return $classMetadata;
  75.     }
  77.     public function getAllClassNames(): array
  78.     {
  79.         $classes = [];
  80.         $metadata = [];
  82.         foreach ($this->managerRegistryList as $managerRegistry) {
  83.             $managers = $managerRegistry->getManagers();
  84.             foreach ($managers as $manager) {
  85.                 $metadata[] = $manager->getMetadataFactory()->getAllMetadata();
  86.             }
  87.         }
  89.         $metadata = \array_merge(...$metadata);
  91.         /** @var \Doctrine\Persistence\Mapping\ClassMetadata $classMeta */
  92.         foreach ($metadata as $classMeta) {
  93.             if ($this->isUploadable(new \ReflectionClass($classMeta->getName()))) {
  94.                 $classes[] = $classMeta->getName();
  95.             }
  96.         }
  98.         return $classes;
  99.     }
  101.     protected function isUploadable(\ReflectionClass $class): bool
  102.     {
  103.         // Support both new Attribute\ and deprecated Annotation\ namespaces
  104.         $uploadable = $this->reader->getClassAttribute($class, Uploadable::class);
  105.         if (null === $uploadable) {
  106.             // Fallback to deprecated Annotation namespace
  107.             $uploadable = $this->reader->getClassAttribute($class, UploadableAnnotation::class);
  108.         }
  110.         return null !== $uploadable;
  111.     }
  112. }