<?php

namespace App\ClinicService\Repository;

use App\ClinicService\Entity\ServiceItem;
use App\ClinicService\Entity\ServiceSection;
use App\Staff\Entity\ClinicStaff;
use Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository;
use Doctrine\Persistence\ManagerRegistry;

class ServiceItemRepository extends ServiceEntityRepository
{
    public function __construct(ManagerRegistry $registry)
    {
        parent::__construct($registry, ServiceItem::class);
    }

    public function findByUuid(string $uuid): ?ServiceItem
    {
        return $this->findOneBy(['uuid' => $uuid]);
    }

    public function findBySection(ServiceSection $section): array
    {
        return $this->createQueryBuilder('i')
            ->where('i.section = :section')
            ->setParameter('section', $section)
            ->orderBy('i.name', 'ASC')
            ->getQuery()
            ->getResult();
    }

    /** همه‌ی سرویس‌های یک owner (در همه‌ی بخش‌ها) — برای انتخاب/جستجوی سراسری. */
    public function findByEntity(string $entityType, int $entityId): array
    {
        return $this->createQueryBuilder('i')
            ->join('i.section', 's')
            ->where('s.entityType = :type')->setParameter('type', $entityType)
            ->andWhere('s.entityId = :id')->setParameter('id', $entityId)
            ->orderBy('i.name', 'ASC')
            ->getQuery()
            ->getResult();
    }

    /**
     * سرویس‌های فعالِ تخصیص‌یافته به یک پرسنل، محدود به محیط خودش.
     *
     * ردیف‌های قبل از مهاجرتِ چندپرسنلی فقط ستون تکیِ `staff` را دارند، پس هر دو
     * سمت رابطه بررسی می‌شود.
     *
     * @return ServiceItem[]
     */
    public function findByStaff(ClinicStaff $staff): array
    {
        return $this->createQueryBuilder('i')
            ->join('i.section', 's')
            ->leftJoin('i.staffMembers', 'm')
            ->where('s.entityType = :type')->setParameter('type', $staff->getEntityType())
            ->andWhere('s.entityId = :id')->setParameter('id', $staff->getEntityId())
            ->andWhere('i.active = true')
            ->andWhere('m = :staff OR i.staff = :staff')->setParameter('staff', $staff)
            ->orderBy('i.name', 'ASC')
            ->getQuery()
            ->getResult();
    }

    /**
     * Count services per section in a single query (avoids N+1 in the section list).
     *
     * @param ServiceSection[] $sections
     * @return array<string,int> section uuid → number of services
     */
    public function countBySections(array $sections): array
    {
        if (empty($sections)) {
            return [];
        }

        $rows = $this->createQueryBuilder('i')
            ->select('s.uuid AS uuid, COUNT(i.id) AS cnt')
            ->join('i.section', 's')
            ->where('i.section IN (:sections)')
            ->setParameter('sections', $sections)
            ->groupBy('s.uuid')
            ->getQuery()
            ->getScalarResult();

        $counts = [];
        foreach ($rows as $row) {
            $counts[$row['uuid']] = (int) $row['cnt'];
        }
        return $counts;
    }

    /**
     * تعداد سرویس‌های فعالِ «نمایش در نوبت‌دهی» (bookable) متعلق به یک entity
     * (پزشک/کلینیک) — از طریق section.entityType/entityId. برای اجبارِ حالت سرویس.
     */
    public function countBookableByEntity(string $entityType, int $entityId): int
    {
        return (int) $this->createQueryBuilder('i')
            ->select('COUNT(i.id)')
            ->join('i.section', 's')
            ->where('s.entityType = :type')
            ->andWhere('s.entityId = :id')
            ->andWhere('i.bookable = true')
            ->andWhere('i.active = true')
            ->setParameter('type', $entityType)
            ->setParameter('id', $entityId)
            ->getQuery()
            ->getSingleScalarResult();
    }

    /**
     * سرویس‌های فعالِ «نمایش در نوبت‌دهی» (bookable) یک entity — برای نمایش عمومیِ
     * انتخاب سرویس در نوبت‌گیری آنلاین.
     *
     * @return ServiceItem[]
     */
    public function findBookableByEntity(string $entityType, int $entityId): array
    {
        return $this->createQueryBuilder('i')
            ->join('i.section', 's')
            ->where('s.entityType = :type')
            ->andWhere('s.entityId = :id')
            ->andWhere('i.bookable = true')
            ->andWhere('i.active = true')
            ->setParameter('type', $entityType)
            ->setParameter('id', $entityId)
            ->orderBy('i.name', 'ASC')
            ->getQuery()
            ->getResult();
    }

    /**
     * نگاشت id → uuid برای مجموعه‌ای از خدمات.
     *
     * عمداً entity هیدریت نمی‌کند: ServiceItem رابطهٔ staffMembers را EAGER دارد،
     * پس هر entity یک کوئری اضافه برای بارگذاری کارکنانش می‌زند و فهرستی که فقط
     * uuid می‌خواهد به N+1 می‌افتد.
     *
     * @param  int[] $ids
     * @return array<int, string>
     */
    public function findUuidsByIds(array $ids): array
    {
        if ($ids === []) {
            return [];
        }

        $rows = $this->createQueryBuilder('i')
            ->select('i.id AS id, i.uuid AS uuid')
            ->where('i.id IN (:ids)')
            ->setParameter('ids', $ids)
            ->getQuery()
            ->getScalarResult();

        return array_column($rows, 'uuid', 'id');
    }

    public function save(ServiceItem $item): void
    {
        $this->getEntityManager()->persist($item);
        $this->getEntityManager()->flush();
    }

    public function remove(ServiceItem $item): void
    {
        $this->getEntityManager()->remove($item);
        $this->getEntityManager()->flush();
    }
}
