From 0a9c79b3b35daae46f818aef1e7595f324f8552a Mon Sep 17 00:00:00 2001
From: Charles Bochet <charles@twenty.com>
Date: Fri, 5 Apr 2024 21:01:03 +0200
Subject: [PATCH] Fix userload on signout

---
 packages/twenty-front/src/modules/auth/hooks/useAuth.ts     | 5 +++++
 .../src/modules/users/components/UserProviderEffect.tsx     | 6 ++++--
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/packages/twenty-front/src/modules/auth/hooks/useAuth.ts b/packages/twenty-front/src/modules/auth/hooks/useAuth.ts
index 4f6e3b681..53bed2162 100644
--- a/packages/twenty-front/src/modules/auth/hooks/useAuth.ts
+++ b/packages/twenty-front/src/modules/auth/hooks/useAuth.ts
@@ -14,6 +14,7 @@ import { isVerifyPendingState } from '@/auth/states/isVerifyPendingState';
 import { workspacesState } from '@/auth/states/workspaces';
 import { authProvidersState } from '@/client-config/states/authProvidersState';
 import { billingState } from '@/client-config/states/billingState';
+import { isClientConfigLoadedState } from '@/client-config/states/isClientConfigLoadedState';
 import { isDebugModeState } from '@/client-config/states/isDebugModeState';
 import { isSignInPrefilledState } from '@/client-config/states/isSignInPrefilledState';
 import { supportChatState } from '@/client-config/states/supportChatState';
@@ -165,8 +166,12 @@ export const useAuth = () => {
         const supportChat = snapshot.getLoadable(supportChatState).getValue();
         const telemetry = snapshot.getLoadable(telemetryState).getValue();
         const isDebugMode = snapshot.getLoadable(isDebugModeState).getValue();
+        const isClientConfigLoaded = snapshot
+          .getLoadable(isClientConfigLoadedState)
+          .getValue();
 
         const initialSnapshot = emptySnapshot.map(({ set }) => {
+          set(isClientConfigLoadedState, isClientConfigLoaded);
           set(iconsState, iconsValue);
           set(authProvidersState, authProvidersValue);
           set(billingState, billing);
diff --git a/packages/twenty-front/src/modules/users/components/UserProviderEffect.tsx b/packages/twenty-front/src/modules/users/components/UserProviderEffect.tsx
index f4f10c417..d7ac742ff 100644
--- a/packages/twenty-front/src/modules/users/components/UserProviderEffect.tsx
+++ b/packages/twenty-front/src/modules/users/components/UserProviderEffect.tsx
@@ -28,16 +28,18 @@ export const UserProviderEffect = () => {
 
   const { loading: queryLoading, data: queryData } = useQuery<{
     currentUser: User;
-  }>(GET_CURRENT_USER, { skip: isCurrentUserLoaded });
+  }>(GET_CURRENT_USER, {
+    skip: isCurrentUserLoaded,
+  });
 
   useEffect(() => {
     if (!queryLoading) {
       setIsLoading(false);
+      setIsCurrentUserLoaded(true);
     }
 
     if (!isDefined(queryData?.currentUser)) return;
 
-    setIsCurrentUserLoaded(true);
     setCurrentUser(queryData.currentUser);
     setCurrentWorkspace(queryData.currentUser.defaultWorkspace);