diff --git a/public/index.php b/public/index.php index 1a926eb..ddca47a 100644 --- a/public/index.php +++ b/public/index.php @@ -29,7 +29,7 @@ if ($match !== null) { } if (!$authorized) { - Container::$request->session()->set('redirect_after_login', $url); + Container::$request->session()->set('redirect_after_login', '/' . $url); $response = new Redirect(Container::$routeCollection->getRoute('login')->generateLink(), IRedirect::TEMPORARY); header('Location: ' . $response->getUrl(), true, $response->getHttpCode()); return; diff --git a/src/Controller/LoginController.php b/src/Controller/LoginController.php index 762491e..8588ebf 100644 --- a/src/Controller/LoginController.php +++ b/src/Controller/LoginController.php @@ -41,7 +41,7 @@ class LoginController return new Redirect(\Container::$routeCollection->getRoute('index')->generateLink(), IRedirect::TEMPORARY); } - return new HtmlContent('login/login'); + return new HtmlContent('login/login', ['redirectUrl' => $this->getRedirectUrl()]); } public function getGoogleLoginRedirect(): IRedirect @@ -91,12 +91,13 @@ class LoginController $user = $this->userRepository->getById($resetter->getUserId()); - return new HtmlContent('login/reset_password', ['success' => true, 'token' => $token, 'email' => $user->getEmail()]); + return new HtmlContent('login/reset_password', ['success' => true, 'token' => $token, 'email' => $user->getEmail(), 'redirectUrl' => $this->getRedirectUrl()]); } public function login(): IContent { if ($this->request->user() !== null) { + $this->deleteRedirectUrl(); return new JsonContent(['success' => true]); } @@ -112,20 +113,17 @@ class LoginController $this->request->setUser($user); + $this->deleteRedirectUrl(); return new JsonContent(['success' => true]); } public function loginWithGoogle() { - $redirectUrl = $this->request->session()->get('redirect_after_login'); - if ($redirectUrl === null) { - $redirectUrl = \Container::$routeCollection->getRoute('index')->generateLink(); - } $defaultError = 'Authentication with Google failed. Please try again!'; if ($this->request->user() !== null) { - $this->request->session()->delete('redirect_after_login'); - return new Redirect($redirectUrl, IRedirect::TEMPORARY); + $this->deleteRedirectUrl(); + return new Redirect($this->getRedirectUrl(), IRedirect::TEMPORARY); } if ($this->request->query('state') !== $this->request->session()->get('oauth_state')) { @@ -160,8 +158,8 @@ class LoginController $this->request->setUser($user); - $this->request->session()->delete('redirect_after_login'); - return new Redirect($redirectUrl, IRedirect::TEMPORARY); + $this->deleteRedirectUrl(); + return new Redirect($this->getRedirectUrl(), IRedirect::TEMPORARY); } public function logout(): IRedirect @@ -174,9 +172,10 @@ class LoginController public function requestPasswordReset(): IContent { if ($this->request->user() !== null) { + $this->deleteRedirectUrl(); return new JsonContent([ 'redirect' => [ - 'target' => '/' . \Container::$routeCollection->getRoute('home')->generateLink() + 'target' => $this->getRedirectUrl() ] ]); } @@ -237,9 +236,10 @@ class LoginController public function resetPassword(): IContent { if ($this->request->user() !== null) { + $this->deleteRedirectUrl(); return new JsonContent([ 'redirect' => [ - 'target' => '/' . \Container::$routeCollection->getRoute('home')->generateLink() + 'target' => $this->getRedirectUrl() ] ]); } @@ -280,6 +280,7 @@ class LoginController $this->request->setUser($user); + $this->deleteRedirectUrl(); return new JsonContent(['success' => true]); } @@ -296,4 +297,18 @@ class LoginController ]); $mail->send(); } + + private function getRedirectUrl(): string + { + $redirectUrl = $this->request->session()->get('redirect_after_login'); + if ($redirectUrl === null) { + return \Container::$routeCollection->getRoute('index')->generateLink(); + } + return $redirectUrl; + } + + private function deleteRedirectUrl(): void + { + $this->request->session()->delete('redirect_after_login'); + } } diff --git a/views/login/login.php b/views/login/login.php index 0a004bb..6c3cc79 100644 --- a/views/login/login.php +++ b/views/login/login.php @@ -3,7 +3,7 @@ @section(main)

Login

-
+

diff --git a/views/login/reset_password.php b/views/login/reset_password.php index 28dc505..f88903f 100644 --- a/views/login/reset_password.php +++ b/views/login/reset_password.php @@ -4,7 +4,7 @@

Reset password

- +