<?php

namespace App\Appointment\Plan\Repository;

use App\Appointment\Plan\Entity\SegmentTemplate;
use App\ClinicService\Entity\ServiceItem;
use Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository;
use Doctrine\Persistence\ManagerRegistry;

/**
 * @extends ServiceEntityRepository<SegmentTemplate>
 */
class SegmentTemplateRepository extends ServiceEntityRepository
{
    public function __construct(ManagerRegistry $registry)
    {
        parent::__construct($registry, SegmentTemplate::class);
    }

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

    /**
     * الگوی بخش‌های یک سرویس، به‌همراه نیازمندی‌ها و نوع منبع و مهارتشان — یک کوئری،
     * نه یکی per بخش. سازندهٔ برنامه همهٔ این‌ها را لازم دارد.
     *
     * @return SegmentTemplate[]
     */
    public function findForService(ServiceItem $service): array
    {
        return $this->createQueryBuilder('s')
            ->addSelect('r', 'rt', 'sk')
            ->leftJoin('s.requirements', 'r')
            ->leftJoin('r.resourceType', 'rt')
            ->leftJoin('r.skill', 'sk')
            ->where('s.service = :service')
            ->setParameter('service', $service)
            ->orderBy('s.sequence', 'ASC')
            ->getQuery()
            ->getResult();
    }

    /**
     * @param int[] $serviceIds
     * @return array<int, SegmentTemplate[]> شناسهٔ سرویس => بخش‌ها
     */
    public function findForServices(array $serviceIds): array
    {
        if ($serviceIds === []) {
            return [];
        }

        $rows = $this->createQueryBuilder('s')
            ->addSelect('r', 'rt', 'sk')
            ->leftJoin('s.requirements', 'r')
            ->leftJoin('r.resourceType', 'rt')
            ->leftJoin('r.skill', 'sk')
            ->where('IDENTITY(s.service) IN (:ids)')
            ->setParameter('ids', $serviceIds)
            ->orderBy('s.sequence', 'ASC')
            ->getQuery()
            ->getResult();

        $byService = [];
        foreach ($rows as $segment) {
            $byService[(int) $segment->getService()->getId()][] = $segment;
        }

        return $byService;
    }

    /**
     * کدام سرویس‌ها اصلاً بخش تعریف‌شده دارند — یک کوئری، نه یکی per سرویس.
     *
     * @param int[] $serviceIds
     * @return array<int, bool> شناسهٔ سرویس => دارد/ندارد
     */
    public function hasSegmentsMap(array $serviceIds): array
    {
        if ($serviceIds === []) {
            return [];
        }

        $rows = $this->createQueryBuilder('s')
            ->select('IDENTITY(s.service) AS service_id', 'COUNT(s.id) AS total')
            ->where('IDENTITY(s.service) IN (:ids)')
            ->setParameter('ids', $serviceIds)
            ->groupBy('service_id')
            ->getQuery()
            ->getArrayResult();

        $map = [];
        foreach ($rows as $row) {
            $map[(int) $row['service_id']] = ((int) $row['total']) > 0;
        }

        return $map;
    }

    public function deleteForService(ServiceItem $service): int
    {
        return (int) $this->createQueryBuilder('s')
            ->delete()
            ->where('s.service = :service')
            ->setParameter('service', $service)
            ->getQuery()
            ->execute();
    }
}
