<?php

namespace App\Repository\TimeLog;

use App\Entity\Campaign;
use App\Entity\Client;
use App\Entity\TimeLog;
use App\Entity\TimeLogType;
use App\Entity\TimeLogTypeCategory;
use App\Service\DataGrid\DataGridInterface;
use Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository;
use Doctrine\Common\Collections\Criteria;
use Doctrine\ORM\OptimisticLockException;
use Doctrine\ORM\ORMException;
use Doctrine\ORM\Query\QueryException;
use Exception;
use Symfony\Bridge\Doctrine\RegistryInterface;

final class DoctrineTimeLogRepository extends ServiceEntityRepository implements TimeLogRepositoryInterface
{
    public function __construct(RegistryInterface $registry)
    {
        parent::__construct($registry, TimeLog::class);
    }

    public function findById(int $id): ?TimeLog
    {
        return $this->findOneBy(['id' => $id]);
    }

    /**
     * @param TimeLog $timeLog
     *
     * @throws ORMException
     * @throws OptimisticLockException
     */
    public function save(TimeLog $timeLog): void
    {
        $this->_em->persist($timeLog);
        $this->_em->flush();
    }

    /**
     * @param TimeLog $timeLog
     *
     * @throws ORMException
     * @throws OptimisticLockException
     */
    public function delete(TimeLog $timeLog): void
    {
        $this->_em->remove($timeLog);
        $this->_em->flush();
    }

    public function findAll()
    {
        return $this->findBy([], ['createdAt' => 'DESC']);
    }

    /*
     * @TODO clean criteria
     */
    public function aggregateByGrid(DataGridInterface $dataGrid): array
    {
        $criteria = [];

        if ($dataGrid->getFilters()->containsKey('campaign') && $dataGrid->getFilters()->get('campaign')->isValid()) {
            $criteria += ['campaign' => $dataGrid->getFilters()->get('campaign')->getValue()];
        }

        if ($dataGrid->getFilters()->containsKey('subjectType') && $dataGrid->getFilters()->get('subjectType')->isValid()) {
            $criteria += ['subjectType' => $dataGrid->getFilters()->get('subjectType')->getValue()];
        }

        if ($dataGrid->getFilters()->containsKey('subjectId') && $dataGrid->getFilters()->get('subjectId')->isValid()) {
            $criteria += ['subjectId' => $dataGrid->getFilters()->get('subjectId')->getValue()];
        }

        if ($dataGrid->getFilters()->containsKey('type') && $dataGrid->getFilters()->get('type')->isValid()) {
            $criteria += ['type' => $dataGrid->getFilters()->get('type')->getValue()];
        }

        if ($dataGrid->getFilters()->containsKey('category') && $dataGrid->getFilters()->get('category')->isValid()) {
            $criteria += ['category' => $dataGrid->getFilters()->get('category')->getValue()];
        }

        if ($dataGrid->getFilters()->containsKey('user') && $dataGrid->getFilters()->get('user')->isValid()) {
            $criteria += ['user' => $dataGrid->getFilters()->get('user')->getValue()];
        }

        if ($dataGrid->getFilters()->containsKey('client') && $dataGrid->getFilters()->get('client')->isValid()) {
            $criteria += ['client' => $dataGrid->getFilters()->get('client')->getValue()];
        }

        if ($dataGrid->getFilters()->containsKey('dateFrom') && $dataGrid->getFilters()->get('dateFrom')->isValid()) {
            $criteria += ['dateFrom' => $dataGrid->getFilters()->get('dateFrom')->getValue()];
        }

        $queryBuilder = $this->createQueryBuilder('timelog');

        $queryBuilder->select('COUNT(1) as total');

        $newCriteria = $this->getCriteria($criteria);

        if (array_key_exists('client', $criteria)) {
            $queryBuilder
                ->innerJoin(Campaign::class, 'ca', 'WITH', 'timelog.campaign = ca.id')
                ->innerJoin(Client::class, 'cl', 'WITH', 'ca.client = cl.id');
            $newCriteria->andWhere(Criteria::expr()->in('cl.id', $criteria['client']));
        }

        if (array_key_exists('category', $criteria)) {
            $queryBuilder
                ->innerJoin(TimeLogType::class, 'type', 'WITH', 'timelog.type = type.id')
                ->innerJoin(TimeLogTypeCategory::class, 'category', 'WITH', 'type.category = category.id');
            $newCriteria->andWhere(Criteria::expr()->in('category.id', $criteria['category']));
        }

        $queryBuilder->addCriteria($newCriteria);

        foreach ($dataGrid->getAggregates() as $aggregate) {
            $queryBuilder->addSelect(sprintf('%s(timelog.%s) as %s', $aggregate->getFunction(), $aggregate->getField(), $aggregate->getId()));
            $queryBuilder->addOrderBy(sprintf('%s', $aggregate->getId()), 'DESC');
            if (!empty($aggregate->getGroupBy())) {
                $queryBuilder->addSelect(sprintf('(timelog.%s) as %s', $aggregate->getGroupBy(), $aggregate->getGroupBy()));
                $queryBuilder->groupBy(sprintf('timelog.%s', $aggregate->getGroupBy()));
            }
        }

        $query = $queryBuilder->getQuery();

        return $query->getResult();
    }

    /**
     * @param array $criteria
     * @param       $orderBy
     * @param       $limit
     * @param       $offset
     *
     * @return Criteria
     */
    private function getCriteria(array $criteria, $orderBy = null, $limit = null, $offset = null): Criteria
    {
        $newCriteria = Criteria::create()
            ->setFirstResult($offset)
            ->setMaxResults($limit);

        if ($orderBy) {
            $newCriteria->orderBy($orderBy);
        }

        if (array_key_exists('campaign', $criteria)) {
            $newCriteria->andWhere(Criteria::expr()->in('campaign', $criteria['campaign']));
        }

        if (array_key_exists('subjectType', $criteria)) {
            $newCriteria->andWhere(Criteria::expr()->in('subjectType', $criteria['subjectType']));
        }

        if (array_key_exists('subjectId', $criteria)) {
            $newCriteria->andWhere(Criteria::expr()->in('subjectId', $criteria['subjectId']));
        }

        if (array_key_exists('type', $criteria)) {
            $newCriteria->andWhere(Criteria::expr()->in('type', $criteria['type']));
        }

        if (array_key_exists('user', $criteria)) {
            $newCriteria->andWhere(Criteria::expr()->in('user', $criteria['user']));
        }

        if (array_key_exists('dateFrom', $criteria)) {
            $newCriteria->andWhere(Criteria::expr()->gte('dateFrom', $criteria['dateFrom'][0]));
            $newCriteria->andWhere(Criteria::expr()->lte('dateFrom', $criteria['dateFrom'][1]));
        }

        return $newCriteria;
    }

    public function findBy(array $criteria, array $orderBy = null, $limit = null, $offset = null)
    {
        return $this->queryByGrid($criteria, $orderBy, $limit, $offset);
    }

    public function count(array $criteria, array $orderBy = null, $limit = null, $offset = null)
    {
        $result = $this->queryByGrid($criteria);

        return count($result);
    }

    /**
     * @param array $criteria
     * @param null  $orderBy
     * @param null  $limit
     * @param null  $offset
     *
     * @return array
     *
     * @throws QueryException
     */
    private function queryByGrid(array $criteria, $orderBy = null, $limit = null, $offset = null): array
    {
        $newCriteria = $this->getCriteria($criteria, $orderBy, $limit, $offset);

        $queryBuilder = $this->createQueryBuilder('timelog');

        if (array_key_exists('client', $criteria)) {
            $queryBuilder
                ->innerJoin(Campaign::class, 'ca', 'WITH', 'timelog.campaign = ca.id')
                ->innerJoin(Client::class, 'cl', 'WITH', 'ca.client = cl.id');
            $newCriteria->andWhere(Criteria::expr()->in('cl.id', $criteria['client']));
        }

        if (array_key_exists('category', $criteria)) {
            $queryBuilder
                ->innerJoin(TimeLogType::class, 'type', 'WITH', 'timelog.type = type.id')
                ->innerJoin(TimeLogTypeCategory::class, 'category', 'WITH', 'type.category = category.id');
            $newCriteria->andWhere(Criteria::expr()->in('category.id', $criteria['category']));
        }

        $queryBuilder->addCriteria($newCriteria);

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

    /**
     * @return array
     *
     * @throws QueryException
     */
    public function getSubjects(): array
    {
        $criteria = (new Criteria())->where(Criteria::expr()->neq('subjectType', null));
        $query = $this->createQueryBuilder('timelog')
            ->select('timelog.subjectType')
            ->addCriteria($criteria)
            ->distinct()
            ->getQuery();

        return $query->getArrayResult();
    }

    /**
     * @return string|null
     */
    public function getFirstDate(): ?string
    {
        try {
            return $this->createQueryBuilder('timelog')
                ->select('timelog.dateFrom')
                ->orderBy('timelog.dateFrom', 'asc')
                ->setMaxResults(1)
                ->getQuery()
                ->getSingleScalarResult();
        } catch (Exception $exception) {
            return null;
        }
    }
}