pest/src/KernelDump.php

<?php

declare(strict_types=1);

namespace Pest;

use Pest\Support\View;
use Symfony\Component\Console\Output\OutputInterface;

final class KernelDump
{
    /**
     * The output buffer, if any.
     */
    private string $buffer = '';

    /**
     * Creates a new Kernel Dump instance.
     */
    public function __construct(
        private readonly OutputInterface $output,
    ) {
        // ...
    }

    /**
     * Enable the output buffering.
     */
    public function enable(): void
    {
        ob_start(function (string $message): string {
            $this->buffer .= $message;

            return '';
        });
    }

    /**
     * Disable the output buffering.
     */
    public function disable(): void
    {
        @ob_clean();

        if ($this->buffer !== '') {
            $this->flush();
        }
    }

    /**
     * Shutdown the output buffering.
     */
    public function shutdown(): void
    {
        $this->disable();
    }

    /**
     * Flushes the buffer.
     */
    private function flush(): void
    {
        View::renderUsing($this->output);

        if ($this->isOpeningHeadline($this->buffer)) {
            $this->buffer = implode(PHP_EOL, array_slice(explode(PHP_EOL, $this->buffer), 2));
        }

        $type = 'INFO';

        if ($this->isInternalError($this->buffer)) {
            $type = 'ERROR';
            $this->buffer = str_replace(
                sprintf('An error occurred inside PHPUnit.%s%sMessage:  ', PHP_EOL, PHP_EOL), '', $this->buffer,
            );
        }

        $this->buffer = trim($this->buffer);
        $this->buffer = rtrim($this->buffer, '.').'.';

        $lines = explode(PHP_EOL, $this->buffer);

        $lines = array_reverse($lines);
        $firstLine = array_pop($lines);
        $lines = array_reverse($lines);

        View::render('components.badge', [
            'type' => $type,
            'content' => $firstLine,
        ]);

        $this->output->writeln($lines);

        $this->buffer = '';
    }

    /**
     * Checks if the given output contains an opening headline.
     */
    private function isOpeningHeadline(string $output): bool
    {
        return str_contains($output, 'by Sebastian Bergmann and contributors.');
    }

    /**
     * Checks if the given output contains an opening headline.
     */
    private function isInternalError(string $output): bool
    {
        return str_contains($output, 'An error occurred inside PHPUnit.')
            || str_contains($output, 'Fatal error');
    }
}