From 69243dee0b19d8629eeb555a05a4c46b132d0197 Mon Sep 17 00:00:00 2001
From: Ivan <8611739+IRBorisov@users.noreply.github.com>
Date: Wed, 29 Jan 2025 21:30:41 +0300
Subject: [PATCH] B: Remove queries on logout / login

---
 rsconcept/frontend/src/backend/auth/useLogin.tsx  | 5 ++++-
 rsconcept/frontend/src/backend/auth/useLogout.tsx | 3 ++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/rsconcept/frontend/src/backend/auth/useLogin.tsx b/rsconcept/frontend/src/backend/auth/useLogin.tsx
index dd640db9..5a0108ef 100644
--- a/rsconcept/frontend/src/backend/auth/useLogin.tsx
+++ b/rsconcept/frontend/src/backend/auth/useLogin.tsx
@@ -1,5 +1,7 @@
 import { useMutation, useQueryClient } from '@tanstack/react-query';
 
+import { libraryApi } from '@/backend/library/api';
+
 import { authApi } from './api';
 
 export const useLogin = () => {
@@ -7,7 +9,8 @@ export const useLogin = () => {
   const mutation = useMutation({
     mutationKey: ['login'],
     mutationFn: authApi.login,
-    onSettled: () => client.invalidateQueries({ queryKey: [authApi.baseKey] })
+    onSettled: () => client.invalidateQueries({ queryKey: [authApi.baseKey] }),
+    onSuccess: () => client.removeQueries({ queryKey: [libraryApi.baseKey] })
   });
   return {
     login: (
diff --git a/rsconcept/frontend/src/backend/auth/useLogout.tsx b/rsconcept/frontend/src/backend/auth/useLogout.tsx
index d4e2f921..1562e180 100644
--- a/rsconcept/frontend/src/backend/auth/useLogout.tsx
+++ b/rsconcept/frontend/src/backend/auth/useLogout.tsx
@@ -7,7 +7,8 @@ export const useLogout = () => {
   const mutation = useMutation({
     mutationKey: ['logout'],
     mutationFn: authApi.logout,
-    onSettled: () => client.invalidateQueries({ queryKey: [authApi.baseKey] })
+    onSettled: () => client.invalidateQueries({ queryKey: [authApi.baseKey] }),
+    onSuccess: () => client.removeQueries()
   });
   return { logout: (onSuccess?: () => void) => mutation.mutate(undefined, { onSuccess }) };
 };