src/Listener/Command/DatabaseListener.php line 23

Open in your IDE?
  1. <?php
  3. namespace App\Listener\Command;
  5. use Symfony\Component\Console\Command\Command;
  6. use Symfony\Component\Console\Event\ConsoleCommandEvent;
  8. /**
  9.  * Class DatabaseListener
  10.  * @package App\Listener\Command
  11.  */
  12. class DatabaseListener
  13. {
  14.     private array $allowedCommands = [
  15.         'doctrine:database:drop',
  16.         'doctrine:database:create'
  17.     ];
  19.     /**
  20.      * @param ConsoleCommandEvent $event
  21.      * @throws \Exception
  22.      */
  23.     public function onConsoleCommand(ConsoleCommandEvent $event)
  24.     {
  25.         $command = $event->getCommand();
  27.         $input = $event->getInput();
  28.         if (!$this->isProperCommand($command)) {
  29.             return;
  30.         }
  32.         if ($input->hasOption('tenant') && $input->getOption('tenant') !== null) {
  33.             $connection = 'tenant';
  34.         } else {
  35.             $connection = 'main';
  36.         }
  38.         $input->setOption('connection', $connection);
  39.         $input->setOption('em', $connection);
  40.         $command->getDefinition()->getOption('connection')->setDefault($connection);
  41.         $command->getDefinition()->getOption('em')->setDefault($connection);
  42.     }
  44.     /**
  45.      * @param Command $command
  46.      * @return bool
  47.      */
  48.     private function isProperCommand(Command $command)
  49.     {
  50.         return in_array($command->getName(), $this->allowedCommands);
  51.     }
  52. }