🐛 Fix login as refetching

src/common/misc/userDataManagment/useChangeLoggedUser.ts

@@ -1,27 +1,30 @@
 import { closeAllChats } from '@/redux/features/openedChatsSlice';
-import { useGetLoggedUserQuery } from '@/redux/services/loggedUserAPI';
+import { useGetLoggedUserQuery, useGetUserChatsQuery } from '@/redux/services/loggedUserAPI';
 import { useState } from 'react';
 import { useDispatch } from 'react-redux';
 
 export default function useChangeLoggedUser(userId?: string) {
   const dispatch = useDispatch();
 
   const { refetch: refetchLoggedUser } = useGetLoggedUserQuery({});
+  const { refetch: refetchChats } = useGetUserChatsQuery({});
+
   const [isLoading, setIsLoading] = useState(false);
 
   async function switchLoggedUser() {
     setIsLoading(true);
 
     dispatch(closeAllChats());
 
-    await refetchLoggedUser();
-
     if (localStorage) {
       if (userId) localStorage.setItem('loggedUser', JSON.stringify(userId));
       // Redux logged user query handles random user fetching if no userId is stored in localStorage
       else localStorage.removeItem('loggedUser');
     }
 
+    await refetchChats();
+    await refetchLoggedUser();
+
     setIsLoading(false);
   }
 

src/components/pages/Home/index.tsx

@@ -1,6 +1,7 @@
 import ContactsSidebar from '@/components/molecules/ContactsSidebar';
 import HomeWall from '@/components/organisms/HomeWall';
 import ShortcutsSidebar from '@/components/organisms/ShortcutsSidebar';
+import { useGetLoggedUserQuery } from '@/redux/services/loggedUserAPI';
 import { BoxProps } from '@mui/material';
 import { StyledRoot } from './styles';
 

src/components/pages/Profile/index.tsx

@@ -16,6 +16,7 @@ import { ProfileProps, TProfileTabs } from './types';
 export default function Profile({ userId, useTabsRouting = true, sx, ...rootProps }: ProfileProps) {
   const theme = useTheme();
   const router = useRouter();
+
   const { data: profileData, refetch } = useUserDataByIdQuery(userId);
 
   const [selectedTab, setSelectedTab] = useState<TProfileTabs>(useTabsRouting ? null : 'posts');

src/pages/_app.tsx

@@ -61,6 +61,7 @@ import { LocalizationProvider } from '@mui/x-date-pickers';
 import { ThemeProvider as NextThemesProvider } from 'next-themes';
 import { Provider as StoreProvider } from 'react-redux';
 
+import { useGetLoggedUserQuery } from '@/redux/services/loggedUserAPI';
 import '@fortawesome/fontawesome-svg-core/styles.css';
 const { library, config } = require('@fortawesome/fontawesome-svg-core');
 

src/pages/_document.tsx

@@ -11,6 +11,7 @@ import createEmotionCache from '@/design/createEmotionCache';
 import createEmotionServer from '@emotion/server/create-instance';
 import { AppType } from 'next/app';
 import { EmotionAppProps } from './_app';
+import { useGetLoggedUserQuery } from '@/redux/services/loggedUserAPI';
 
 interface EmotionDocumentProps extends DocumentProps {
   emotionStyleTags: JSX.Element[];