yiisoft / yii-debug

src/Collector/CommandCollector.php

<?php

declare(strict_types=1);

namespace Yiisoft\Yii\Debug\Collector;

use RuntimeException;
use Symfony\Component\Console\Command\Command;
use Symfony\Component\Console\Event\ConsoleCommandEvent;
use Symfony\Component\Console\Event\ConsoleErrorEvent;
use Symfony\Component\Console\Event\ConsoleEvent;
use Symfony\Component\Console\Event\ConsoleTerminateEvent;

use function is_object;

final class CommandCollector implements CollectorInterface, IndexCollectorInterface
{
    use CollectorTrait;

    /**
     * Let -1 mean that it was not set during the process.
     */
    private const UNDEFINED_EXIT_CODE = -1;
    private array $commands = [];

    public function getCollected(): array
    {
        return $this->commands;
    }

    public function collect(ConsoleEvent|ConsoleErrorEvent|ConsoleTerminateEvent $event): void
    {
        if (!is_object($event) || !$this->isActive()) {
            return;
        }

        $command = $event->getCommand();

        if ($event instanceof ConsoleErrorEvent) {
            $this->commands[$event::class] = [
                'name' => $event->getInput()->getFirstArgument() ?? '',
                'command' => $command,
                'input' => $event->getInput()->__toString(),
                'output' => $event->getOutput()->fetch(),
                'error' => $event->getError()->getMessage(),
                'exitCode' => $event->getExitCode(),
            ];

            return;
        }

        if ($event instanceof ConsoleTerminateEvent) {
            $this->commands[$event::class] = [
                'name' => $command->getName(),
                'command' => $command,
                'input' => $event->getInput()->__toString(),
                'output' => $event->getOutput()->fetch(),
                'exitCode' => $event->getExitCode(),
            ];
            return;
        }

        if ($event instanceof ConsoleEvent) {
            $this->commands[$event::class] = [
                'name' => $command->getName(),
                'command' => $command,
                'input' => $event->getInput()->__toString(),
                'output' => $event->getOutput()->fetch(),
                'arguments' => $command->getDefinition()->getArguments(),
                'options' => $command->getDefinition()->getOptions(),
            ];
        }
    }

    public function getIndexData(): array
    {
        $eventTypes = [
            ConsoleErrorEvent::class,
            ConsoleTerminateEvent::class,
            ConsoleCommandEvent::class,
        ];

        $commandEvent = null;
        foreach ($eventTypes as $eventType) {
            if (!array_key_exists($eventType, $this->commands)) {
                continue;
            }

            $commandEvent = $this->commands[$eventType];
            break;
        }

        if ($commandEvent === null) {
            $types = array_keys($this->commands);
            throw new RuntimeException(
                sprintf(
                    'Unsupported event type encountered among "%s". Supported only "%s"',
                    implode('", "', $types),
                    implode('", "', $eventTypes),
                )
            );
        }

        return [
            'command' => [
                'name' => $commandEvent['name'],
                'class' => $commandEvent['command'] instanceof Command ? $commandEvent['command']::class : null,
                'input' => $commandEvent['input'],
                'exitCode' => $commandEvent['exitCode'] ?? self::UNDEFINED_EXIT_CODE,
            ],
        ];
    }

    private function reset(): void

The method reset() is not used, and could be removed.

    {
        $this->commands = [];
    }
}