feat(AbstractAdminController): Made AbstractAdminController getReposi…

…tory method overridable
roadiz · Sep 14, 2023 · a46157d · a46157d
1 parent 8624ec4
commit a46157d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 7 deletions.
diff --git a/lib/Rozier/src/Controllers/AbstractAdminController.php b/lib/Rozier/src/Controllers/AbstractAdminController.php
@@ -4,6 +4,7 @@
 
 namespace Themes\Rozier\Controllers;
 
+use Doctrine\Persistence\ObjectRepository;
 use JMS\Serializer\SerializationContext;
 use JMS\Serializer\SerializerInterface;
 use RZ\Roadiz\Core\AbstractEntities\PersistableInterface;
@@ -62,6 +63,11 @@ protected function prepareWorkingItem(PersistableInterface $item): void
         // Add or modify current working item.
     }
 
+    protected function getRepository(): ObjectRepository
+    {
+        return $this->em()->getRepository($this->getEntityClass());
+    }
+
     /**
      * @param Request $request
      * @return Response|null
@@ -214,7 +220,7 @@ public function exportAction(Request $request): JsonResponse
         $this->denyAccessUnlessGranted($this->getRequiredRole());
         $this->additionalAssignation($request);
 
-        $items = $this->em()->getRepository($this->getEntityClass())->findAll();
+        $items = $this->getRepository()->findAll();
 
         return new JsonResponse(
             $this->serializer->serialize(

diff --git a/lib/Rozier/src/Controllers/AbstractAdminWithBulkController.php b/lib/Rozier/src/Controllers/AbstractAdminWithBulkController.php
@@ -105,7 +105,7 @@ public function bulkDeleteAction(Request $request): Response
             if (count($ids) < 1) {
                 $bulkDeleteForm->addError(new FormError('No item selected.'));
             } else {
-                $items = $this->em()->getRepository($this->getEntityClass())->findBy([
+                $items = $this->getRepository()->findBy([
                     'id' => $ids,
                 ]);
                 $deleteForm = $this->createDeleteBulkForm(false, [
@@ -123,7 +123,7 @@ public function bulkDeleteAction(Request $request): Response
                 $deleteForm->addError(new FormError('No item selected.'));
             } else {
                 /** @var PersistableInterface[] $items */
-                $items = $this->em()->getRepository($this->getEntityClass())->findBy([
+                $items = $this->getRepository()->findBy([
                     'id' => $ids,
                 ]);
                 foreach ($items as $item) {
@@ -166,7 +166,7 @@ public function bulkPublishAction(Request $request): Response
             if (count($ids) < 1) {
                 $bulkPublishForm->addError(new FormError('No item selected.'));
             } else {
-                $items = $this->em()->getRepository($this->getEntityClass())->findBy([
+                $items = $this->getRepository()->findBy([
                     'id' => $ids,
                 ]);
                 $publishForm = $this->createPublishBulkForm(false, [
@@ -184,7 +184,7 @@ public function bulkPublishAction(Request $request): Response
                 $publishForm->addError(new FormError('No item selected.'));
             } else {
                 /** @var PersistableInterface[] $items */
-                $items = $this->em()->getRepository($this->getEntityClass())->findBy([
+                $items = $this->getRepository()->findBy([
                     'id' => $ids,
                 ]);
                 foreach ($items as $item) {
@@ -229,7 +229,7 @@ public function bulkUnpublishAction(Request $request): Response
             if (count($ids) < 1) {
                 $bulkUnpublishForm->addError(new FormError('No item selected.'));
             } else {
-                $items = $this->em()->getRepository($this->getEntityClass())->findBy([
+                $items = $this->getRepository()->findBy([
                     'id' => $ids,
                 ]);
                 $unpublishForm = $this->createUnpublishBulkForm(false, [
@@ -247,7 +247,7 @@ public function bulkUnpublishAction(Request $request): Response
                 $unpublishForm->addError(new FormError('No item selected.'));
             } else {
                 /** @var PersistableInterface[] $items */
-                $items = $this->em()->getRepository($this->getEntityClass())->findBy([
+                $items = $this->getRepository()->findBy([
                     'id' => $ids,
                 ]);
                 foreach ($items as $item) {