src/Repository/ExclusionsVideoRepository.php line 23

Open in your IDE?
  1. <?php
  3. namespace App\Repository;
  5. use App\Entity\Company;
  6. use App\Entity\ExclusionsVideo;
  7. use App\Entity\Video;
  8. use App\Repository\Model\IExclusionRepository;
  9. use Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository;
  10. use Doctrine\ORM\NonUniqueResultException;
  11. use Doctrine\Persistence\ManagerRegistry;
  13. /**
  14.  * @method ExclusionsVideo|null find($id, $lockMode = null, $lockVersion = null)
  15.  * @method ExclusionsVideo|null findOneBy(array $criteria, array $orderBy = null)
  16.  * @method ExclusionsVideo[]    findAll()
  17.  * @method ExclusionsVideo[]    findBy(array $criteria, array $orderBy = null, $limit = null, $offset = null)
  18.  */
  19. class ExclusionsVideoRepository extends ServiceEntityRepository implements IExclusionRepository
  20. {
  21.     public function __construct(ManagerRegistry $registry)
  22.     {
  23.         parent::__construct($registry, ExclusionsVideo::class);
  24.     }
  26.     /**
  27.      * Checks if an exclusion for said company exists
  28.      * @param Company $company
  29.      * @param Video $video
  30.      * @return bool
  31.      */
  32.     public function isExclusion(Company $company, Video $video): bool
  33.     {
  34.         try {
  35.             $result = $this->createQueryBuilder('e')
  36.                 ->andWhere('e.company = :company')
  37.                 ->andWhere('e.video = :video')
  38.                 ->setParameter('company', $company)
  39.                 ->setParameter('video', $video)
  40.                 ->getQuery()
  41.                 ->getOneOrNullResult();
  42.         } catch (NonUniqueResultException $e) {
  43.             $result = null;
  44.         }
  45.         return ($result instanceof ExclusionsVideo);
  46.     }
  48.     /**
  49.      * @param Company $company
  50.      * @return int|mixed|string|null
  51.      */
  52.     public function getExcludedIds(Company $company): array
  53.     {
  54.         $result = $this->createQueryBuilder('e')
  55.             ->select('e')
  56.             ->andWhere('e.company = :company')
  57.             ->setParameter('company', $company)
  58.             ->getQuery()
  59.             ->getResult();
  60.         foreach ($result as $key=>$item){
  61.             $result[$key] = $item->getVideo()->getId();
  62.         }
  63.         return $result;
  64.     }
  65. }