<?php
declare(strict_types=1);

namespace Shlinkio\Shlink\Common\IpGeolocation;

use GeoIp2\Database\Reader;
use GeoIp2\Exception\AddressNotFoundException;
use GeoIp2\Model\City;
use GeoIp2\Record\Subdivision;
use MaxMind\Db\Reader\InvalidDatabaseException;
use Shlinkio\Shlink\Common\Exception\WrongIpException;
use function Functional\first;

class GeoLite2LocationResolver implements IpLocationResolverInterface
{
    /** @var Reader */
    private $geoLiteDbReader;

    public function __construct(Reader $geoLiteDbReader)
    {
        $this->geoLiteDbReader = $geoLiteDbReader;
    }

    /**
     * @throws WrongIpException
     */
    public function resolveIpLocation(string $ipAddress): array
    {
        try {
            $city = $this->geoLiteDbReader->city($ipAddress);
            return $this->mapFields($city);
        } catch (AddressNotFoundException $e) {
            throw WrongIpException::fromIpAddress($ipAddress, $e);
        } catch (InvalidDatabaseException $e) {
            throw new WrongIpException('Provided GeoLite2 db file is invalid', 0, $e);
        }
    }

    private function mapFields(City $city): array
    {
        /** @var Subdivision $region */
        $region = first($city->subdivisions);

        return [
            'country_code' => $city->country->isoCode ?? '',
            'country_name' => $city->country->name ?? '',
            'region_name' => $region->name ?? '',
            'city' => $city->city->name ?? '',
            'latitude' => $city->location->latitude ?? '',
            'longitude' => $city->location->longitude ?? '',
            'time_zone' => $city->location->timeZone ?? '',
        ];
    }
}