<?php

namespace App\Repository\Billing;

use App\Entity\Billing;
use App\Entity\Campaign;
use App\Entity\Content;
use App\Entity\Link;
use App\Entity\Prospect;
use App\Service\DataGrid\DataGridInterface;
use Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository;
use Doctrine\Common\Collections\Criteria;
use Symfony\Bridge\Doctrine\RegistryInterface;

final class DoctrineBillingRepository extends ServiceEntityRepository implements BillingRepositoryInterface
{
    public function __construct(RegistryInterface $registry)
    {
        parent::__construct($registry, Billing::class);
    }

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

    /**
     * @param Billing $billing
     *
     * @throws \Doctrine\ORM\ORMException
     * @throws \Doctrine\ORM\OptimisticLockException
     */
    public function save(Billing $billing): void
    {
        $this->_em->persist($billing);
        $this->_em->flush();
    }

    /**
     * @param Billing $billing
     *
     * @throws \Doctrine\ORM\ORMException
     * @throws \Doctrine\ORM\OptimisticLockException
     */
    public function delete(Billing $billing): void
    {
        $this->_em->remove($billing);
        $this->_em->flush();
    }

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

    public function aggregateByGrid(DataGridInterface $dataGrid): array
    {
        $criteria = [];

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

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

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

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

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

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

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

        if (array_key_exists('prospect', $criteria)) {
            $queryBuilder
                ->innerJoin(Link::class, 'l1', 'WITH', 'billing.link = l1.id')
                ->innerJoin(Prospect::class, 'p', 'WITH', 'l1.prospect = p.id');
            $newCriteria->andWhere(Criteria::expr()->in('p.id', $criteria['prospect']));
        }

        if (array_key_exists('campaign', $criteria)) {
            $queryBuilder
                ->innerJoin(Link::class, 'l2', 'WITH', 'billing.link = l2.id')
                ->innerJoin(Content::class, 'co', 'WITH', 'l2.content = co.id')
                ->innerJoin(Campaign::class, 'ca', 'WITH', 'co.campaign = ca.id');
            $newCriteria->andWhere(Criteria::expr()->in('ca.id', $criteria['campaign']));
        }

        $queryBuilder->addCriteria($newCriteria);

        foreach ($dataGrid->getAggregates() as $aggregate) {
            $queryBuilder->addSelect(sprintf('%s(billing.%s) as %s', $aggregate->getFunction(), $aggregate->getField(), $aggregate->getId()));
        }

        $query = $queryBuilder->getQuery();

        return $query->getResult();
    }

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

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

        if (array_key_exists('prospect', $criteria)) {
            $queryBuilder
                ->innerJoin(Link::class, 'l1', 'WITH', 'billing.link = l1.id')
                ->innerJoin(Prospect::class, 'p', 'WITH', 'l1.prospect = p.id');
            $newCriteria->andWhere(Criteria::expr()->in('p.id', $criteria['prospect']));
        }

        if (array_key_exists('campaign', $criteria)) {
            $queryBuilder
                ->innerJoin(Link::class, 'l2', 'WITH', 'billing.link = l2.id')
                ->innerJoin(Content::class, 'co', 'WITH', 'l2.content = co.id')
                ->innerJoin(Campaign::class, 'ca', 'WITH', 'co.campaign = ca.id');
            $newCriteria->andWhere(Criteria::expr()->in('ca.id', $criteria['campaign']));
        }

        $queryBuilder->addCriteria($newCriteria);

        $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('link', $criteria)) {
            $newCriteria->andWhere(Criteria::expr()->in('link', $criteria['link']));
        }

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

        return $newCriteria;
    }
}