92 lines
2.2 KiB
TypeScript
92 lines
2.2 KiB
TypeScript
import { toast } from 'sonner'
|
|
import { create } from 'zustand'
|
|
|
|
import { getSysUserInfo, logout } from '@/api'
|
|
import { getExpandedMenus, sleep } from '@/lib'
|
|
import type { SysUserInfo } from '@/schemas'
|
|
|
|
const IS_LOGIN_KEY = 'is_login'
|
|
|
|
interface Store {
|
|
token: string | null
|
|
tokenExp: string | null
|
|
userInfo: SysUserInfo | null
|
|
expandedMenus: number[]
|
|
isLogin: boolean
|
|
setToken: (token: string | null, tokenExp?: string | null) => void
|
|
logout: () => void
|
|
init: () => Promise<void>
|
|
toggleMenu: (id: number) => void
|
|
}
|
|
|
|
const isLoginPage = () => {
|
|
return typeof window !== 'undefined' && window.location.pathname === '/login'
|
|
}
|
|
|
|
const getIsLogin = () => {
|
|
return typeof window !== 'undefined' && localStorage.getItem(IS_LOGIN_KEY) === 'true'
|
|
}
|
|
|
|
const setLoginFlag = (isLogin: boolean) => {
|
|
if (isLogin) {
|
|
localStorage.setItem(IS_LOGIN_KEY, 'true')
|
|
} else {
|
|
localStorage.removeItem(IS_LOGIN_KEY)
|
|
}
|
|
}
|
|
|
|
export const useStore = create<Store>((set) => {
|
|
return {
|
|
token: null,
|
|
tokenExp: null,
|
|
userInfo: null,
|
|
expandedMenus: [],
|
|
isLogin: getIsLogin(),
|
|
setToken: (token, tokenExp = null) => {
|
|
setLoginFlag(Boolean(token))
|
|
|
|
set({ token, tokenExp })
|
|
},
|
|
logout: async () => {
|
|
try {
|
|
await logout()
|
|
} catch (error) {
|
|
console.error(error)
|
|
toast.error('退出登录失败，请稍后重试', { richColors: true })
|
|
return
|
|
}
|
|
|
|
set({ token: null, tokenExp: null, userInfo: null, isLogin: false })
|
|
setLoginFlag(false)
|
|
location.replace('/login')
|
|
},
|
|
init: async () => {
|
|
if (isLoginPage()) {
|
|
return
|
|
}
|
|
|
|
try {
|
|
const [{ data }] = await Promise.all([getSysUserInfo(), sleep(500)])
|
|
set({
|
|
userInfo: data,
|
|
expandedMenus: getExpandedMenus(data.menus),
|
|
})
|
|
} catch (error) {
|
|
console.error(error)
|
|
set({
|
|
userInfo: null,
|
|
expandedMenus: [],
|
|
})
|
|
}
|
|
},
|
|
toggleMenu: (id: number) => {
|
|
set((state) => {
|
|
const expandedMenus = state.expandedMenus.includes(id)
|
|
? state.expandedMenus.filter((item) => item !== id)
|
|
: [...state.expandedMenus, id]
|
|
return { expandedMenus }
|
|
})
|
|
},
|
|
}
|
|
})
|