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

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: Remove this command.
  27.  *
  28.  * @deprecated since sonata-project/user-bundle 4.14, it will be removed on 5.0.
  29.  */
  30. class TwoStepVerificationCommand extends ContainerAwareCommand
  31. {
  32.     /**
  33.      * @var ?Helper
  34.      */
  35.     private $helper;
  37.     /**
  38.      * @var ?UserManagerInterface
  39.      */
  40.     private $userManager;
  42.     /**
  43.      * NEXT_MAJOR: make $helper and $userManager mandatory (but still nullable).
  44.      */
  45.     public function __construct(
  46.         ?string $name,
  47.         ?Helper $helper = null,
  48.         ?UserManagerInterface $userManager = null
  49.     ) {
  50.         parent::__construct($name);
  52.         $this->helper = $helper;
  53.         $this->userManager = $userManager;
  54.     }
  56.     public function configure(): void
  57.     {
  58.         $this->setName('sonata:user:two-step-verification');
  59.         $this->addArgument(
  60.             'username',
  61.             InputArgument::REQUIRED,
  62.             'The username to protect with a two step verification process'
  63.         );
  64.         $this->addOption('reset', null, InputOption::VALUE_NONE, 'Reset the current two step verification token');
  65.         $this->setDescription(
  66.             'Generate a two step verification process to secure an access (Ideal for super admin protection)'
  67.         );
  68.     }
  70.     public function execute(InputInterface $input, OutputInterface $output): void
  71.     {
  72.         if (null === $this->helper && !$this->getContainer()->has('sonata.user.google.authenticator.provider')) {
  73.             throw new \RuntimeException('Two Step Verification process is not enabled');
  74.         }
  76.         if (null === $this->helper) {
  77.             @trigger_error(sprintf(
  78.                 'Not providing the $helper argument of "%s::__construct()" is deprecated since 4.3.0 and will no longer be possible in 5.0',
  79.                 __CLASS__
  80.             ), \E_USER_DEPRECATED);
  81.             $helper = $this->getContainer()->get('sonata.user.google.authenticator.provider');
  82.             \assert($helper instanceof Helper);
  83.             $this->helper = $helper;
  84.         }
  86.         if (null === $this->userManager) {
  87.             @trigger_error(sprintf(
  88.                 'Not providing the $userManager argument of "%s::__construct()" is deprecated since 4.3.0 and will no longer be possible in 5.0',
  89.                 __CLASS__
  90.             ), \E_USER_DEPRECATED);
  91.             $manager = $this->getContainer()->get('fos_user.user_manager');
  92.             \assert($manager instanceof UserManagerInterface);
  93.             $this->userManager = $manager;
  94.         }
  96.         $user = $this->userManager->findUserByUsernameOrEmail($input->getArgument('username'));
  97.         \assert($user instanceof UserInterface);
  99.         if (!$user) {
  100.             throw new \RuntimeException(sprintf('Unable to find the username : %s', $input->getArgument('username')));
  101.         }
  103.         if (!$user->getTwoStepVerificationCode() || $input->getOption('reset')) {
  104.             $user->setTwoStepVerificationCode($this->helper->generateSecret());
  105.             $this->userManager->updateUser($user);
  106.         }
  108.         $output->writeln([
  109.             sprintf('<info>Username</info> : %s', $input->getArgument('username')),
  110.             sprintf('<info>Secret</info> : %s', $user->getTwoStepVerificationCode()),
  111.             sprintf('<info>Url</info> : %s', $this->helper->getUrl($user)),
  112.         ]);
  113.     }
  114. }