ConsoleLifecycleSubscriber.php

<?php

namespace Perfbase\Drupal\EventSubscriber;

use Perfbase\Drupal\Config\ConfigResolver;
use Perfbase\Drupal\Lifecycle\ConsoleCommandLifecycle;
use Perfbase\Drupal\Runtime\ActiveLifecycleRegistry;
use Perfbase\Drupal\Runtime\PerfbaseFactoryInterface;
use Perfbase\Drupal\Support\ErrorHandler;
use Symfony\Component\Console\Command\Command;
use Symfony\Component\Console\ConsoleEvents;
use Symfony\Component\Console\Event\ConsoleCommandEvent;
use Symfony\Component\Console\Event\ConsoleErrorEvent;
use Symfony\Component\Console\Event\ConsoleTerminateEvent;
use Symfony\Component\EventDispatcher\EventSubscriberInterface;

/**
 *
 */
class ConsoleLifecycleSubscriber implements EventSubscriberInterface {
  private const KEY = 'console.current';

  private PerfbaseFactoryInterface $factory;

  private ConfigResolver $configResolver;

  private ErrorHandler $errorHandler;

  private ActiveLifecycleRegistry $registry;

  public function __construct(
    PerfbaseFactoryInterface $factory,
    ConfigResolver $configResolver,
    ErrorHandler $errorHandler,
    ActiveLifecycleRegistry $registry,
  ) {
    $this->factory = $factory;
    $this->configResolver = $configResolver;
    $this->errorHandler = $errorHandler;
    $this->registry = $registry;
  }

  /**
   * @return array<string, string>
   */
  public static function getSubscribedEvents(): array {
    return [
      ConsoleEvents::COMMAND => 'onCommand',
      ConsoleEvents::ERROR => 'onError',
      ConsoleEvents::TERMINATE => 'onTerminate',
    ];
  }

  /**
   *
   */
  public function onCommand(ConsoleCommandEvent $event): void {
    $commandName = $this->resolveCommandName($event);
    $lifecycle = new ConsoleCommandLifecycle(
          $commandName,
          $this->factory,
          $this->configResolver,
          $this->errorHandler
      );

    $this->registry->put(self::KEY, $lifecycle);
    $lifecycle->startProfiling();
  }

  /**
   *
   */
  public function onError(ConsoleErrorEvent $event): void {
    $lifecycle = $this->registry->get(self::KEY);
    if ($lifecycle instanceof ConsoleCommandLifecycle) {
      $lifecycle->setException($event->getError()->getMessage());
    }
  }

  /**
   *
   */
  public function onTerminate(ConsoleTerminateEvent $event): void {
    $lifecycle = $this->registry->pull(self::KEY);
    if ($lifecycle instanceof ConsoleCommandLifecycle) {
      $lifecycle->setExitCode($event->getExitCode());
      $lifecycle->stopProfiling();
    }
  }

  /**
   *
   */
  private function resolveCommandName(ConsoleCommandEvent $event): string {
    $command = $event->getCommand();
    if ($command instanceof Command && is_string($command->getName()) && $command->getName() !== '') {
      return $command->getName();
    }

    $firstArgument = $event->getInput()->getFirstArgument();

    return is_string($firstArgument) && $firstArgument !== '' ? $firstArgument : 'unknown';
  }

}