EmanueleMinotto\GuzzleSnapshot\SnapshotMiddleware

Complexity

Total Complexity

6

Size/Duplication

Total Lines	46
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__invoke()	0	14	3
A	onFulfilled()	0	6	1
A	setRequestStorage()	0	3	1
A	__construct()	0	3	1

<?php

declare(strict_types=1);

namespace EmanueleMinotto\GuzzleSnapshot;

use EmanueleMinotto\GuzzleSnapshot\Strategy\StrategyInterface;
use GuzzleHttp\Promise\PromiseInterface;
use Psr\Http\Message\RequestInterface;
use Psr\Http\Message\ResponseInterface;

class SnapshotMiddleware
{
    /**
     * @var StrategyInterface
     */
    private $strategy;

    /**
     * @var bool
     */
    private $requestStorage = false;

    public function __construct(StrategyInterface $strategy)
    {
        $this->strategy = $strategy;
    }

    public function __invoke(callable $handler): callable
    {
        return function (RequestInterface $request, array $options) use ($handler) {
            $promise = $this->strategy->getPromise($request);
            if ($promise instanceof PromiseInterface) {
                return $promise;
            }

            if ($this->requestStorage) {
                $this->strategy->storeRequest($request);
            }

            return $handler($request, $options)->then(
                $this->onFulfilled($request)
            );
        };
    }

    public function setRequestStorage(bool $requestStorage): void
    {
        $this->requestStorage = $requestStorage;
    }

    private function onFulfilled(RequestInterface $request): callable
    {
        return function (ResponseInterface $response) use ($request) {
            $this->strategy->storeResponse($request, $response);

            return $response;
        };
    }
}