added redirections after logging in or out (#253)

* added redirections after logging in or out * added redirections after logging in or out
ZdruzenieSTROM · Dec 9, 2023 · 72a0ba6 · 72a0ba6
1 parent e1731f1
commit 72a0ba6
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/src/components/PageLayout/Authentication/Authentication.tsx b/src/components/PageLayout/Authentication/Authentication.tsx
@@ -1,4 +1,5 @@
 import Link from 'next/link'
+import {useRouter} from 'next/router'
 import {FC, useState} from 'react'
 
 import {AuthContainer} from '@/utils/AuthContainer'
@@ -18,6 +19,15 @@ export const Authentication: FC = () => {
 
   const {seminar} = useSeminarInfo()
 
+  const router = useRouter()
+
+  const redirectLogout = () => {
+    logout()
+    if (router.asPath.endsWith('profil') || router.asPath.endsWith('profil/uprava')) {
+      router.push('/')
+    }
+  }
+
   if (!isAuthed) {
     return (
       <>
@@ -38,7 +48,7 @@ export const Authentication: FC = () => {
     return (
       <div className={styles.authenticationDisplayButtons}>
         <Link href={`/${seminar}/profil`}>Profil</Link>
-        <span onClick={() => logout()}>Odhlásiť</span>
+        <span onClick={redirectLogout}>Odhlásiť</span>
       </div>
     )
   }

diff --git a/src/components/PageLayout/LoginForm/LoginForm.tsx b/src/components/PageLayout/LoginForm/LoginForm.tsx
@@ -1,3 +1,4 @@
+import {useRouter} from 'next/router'
 import {FC} from 'react'
 import {SubmitHandler, useForm} from 'react-hook-form'
 
@@ -24,8 +25,17 @@ export const LoginForm: FC<ILoginForm> = ({closeOverlay}) => {
   const {login} = AuthContainer.useContainer()
   const {handleSubmit, control} = useForm<LoginFormValues>({defaultValues})
 
+  const router = useRouter()
+
+  const redirectClose = () => {
+    closeOverlay()
+    if (router.asPath.endsWith('registracia')) {
+      router.push('/')
+    }
+  }
+
   const onSubmit: SubmitHandler<LoginFormValues> = (data) => {
-    login({data, onSuccess: closeOverlay})
+    login({data, onSuccess: redirectClose})
   }
 
   const requiredRule = {required: '* Toto pole nemôže byť prázdne.'}