ap2i/src/Doctrine/Billing/CurrentResidenceAreaExtension.php

<?php

declare(strict_types=1);

namespace App\Doctrine\Billing;

use ApiPlatform\Metadata\Operation;
use App\Doctrine\AbstractExtension;
use App\Entity\Access\Access;
use App\Entity\Billing\ResidenceArea;
use Doctrine\ORM\QueryBuilder;
use Symfony\Bundle\SecurityBundle\Security;

/**
 * Class CurrentResidenceAreaExtension : Filtre de sécurité par défaut pour une resource ResidenceArea.
 */
final class CurrentResidenceAreaExtension extends AbstractExtension
{
    public function __construct(private Security $security)
    {
    }

    public function supports(string $resourceClass, ?Operation $operation): bool
    {
        return $resourceClass === ResidenceArea::class;
    }

    protected function addWhere(QueryBuilder $queryBuilder, string $resourceClass, ?Operation $operation): void
    {
        /** @var Access $currentUser */
        $currentUser = $this->security->getUser();
        if ($currentUser === null || $currentUser->getOrganization() === null) {
            return;
        }
        $rootAlias = $queryBuilder->getRootAliases()[0];
        $queryBuilder
            ->andWhere(sprintf('%s.billingSetting = :billingSetting', $rootAlias))
            ->setParameter('billingSetting', $currentUser->getOrganization()->getBillingSetting())
        ;
    }
}