From feade5d02988cdfb42939dd7dc693d513d872267 Mon Sep 17 00:00:00 2001
From: Tuan Dang <dangtony98@gmail.com>
Date: Mon, 9 Oct 2023 09:00:31 +0100
Subject: [PATCH] Clear react-query cache upon user logout, delete account

---
 frontend/src/hooks/api/users/index.tsx   |  3 ++-
 frontend/src/hooks/api/users/queries.tsx | 15 ++++++++-------
 2 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/frontend/src/hooks/api/users/index.tsx b/frontend/src/hooks/api/users/index.tsx
index 338c3b069..ce688fb0a 100644
--- a/frontend/src/hooks/api/users/index.tsx
+++ b/frontend/src/hooks/api/users/index.tsx
@@ -18,4 +18,5 @@ export {
   useRevokeMySessions,
   useUpdateMfaEnabled,
   useUpdateOrgUserRole,
-  useUpdateUserAuthMethods} from "./queries";
\ No newline at end of file
+  useUpdateUserAuthMethods
+} from "./queries";
\ No newline at end of file
diff --git a/frontend/src/hooks/api/users/queries.tsx b/frontend/src/hooks/api/users/queries.tsx
index 04e620a9e..7949a59e5 100644
--- a/frontend/src/hooks/api/users/queries.tsx
+++ b/frontend/src/hooks/api/users/queries.tsx
@@ -51,11 +51,7 @@ export const useDeleteUser = () => {
       return user;
     },
     onSuccess: () => {
-      queryClient.invalidateQueries(userKeys.getUser);
-      queryClient.invalidateQueries(userKeys.userAction);
-      queryClient.invalidateQueries(userKeys.myIp);
-      queryClient.invalidateQueries(userKeys.myAPIKeys);
-      queryClient.invalidateQueries(userKeys.mySessions);
+      queryClient.clear();
     }
   });
 };
@@ -226,8 +222,9 @@ export const useRegisterUserAction = () => {
   });
 };
 
-export const useLogoutUser = () =>
-  useMutation({
+export const useLogoutUser = () => {
+  const queryClient = useQueryClient();
+  return useMutation({
     mutationFn: async () => {
       await apiRequest.post("/api/v1/auth/logout");
     },
@@ -244,8 +241,12 @@ export const useLogoutUser = () =>
       localStorage.removeItem("PRIVATE_KEY");
       localStorage.removeItem("orgData.id");
       localStorage.removeItem("projectData.id");
+      
+      queryClient.clear();
     }
   });
+}
+
 
 export const useGetMyIp = () => {
   return useQuery({