<?php

namespace App\BusProNet\DataProvider;

use App\BusProNet\Client\ReferenceDataClient;
use App\BusProNet\MessageType;
use App\BusProNet\Model\Country;
use Psr\Cache\InvalidArgumentException;
use Symfony\Contracts\Cache\CacheInterface;
use Symfony\Contracts\Cache\ItemInterface;

class CountryDataProvider
{
    public function __construct(private readonly ReferenceDataClient $referenceDataClient, private readonly CacheInterface $cache)
    {
    }

    /** @return array<string, Country> */
    public function getAll(): array
    {
        try {
            return $this->cache->get('bpn_countries', function (ItemInterface $item) {
                $item->expiresAfter(6 * 60 * 60);

                return $this
                    ->referenceDataClient
                    ->getBaseData(MessageType::Countries)
                    ->getItems()
                ;
            });
        } catch (InvalidArgumentException $e) {
            return [];
        }
    }

    public function get(?string $token): ?Country
    {
        if (null === $token) {
            return null;
        }

        return $this->getAll()[$token] ?? null;
    }
}