chore: auto login function change

pull/17608/head
keting lu 1 year ago
parent d85ea39e06
commit 07e9eea21c

@ -1,12 +1,17 @@
'use client' 'use client'
import { useRouter } from 'next/navigation'
import { useEffect } from 'react' import { useEffect } from 'react'
const MessageListener = () => { const MessageListener = () => {
const router = useRouter()
useEffect(() => { useEffect(() => {
localStorage.removeItem('disable_log_out')
const handleMessage = (event: MessageEvent) => { const handleMessage = (event: MessageEvent) => {
console.log('Received message from A:', event.data)
if (event.data.action === 'auto-login') { if (event.data.action === 'auto-login') {
localStorage.setItem('auto-login', JSON.stringify(event.data.data))
event.source.postMessage('got', event.origin) 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)
} }
} }

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

Loading…
Cancel
Save