fix logout

pages/api/users/logout.ts

@@ -21,11 +21,7 @@ router.all(async (req, res) => {
 
   removeTokenCookie(res);
 
-  res.status(200).json({
-    message: 'Logged out.',
-    statusCode: 200,
-    success: true,
-  });
+  res.redirect('/');
 });
 
 const handler = router.handler(NextConnectOptions);

pages/api/users/register.ts

@@ -1,3 +1,4 @@
+import { setLoginSession } from '@/config/auth/session';
 import { NextApiRequest, NextApiResponse } from 'next';
 import { z } from 'zod';
 import ServerError from '@/config/util/ServerError';
@@ -35,6 +36,11 @@ const registerUser = async (req: RegisterUserRequest, res: NextApiResponse) => {
   }
 
   const user = await createNewUser(req.body);
+
+  await setLoginSession(res, {
+    id: user.id,
+    username: user.username,
+  });
   res.status(201).json({
     message: 'User created successfully.',
     payload: user,

pages/logout/index.tsx

@@ -1,16 +0,0 @@
-import { NextPage } from 'next';
-import { useRouter } from 'next/router';
-import { useEffect } from 'react';
-
-const LogoutPage: NextPage = () => {
-  const router = useRouter();
-
-  useEffect(() => {
-    document.cookie = 'token=; expires=Thu, 01 Jan 1970 00:00:01 GMT;';
-    router.reload();
-    router.push('/');
-  }, [router]);
-  return null;
-};
-
-export default LogoutPage;

pages/register/index.tsx

@@ -10,6 +10,7 @@ import sendRegisterUserRequest from '@/requests/sendRegisterUserRequest';
 import CreateUserValidationSchema from '@/services/User/schema/CreateUserValidationSchema';
 import { zodResolver } from '@hookform/resolvers/zod';
 import { NextPage } from 'next';
+import { useRouter } from 'next/router';
 import { useState } from 'react';
 import { useForm } from 'react-hook-form';
 import { FaUserCircle } from 'react-icons/fa';
@@ -18,6 +19,7 @@ import { z } from 'zod';
 interface RegisterUserProps {}
 
 const RegisterUserPage: NextPage<RegisterUserProps> = () => {
+  const router = useRouter();
   const { reset, register, handleSubmit, formState } = useForm<
     z.infer<typeof CreateUserValidationSchema>
   >({
@@ -31,7 +33,7 @@ const RegisterUserPage: NextPage<RegisterUserProps> = () => {
   const onSubmit = async (data: z.infer<typeof CreateUserValidationSchema>) => {
     try {
       await sendRegisterUserRequest(data);
-      reset();
+      router.push('/', undefined, { shallow: true });
     } catch (error) {
       setServerResponseError(
         error instanceof Error