From 07e9eea21c6aecac2a0b254f6696415e151e3935 Mon Sep 17 00:00:00 2001
From: keting lu <luketing1999@gmail.com>
Date: Fri, 14 Mar 2025 17:37:21 +0800
Subject: [PATCH] chore: auto login function change

---
 web/app/components/base/message-listener.tsx         | 11 ++++++++---
 web/app/components/header/account-dropdown/index.tsx |  6 +++++-
 2 files changed, 13 insertions(+), 4 deletions(-)

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) {
                       )
                     }
                   </div>
-                  <Menu.Item>
+                  {
+                    !sessionStorage.getItem('disable_log_out')
+                  && <Menu.Item>
                     {({ active }) => <div className='p-1' onClick={() => handleLogout()}>
                       <div
                         className={
@@ -202,6 +204,8 @@ export default function AppSelector({ isMobile }: IAppSelector) {
                       </div>
                     </div>}
                   </Menu.Item>
+                  }
+
                 </Menu.Items>
               </Transition>
             </>