mirror of
https://github.com/langgenius/dify.git
synced 2026-05-04 17:38:04 +08:00
Feat/api jwt (#1212)
This commit is contained in:
@ -1,7 +1,9 @@
|
||||
'use client'
|
||||
|
||||
import { SWRConfig } from 'swr'
|
||||
import { useEffect, useState } from 'react'
|
||||
import type { ReactNode } from 'react'
|
||||
import { useRouter, useSearchParams } from 'next/navigation'
|
||||
|
||||
type SwrInitorProps = {
|
||||
children: ReactNode
|
||||
@ -9,13 +11,32 @@ type SwrInitorProps = {
|
||||
const SwrInitor = ({
|
||||
children,
|
||||
}: SwrInitorProps) => {
|
||||
return (
|
||||
<SWRConfig value={{
|
||||
shouldRetryOnError: false,
|
||||
}}>
|
||||
{children}
|
||||
</SWRConfig>
|
||||
)
|
||||
const router = useRouter()
|
||||
const searchParams = useSearchParams()
|
||||
const consoleToken = searchParams.get('console_token')
|
||||
const consoleTokenFromLocalStorage = localStorage?.getItem('console_token')
|
||||
const [init, setInit] = useState(false)
|
||||
|
||||
useEffect(() => {
|
||||
if (!(consoleToken || consoleTokenFromLocalStorage))
|
||||
router.replace('/signin')
|
||||
|
||||
if (consoleToken) {
|
||||
localStorage?.setItem('console_token', consoleToken!)
|
||||
router.replace('/apps', { forceOptimisticNavigation: false })
|
||||
}
|
||||
setInit(true)
|
||||
}, [])
|
||||
|
||||
return init
|
||||
? (
|
||||
<SWRConfig value={{
|
||||
shouldRetryOnError: false,
|
||||
}}>
|
||||
{children}
|
||||
</SWRConfig>
|
||||
)
|
||||
: null
|
||||
}
|
||||
|
||||
export default SwrInitor
|
||||
|
||||
@ -8,6 +8,10 @@ import I18n from '@/context/i18n'
|
||||
|
||||
const Header = () => {
|
||||
const { locale, setLocaleOnClient } = useContext(I18n)
|
||||
|
||||
if (localStorage?.getItem('console_token'))
|
||||
localStorage.removeItem('console_token')
|
||||
|
||||
return <div className='flex items-center justify-between p-6 w-full'>
|
||||
<div className={style.logo}></div>
|
||||
<Select
|
||||
|
||||
@ -89,7 +89,7 @@ const NormalForm = () => {
|
||||
}
|
||||
try {
|
||||
setIsLoading(true)
|
||||
await login({
|
||||
const res = await login({
|
||||
url: '/login',
|
||||
body: {
|
||||
email,
|
||||
@ -97,7 +97,8 @@ const NormalForm = () => {
|
||||
remember_me: true,
|
||||
},
|
||||
})
|
||||
router.push('/apps')
|
||||
localStorage.setItem('console_token', res.data)
|
||||
router.replace('/apps')
|
||||
}
|
||||
finally {
|
||||
setIsLoading(false)
|
||||
|
||||
Reference in New Issue
Block a user