vendor/api-platform/core/src/Doctrine/Orm/State/ItemProvider.php line 40

  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\Orm\State;
  16. use ApiPlatform\Doctrine\Orm\Extension\QueryItemExtensionInterface;
  17. use ApiPlatform\Doctrine\Orm\Extension\QueryResultItemExtensionInterface;
  18. use ApiPlatform\Doctrine\Orm\Util\QueryNameGenerator;
  19. use ApiPlatform\Exception\RuntimeException;
  20. use ApiPlatform\Metadata\Operation;
  21. use ApiPlatform\Metadata\Resource\Factory\ResourceMetadataCollectionFactoryInterface;
  22. use ApiPlatform\State\ProviderInterface;
  23. use Doctrine\ORM\EntityManagerInterface;
  24. use Doctrine\Persistence\ManagerRegistry;
  25. use Psr\Container\ContainerInterface;
  27. /**
  28.  * Item state provider using the Doctrine ORM.
  29.  *
  30.  * @author Kévin Dunglas <kevin@dunglas.fr>
  31.  * @author Samuel ROZE <samuel.roze@gmail.com>
  32.  */
  33. final class ItemProvider implements ProviderInterface
  34. {
  35.     use LinksHandlerTrait;
  37.     /**
  38.      * @param QueryItemExtensionInterface[] $itemExtensions
  39.      */
  40.     public function __construct(ResourceMetadataCollectionFactoryInterface $resourceMetadataCollectionFactory, private readonly ManagerRegistry $managerRegistry, private readonly iterable $itemExtensions = [], ContainerInterface $handleLinksLocator = null)
  41.     {
  42.         $this->resourceMetadataCollectionFactory = $resourceMetadataCollectionFactory;
  43.         $this->handleLinksLocator = $handleLinksLocator;
  44.     }
  46.     public function provide(Operation $operation, array $uriVariables = [], array $context = []): ?object
  47.     {
  48.         $entityClass = $operation->getClass();
  49.         if (($options = $operation->getStateOptions()) && $options instanceof Options && $options->getEntityClass()) {
  50.             $entityClass = $options->getEntityClass();
  51.         }
  53.         /** @var EntityManagerInterface $manager */
  54.         $manager = $this->managerRegistry->getManagerForClass($entityClass);
  56.         $fetchData = $context['fetch_data'] ?? true;
  57.         if (!$fetchData && \array_key_exists('id', $uriVariables)) {
  58.             // todo : if uriVariables don't contain the id, this fails. This should behave like it does in the following code
  59.             return $manager->getReference($entityClass, $uriVariables);
  60.         }
  62.         $repository = $manager->getRepository($entityClass);
  63.         if (!method_exists($repository, 'createQueryBuilder')) {
  64.             throw new RuntimeException('The repository class must have a "createQueryBuilder" method.');
  65.         }
  67.         $queryBuilder = $repository->createQueryBuilder('o');
  68.         $queryNameGenerator = new QueryNameGenerator();
  70.         if ($handleLinks = $this->getLinksHandler($operation)) {
  71.             $handleLinks($queryBuilder, $uriVariables, $queryNameGenerator, ['entityClass' => $entityClass, 'operation' => $operation] + $context);
  72.         } else {
  73.             $this->handleLinks($queryBuilder, $uriVariables, $queryNameGenerator, $context, $entityClass, $operation);
  74.         }
  76.         foreach ($this->itemExtensions as $extension) {
  77.             $extension->applyToItem($queryBuilder, $queryNameGenerator, $entityClass, $uriVariables, $operation, $context);
  79.             if ($extension instanceof QueryResultItemExtensionInterface && $extension->supportsResult($entityClass, $operation, $context)) {
  80.                 return $extension->getResult($queryBuilder, $entityClass, $operation, $context);
  81.             }
  82.         }
  84.         return $queryBuilder->getQuery()->getOneOrNullResult();
  85.     }
  86. }