refactor(thumbnail): replace ManagerRegistry with specific repositories

Author: roadiz-ci

src/EntityThumbnail/Provider/DocumentThumbnailProvider.php

@@ -28,13 +28,13 @@ public function supports(string $entityClass, int|string $identifier): bool
         return $this->isClassSupported($entityClass, Document::class);
     }
 
-    public function getDocumentUrl(BaseDocumentInterface $document): ?string
+    public function getDocumentUrl(BaseDocumentInterface $document, int $size = 64): ?string
     {
         $url = $this->documentUrlGenerator
             ->setDocument($document)
             ->setOptions([
-                'width' => 64,
-                'height' => 64,
+                'width' => $size,
+                'height' => $size,
                 'crop' => '1:1',
                 'quality' => 80,
                 'sharpen' => 3,
@@ -70,12 +70,14 @@ public function getThumbnail(string $entityClass, int|string $identifier): ?Enti
             );
         }
 
+        $size = 64;
+
         return new EntityThumbnail(
-            url: $this->getDocumentUrl($document),
+            url: $this->getDocumentUrl($document, $size),
             alt: $document->getAlternativeText() ?? '',
             title: $document->getFilename(),
-            width: 64,
-            height: 64,
+            width: $size,
+            height: $size,
         );
     }
 }

src/EntityThumbnail/Provider/NodeThumbnailProvider.php

@@ -4,10 +4,10 @@
 
 namespace RZ\Roadiz\RozierBundle\EntityThumbnail\Provider;
 
-use Doctrine\Persistence\ManagerRegistry;
 use RZ\Roadiz\Core\AbstractEntities\NodeInterface;
 use RZ\Roadiz\CoreBundle\Entity\Node;
 use RZ\Roadiz\CoreBundle\Entity\NodesSources;
+use RZ\Roadiz\CoreBundle\Repository\NotPublishedNodeRepository;
 use RZ\Roadiz\RozierBundle\EntityThumbnail\AbstractEntityThumbnailProvider;
 use RZ\Roadiz\RozierBundle\EntityThumbnail\EntityThumbnail;
 
@@ -17,7 +17,7 @@
 final readonly class NodeThumbnailProvider extends AbstractEntityThumbnailProvider
 {
     public function __construct(
-        private ManagerRegistry $managerRegistry,
+        private NotPublishedNodeRepository $nodeRepository,
         private NodesSourcesThumbnailProvider $nodesSourcesThumbnailProvider,
     ) {
     }
@@ -35,8 +35,7 @@ public function getThumbnail(string $entityClass, int|string $identifier): ?Enti
             return null;
         }
 
-        $repository = $this->managerRegistry->getRepository($entityClass);
-        $node = $repository->find($identifier);
+        $node = $this->nodeRepository->find($identifier);
 
         if (!$node instanceof Node) {
             return null;

src/EntityThumbnail/Provider/NodesSourcesThumbnailProvider.php

@@ -4,9 +4,9 @@
 
 namespace RZ\Roadiz\RozierBundle\EntityThumbnail\Provider;
 
-use Doctrine\Persistence\ManagerRegistry;
-use RZ\Roadiz\CoreBundle\Entity\Document;
 use RZ\Roadiz\CoreBundle\Entity\NodesSources;
+use RZ\Roadiz\CoreBundle\Repository\DocumentRepository;
+use RZ\Roadiz\CoreBundle\Repository\NotPublishedNodesSourcesRepository;
 use RZ\Roadiz\CoreBundle\Security\Authorization\Voter\NodeVoter;
 use RZ\Roadiz\RozierBundle\EntityThumbnail\AbstractEntityThumbnailProvider;
 use RZ\Roadiz\RozierBundle\EntityThumbnail\EntityThumbnail;
@@ -18,7 +18,8 @@
 final readonly class NodesSourcesThumbnailProvider extends AbstractEntityThumbnailProvider
 {
     public function __construct(
-        private ManagerRegistry $managerRegistry,
+        private NotPublishedNodesSourcesRepository $nodesSourcesRepository,
+        private DocumentRepository $documentRepository,
         private DocumentThumbnailProvider $documentThumbnailProvider,
         private Security $security,
     ) {
@@ -37,8 +38,7 @@ public function getThumbnail(string $entityClass, int|string $identifier): ?Enti
             return null;
         }
 
-        $repository = $this->managerRegistry->getRepository($entityClass);
-        $nodeSource = $repository->find($identifier);
+        $nodeSource = $this->nodesSourcesRepository->find($identifier);
 
         if (!$nodeSource instanceof NodesSources) {
             return null;
@@ -51,23 +51,18 @@ public function getThumbnail(string $entityClass, int|string $identifier): ?Enti
         $title = $nodeSource->getTitle() ?? 'Node';
 
         // Get the first displayable document associated with this NodesSources
-        $documentDto = $this->managerRegistry
-            ->getRepository(Document::class)
-            ->findOneDisplayableDtoByNodeSource($nodeSource);
+        $documentDto = $this->documentRepository->findOneDisplayableDtoByNodeSource($nodeSource);
 
-        if (null === $documentDto || $documentDto->isPrivate()) {
-            return new EntityThumbnail(
-                title: $title,
-            );
-        }
-
-        $url = null;
-        if ($documentDto->isImage() || $documentDto->isSvg()) {
-            $url = $this->documentThumbnailProvider->getDocumentUrl($documentDto);
+        if (
+            null === $documentDto
+            || $documentDto->isPrivate()
+            || !($documentDto->isImage() || $documentDto->isSvg())
+        ) {
+            return null;
         }
 
         return new EntityThumbnail(
-            url: $url,
+            url: $this->documentThumbnailProvider->getDocumentUrl($documentDto),
             alt: $title,
             title: $title,
             width: 64,

src/EntityThumbnail/Provider/UserThumbnailProvider.php

@@ -4,8 +4,8 @@
 
 namespace RZ\Roadiz\RozierBundle\EntityThumbnail\Provider;
 
-use Doctrine\Persistence\ManagerRegistry;
 use RZ\Roadiz\CoreBundle\Entity\User;
+use RZ\Roadiz\CoreBundle\Repository\UserRepository;
 use RZ\Roadiz\RozierBundle\EntityThumbnail\AbstractEntityThumbnailProvider;
 use RZ\Roadiz\RozierBundle\EntityThumbnail\EntityThumbnail;
 
@@ -16,7 +16,7 @@
 final readonly class UserThumbnailProvider extends AbstractEntityThumbnailProvider
 {
     public function __construct(
-        private ManagerRegistry $managerRegistry,
+        private UserRepository $userRepository,
     ) {
     }
 
@@ -33,15 +33,12 @@ public function getThumbnail(string $entityClass, int|string $identifier): ?Enti
             return null;
         }
 
-        // Try to fetch user by ID or email
-        $repository = $this->managerRegistry->getRepository($entityClass);
-
         // If identifier is numeric, try by ID first
         if (is_numeric($identifier)) {
-            $user = $repository->find((int) $identifier);
+            $user = $this->userRepository->find((int) $identifier);
         } else {
             // Otherwise try by email
-            $user = $repository->findOneBy(['email' => $identifier]);
+            $user = $this->userRepository->findOneBy(['email' => $identifier]);
         }
 
         if (!$user instanceof User) {