Pfilsx\DataGrid\Grid\Providers\RepositoryDataProvider

Complexity

Total Complexity

17

Size/Duplication

Total Lines	126
Duplicated Lines	0 %

Importance

Changes	4
Bugs	0	Features	0

14 Methods

Rating	Name	Duplication	Size	Complexity
A	gteDate()	0	5	1
A	gtDate()	0	5	1
A	addCustomFilter()	0	5	1
A	equalDate()	0	7	1
A	lteDate()	0	5	1
A	ltDate()	0	5	1
A	getCriteria()	0	3	1
A	setSort()	0	4	1
A	addEqualFilter()	0	8	2
A	addLikeFilter()	0	4	1
A	getTotalCount()	0	3	1
A	__construct()	0	4	1
A	addRelationFilter()	0	10	2
A	getItems()	0	14	2

<?php


namespace Pfilsx\DataGrid\Grid\Providers;


use DateTime;
use Doctrine\Common\Collections\Criteria;
use Doctrine\ORM\EntityRepository;
use Pfilsx\DataGrid\Grid\Items\EntityGridItem;
use Symfony\Bridge\Doctrine\ManagerRegistry;

class RepositoryDataProvider extends DataProvider
{

    /**
     * @var EntityRepository
     */
    protected $repository;

    /**
     * @var Criteria
     */
    protected $criteria;


    public function __construct(EntityRepository $repository, ManagerRegistry $manager)
    {
        $this->repository = $repository;
        parent::__construct($manager);
    }


    public function getItems(): array
    {
        $this->getCriteria()
            ->setMaxResults($this->getPager()->getLimit())
            ->setFirstResult($this->getPager()->getFirst());
        $result = $this->repository->matching($this->getCriteria())->toArray();
        if (!empty($result)){
            $identifier = $this->getEntityIdentifier(get_class($result[0]));
            return array_map(function ($entity) use ($identifier) {
                $item = new EntityGridItem($entity, $identifier);
                return $item;
            }, $result);
        }
        return $result;
    }

    public function getTotalCount(): int
    {
        return $this->repository->matching($this->getCriteria())->count();
    }

    /**
     * @return Criteria
     */
    public function getCriteria()
    {
        return $this->criteria ?? ($this->criteria = Criteria::create());
    }

    public function setSort(array $sort): DataProviderInterface
    {
        $this->getCriteria()->orderBy($sort);
        return $this;
    }

    public function addEqualFilter(string $attribute, $value): DataProviderInterface
    {
        if ($value === null) {
            $this->getCriteria()->andWhere(Criteria::expr()->isNull($attribute));
        } else {
            $this->getCriteria()->andWhere(Criteria::expr()->eq($attribute, $value));
        }
        return $this;
    }

    public function addLikeFilter(string $attribute, $value): DataProviderInterface
    {
        $this->getCriteria()->andWhere(Criteria::expr()->contains($attribute, $value));
        return $this;
    }

    public function addRelationFilter(string $attribute, $value, string $relationClass): DataProviderInterface
    {
        $identifier = $this->getEntityIdentifier($relationClass);
        if ($identifier !== null) {
            $repository = $this->entityManager->getRepository($relationClass);
            $entity = $repository->findOneBy([$identifier => $value]);
            $this->getCriteria()->andWhere(Criteria::expr()->eq($attribute, $entity));
        }

        return $this;
    }

    public function addCustomFilter(string $attribute, $value, callable $callback): DataProviderInterface
    {
        $criteria = $this->getCriteria();
        call_user_func_array($callback, [&$criteria, $attribute, $value]);
        return $this;
    }

    protected function equalDate($attribute, $value): void
    {
        $date = new DateTime($value);
        $nextDate = (clone $date)->modify('+1 day');
        $this->getCriteria()
            ->andWhere(Criteria::expr()->gte($attribute, $date))
            ->andWhere(Criteria::expr()->lt($attribute, $nextDate));
    }

    protected function ltDate($attribute, $value): void
    {
        $date = new DateTime($value);
        $this->getCriteria()
            ->andWhere(Criteria::expr()->lt($attribute, $date));
    }

    protected function lteDate($attribute, $value): void
    {
        $date = (new DateTime($value))->modify('+1 day');
        $this->getCriteria()
            ->andWhere(Criteria::expr()->lt($attribute, $date));
    }

    protected function gtDate($attribute, $value): void
    {
        $date = (new DateTime($value))->modify('+1 day');
        $this->getCriteria()
            ->andWhere(Criteria::expr()->gte($attribute, $date));
    }

    protected function gteDate($attribute, $value): void
    {
        $date = new DateTime($value);
        $this->getCriteria()
            ->andWhere(Criteria::expr()->gte($attribute, $date));
    }
}