Merge pull request #13 from Dialogue-Bot/DIAL-19-logout

DIAL-19-logout

Author: hi02102002

client/src/apis/auth.ts

@@ -37,6 +37,10 @@ class Auth {
    loginWithIdToken(idToken: string): Promise<TBaseResponse<TToken>> {
       return http_client.post(ENDPOINTS.AUTH.WITH_ID_TOKEN, { idToken });
    }
+
+   logout(): Promise<TBaseResponse<null>> {
+      return http_client.post(ENDPOINTS.AUTH.LOGOUT);
+   }
 }
 
 export const auth = new Auth();

client/src/constants/index.ts

@@ -9,6 +9,7 @@ export const ENDPOINTS = {
       FORGOT_PASSWORD: '/auth/forgot-password',
       RESET_PASSWORD: '/auth/reset-password',
       WITH_ID_TOKEN: '/auth/with-id-token',
+      LOGOUT: '/auth/logout',
    },
    UPLOAD: {
       SINGLE: '/upload/single',

client/src/hooks/auth/index.ts

@@ -2,5 +2,6 @@ export * from './use-forgot-pass';
 export * from './use-login';
 export * from './use-login-with-id-token';
 export * from './use-login-with-provider';
+export * from './use-logout';
 export * from './use-register';
 export * from './use-set-pass';

client/src/hooks/auth/use-login-with-provider.ts

@@ -20,9 +20,7 @@ export const useLoginWithProvider = (provider: AuthProvider) => {
 
          return auth.loginWithIdToken(idToken);
       },
-      onSuccess(data) {
-         toast.success(data.message);
-
+      async onSuccess() {
          window.location.href = search.redirect || '/dashboard';
       },
       onError(err: any) {

client/src/hooks/auth/use-login.ts

@@ -15,9 +15,7 @@ export const useLogin = () => {
       mutationFn: (data: TLogin) => {
          return auth.login(data);
       },
-      onSuccess(data) {
-         toast.success(data.message);
-
+      async onSuccess() {
          window.location.href = search.redirect || '/dashboard';
       },
       onError(err: any) {

client/src/hooks/auth/use-logout.ts

@@ -0,0 +1,23 @@
+import { auth } from '@/apis/auth';
+import { auth as fAuth } from '@/lib/firebase';
+import { useUserStore } from '@/store/use-user';
+import { useMutation } from '@tanstack/react-query';
+
+export const useLogout = () => {
+   const { user } = useUserStore();
+   return useMutation({
+      mutationFn: async () => {
+         if (user?.provider !== 'local') {
+            await fAuth.signOut();
+         }
+
+         await auth.logout();
+      },
+      async onSuccess() {
+         window.location.href = '/login';
+      },
+      onError() {
+         window.location.href = '/login';
+      },
+   });
+};

client/src/routes/_private/chatbots.tsx

@@ -1,10 +1,23 @@
+import { useLogout } from '@/hooks/auth';
 import { useUserStore } from '@/store/use-user';
 import { createFileRoute } from '@tanstack/react-router';
 
 const Chatbots = () => {
    const { user } = useUserStore();
+   const logoutMutaion = useLogout();
 
-   return <div>{JSON.stringify(user)}</div>;
+   return (
+      <div>
+         {JSON.stringify(user)}
+         <button
+            onClick={() => {
+               logoutMutaion.mutate();
+            }}
+         >
+            Logout
+         </button>
+      </div>
+   );
 };
 
 export const Route = createFileRoute('/_private/chatbots')({

client/src/store/use-user.ts

@@ -4,7 +4,7 @@ import { devtools } from 'zustand/middleware';
 
 type TUserState = {
    user: TUser | null;
-   setUser: (user: TUser) => void;
+   setUser: (user: TUser | null) => void;
 };
 
 export const useUserStore = create<TUserState>()(

server/src/constants/index.ts

@@ -21,6 +21,7 @@ export const ENDPOINTS = {
       FORGOT_PASSWORD: '/auth/forgot-password',
       RESET_PASSWORD: '/auth/reset-password',
       WITH_ID_TOKEN: '/auth/with-id-token',
+      LOGOUT: '/auth/logout',
    },
    UPLOAD: {
       SINGLE: '/upload/single',

server/src/controllers/auth.controller.ts

@@ -99,6 +99,18 @@ export class AuthController {
       });
    });
 
+   public logout = catchAsync(async (req: RequestWithUser, res) => {
+
+      await this.authService.logout(req.user?.id as string)
+
+      this.clearTokensCookie(res);
+
+      res.status(StatusCodes.OK).json({
+         message: this.localeService.i18n().AUTH.LOGOUT_SUCCESS(),
+         data: null,
+      });
+   });
+
    private setTokensCookie = (res: Response, tokens: TTokenData) => {
       const { accessToken, refreshToken } = tokens;
       res.cookie('access_token', accessToken, {
@@ -115,4 +127,9 @@ export class AuthController {
          domain: 'localhost', //TODO: THIS WILL BE CHANGED IN PRODUCTION
       });
    };
+
+   private clearTokensCookie = (res: Response) => {
+      res.clearCookie('access_token',);
+      res.clearCookie('refresh_token');
+   };
 }