src/Security/GoogleAuthenticator.php line 21

Open in your IDE?
  1. <?php
  3. namespace App\Security;
  5. use App\Kernel;
  6. use App\Entity\User; // your user entity
  7. use Doctrine\ORM\EntityManagerInterface;
  8. use Symfony\Component\HttpFoundation\Request;
  9. use Symfony\Component\HttpFoundation\Response;
  10. use Symfony\Component\Routing\RouterInterface;
  11. use League\OAuth2\Client\Provider\GoogleUser;
  12. use KnpU\OAuth2ClientBundle\Client\ClientRegistry;
  13. use Symfony\Component\HttpFoundation\RedirectResponse;
  14. use Symfony\Component\Security\Http\Authenticator\Passport\Passport;
  15. use Symfony\Component\Security\Core\Exception\AuthenticationException;
  16. use KnpU\OAuth2ClientBundle\Security\Authenticator\OAuth2Authenticator;
  17. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  18. use Symfony\Component\Security\Http\Authenticator\Passport\Badge\UserBadge;
  19. use Symfony\Component\Security\Http\Authenticator\Passport\SelfValidatingPassport;
  21. class GoogleAuthenticator extends OAuth2Authenticator
  22. {
  23.     private $clientRegistry;
  24.     private $entityManager;
  25.     private $router;
  26.     private $kernel;
  28.     public function __construct(ClientRegistry $clientRegistry, EntityManagerInterface $entityManager, RouterInterface $router, Kernel $kernel)
  29.     {
  30.         $this->clientRegistry = $clientRegistry;
  31.         $this->entityManager = $entityManager;
  32.         $this->router = $router;
  33.         $this->kernel = $kernel;
  34.     }
  36.     public function supports(Request $request): ?bool
  37.     {
  38.         // continue ONLY if the current ROUTE matches the check ROUTE
  39.         return $request->attributes->get('_route') === 'connect_google_check';
  40.     }
  42.     public function authenticate(Request $request): Passport
  43.     {
  44.         $client = $this->clientRegistry->getClient('google_main');
  45.         $accessToken = $this->fetchAccessToken($client);
  47.         return new SelfValidatingPassport(
  48.             new UserBadge($accessToken->getToken(), function() use ($accessToken, $client) {
  49.                 /** @var GoogleUser $googleUser */
  50.                 $googleUser = $client->fetchUserFromToken($accessToken);
  52.                 $email = $googleUser->getEmail();
  54.                 // Если в базе нет пользователя с таким email, то сохраняем его
  55.                 /** @var User $user */
  56.                 $user = $this->entityManager->getRepository(User::class)->findOneBy(['email' => $email]);
  57.                 
  58.                 if(!$user->getId()) {
  59.                     $user = new User();
  60.                 }
  61.                                 
  62.                 if ($googleUser->getId() != $user->getGoogleId()) {
  63.                     $user->setGoogleId($googleUser->getId());
  64.                 }
  65.                 
  66.                 if (!$user->getName()) {
  67.                     $user->setName($googleUser->getFirstName());
  68.                 }
  70.                 if (!$user->getSurname()) {
  71.                     $user->setSurname($googleUser->getLastName());
  72.                 }
  74.                 $this->entityManager->persist($user);
  75.                 $this->entityManager->flush();
  77.                 // Если в базе еще нет фото, а в фейсбук есть фото, то сохраняем его
  78.                 // $user_pic = $this->kernel->getProjectDir()."/public/pic/user/".$user->getId().".jpg";
  79.                 // if ($googleUser->getAvatar() && !file_exists($user_pic)) {
  80.                 //     copy($googleUser->getAvatar(), $user_pic);
  81.                 // }
  83.                 return $user;
  84.             })
  85.         );
  86.     }
  88.     public function onAuthenticationSuccess(Request $request, TokenInterface $token, string $firewallName): ?Response
  89.     {
  90.         $targetUrl = $this->router->generate('user');
  91.         return new RedirectResponse($targetUrl);
  92.     }
  94.     public function onAuthenticationFailure(Request $request, AuthenticationException $exception): ?Response
  95.     {
  96.         $message = strtr($exception->getMessageKey(), $exception->getMessageData());
  98.         return new Response($message, Response::HTTP_FORBIDDEN);
  99.     }
  100. }