LAG\AdminBundle\Bridge\Doctrine\ORM\Results\ResultsHandler

Complexity

Total Complexity

10

Size/Duplication

Total Lines	44
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes	1
Bugs	0	Features	1

2 Methods

Rating	Name	Duplication	Size	Complexity
A	handle()	0	21	4
A	getAdapter()	0	19	6

<?php

namespace LAG\AdminBundle\Bridge\Doctrine\ORM\Results;

use Doctrine\Common\Collections\Collection;
use Doctrine\ORM\Query;
use Doctrine\ORM\QueryBuilder;
use LAG\AdminBundle\Exception\Exception;
use Pagerfanta\Adapter\AdapterInterface;
use Pagerfanta\Adapter\ArrayAdapter;
use Pagerfanta\Adapter\DoctrineCollectionAdapter;
use Pagerfanta\Adapter\DoctrineORMAdapter;
use Pagerfanta\Pagerfanta;

class ResultsHandler implements ResultsHandlerInterface
{
    public function handle($data, bool $pagination, int $page = 1, int $maxPerPage = 25)
    {
        if ($pagination) {
            $adapter = $this->getAdapter($data);

            $pager = new Pagerfanta($adapter);
            $pager->setCurrentPage($page);
            $pager->setMaxPerPage($maxPerPage);

            return $pager;
        }

        if ($data instanceof Query) {
            return $data->getResult();
        }

        if ($data instanceof QueryBuilder) {
            return $data->getQuery()->getResult();
        }

        return $data;
    }

    private function getAdapter($data): AdapterInterface
    {
        if ($data instanceof QueryBuilder || $data instanceof Query) {
            return new DoctrineORMAdapter($data, true, true);
        }

        if (is_array($data)) {
            return new ArrayAdapter($data);
        }

        if (is_iterable($data)) {
            return new ArrayAdapter(iterator_to_array($data));
        }

        if ($data instanceof Collection) {
            return new DoctrineCollectionAdapter($data);
        }

        throw new Exception('Unable to find an adapter for type "'.gettype($data).'"');
    }
}