src/Listener/Command/CommandListener.php line 47

Open in your IDE?
  1. <?php
  3. namespace App\Listener\Command;
  5. use App\Doctrine\DBAL\TenantConnection;
  6. use App\Entity\Main\Tenant;
  7. use App\Repository\Main\TenantRepository;
  8. use Exception;
  9. use Symfony\Component\Console\Command\Command;
  10. use Symfony\Component\Console\Event\ConsoleCommandEvent;
  11. use Symfony\Component\Console\Input\InputArgument;
  12. use Symfony\Component\Console\Input\InputOption;
  14. /**
  15.  * Class CommandListener
  16.  * @package App\Listener\Command
  17.  */
  18. class CommandListener
  19. {
  20.     private array $allowedCommands;
  22.     private TenantConnection $tenantConnection;
  24.     private TenantRepository $tenantRepository;
  26.     /**
  27.      * CommandListener constructor.
  28.      *
  29.      * @param TenantConnection $tenantConnection
  30.      * @param TenantRepository $tenantRepository
  31.      * @param array $allowedCommands
  32.      */
  33.     public function __construct(
  34.         TenantConnection $tenantConnection,
  35.         TenantRepository $tenantRepository,
  36.         $allowedCommands = []
  37.     ) {
  38.         $this->tenantConnection = $tenantConnection;
  39.         $this->tenantRepository = $tenantRepository;
  40.         $this->allowedCommands = $allowedCommands;
  41.     }
  43.     /**
  44.      * @param ConsoleCommandEvent $event
  45.      * @throws Exception
  46.      */
  47.     public function onConsoleCommand(ConsoleCommandEvent $event)
  48.     {
  49.         $command = $event->getCommand();
  50.         $input = $event->getInput();
  52.         if (!$this->isProperCommand($command)) {
  53.             return;
  54.         }
  56.         $command->addArgument('tenant', InputArgument::OPTIONAL, 'Tenant ID');
  58.         if (!$command->getDefinition()->hasOption('em')) {
  59.             $command->getDefinition()->addOption(
  60.                 new InputOption('em', null, InputOption::VALUE_OPTIONAL, 'The entity manager to use for this command')
  61.             );
  62.         }
  64.         $input->bind($command->getDefinition());
  66.         if (is_null($input->getArgument('tenant')) || strpos($input->getArgument('tenant'), 'tenant=') === false) {
  67.             return;
  68.         }
  70.         $tenantID = (int) str_replace('tenant=', '', $input->getArgument('tenant'));
  73.         /** @var Tenant $tenant */
  74.         $tenant = $this->tenantRepository->find($tenantID);
  76.         if (!$tenant instanceof Tenant) {
  77.             throw new Exception(sprintf('Tenant id° %s does not exists', $tenantID));
  78.         }
  80.         $input->setOption('em', 'tenant');
  81.         $command->getDefinition()->getOption('em')->setDefault('tenant');
  83.         $this->tenantConnection->changeParams($tenant->getDbName(), $tenant->getDbUser(), $tenant->getDbPassword());
  84.     }
  86.     /**
  87.      * @param Command $command
  88.      * @return bool
  89.      */
  90.     private function isProperCommand(
  91.         Command $command
  92.     ): bool {
  93.         return in_array($command->getName(), $this->allowedCommands);
  94.     }
  95. }