src/Doctrine/ExclusionsFilter.php line 38

Open in your IDE?
  1. <?php
  4. namespace App\Doctrine;
  7. use App\Annotation\Exclusions;
  8. use App\Entity\Company;
  9. use App\Entity\User;
  10. use App\Repository\Model\IExclusionRepository;
  11. use Doctrine\Common\Annotations\AnnotationReader;
  12. use Doctrine\ORM\EntityManagerInterface;
  13. use Doctrine\ORM\Mapping\ClassMetadata;
  14. use Doctrine\ORM\Query\Filter\SQLFilter;
  15. use Psr\Log\LoggerInterface;
  16. use Symfony\Component\Serializer\NameConverter\CamelCaseToSnakeCaseNameConverter;
  18. class ExclusionsFilter extends SQLFilter
  19. {
  20.     /**
  21.      * @var User
  22.      */
  23.     private $user;
  25.     /**
  26.      * @var EntityManagerInterface
  27.      */
  28.     private $em;
  30.     /**
  31.      * @var LoggerInterface
  32.      */
  33.     private $logger;
  35.     /**
  36.      * @inheritDoc
  37.      */
  38.     public function addFilterConstraint(ClassMetadata $targetEntity, $targetTableAlias): string
  39.     {
  40.         $result = "";
  41.         $reader = new AnnotationReader();
  42.         $annotation = $reader->getClassAnnotation($targetEntity->getReflectionClass(), Exclusions::class);
  43.         if ($annotation instanceof Exclusions && $this->user instanceof User && $this->user->hasRole(User::ROLE_CLIENT)) {
  44.             // $user = $this->em->getRepository(User::class)->find($this->user->getId());
  45.             $company = $this->user->getClient()->getCompany();
  46.             $repo = $this->em->getRepository($annotation->exclusionsClass);
  47.             if ($repo instanceof IExclusionRepository && $company instanceof Company) {
  48.                 $nameConverter = new CamelCaseToSnakeCaseNameConverter();
  49.                 $name = $nameConverter->normalize($targetEntity->getReflectionClass()->getShortName() . '_id');
  50.                 $class = $nameConverter->normalize((new \ReflectionClass($annotation->exclusionsClass))->getShortName());
  51.                 $result = "$targetTableAlias.id NOT IN((SELECT x0_.$name FROM $class AS x0_ WHERE x0_.company_id = {$company->getId()}))";
  52.             }
  53.         }
  54.         return $result;
  55.     }
  57.     /**
  58.      * @return mixed
  59.      */
  60.     public function getUser()
  61.     {
  62.         return $this->user;
  63.     }
  65.     /**
  66.      * @param mixed $user
  67.      * @return ExclusionsFilter
  68.      */
  69.     public function setUser($user): self
  70.     {
  71.         $this->user = $user;
  72.         return $this;
  73.     }
  75.     /**
  76.      * @param EntityManagerInterface $em
  77.      * @return $this
  78.      */
  79.     public function setEntityManager(EntityManagerInterface $em): self
  80.     {
  81.         $this->em = $em;
  82.         return $this;
  83.     }
  85.     /**
  86.      * @param mixed $logger
  87.      * @return ExclusionsFilter
  88.      */
  89.     public function setLogger($logger): self
  90.     {
  91.         $this->logger = $logger;
  92.         return $this;
  93.     }
  95.     /**
  96.      * @return mixed
  97.      */
  98.     public function getLogger()
  99.     {
  100.         return $this->logger;
  101.     }
  102. }