<?php

declare(strict_types=1);

namespace App\Entity\Repository;

use App\Container\ContainerAwareTrait;
use App\Doctrine\Repository;
use App\Entity\Enums\StorageLocationAdapters;
use App\Entity\Enums\StorageLocationTypes;
use App\Entity\Station;
use App\Entity\StorageLocation;
use App\Entity\StorageLocationAdapter\StorageLocationAdapterInterface;
use InvalidArgumentException;

/**
 * @extends Repository<StorageLocation>
 */
final class StorageLocationRepository extends Repository
{
    use ContainerAwareTrait;

    protected string $entityClass = StorageLocation::class;

    public function findByType(
        string|StorageLocationTypes $type,
        int $id
    ): ?StorageLocation {
        if ($type instanceof StorageLocationTypes) {
            $type = $type->value;
        }

        return $this->repository->findOneBy(
            [
                'type' => $type,
                'id' => $id,
            ]
        );
    }

    /**
     * @param string|StorageLocationTypes $type
     *
     * @return StorageLocation[]
     */
    public function findAllByType(string|StorageLocationTypes $type): array
    {
        if ($type instanceof StorageLocationTypes) {
            $type = $type->value;
        }

        return $this->repository->findBy(
            [
                'type' => $type,
            ]
        );
    }

    /**
     * @param string|StorageLocationTypes $type
     * @param bool $addBlank
     * @param string|null $emptyString
     *
     * @return string[]
     */
    public function fetchSelectByType(
        string|StorageLocationTypes $type,
        bool $addBlank = false,
        ?string $emptyString = null
    ): array {
        $select = [];

        if ($addBlank) {
            $emptyString ??= __('None');
            $select[''] = $emptyString;
        }

        foreach ($this->findAllByType($type) as $storageLocation) {
            $select[$storageLocation->id] = (string)$storageLocation;
        }

        return $select;
    }

    public function createDefaultStorageLocations(): void
    {
        $backupLocations = $this->findAllByType(StorageLocationTypes::Backup);

        if (0 === count($backupLocations)) {
            $record = new StorageLocation(
                StorageLocationTypes::Backup,
                StorageLocationAdapters::Local
            );
            $record->path = StorageLocation::DEFAULT_BACKUPS_PATH;
            $this->em->persist($record);
        }

        $this->em->flush();
    }

    /**
     * @param StorageLocation $storageLocation
     *
     * @return Station[]
     */
    public function getStationsUsingLocation(StorageLocation $storageLocation): array
    {
        $qb = $this->em->createQueryBuilder()
            ->select('s')
            ->from(Station::class, 's');

        switch ($storageLocation->type) {
            case StorageLocationTypes::StationMedia:
                $qb->where('s.media_storage_location = :storageLocation')
                    ->setParameter('storageLocation', $storageLocation);
                break;

            case StorageLocationTypes::StationRecordings:
                $qb->where('s.recordings_storage_location = :storageLocation')
                    ->setParameter('storageLocation', $storageLocation);
                break;

            case StorageLocationTypes::StationPodcasts:
                $qb->where('s.podcasts_storage_location = :storageLocation')
                    ->setParameter('storageLocation', $storageLocation);
                break;

            case StorageLocationTypes::Backup:
                return [];
        }

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

    public function getAdapter(StorageLocation $storageLocation): StorageLocationAdapterInterface
    {
        $adapterClass = $storageLocation->adapter->getAdapterClass();

        if (!$this->di->has($adapterClass)) {
            throw new InvalidArgumentException(sprintf('Class not found: %s', $adapterClass));
        }

        /** @var StorageLocationAdapterInterface $adapter */
        $adapter = $this->di->get($adapterClass);
        return $adapter->withStorageLocation($storageLocation);
    }
}
