Added collection to Criteria event

Author: TomHAnderson

docs/events.rst

@@ -109,11 +109,13 @@ the association if you assigned an event name in the attributes.
       },
   );
 
-The ``Criteria`` event has one function in addition to getters for
+The ``Criteria`` event has two functions in addition to getters for
 all resolve parameters:
 
 * ``getCriteria`` - Will return a Criteria object with the user specified
   filters already applied.
+* ``getCollection`` - Will return the unfetched collection object.  This is useful
+  if you need to fetch the collection to apply additional criteria.
 
 
 Modify an Entity Definition

src/Event/Criteria.php

@@ -4,7 +4,9 @@
 
 namespace ApiSkeletons\Doctrine\ORM\GraphQL\Event;
 
+use Doctrine\Common\Collections\Collection;
 use Doctrine\Common\Collections\Criteria as DoctrineCriteria;
+use Doctrine\ORM\PersistentCollection;
 use GraphQL\Type\Definition\ResolveInfo;
 use League\Event\HasEventName;
 
@@ -15,10 +17,14 @@
 class Criteria implements
     HasEventName
 {
-    /** @param mixed[] $args */
+    /**
+     * @param PersistentCollection<array-key, mixed> $collection
+     * @param mixed[]                                $args
+     */
     public function __construct(
         protected readonly DoctrineCriteria $criteria,
         protected readonly string $eventName,
+        protected readonly Collection $collection,
         protected readonly mixed $objectValue,
         protected readonly array $args,
         protected readonly mixed $context,
@@ -36,6 +42,12 @@ public function getCriteria(): DoctrineCriteria
         return $this->criteria;
     }
 
+    /** @return PersistentCollection<array-key, mixed> */
+    public function getCollection(): Collection
+    {
+        return $this->collection;
+    }
+
     public function getObjectValue(): mixed
     {
         return $this->objectValue;

src/Resolve/ResolveCollectionFactory.php

@@ -168,6 +168,7 @@ protected function buildPagination(
                 new CriteriaEvent(
                     $criteria,
                     $criteriaEventName,
+                    $collection,
                     ...$resolve,
                 ),
             );

test/Feature/Event/CriteriaTest.php

@@ -9,6 +9,7 @@
 use ApiSkeletons\Doctrine\ORM\GraphQL\Event\Criteria as CriteriaEvent;
 use ApiSkeletonsTest\Doctrine\ORM\GraphQL\AbstractTest;
 use ApiSkeletonsTest\Doctrine\ORM\GraphQL\Entity\Artist;
+use Doctrine\Common\Collections\Collection;
 use Doctrine\Common\Collections\Criteria;
 use GraphQL\GraphQL;
 use GraphQL\Type\Definition\ObjectType;
@@ -33,6 +34,7 @@ function (CriteriaEvent $event): void {
                     $event->getCriteria()->expr()->eq('venue', 'Delta Center'),
                 );
 
+                $this->assertInstanceOf(Collection::class, $event->getCollection());
                 $this->assertInstanceOf(Artist::class, $event->getObjectValue());
                 $this->assertEquals('contextTest', $event->getContext());
                 $this->assertIsArray($event->getArgs());