diff --git a/web/app/components/base/message-listener.tsx b/web/app/components/base/message-listener.tsx index fd2a860884..2c65945d99 100644 --- a/web/app/components/base/message-listener.tsx +++ b/web/app/components/base/message-listener.tsx @@ -1,12 +1,17 @@ 'use client' +import { useRouter } from 'next/navigation' import { useEffect } from 'react' const MessageListener = () => { + const router = useRouter() useEffect(() => { + localStorage.removeItem('disable_log_out') const handleMessage = (event: MessageEvent) => { - console.log('Received message from A:', event.data) - if(event.data.action === 'auto-login') { - localStorage.setItem('auto-login', JSON.stringify(event.data.data)) + if (event.data.action === 'auto-login') { event.source.postMessage('got', event.origin) + localStorage.setItem('console_token', event.data.data.token) + localStorage.setItem('refresh_token', event.data.data.refreshToken) + sessionStorage.setItem('disable_log_out', true) + router.replace(event.data.src) } } diff --git a/web/app/components/header/account-dropdown/index.tsx b/web/app/components/header/account-dropdown/index.tsx index a54fa64838..4fa1b7c099 100644 --- a/web/app/components/header/account-dropdown/index.tsx +++ b/web/app/components/header/account-dropdown/index.tsx @@ -190,7 +190,9 @@ export default function AppSelector({ isMobile }: IAppSelector) { ) } -