refetch user balance after unified login

blockscout · Feb 3, 2025 · a8738a8 · a8738a8
1 parent 4c42918
commit a8738a8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/ui/rewards/login/RewardsLoginModal.tsx b/ui/rewards/login/RewardsLoginModal.tsx
@@ -23,7 +23,7 @@ const MIXPANEL_CONFIG = {
 const RewardsLoginModal = () => {
   const { isOpen: isWalletModalOpen } = useWallet({ source: 'Merits' });
   const isMobile = useIsMobile();
-  const { isLoginModalOpen, closeLoginModal, saveApiToken } = useRewardsContext();
+  const { isLoginModalOpen, closeLoginModal } = useRewardsContext();
 
   const [ isLoginStep, setIsLoginStep ] = useBoolean(true);
   const [ isReferral, setIsReferral ] = useBoolean(false);
@@ -51,12 +51,11 @@ const RewardsLoginModal = () => {
 
   const handleAuthModalClose = useCallback((isSuccess?: boolean, rewardsApiToken?: string) => {
     if (isSuccess && rewardsApiToken) {
-      saveApiToken(rewardsApiToken);
       goNext(false);
     }
     setAuthModalInitialScreen(undefined);
     authModal.onClose();
-  }, [ authModal, setAuthModalInitialScreen, goNext, saveApiToken ]);
+  }, [ authModal, setAuthModalInitialScreen, goNext ]);
 
   return (
     <>

diff --git a/ui/snippets/auth/AuthModal.tsx b/ui/snippets/auth/AuthModal.tsx
@@ -7,7 +7,7 @@ import type { Screen, ScreenSuccess } from './types';
 
 import config from 'configs/app';
 import { getResourceKey } from 'lib/api/useApiQuery';
-import * as cookies from 'lib/cookies';
+import { useRewardsContext } from 'lib/contexts/rewards';
 import useGetCsrfToken from 'lib/hooks/useGetCsrfToken';
 import * as mixpanel from 'lib/mixpanel';
 import IconSvg from 'ui/shared/IconSvg';
@@ -40,6 +40,8 @@ const AuthModal = ({ initialScreen, onClose, mixpanelConfig, closeOnError }: Pro
   const [ isSuccess, setIsSuccess ] = React.useState(false);
   const [ rewardsApiToken, setRewardsApiToken ] = React.useState<string | undefined>(undefined);
 
+  const { saveApiToken } = useRewardsContext();
+
   const router = useRouter();
   const csrfQuery = useGetCsrfToken();
   const queryClient = useQueryClient();
@@ -92,11 +94,11 @@ const AuthModal = ({ initialScreen, onClose, mixpanelConfig, closeOnError }: Pro
 
     if ('rewardsToken' in screen && screen.rewardsToken) {
       setRewardsApiToken(screen.rewardsToken);
-      cookies.set(cookies.NAMES.REWARDS_API_TOKEN, screen.rewardsToken, { expires: 365 });
+      saveApiToken(screen.rewardsToken);
     }
 
     onNextStep(screen);
-  }, [ initialScreen, mixpanelConfig?.account_link_info.source, onNextStep, csrfQuery, queryClient ]);
+  }, [ initialScreen, mixpanelConfig?.account_link_info.source, onNextStep, csrfQuery, queryClient, saveApiToken ]);
 
   const onModalClose = React.useCallback(() => {
     onClose(isSuccess, rewardsApiToken);