Shlinkio\Shlink\Rest\Middleware\ShortCode\CreateShortCodeContentNegotiationMiddleware

Complexity

Total Complexity

8

Size/Duplication

Total Lines	49
Duplicated Lines	0 %

Coupling/Cohesion

Components	0
Dependencies	0

Test Coverage

Coverage

100%

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	process()	0	20	3
A	determineAcceptTypeFromQuery()	0	9	3
A	determineAcceptTypeFromHeader()	0	6	2

<?php
declare(strict_types=1);

namespace Shlinkio\Shlink\Rest\Middleware\ShortCode;

use Psr\Http\Message\ResponseInterface;
use Psr\Http\Message\ServerRequestInterface;
use Psr\Http\Server\MiddlewareInterface;
use Psr\Http\Server\RequestHandlerInterface;
use Zend\Diactoros\Response;
use Zend\Diactoros\Response\JsonResponse;

class CreateShortCodeContentNegotiationMiddleware implements MiddlewareInterface
{
    private const PLAIN_TEXT = 'text';
    private const JSON = 'json';

    /**
     * Process an incoming server request and return a response, optionally delegating
     * response creation to a handler.
     * @throws \RuntimeException
     * @throws \InvalidArgumentException
     */
    public function process(ServerRequestInterface $request, RequestHandlerInterface $handler): ResponseInterface
    {
        /** @var JsonResponse $response */
        $response = $handler->handle($request);
        $acceptedType = $request->hasHeader('Accept')
            ? $this->determineAcceptTypeFromHeader($request->getHeaderLine('Accept'))
            : $this->determineAcceptTypeFromQuery($request->getQueryParams());

        // If JSON was requested, return the response from next handler as is
        if ($acceptedType === self::JSON) {
            return $response;
        }

        // If requested, return a plain text response containing the short URL only
        $resp = (new Response())->withHeader('Content-Type', 'text/plain');
        $body = $resp->getBody();
        $body->write($response->getPayload()['shortUrl'] ?? '');
        $body->rewind();
        return $resp;
    }

    private function determineAcceptTypeFromQuery(array $query): string
    {
        if (! isset($query['format'])) {
            return self::JSON;
        }

        $format = \strtolower((string) $query['format']);
        return $format === 'txt' ? self::PLAIN_TEXT : self::JSON;
    }

    private function determineAcceptTypeFromHeader(string $acceptValue): string
    {
        $accepts = \explode(',', $acceptValue);
        $accept = \strtolower(\array_shift($accepts));
        return \strpos($accept, 'text/plain') !== false ? self::PLAIN_TEXT : self::JSON;
    }
}