46 lines
1.6 KiB
PHP
46 lines
1.6 KiB
PHP
<?php
|
|
|
|
namespace App\Doctrine;
|
|
|
|
use ApiPlatform\Doctrine\Orm\Extension\QueryCollectionExtensionInterface;
|
|
use ApiPlatform\Doctrine\Orm\Extension\QueryItemExtensionInterface;
|
|
use ApiPlatform\Doctrine\Orm\Util\QueryNameGeneratorInterface;
|
|
use ApiPlatform\Metadata\Operation;
|
|
use App\Entity\User;
|
|
use App\Entity\View;
|
|
use Doctrine\ORM\QueryBuilder;
|
|
use Symfony\Bundle\SecurityBundle\Security;
|
|
|
|
final readonly class UserViewExtension implements QueryCollectionExtensionInterface, QueryItemExtensionInterface
|
|
{
|
|
public function __construct(
|
|
private Security $security,
|
|
)
|
|
{
|
|
}
|
|
|
|
public function applyToCollection(QueryBuilder $queryBuilder, QueryNameGeneratorInterface $queryNameGenerator, string $resourceClass, ?Operation $operation = null, array $context = []): void
|
|
{
|
|
$this->addWhere($queryBuilder, $resourceClass);
|
|
|
|
}
|
|
|
|
public function applyToItem(QueryBuilder $queryBuilder, QueryNameGeneratorInterface $queryNameGenerator, string $resourceClass, array $identifiers, ?Operation $operation = null, array $context = []): void
|
|
{
|
|
$this->addWhere($queryBuilder, $resourceClass);
|
|
}
|
|
|
|
private function addWhere(QueryBuilder $queryBuilder, string $resourceClass): void
|
|
{
|
|
if (View::class !== $resourceClass ) {
|
|
return;
|
|
}
|
|
|
|
/** @var User $user */
|
|
$user = $this->security->getUser();
|
|
|
|
$rootAlias = $queryBuilder->getRootAliases()[0];
|
|
$queryBuilder->andWhere(sprintf('%s.user = :current_user', $rootAlias));
|
|
$queryBuilder->setParameter('current_user', $user->getId());
|
|
}
|
|
} |