src/EventListener/JWTAuthenticatedListener.php line 35

Open in your IDE?
  1. <?php
  2. /**
  3.  * Custom response success auth
  4.  *
  5.  * @package RMCS
  6.  * @author Vlad Shashkov  <vlad.s@zimalab.com>
  7.  * @copyright 2014 - 2019 The Zimalab
  8.  */
  9. declare(strict_types=1);
  11. namespace App\EventListener;
  13. use Doctrine\Persistence\{ManagerRegistry, ObjectManager};
  14. use Gesdinet\JWTRefreshTokenBundle\Entity\RefreshToken;
  15. use Lexik\Bundle\JWTAuthenticationBundle\Event\AuthenticationSuccessEvent;
  16. use Symfony\Component\Security\Core\User\UserInterface;
  18. class JWTAuthenticatedListener
  19. {
  20.     /**
  21.      * @var ObjectManager
  22.      */
  23.     private $em;
  25.     public function __construct(ManagerRegistry $registry)
  26.     {
  27.         $this->em = $registry->getManager('default');
  28.     }
  30.     /**
  31.      * Event auth success
  32.      *
  33.      * @param AuthenticationSuccessEvent $event
  34.      */
  35.     public function onAuthenticationSuccessResponse(AuthenticationSuccessEvent $event):void
  36.     {
  37.         $this->removeOldToken();
  38.         $this->setRolesResponse($event);
  39.     }
  41.     /**
  42.      * Remove old refresh token
  43.      */
  44.     private function removeOldToken():void
  45.     {
  46.         $refreshTokens = $this->em
  47.             ->getRepository(RefreshToken::class)
  48.             ->findInvalid();
  49.         if (!empty($refreshTokens)) {
  50.             foreach ($refreshTokens as $refreshToken) {
  51.                 $this->em->remove($refreshToken);
  52.             }
  53.             $this->em->flush();
  54.         }
  55.     }
  57.     /**
  58.      * Set response role
  59.      *
  60.      * @param AuthenticationSuccessEvent $event
  61.      */
  62.     private function setRolesResponse(AuthenticationSuccessEvent $event):void
  63.     {
  64.         $data = $event->getData();
  65.         $user = $event->getUser();
  67.         if (!$user instanceof UserInterface) {
  68.             return;
  69.         }
  70.         $data['roles'] = $user->getRoles();
  72.         $event->setData($data);
  73.     }
  74. }