AccessExtension.php 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. <?php
  2. namespace App\Doctrine\Access;
  3. use ApiPlatform\Core\Bridge\Doctrine\Orm\Extension\QueryCollectionExtensionInterface;
  4. use ApiPlatform\Core\Bridge\Doctrine\Orm\Extension\QueryItemExtensionInterface;
  5. use ApiPlatform\Core\Bridge\Doctrine\Orm\Util\QueryNameGeneratorInterface;
  6. use App\Entity\Access\Access;
  7. use Doctrine\ORM\QueryBuilder;
  8. use Symfony\Component\Security\Core\Security;
  9. final class AccessExtension implements QueryCollectionExtensionInterface, QueryItemExtensionInterface
  10. {
  11. private $security;
  12. public function __construct(Security $security)
  13. {
  14. $this->security = $security;
  15. }
  16. public function applyToCollection(QueryBuilder $queryBuilder, QueryNameGeneratorInterface $queryNameGenerator, string $resourceClass, string $operationName = null): void
  17. {
  18. $this->addWhere($queryBuilder, $resourceClass);
  19. }
  20. public function applyToItem(QueryBuilder $queryBuilder, QueryNameGeneratorInterface $queryNameGenerator, string $resourceClass, array $identifiers, string $operationName = null, array $context = []): void
  21. {
  22. $this->addWhere($queryBuilder, $resourceClass);
  23. }
  24. private function addWhere(QueryBuilder $queryBuilder, string $resourceClass): void
  25. {
  26. if (Access::class !== $resourceClass) {
  27. return;
  28. }
  29. /** @var Access $currentUser */
  30. $currentUser = $this->security->getUser();
  31. $rootAlias = $queryBuilder->getRootAliases()[0];
  32. $queryBuilder->andWhere(sprintf('%s.organization = :current_organization', $rootAlias));
  33. $queryBuilder->setParameter('current_organization', $currentUser->getOrganization());
  34. }
  35. }