From d77b3d9ee9355be41ba73aacd9fd9eecd1344a64 Mon Sep 17 00:00:00 2001
From: Vycheslav Mankevich <sla6843@yandex.ru>
Date: Fri, 2 Jul 2021 11:09:17 +0300
Subject: [PATCH] Added specialized exceptions (#239)

* Added specialized exceptions

* Fixed styling

* Fixed styling

* Fixed styling

Co-authored-by: Vyacheslav Mankevich <vyacheslav.mankevich@team.bumble.com>
---
 Exception/InvalidRefreshTokenException.php           | 9 +++++++++
 Exception/UnknownRefreshTokenException.php           | 9 +++++++++
 Exception/UnknownUserFromRefreshTokenException.php   | 9 +++++++++
 Security/Authenticator/RefreshTokenAuthenticator.php | 6 ++++--
 Service/RefreshToken.php                             | 3 ++-
 5 files changed, 33 insertions(+), 3 deletions(-)
 create mode 100644 Exception/InvalidRefreshTokenException.php
 create mode 100644 Exception/UnknownRefreshTokenException.php
 create mode 100644 Exception/UnknownUserFromRefreshTokenException.php

diff --git a/Exception/InvalidRefreshTokenException.php b/Exception/InvalidRefreshTokenException.php
new file mode 100644
index 00000000..7c9506f9
--- /dev/null
+++ b/Exception/InvalidRefreshTokenException.php
@@ -0,0 +1,9 @@
+<?php
+
+namespace Gesdinet\JWTRefreshTokenBundle\Exception;
+
+use Symfony\Component\Security\Core\Exception\AuthenticationException;
+
+class InvalidRefreshTokenException extends AuthenticationException
+{
+}
diff --git a/Exception/UnknownRefreshTokenException.php b/Exception/UnknownRefreshTokenException.php
new file mode 100644
index 00000000..5e2e570f
--- /dev/null
+++ b/Exception/UnknownRefreshTokenException.php
@@ -0,0 +1,9 @@
+<?php
+
+namespace Gesdinet\JWTRefreshTokenBundle\Exception;
+
+use Symfony\Component\Security\Core\Exception\AuthenticationException;
+
+class UnknownRefreshTokenException extends AuthenticationException
+{
+}
diff --git a/Exception/UnknownUserFromRefreshTokenException.php b/Exception/UnknownUserFromRefreshTokenException.php
new file mode 100644
index 00000000..a219b30a
--- /dev/null
+++ b/Exception/UnknownUserFromRefreshTokenException.php
@@ -0,0 +1,9 @@
+<?php
+
+namespace Gesdinet\JWTRefreshTokenBundle\Exception;
+
+use Symfony\Component\Security\Core\Exception\AuthenticationException;
+
+class UnknownUserFromRefreshTokenException extends AuthenticationException
+{
+}
diff --git a/Security/Authenticator/RefreshTokenAuthenticator.php b/Security/Authenticator/RefreshTokenAuthenticator.php
index 1185b128..8f0c7625 100644
--- a/Security/Authenticator/RefreshTokenAuthenticator.php
+++ b/Security/Authenticator/RefreshTokenAuthenticator.php
@@ -12,6 +12,8 @@
 namespace Gesdinet\JWTRefreshTokenBundle\Security\Authenticator;
 
 use Gesdinet\JWTRefreshTokenBundle\Request\RequestRefreshToken;
+use Gesdinet\JWTRefreshTokenBundle\Exception\UnknownRefreshTokenException;
+use Gesdinet\JWTRefreshTokenBundle\Exception\UnknownUserFromRefreshTokenException;
 use Symfony\Component\HttpFoundation\JsonResponse;
 use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
 use Symfony\Component\Security\Core\Exception\AuthenticationException;
@@ -76,13 +78,13 @@ public function getUser($credentials, UserProviderInterface $userProvider)
         $username = $userProvider->getUsernameForRefreshToken($refreshToken);
 
         if (null === $username) {
-            throw new AuthenticationException(sprintf('Refresh token "%s" does not exist.', $refreshToken));
+            throw new UnknownRefreshTokenException(sprintf('Refresh token "%s" does not exist.', $refreshToken));
         }
 
         $user = $userProvider->loadUserByUsername($username);
 
         if (null === $user) {
-            throw new AuthenticationException(sprintf('User with refresh token "%s" does not exist.', $refreshToken));
+            throw new UnknownUserFromRefreshTokenException(sprintf('User with refresh token "%s" does not exist.', $refreshToken));
         }
 
         $this->userChecker->checkPreAuth($user);
diff --git a/Service/RefreshToken.php b/Service/RefreshToken.php
index 1ccb38a9..72f7a779 100644
--- a/Service/RefreshToken.php
+++ b/Service/RefreshToken.php
@@ -13,6 +13,7 @@
 
 use Gesdinet\JWTRefreshTokenBundle\Event\RefreshEvent;
 use Gesdinet\JWTRefreshTokenBundle\Security\Authenticator\RefreshTokenAuthenticator;
+use Gesdinet\JWTRefreshTokenBundle\Exception\InvalidRefreshTokenException;
 use Symfony\Component\EventDispatcher\EventDispatcherInterface;
 use Symfony\Contracts\EventDispatcher\EventDispatcherInterface as ContractsEventDispatcherInterface;
 use InvalidArgumentException;
@@ -133,7 +134,7 @@ public function refresh(Request $request)
         if (null === $refreshToken || !$refreshToken->isValid()) {
             return $this->failureHandler->onAuthenticationFailure(
                 $request,
-                new AuthenticationException(
+                new InvalidRefreshTokenException(
                     sprintf('Refresh token "%s" is invalid.', $refreshToken)
                 )
             );