WyriHaximus\React\Http\Middleware\ResponseCacheMiddleware

Complexity

Total Complexity

6

Size/Duplication

Total Lines	47
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	2
B	__invoke()	0	25	4

<?php declare(strict_types=1);

namespace WyriHaximus\React\Http\Middleware;

use Psr\Http\Message\ResponseInterface;
use Psr\Http\Message\ServerRequestInterface;
use React\Cache\ArrayCache;
use React\Cache\CacheInterface;
use React\Http\Io\HttpBodyStream;
use function React\Promise\resolve;
use function RingCentral\Psr7\stream_for;

final class ResponseCacheMiddleware
{
    /**
     * @var CacheConfigurationInterface
     */
    private $cacheConfiguration;

    /**
     * @var CacheInterface
     */
    private $cache;

    /**
     * @param CacheConfigurationInterface $cacheConfiguration
     * @param CacheInterface|null         $cache
     */
    public function __construct(CacheConfigurationInterface $cacheConfiguration, CacheInterface $cache = null)
    {
        $this->cacheConfiguration = $cacheConfiguration;
        $this->cache = $cache instanceof CacheInterface ? $cache : new ArrayCache();
    }

    public function __invoke(ServerRequestInterface $request, callable $next)
    {
        if (!$this->cacheConfiguration->requestIsCacheable($request)) {
            return resolve($next($request));
        }

        $key = $this->cacheConfiguration->cacheKey($request);

        return $this->cache->get($key)->then(function ($json) {
            return $this->cacheConfiguration->cacheDecode($json);
        }, function () use ($next, $request, $key) {
            return resolve($next($request))->then(function (ResponseInterface $response) use ($request, $key) {
                if ($response->getBody() instanceof HttpBodyStream) {
                    return $response;
                }

                if (!$this->cacheConfiguration->responseIsCacheable($request, $response)) {
                    return $response;
                }

                $body = (string)$response->getBody();
                $encodedResponse = $this->cacheConfiguration->cacheEncode($response->withBody(stream_for($body)));
                $this->cache->set($key, $encodedResponse);

                return $response->withBody(stream_for($body));
            });
        });
    }
}