Add preBatchAction to Admin Extension and add a pre_batch_action event

docs/reference/events.rst

@@ -46,3 +46,10 @@ Block events help you customize your templates. Available events are :
 
 If you want more information about block events, you should check the
 `"Event" section of block bundle documentation <https://docs.sonata-project.org/projects/SonataBlockBundle/en/3.x/reference/events>`_.
+
+BatchActionEvent
+^^^^^^^^^^^^^^^^
+
+This event is dispatched when a batch action is being executed. The event name is:
+
+- ``sonata.admin.event.batch_action.pre_batch_action``

src/Admin/AbstractAdminExtension.php

@@ -96,6 +96,15 @@ public function configureBatchActions(AdminInterface $admin, array $actions): ar
         return $actions;
     }
 
+    // NEXT_MAJOR: Remove the PHPDoc block as the interface will then specify the types
+    /**
+     * @param mixed[] $idx
+     * @phpstan-param AdminInterface<T> $admin
+     */
+    public function preBatchAction(AdminInterface $admin, string $actionName, ProxyQueryInterface $query, array &$idx, bool $allElements): void
+    {
+    }
+
     public function configureExportFields(AdminInterface $admin, array $fields): array
     {
         return $fields;

src/Admin/AdminExtensionInterface.php

@@ -24,6 +24,10 @@
 /**
  * @author Thomas Rabaix <[email protected]>
  *
+ * NEXT_MAJOR: Uncomment the actual definition of below methods inside the interface and remove these annotations.
+ *
+ * @method void preBatchAction(AdminInterface $admin, string $actionName, ProxyQueryInterface $query, array &$idx, bool $allElements)
+ *
  * @phpstan-template T of object
  */
 interface AdminExtensionInterface
@@ -127,6 +131,13 @@ public function getAccessMapping(AdminInterface $admin): array;
      */
     public function configureBatchActions(AdminInterface $admin, array $actions): array;
 
+    // NEXT_MAJOR: Uncomment the method definition
+    ///**
+    // * @param mixed[] $idx
+    // * @phpstan-param AdminInterface<T> $admin
+    // */
+    //public function preBatchAction(AdminInterface $admin, string $actionName, ProxyQueryInterface $query, array &$idx, bool $allElements): void;
+
     /**
      * Get a chance to modify export fields.
      *

src/Controller/CRUDController.php

@@ -469,6 +469,13 @@ public function batchAction(Request $request): Response
         $query->setMaxResults(null);
 
         $this->admin->preBatchAction($action, $query, $idx, $allElements);
+        foreach ($this->admin->getExtensions() as $extension) {
+            // NEXT_MAJOR: Remove the if-statement around the call to `$extension->preBatchAction()`
+            // @phpstan-ignore-next-line
+            if (method_exists($extension, 'preBatchAction')) {
+                $extension->preBatchAction($this->admin, $action, $query, $idx, $allElements);
+            }
+        }
 
         if (\count($idx) > 0) {
             $this->admin->getModelManager()->addIdentifiersToQuery($this->admin->getClass(), $query, $idx);

src/Event/AdminEventExtension.php

@@ -126,4 +126,12 @@ public function postRemove(AdminInterface $admin, object $object): void
             'sonata.admin.event.persistence.post_remove'
         );
     }
+
+    public function preBatchAction(AdminInterface $admin, string $actionName, ProxyQueryInterface $query, array &$idx, bool $allElements): void
+    {
+        $this->eventDispatcher->dispatch(
+            new BatchActionEvent($admin, BatchActionEvent::TYPE_PRE_BATCH_ACTION, $actionName, $query, $idx, $allElements),
+            'sonata.admin.event.batch_action.pre_batch_action'
+        );
+    }
 }

src/Event/BatchActionEvent.php

@@ -0,0 +1,120 @@
+<?php
+
+declare(strict_types=1);
+
+/*
+ * This file is part of the Sonata Project package.
+ *
+ * (c) Thomas Rabaix <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Sonata\AdminBundle\Event;
+
+use Sonata\AdminBundle\Admin\AdminInterface;
+use Sonata\AdminBundle\Datagrid\ProxyQueryInterface;
+use Symfony\Contracts\EventDispatcher\Event;
+
+/**
+ * This event is sent by hook:
+ *   - preBatchAction.
+ *
+ * You can register the listener to the event dispatcher by using:
+ *   - sonata.admin.event.batch_action.pre_batch_action)
+ *
+ * @author Jochem Klaver <[email protected]>
+ *
+ * @phpstan-template T of object
+ */
+final class BatchActionEvent extends Event
+{
+    public const TYPE_PRE_BATCH_ACTION = 'pre_batch_action';
+
+    /**
+     * @var AdminInterface<object>
+     * @phpstan-var AdminInterface<T>
+     */
+    private $admin;
+
+    /**
+     * @var string
+     * @phpstan-var self::TYPE_*
+     */
+    private $type;
+
+    /**
+     * @var string
+     */
+    private $actionName;
+
+    /**
+     * @var ProxyQueryInterface
+     */
+    private $proxyQuery;
+
+    /**
+     * @var mixed[]
+     */
+    private $idx;
+
+    /**
+     * @var bool
+     */
+    private $allElements;
+
+    /**
+     * @param mixed[] $idx
+     * @phpstan-param AdminInterface<T> $admin
+     * @phpstan-param self::TYPE_* $type
+     */
+    public function __construct(AdminInterface $admin, string $type, string $actionName, ProxyQueryInterface $proxyQuery, array &$idx, bool $allElements)
+    {
+        $this->admin = $admin;
+        $this->type = $type;
+        $this->actionName = $actionName;
+        $this->proxyQuery = $proxyQuery;
+        $this->idx = &$idx;
+        $this->allElements = $allElements;
+    }
+
+    /**
+     * @phpstan-return AdminInterface<T>
+     */
+    public function getAdmin(): AdminInterface
+    {
+        return $this->admin;
+    }
+
+    /**
+     * @phpstan-return self::TYPE_*
+     */
+    public function getType(): string
+    {
+        return $this->type;
+    }
+
+    public function getActionName(): string
+    {
+        return $this->actionName;
+    }
+
+    public function getProxyQuery(): ProxyQueryInterface
+    {
+        return $this->proxyQuery;
+    }
+
+    /**
+     * @return mixed[]
+     */
+    public function &getIdx(): array
+    {
+        return $this->idx;
+    }
+
+    public function isAllElements(): bool
+    {
+        return $this->allElements;
+    }
+}

tests/Event/AdminEventExtensionTest.php

@@ -24,6 +24,7 @@
 use Sonata\AdminBundle\Datagrid\ListMapper;
 use Sonata\AdminBundle\Datagrid\ProxyQueryInterface;
 use Sonata\AdminBundle\Event\AdminEventExtension;
+use Sonata\AdminBundle\Event\BatchActionEvent;
 use Sonata\AdminBundle\Event\ConfigureEvent;
 use Sonata\AdminBundle\Event\ConfigureQueryEvent;
 use Sonata\AdminBundle\Event\PersistenceEvent;
@@ -189,4 +190,38 @@ public function testPostRemove(): void
             static::equalTo('sonata.admin.event.persistence.post_remove'),
         ])->postRemove($this->createMock(AdminInterface::class), new \stdClass());
     }
+
+    public function testPreBatchAction(): void
+    {
+        $admin = $this->createMock(AdminInterface::class);
+        $proxyQuery = $this->createMock(ProxyQueryInterface::class);
+        $idx = [1, 2, 3];
+
+        $this->getExtension([
+            static::callback(
+                static function (Event $event) use (&$idx): bool {
+                    if (!$event instanceof BatchActionEvent) {
+                        return false;
+                    }
+
+                    // @phpstan-ignore-next-line
+                    if (BatchActionEvent::TYPE_PRE_BATCH_ACTION !== $event->getType()) {
+                        return false;
+                    }
+
+                    if ('delete' !== $event->getActionName()) {
+                        return false;
+                    }
+
+                    $idx[] = 4; // Test if this was passed by reference correctly everywhere
+                    if ($event->getIdx() !== $idx) {
+                        return false;
+                    }
+
+                    return true;
+                }
+            ),
+            static::equalTo('sonata.admin.event.batch_action.pre_batch_action'),
+        ])->preBatchAction($admin, 'delete', $proxyQuery, $idx, false);
+    }
 }