Use IpGeolocationMiddleware to geolocate visitors instead of LocateVisit event

module/Core/src/Geolocation/Middleware/IpGeolocationMiddleware.php

@@ -0,0 +1,58 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Shlinkio\Shlink\Core\Geolocation\Middleware;
+
+use Psr\Http\Message\ResponseInterface;
+use Psr\Http\Message\ServerRequestInterface;
+use Psr\Http\Server\MiddlewareInterface;
+use Psr\Http\Server\RequestHandlerInterface;
+use Psr\Log\LoggerInterface;
+use Shlinkio\Shlink\Core\Config\Options\TrackingOptions;
+use Shlinkio\Shlink\IpGeolocation\Exception\WrongIpException;
+use Shlinkio\Shlink\IpGeolocation\GeoLite2\DbUpdaterInterface;
+use Shlinkio\Shlink\IpGeolocation\Model\Location;
+use Shlinkio\Shlink\IpGeolocation\Resolver\IpLocationResolverInterface;
+use Throwable;
+
+use function Shlinkio\Shlink\Core\ipAddressFromRequest;
+
+readonly class IpGeolocationMiddleware implements MiddlewareInterface
+{
+    public function __construct(
+        private IpLocationResolverInterface $ipLocationResolver,
+        private DbUpdaterInterface $dbUpdater,
+        private LoggerInterface $logger,
+        private TrackingOptions $trackingOptions,
+    ) {
+    }
+
+    public function process(ServerRequestInterface $request, RequestHandlerInterface $handler): ResponseInterface
+    {
+        if (! $this->trackingOptions->isGeolocationRelevant()) {
+            return $handler->handle($request);
+        }
+
+        if (! $this->dbUpdater->databaseFileExists()) {
+            $this->logger->warning('Tried to geolocate IP address, but a GeoLite2 db was not found.');
+            return $handler->handle($request);
+        }
+
+        $location = $this->geolocateIpAddress(ipAddressFromRequest($request));
+        return $handler->handle($request->withAttribute(Location::class, $location));
+    }
+
+    private function geolocateIpAddress(string|null $ipAddress): Location
+    {
+        try {
+            return $ipAddress === null ? Location::empty() : $this->ipLocationResolver->resolveIpLocation($ipAddress);
+        } catch (WrongIpException $e) {
+            $this->logger->warning('Tried to locate IP address, but it seems to be wrong. {e}', ['e' => $e]);
+            return Location::empty();
+        } catch (Throwable $e) {
+            $this->logger->error('An unexpected error occurred while trying to locate IP address. {e}', ['e' => $e]);
+            return Location::empty();
+        }
+    }
+}