<?php

namespace App\Repository\Campaign\Tasklist\Task;

use App\Entity\Campaign\Tasklist\Task;
use App\Service\Campaign\Tasklist\TaskRepositoryInterface;
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 Exception;
use Symfony\Bridge\Doctrine\RegistryInterface;

final class DoctrineTaskRepository extends ServiceEntityRepository implements TaskRepositoryInterface
{
    public function __construct(RegistryInterface $registry)
    {
        parent::__construct($registry, Task::class);
    }

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

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

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

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

        return count($result);
    }

    /**
     * @param array|null $orderBy
     * @param int|null $limit
     * @param int|null $offset
     */
    private function queryByGrid(array $criteria, ?array $orderBy = null, ?int $limit = null, ?int $offset = null)
    {
        $newCriteria = $this->getCriteria($criteria, $orderBy, $limit, $offset);

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

        if (array_key_exists('campaign', $criteria) || array_key_exists('template', $criteria)) {
            $queryBuilder
                ->innerJoin('task.tasklist', 'tl');
            if (array_key_exists('campaign', $criteria)) {
                $newCriteria->andWhere(Criteria::expr()->in('tl.campaign', $criteria['campaign']));
            }
            if (array_key_exists('template', $criteria) && is_bool($criteria['template'])) {
                $newCriteria->andWhere(Criteria::expr()->eq('tl.template', $criteria['template']));
            }
        }

        $queryBuilder->addCriteria($newCriteria);

        if ($orderBy && array_key_exists('priority', $orderBy)) {
            $queryBuilder->addSelect('-task.dueAt as HIDDEN due');
            $queryBuilder->orderBy('task.priority', $orderBy['priority']);
            $queryBuilder->addOrderBy('due', 'desc');
        }

        $query = $queryBuilder->getQuery();

        return $query->getResult();
    }

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

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

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

        if (array_key_exists('q', $criteria)) {
            $newCriteria->where(Criteria::expr()->contains('title', $criteria['q']))
                ->orWhere(Criteria::expr()->contains('description', $criteria['q']));
        }

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

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

        if (array_key_exists('completed', $criteria)) {
            if ($criteria['completed']) {
                $newCriteria->andWhere(Criteria::expr()->eq('completed', true));
            } else {
                $newCriteria->andWhere(Criteria::expr()->eq('completed', false));
            }
        }

        return $newCriteria;
    }

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

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

    public function aggregateByGrid(array $criteria, DataGridInterface $dataGrid): array
    {
        $queryBuilder = $this->createQueryBuilder('task');

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

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

        $queryBuilder
            ->innerJoin('task.tasklist', 'tasklist');

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

        $queryBuilder->addCriteria($newCriteria);

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

        $query = $queryBuilder->getQuery();

        return $query->getResult();
    }

    /**
     * @return string|null
     */
    public function getFirstCompletedAt(): ?string
    {
        try {
            $criteria = (new Criteria())->where(Criteria::expr()->neq('t.completedAt', null));

            return $this->createQueryBuilder('t')
                ->select('t.completedAt')
                ->addCriteria($criteria)
                ->orderBy('t.completedAt', 'asc')
                ->setMaxResults(1)
                ->getQuery()
                ->getSingleScalarResult();
        } catch (Exception $exception) {
            return null;
        }
    }

    public function moveTasksBetweenUsers(int $from, int $to)
    {
        $qb = $this->createQueryBuilder('t');
        $qb
            ->update()
            ->set('t.user', $to)
            ->where('t.user = :from');

        $qb->setParameter('from', $from);

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