TechDeCo\ElasticApmAgent\Convenience\Middleware\OpenTransactionRequestEnricher\ResponseHeaderBlacklistEnricher

Complexity

Total Complexity

5

Size/Duplication

Total Lines	36
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	4

Test Coverage

Coverage

100%

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
A	enrichFromResponse()	0	10	1
A	addHeadersToApmResponse()	0	12	3

<?php
declare(strict_types=1);

namespace TechDeCo\ElasticApmAgent\Convenience\Middleware\OpenTransactionRequestEnricher;

use Psr\Http\Message\ResponseInterface;
use TechDeCo\ElasticApmAgent\Convenience\Middleware\OpenTransactionResponseEnricher;
use TechDeCo\ElasticApmAgent\Convenience\OpenTransaction;
use TechDeCo\ElasticApmAgent\Message\Response;
use function array_map;
use function implode;
use function in_array;
use function strtolower;

final class ResponseHeaderBlacklistEnricher implements OpenTransactionResponseEnricher
{
    /**
     * @var string[]
     */
    private $headerBlacklist;

    public function __construct(string ...$headerBlacklist)
    {
        $this->headerBlacklist = array_map('strtolower', $headerBlacklist);
    }

    public function enrichFromResponse(OpenTransaction $transaction, ResponseInterface $response): void
    {
        $apmResponse = $this->addHeadersToApmResponse(
            $response,
            $transaction->getContext()->getResponse() ?? Response::fromHttpResponse($response)
        );
        $transaction->setContext(
            $transaction->getContext()->withResponse($apmResponse)
        );
    }

    private function addHeadersToApmResponse(ResponseInterface $httpResponse, Response $apmResponse): Response
    {
        foreach ($httpResponse->getHeaders() as $name => $valueList) {
            if (in_array(strtolower($name), $this->headerBlacklist, true)) {
                continue;
            }

            $apmResponse = $apmResponse->withHeader($name, implode(',', $valueList));

Consider using a different name than the parameter $apmResponse. This often makes code more readable.

        }

        return $apmResponse;
    }
}