| 123456789101112131415161718192021222324252627282930313233343536373839404142 |
- <?php
- namespace App\Doctrine\Access;
- use ApiPlatform\Core\Bridge\Doctrine\Orm\Extension\QueryCollectionExtensionInterface;
- use ApiPlatform\Core\Bridge\Doctrine\Orm\Extension\QueryItemExtensionInterface;
- use ApiPlatform\Core\Bridge\Doctrine\Orm\Util\QueryNameGeneratorInterface;
- use App\Entity\Access\Access;
- use Doctrine\ORM\QueryBuilder;
- use Symfony\Component\Security\Core\Security;
- final class AccessExtension implements QueryCollectionExtensionInterface, QueryItemExtensionInterface
- {
- private $security;
- public function __construct(Security $security)
- {
- $this->security = $security;
- }
- public function applyToCollection(QueryBuilder $queryBuilder, QueryNameGeneratorInterface $queryNameGenerator, string $resourceClass, string $operationName = null): void
- {
- $this->addWhere($queryBuilder, $resourceClass);
- }
- public function applyToItem(QueryBuilder $queryBuilder, QueryNameGeneratorInterface $queryNameGenerator, string $resourceClass, array $identifiers, string $operationName = null, array $context = []): void
- {
- $this->addWhere($queryBuilder, $resourceClass);
- }
- private function addWhere(QueryBuilder $queryBuilder, string $resourceClass): void
- {
- if (Access::class !== $resourceClass) {
- return;
- }
- /** @var Access $currentUser */
- $currentUser = $this->security->getUser();
- $rootAlias = $queryBuilder->getRootAliases()[0];
- $queryBuilder->andWhere(sprintf('%s.organization = :current_organization', $rootAlias));
- $queryBuilder->setParameter('current_organization', $currentUser->getOrganization());
- }
- }
|