vendor/sonata-project/user-bundle/src/Command/TwoStepVerificationCommand.php line 28

Open in your IDE?
  1. <?php
  3. declare(strict_types=1);
  5. /*
  6.  * This file is part of the Sonata Project package.
  7.  *
  8.  * (c) Thomas Rabaix <thomas.rabaix@sonata-project.org>
  9.  *
  10.  * For the full copyright and license information, please view the LICENSE
  11.  * file that was distributed with this source code.
  12.  */
  14. namespace Sonata\UserBundle\Command;
  16. use FOS\UserBundle\Model\UserManagerInterface;
  17. use Sonata\UserBundle\GoogleAuthenticator\Helper;
  18. use Sonata\UserBundle\Model\UserInterface;
  19. use Symfony\Bundle\FrameworkBundle\Command\ContainerAwareCommand;
  20. use Symfony\Component\Console\Input\InputArgument;
  21. use Symfony\Component\Console\Input\InputInterface;
  22. use Symfony\Component\Console\Input\InputOption;
  23. use Symfony\Component\Console\Output\OutputInterface;
  25. /**
  26.  * NEXT_MAJOR: stop extending ContainerAwareCommand.
  27.  */
  28. class TwoStepVerificationCommand extends ContainerAwareCommand
  29. {
  30.     /**
  31.      * @var ?Helper
  32.      */
  33.     private $helper;
  35.     /**
  36.      * @var ?UserManagerInterface
  37.      */
  38.     private $userManager;
  40.     /**
  41.      * NEXT_MAJOR: make $helper and $userManager mandatory (but still nullable).
  42.      */
  43.     public function __construct(
  44.         ?string $name,
  45.         ?Helper $helper = null,
  46.         ?UserManagerInterface $userManager = null
  47.     ) {
  48.         parent::__construct($name);
  50.         $this->helper = $helper;
  51.         $this->userManager = $userManager;
  52.     }
  54.     /**
  55.      * {@inheritdoc}
  56.      */
  57.     public function configure(): void
  58.     {
  59.         $this->setName('sonata:user:two-step-verification');
  60.         $this->addArgument(
  61.             'username',
  62.             InputArgument::REQUIRED,
  63.             'The username to protect with a two step verification process'
  64.         );
  65.         $this->addOption('reset', null, InputOption::VALUE_NONE, 'Reset the current two step verification token');
  66.         $this->setDescription(
  67.             'Generate a two step verification process to secure an access (Ideal for super admin protection)'
  68.         );
  69.     }
  71.     /**
  72.      * {@inheritdoc}
  73.      */
  74.     public function execute(InputInterface $input, OutputInterface $output): void
  75.     {
  76.         if (null === $this->helper && !$this->getContainer()->has('sonata.user.google.authenticator.provider')) {
  77.             throw new \RuntimeException('Two Step Verification process is not enabled');
  78.         }
  80.         if (null === $this->helper) {
  81.             @trigger_error(sprintf(
  82.                 'Not providing the $helper argument of "%s::__construct()" is deprecated since 4.3.0 and will no longer be possible in 5.0',
  83.                 __CLASS__
  84.             ), E_USER_DEPRECATED);
  85.             $helper = $this->getContainer()->get('sonata.user.google.authenticator.provider');
  86.             \assert($helper instanceof Helper);
  87.             $this->helper = $helper;
  88.         }
  90.         if (null === $this->userManager) {
  91.             @trigger_error(sprintf(
  92.                 'Not providing the $userManager argument of "%s::__construct()" is deprecated since 4.3.0 and will no longer be possible in 5.0',
  93.                 __CLASS__
  94.             ), E_USER_DEPRECATED);
  95.             $manager = $this->getContainer()->get('fos_user.user_manager');
  96.             \assert($manager instanceof UserManagerInterface);
  97.             $this->userManager = $manager;
  98.         }
  100.         $user = $this->userManager->findUserByUsernameOrEmail($input->getArgument('username'));
  101.         \assert($user instanceof UserInterface);
  103.         if (!$user) {
  104.             throw new \RuntimeException(sprintf('Unable to find the username : %s', $input->getArgument('username')));
  105.         }
  107.         if (!$user->getTwoStepVerificationCode() || $input->getOption('reset')) {
  108.             $user->setTwoStepVerificationCode($this->helper->generateSecret());
  109.             $this->userManager->updateUser($user);
  110.         }
  112.         $output->writeln([
  113.             sprintf('<info>Username</info> : %s', $input->getArgument('username')),
  114.             sprintf('<info>Secret</info> : %s', $user->getTwoStepVerificationCode()),
  115.             sprintf('<info>Url</info> : %s', $this->helper->getUrl($user)),
  116.         ]);
  117.     }
  118. }