From 9a428b6e2773e2ab5be9c2269bf579e676fe5be9 Mon Sep 17 00:00:00 2001
From: g4titanx <g4titan1@gmail.com>
Date: Thu, 22 May 2025 17:40:18 +0100
Subject: [PATCH] fix: fetch user profile via GraphQL in authStore

---
 src/providers/DynamicProvider.tsx |  4 +--
 src/store/authStore.ts            | 55 ++++++++++++++++++++++++++++---
 src/utils/hostname.ts             |  2 +-
 3 files changed, 54 insertions(+), 7 deletions(-)

diff --git a/src/providers/DynamicProvider.tsx b/src/providers/DynamicProvider.tsx
index 8ad2820..7d4ece0 100644
--- a/src/providers/DynamicProvider.tsx
+++ b/src/providers/DynamicProvider.tsx
@@ -55,9 +55,9 @@ const DynamicUtils = ({
 
   useEffect(() => {
     if (!accessToken) return;
-    
+
     setIsLoggedIn(isLoggedIn);
-  }, [accessToken, isLoggedIn])
+  }, [accessToken, isLoggedIn, setIsLoggedIn]);
 
   useDynamicEvents('authFlowOpen', async () => {
     setAuthModalOpen(true);
diff --git a/src/store/authStore.ts b/src/store/authStore.ts
index 9490659..8fd7416 100644
--- a/src/store/authStore.ts
+++ b/src/store/authStore.ts
@@ -30,7 +30,7 @@ export interface AuthStore {
   triggerLoginModal?: TriggerLoginModal;
   triggerLogout?: TriggerLogout;
   reinitializeSdk?: ReinitializeSdk;
-  setAccessToken: (value: string) => void;
+  setAccessToken: (value: string) => Promise<void>;
   setAuthToken: (value: string) => void;
   setIsLoggingIn: (isLoggingIn: boolean) => void;
   setIsLoggedIn: (isLoggedIn: boolean) => void;
@@ -71,19 +71,66 @@ export const initialState: AuthState = {
   userProfile: undefined,
 };
 
+// Initialize state with dynamic_authentication_token if present
+const token = localStorage.getItem('dynamic_authentication_token') || '';
+const { projectId: initialProjectId = '' } = token ? decodeAccessToken(token) : {};
+
 export const useAuthStore = create<AuthStore>()((set, get) => ({
   ...initialState,
-  setAccessToken: (accessToken: string) => {
+  accessToken: token,
+  projectId: initialProjectId,
+  isLoggedIn: !!token,
+  setAccessToken: async (accessToken: string) => {
     const { projectId } = decodeAccessToken(accessToken);
 
     set({
       accessToken,
       projectId,
       isLoggingIn: false,
+      isLoggedIn: true,
     });
-
     cookies.set('accessToken', accessToken);
     cookies.set('projectId', projectId);
+
+    // Fetch user profile after setting accessToken
+    try {
+      const { graphqlApiUrl } = useConfigStore.getState();
+      if (!graphqlApiUrl) {
+        throw new Error('GraphQL API URL is required to fetch user profile');
+      }
+      const response = await fetch(graphqlApiUrl, {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+          Authorization: `Bearer ${accessToken}`,
+        },
+        body: JSON.stringify({
+          query: `
+            query Me {
+              me {
+                userId
+                username
+                email
+                avatar
+              }
+            }
+          `,
+        }),
+      });
+      const data = await response.json();
+      if (data.errors) {
+        throw new Error(data.errors[0]?.message || 'Failed to fetch user profile');
+      }
+      const userProfile = data?.data?.me;
+      if (userProfile) {
+        set({ userProfile });
+      } else {
+        throw new Error('User profile data not found');
+      }
+    } catch (err) {
+      console.error('Failed to fetch user profile:', err);
+      set({ userProfile: undefined });
+    }
   },
   setAuthToken: (authToken: string) => {
     set({ authToken });
@@ -127,7 +174,7 @@ export const useAuthStore = create<AuthStore>()((set, get) => ({
 
       if (decodedProjectId !== projectId) throw Error('Found a project mismatch');
 
-      setAccessToken(accessToken);
+      await setAccessToken(accessToken);
     } catch (err) {
       console.error('Failed to update access token:', err);
 
diff --git a/src/utils/hostname.ts b/src/utils/hostname.ts
index 5ee6878..3e9596a 100644
--- a/src/utils/hostname.ts
+++ b/src/utils/hostname.ts
@@ -23,6 +23,6 @@ export const getDomain = (url: string) => {
   }
 
   return getTopLevelDomain(url);
-}
+};
 
 export const getHostname = (url: string) => window.location.hostname;