src/EventSubscriber/CheckVerifiedUserSubscriber.php line 99

  1. <?php
  2. /**
  3.  * CheckVerifiedUserSubscriber.
  4.  *
  5.  * @author  SOLTIX <lukasz.m@soltix.pl>
  6.  * @license <https://opensource.org/licenses/MIT> MIT
  7.  */
  8. declare(strict_types=1);
  11. namespace App\EventSubscriber;
  13. use App\Entity\User;
  14. use Doctrine\ORM\EntityManager;
  15. use Doctrine\Persistence\ManagerRegistry;
  16. use Scheb\TwoFactorBundle\Security\Authentication\Token\TwoFactorToken;
  17. use Scheb\TwoFactorBundle\Security\TwoFactor\Event\TwoFactorAuthenticationEvent;
  18. use Scheb\TwoFactorBundle\Security\TwoFactor\Event\TwoFactorAuthenticationEvents;
  19. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  20. use Symfony\Component\HttpFoundation\Session\Session;
  21. use Symfony\Component\Security\Http\Event\LoginSuccessEvent;
  23. class CheckVerifiedUserSubscriber implements EventSubscriberInterface
  24. {
  26.     /**
  27.      * @var ManagerRegistry
  28.      */
  29.     private ManagerRegistry $managerRegistry;
  32.     /**
  33.      * @param ManagerRegistry $managerRegistry
  34.      */
  35.     public function __construct(ManagerRegistry $managerRegistry)
  36.     {
  37.         $this->managerRegistry = $managerRegistry;
  39.     }//end __construct()
  42.     /**
  43.      * Returns an array of event names this subscriber wants to listen to.
  44.      *
  45.      * The array keys are event names and the value can be:
  46.      *
  47.      *  * The method name to call (priority defaults to 0)
  48.      *  * An array composed of the method name to call and the priority
  49.      *  * An array of arrays composed of the method names to call and respective
  50.      *    priorities, or 0 if unset
  51.      *
  52.      * For instance:
  53.      *
  54.      *  * ['eventName' => 'methodName']
  55.      *  * ['eventName' => ['methodName', $priority]]
  56.      *  * ['eventName' => [['methodName1', $priority], ['methodName2']]]
  57.      *
  58.      * The code must not depend on runtime state as it will only be called at compile time.
  59.      * All logic depending on runtime state must be put into the individual methods handling the events.
  60.      *
  61.      * @return array<string, string|array{0: string, 1: int}|list<array{0: string, 1?: int}>>
  62.      */
  63.     public static function getSubscribedEvents()
  64.     {
  65.         return [
  66.             LoginSuccessEvent::class                => 'onSuccessLogin',
  67.             TwoFactorAuthenticationEvents::COMPLETE => 'onSuccessTwoFactorAuthentication',
  68.         ];
  70.     }//end getSubscribedEvents()
  73.     /**
  74.      * @param LoginSuccessEvent $event
  75.      *
  76.      * @return void
  77.      */
  78.     public function onSuccessLogin(LoginSuccessEvent $event): void
  79.     {
  80.         $session = new Session();
  81.         $user    = $event->getUser();
  83.         if ($user instanceof User && $user->isFirstLogin() === true) {
  84.             $session->set('is_change_password', true);
  85.             $user->setFirstLogin(false);
  86.             $this->managerRegistry->getManager()->flush();
  87.         } else {
  88.             $session->set('is_change_password', false);
  89.         }
  91.     }//end onSuccessLogin()
  94.     /**
  95.      * @param TwoFactorAuthenticationEvent $event
  96.      *
  97.      * @return void
  98.      */
  99.     public function onSuccessTwoFactorAuthentication(TwoFactorAuthenticationEvent $event): void
  100.     {
  101.         $token = $event->getToken();
  102.         $user  = $token->getUser();
  103.         $user?->setUse2FaCode(true);
  104.         $this->managerRegistry->getManager()->flush();
  106.     }//end onSuccessTwoFactorAuthentication()
  109. }//end class