vendor/api-platform/core/src/Doctrine/Odm/State/CollectionProvider.php line 41

  1. <?php
  3. /*
  4.  * This file is part of the API Platform project.
  5.  *
  6.  * (c) Kévin Dunglas <dunglas@gmail.com>
  7.  *
  8.  * For the full copyright and license information, please view the LICENSE
  9.  * file that was distributed with this source code.
  10.  */
  12. declare(strict_types=1);
  14. namespace ApiPlatform\Doctrine\Odm\State;
  16. use ApiPlatform\Doctrine\Odm\Extension\AggregationCollectionExtensionInterface;
  17. use ApiPlatform\Doctrine\Odm\Extension\AggregationResultCollectionExtensionInterface;
  18. use ApiPlatform\Exception\RuntimeException;
  19. use ApiPlatform\Metadata\Operation;
  20. use ApiPlatform\Metadata\Resource\Factory\ResourceMetadataCollectionFactoryInterface;
  21. use ApiPlatform\State\ProviderInterface;
  22. use Doctrine\ODM\MongoDB\DocumentManager;
  23. use Doctrine\ODM\MongoDB\Repository\DocumentRepository;
  24. use Doctrine\Persistence\ManagerRegistry;
  26. /**
  27.  * Collection state provider using the Doctrine ODM.
  28.  */
  29. final class CollectionProvider implements ProviderInterface
  30. {
  31.     use LinksHandlerTrait;
  33.     /**
  34.      * @param AggregationCollectionExtensionInterface[] $collectionExtensions
  35.      */
  36.     public function __construct(ResourceMetadataCollectionFactoryInterface $resourceMetadataCollectionFactory, private readonly ManagerRegistry $managerRegistry, private readonly iterable $collectionExtensions = [])
  37.     {
  38.         $this->resourceMetadataCollectionFactory = $resourceMetadataCollectionFactory;
  39.     }
  41.     public function provide(Operation $operation, array $uriVariables = [], array $context = []): iterable
  42.     {
  43.         $resourceClass = $operation->getClass();
  44.         /** @var DocumentManager $manager */
  45.         $manager = $this->managerRegistry->getManagerForClass($resourceClass);
  47.         $repository = $manager->getRepository($resourceClass);
  48.         if (!$repository instanceof DocumentRepository) {
  49.             throw new RuntimeException(sprintf('The repository for "%s" must be an instance of "%s".', $resourceClass, DocumentRepository::class));
  50.         }
  52.         $aggregationBuilder = $repository->createAggregationBuilder();
  54.         $this->handleLinks($aggregationBuilder, $uriVariables, $context, $resourceClass, $operation);
  56.         foreach ($this->collectionExtensions as $extension) {
  57.             $extension->applyToCollection($aggregationBuilder, $resourceClass, $operation, $context);
  59.             if ($extension instanceof AggregationResultCollectionExtensionInterface && $extension->supportsResult($resourceClass, $operation, $context)) {
  60.                 return $extension->getResult($aggregationBuilder, $resourceClass, $operation, $context);
  61.             }
  62.         }
  64.         $attribute = $operation->getExtraProperties()['doctrine_mongodb'] ?? [];
  65.         $executeOptions = $attribute['execute_options'] ?? [];
  67.         return $aggregationBuilder->hydrate($resourceClass)->execute($executeOptions);
  68.     }
  69. }