Moved NotFoundHandler to ErrorHandler namespace

module/Core/src/ErrorHandler/NotFoundHandler.php

@@ -0,0 +1,92 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Shlinkio\Shlink\Core\ErrorHandler;
+
+use Fig\Http\Message\StatusCodeInterface;
+use InvalidArgumentException;
+use Psr\Http\Message\ResponseInterface;
+use Psr\Http\Message\ServerRequestInterface;
+use Psr\Http\Message\UriInterface;
+use Psr\Http\Server\RequestHandlerInterface;
+use Shlinkio\Shlink\Core\Action\RedirectAction;
+use Shlinkio\Shlink\Core\Options\NotFoundRedirectOptions;
+use Zend\Diactoros\Response;
+use Zend\Expressive\Router\RouteResult;
+use Zend\Expressive\Template\TemplateRendererInterface;
+
+use function array_shift;
+use function explode;
+use function rtrim;
+
+class NotFoundHandler implements RequestHandlerInterface
+{
+    public const NOT_FOUND_TEMPLATE = 'ShlinkCore::error/404';
+    public const INVALID_SHORT_CODE_TEMPLATE = 'ShlinkCore::invalid-short-code';
+
+    /** @var TemplateRendererInterface */
+    private $renderer;
+    /** @var NotFoundRedirectOptions */
+    private $redirectOptions;
+    /** @var string */
+    private $shlinkBasePath;
+
+    public function __construct(
+        TemplateRendererInterface $renderer,
+        NotFoundRedirectOptions $redirectOptions,
+        string $shlinkBasePath
+    ) {
+        $this->renderer = $renderer;
+        $this->redirectOptions = $redirectOptions;
+        $this->shlinkBasePath = $shlinkBasePath;
+    }
+
+    /**
+     * Dispatch the next available middleware and return the response.
+     *
+     * @param ServerRequestInterface $request
+     *
+     * @return ResponseInterface
+     * @throws InvalidArgumentException
+     */
+    public function handle(ServerRequestInterface $request): ResponseInterface
+    {
+        /** @var RouteResult $routeResult */
+        $routeResult = $request->getAttribute(RouteResult::class, RouteResult::fromRouteFailure(null));
+        $redirectResponse = $this->createRedirectResponse($routeResult, $request->getUri());
+        if ($redirectResponse !== null) {
+            return $redirectResponse;
+        }
+
+        $accepts = explode(',', $request->getHeaderLine('Accept'));
+        $accept = array_shift($accepts);
+        $status = StatusCodeInterface::STATUS_NOT_FOUND;
+
+        $template = $routeResult->isFailure() ? self::NOT_FOUND_TEMPLATE : self::INVALID_SHORT_CODE_TEMPLATE;
+        return new Response\HtmlResponse($this->renderer->render($template), $status);
+    }
+
+    private function createRedirectResponse(RouteResult $routeResult, UriInterface $uri): ?ResponseInterface
+    {
+        $isBaseUrl = rtrim($uri->getPath(), '/') === $this->shlinkBasePath;
+
+        if ($isBaseUrl && $this->redirectOptions->hasBaseUrlRedirect()) {
+            return new Response\RedirectResponse($this->redirectOptions->getBaseUrlRedirect());
+        }
+
+        if (!$isBaseUrl && $routeResult->isFailure() && $this->redirectOptions->hasRegular404Redirect()) {
+            return new Response\RedirectResponse($this->redirectOptions->getRegular404Redirect());
+        }
+
+        if (
+            $routeResult->isSuccess() &&
+            $routeResult->getMatchedRouteName() === RedirectAction::class &&
+            $this->redirectOptions->hasInvalidShortUrlRedirect()
+        ) {
+            return new Response\RedirectResponse($this->redirectOptions->getInvalidShortUrlRedirect());
+        }
+
+        return null;
+    }
+}