<?php

namespace App\Repository\User;

use App\Entity\Campaign;
use App\Entity\Campaign\TeamMember;
use App\Entity\User;
use Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository;
use Doctrine\Common\Collections\Criteria;
use Doctrine\Common\Persistence\ManagerRegistry;

final class DoctrineUserRepository extends ServiceEntityRepository implements UserRepositoryInterface
{
    public function __construct(ManagerRegistry $registry)
    {
        // The second parameter is the Entity this Repository uses
        parent::__construct($registry, User::class);
    }

    /**
     * @param int $userId
     *
     * @return User
     */
    public function findById(int $userId): ?User
    {
        return $this->findOneBy(['id' => $userId]);
    }

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

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

    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
    {
        $queryBuilder = $this->createQueryBuilder('user');

        $newCriteria = $this->getCriteria($criteria, $orderBy, $limit, $offset);

        if (array_key_exists('campaign', $criteria)) {
            $queryBuilder
                ->innerJoin(TeamMember::class, 't', 'WITH', 'user.id = t.user')
                ->innerJoin(Campaign::class, 'c', 'WITH', 't.campaign = c.id');

            $newCriteria->andWhere(Criteria::expr()->in('c.id', $criteria['campaign']));
        }

        $queryBuilder->addCriteria($newCriteria);

        $query = $queryBuilder->getQuery();

        return $query->getResult(1);
    }

    /**
     * @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, ?int $limit, ?int $offset): Criteria
    {
        $newCriteria = Criteria::create()
            ->setFirstResult($offset)
            ->setMaxResults($limit);

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

        if (array_key_exists('q', $criteria)) {
            $newCriteria->where(Criteria::expr()->contains('firstName', $criteria['q']))
                ->orWhere(Criteria::expr()->contains('lastName', $criteria['q']))
                ->orWhere(Criteria::expr()->contains('jobTitle', $criteria['q']))
                ->orWhere(Criteria::expr()->contains('email', $criteria['q']));
        }

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

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

        return $newCriteria;
    }
}