Added remove method in DoctrineAdminRepository and load admin repository in Entity from customAction in AdminController

Author: david

src/LIN3S/AdminBundle/Configuration/Model/Entity.php

@@ -29,6 +29,7 @@ final class Entity
     private $listOrderByDefault;
     private $printNames;
     private $repositoryServiceId;
+    private $repository;
 
     public function __construct(
         $name,
@@ -281,13 +282,24 @@ public function printNames()
         return $this->printNames;
     }
 
-    /**
-     * Gets the repository service id.
-     *
-     * @return string
-     */
     public function repositoryServiceId()
     {
         return $this->repositoryServiceId;
     }
+
+    public function repository()
+    {
+        if (!$this->repository instanceof AdminRepository) {
+            throw new \Exception(
+                sprintf('The repository is not loaded yet')
+            );
+        }
+
+        return $this->repository;
+    }
+
+    public function loadRepository(AdminRepository $repository)
+    {
+        $this->repository = $repository;
+    }
 }

src/LIN3S/AdminBundle/Controller/AdminController.php

@@ -57,8 +57,11 @@ public function customAction($entity, $action, $id = null, Request $request)
     {
         $entityConfig = $this->get('lin3s_admin.configuration.factory.entity')->createFor($entity);
         $entityObject = null;
+        $repository = $this->get($entityConfig->repositoryServiceId());
+        $entityConfig->loadRepository($repository);
+
         if ($id) {
-            $entityObject = $this->get($entityConfig->repositoryServiceId())->find($entityConfig, $id);
+            $entityObject = $repository->find($entityConfig, $id);
         }
         if ($id && !$entityObject) {
             throw $this->createNotFoundException(

src/LIN3S/AdminBundle/Repository/AdminRepository.php

@@ -21,6 +21,8 @@ interface AdminRepository
 {
     public function find(Entity $config, $id);
 
+    public function remove($entity);
+
     public function findByRequest(Request $request, Entity $config);
 
     public function countAll(Request $request, Entity $config);

src/LIN3S/AdminBundle/Repository/DoctrineAdminRepository.php

@@ -34,6 +34,12 @@ public function find(Entity $config, $id)
         return $this->manager->find($config->className(), $id);
     }
 
+    public function remove($entity)
+    {
+        $this->manager->remove($entity);
+        $this->manager->flush();
+    }
+
     public function findByRequest(Request $request, Entity $config)
     {
         $queryBuilder = $this->queryBuilder->generate($request, $config);