Yiisoft\VarDumper\Handler\StreamHandler

Complexity

Total Complexity

20

Size/Duplication

Total Lines	100
Duplicated Lines	0 %

Test Coverage

Coverage

93.33%

Importance

Changes	5
Bugs	1	Features	1

6 Methods

Rating	Name	Duplication	Size	Complexity
A	withEncoder()	0	5	1
A	__construct()	0	12	4
A	handle()	0	18	3
A	__destruct()	0	6	3
A	writeToStream()	0	6	2
B	initializeStream()	0	19	7

<?php

declare(strict_types=1);

namespace Yiisoft\VarDumper\Handler;

use InvalidArgumentException;
use RuntimeException;
use Socket;
use Yiisoft\VarDumper\HandlerInterface;

use function fsockopen;
use function fwrite;
use function get_debug_type;
use function is_resource;
use function is_string;

final class StreamHandler implements HandlerInterface
{
    /**
     * @var callable|null
     */
    private $encoder = null;
    /**
     * @var resource|Socket|null
     */
    private $stream = null;

    /**
     * @var resource|Socket|string|null
     */
    private $uri = null;

    /**
     * @param mixed|resource|string $uri
     */
    public function __construct(
        mixed $uri = 'udp://127.0.0.1:8890'
    ) {
        if (!is_string($uri) && !is_resource($uri) && !$uri instanceof Socket) {
            throw new InvalidArgumentException(
                sprintf(
                    'Argument $uri must be a string or a resource, "%s" given.',
                    get_debug_type($uri)
                )
            );
        }
        $this->uri = $uri;
    }

    /**
     * Encodes with {@see self::$encoder} {@param $variable} and sends the result to the stream.

The type Yiisoft\VarDumper\Handler\and was not found. Maybe you did not declare it correctly or list all dependencies?

     */
    public function handle(mixed $variable, int $depth, bool $highlight = false): void
    {
        $data = ($this->encoder ?? '\json_encode')($variable);
        if (!is_string($data)) {
            throw new RuntimeException(
                sprintf(
                    'Encoder must return a string, "%s" returned.',
                    get_debug_type($data)
                )
            );
        }

        $this->initializeStream();

        if (!$this->writeToStream($data)) {
            $this->initializeStream();

            $this->writeToStream($data);
        }
    }

    public function withEncoder(callable $encoder): HandlerInterface
    {
        $new = clone $this;
        $new->encoder = $encoder;
        return $new;
    }

    private function initializeStream(): void
    {
        if (is_string($this->uri)) {
            if (
                str_starts_with($this->uri, 'udp://') ||
                str_starts_with($this->uri, 'udg://') ||
                str_starts_with($this->uri, 'tcp://') ||
                str_starts_with($this->uri, 'unix://')
            ) {
                $this->stream = fsockopen($this->uri);
            } else {
                $this->stream = fopen($this->uri, 'w+');
            }
        } else {
            $this->stream = $this->uri;
        }

        if (!is_resource($this->stream)) {
            throw new RuntimeException('Cannot initialize a stream.');
        }
    }

    private function writeToStream(string $data): bool
    {
        if (!is_resource($this->stream)) {
            return false;
        }
        return @fwrite($this->stream, $data) !== false;
    }

    public function __destruct()
    {
        if (!is_string($this->uri) || !is_resource($this->stream)) {
            return;
        }
        fclose($this->stream);
    }
}