chill-bundles/src/Bundle/ChillCalendarBundle/Repository/CalendarACLAwareRepository.php

<?php

/**
 * Chill is a software for social workers.
 *
 * For the full copyright and license information, please view
 * the LICENSE file that was distributed with this source code.
 */

declare(strict_types=1);

/*
 * Chill is a software for social workers
 *
 * For the full copyright and license information, please view
 * the LICENSE file that was distributed with this source code.
 */

namespace Chill\CalendarBundle\Repository;

use Chill\CalendarBundle\Entity\Calendar;
use Chill\PersonBundle\Entity\AccompanyingPeriod;
use DateTimeImmutable;
use Doctrine\ORM\EntityManagerInterface;
use Doctrine\ORM\QueryBuilder;

class CalendarACLAwareRepository implements CalendarACLAwareRepositoryInterface
{
    private EntityManagerInterface $em;

    public function __construct(EntityManagerInterface $em)
    {
        $this->em = $em;
    }

    public function buildQueryByAccompanyingPeriod(AccompanyingPeriod $period, ?DateTimeImmutable $startDate, ?DateTimeImmutable $endDate): QueryBuilder
    {
        $qb = $this->em->createQueryBuilder();
        $qb->from(Calendar::class, 'c');

        $andX = $qb->expr()->andX($qb->expr()->eq('c.accompanyingPeriod', ':period'));
        $qb->setParameter('period', $period);

        if (null !== $startDate) {
            $andX->add($qb->expr()->gte('c.startDate', ':startDate'));
            $qb->setParameter('startDate', $startDate);
        }

        if (null !== $endDate) {
            $andX->add($qb->expr()->lte('c.endDate', ':endDate'));
            $qb->setParameter('endDate', $endDate);
        }

        $qb->where($andX);

        return $qb;
    }

    public function countByAccompanyingPeriod(AccompanyingPeriod $period, ?DateTimeImmutable $startDate, ?DateTimeImmutable $endDate): int
    {
        $qb = $this->buildQueryByAccompanyingPeriod($period, $startDate, $endDate)->select('count(c)');

        return $qb->getQuery()->getSingleScalarResult();
    }

    /**
     * @return array|Calendar[]
     */
    public function findByAccompanyingPeriod(AccompanyingPeriod $period, ?DateTimeImmutable $startDate, ?DateTimeImmutable $endDate, ?array $orderBy = [], ?int $offset = null, ?int $limit = null): array
    {
        $qb = $this->buildQueryByAccompanyingPeriod($period, $startDate, $endDate)->select('c');

        foreach ($orderBy as $sort => $order) {
            $qb->addOrderBy('c.' . $sort, $order);
        }

        if (null !== $offset) {
            $qb->setFirstResult($offset);
        }

        if (null !== $limit) {
            $qb->setMaxResults($limit);
        }

        return $qb->getQuery()->getResult();
    }
}