93 lines
2.2 KiB
TypeScript
93 lines
2.2 KiB
TypeScript
import { useCache } from '@/hooks/web/useCache'
|
||
import { TokenType } from '@/api/login/types'
|
||
import { decrypt, encrypt } from '@/utils/jsencrypt'
|
||
|
||
const { wsCache } = useCache()
|
||
|
||
const AccessTokenKey = 'ACCESS_TOKEN'
|
||
const RefreshTokenKey = 'REFRESH_TOKEN'
|
||
|
||
// 获取token
|
||
export const getAccessToken = () => {
|
||
// 此处与TokenKey相同，此写法解决初始化时Cookies中不存在TokenKey报错
|
||
return wsCache.get(AccessTokenKey) ? wsCache.get(AccessTokenKey) : wsCache.get('ACCESS_TOKEN')
|
||
}
|
||
|
||
// 刷新token
|
||
export const getRefreshToken = () => {
|
||
return wsCache.get(RefreshTokenKey)
|
||
}
|
||
|
||
// 设置token
|
||
export const setToken = (token: TokenType) => {
|
||
wsCache.set(RefreshTokenKey, token.refreshToken)
|
||
wsCache.set(AccessTokenKey, token.accessToken)
|
||
}
|
||
|
||
// 删除token
|
||
export const removeToken = () => {
|
||
wsCache.delete(AccessTokenKey)
|
||
wsCache.delete(RefreshTokenKey)
|
||
}
|
||
|
||
/** 格式化token（jwt格式） */
|
||
export const formatToken = (token: string): string => {
|
||
return 'Bearer ' + token
|
||
}
|
||
// ========== 账号相关 ==========
|
||
|
||
const LoginFormKey = 'LOGINFORM'
|
||
|
||
export type LoginFormType = {
|
||
tenantName: string
|
||
username: string
|
||
password: string
|
||
rememberMe: boolean
|
||
}
|
||
|
||
export const getLoginForm = () => {
|
||
const loginForm: LoginFormType = wsCache.get(LoginFormKey)
|
||
if (loginForm) {
|
||
loginForm.password = decrypt(loginForm.password) as string
|
||
}
|
||
return loginForm
|
||
}
|
||
|
||
export const setLoginForm = (loginForm: LoginFormType) => {
|
||
loginForm.password = encrypt(loginForm.password) as string
|
||
wsCache.set(LoginFormKey, loginForm, { exp: 30 * 24 * 60 * 60 })
|
||
}
|
||
|
||
export const removeLoginForm = () => {
|
||
wsCache.delete(LoginFormKey)
|
||
}
|
||
|
||
// ========== 租户相关 ==========
|
||
|
||
const TenantIdKey = 'TENANT_ID'
|
||
const TenantNameKey = 'TENANT_NAME'
|
||
|
||
export const getTenantName = () => {
|
||
return wsCache.get(TenantNameKey)
|
||
}
|
||
|
||
export const setTenantName = (tenantName: string) => {
|
||
wsCache.set(TenantNameKey, tenantName, { exp: 30 * 24 * 60 * 60 })
|
||
}
|
||
|
||
export const removeTenantName = () => {
|
||
wsCache.delete(TenantNameKey)
|
||
}
|
||
|
||
export const getTenantId = () => {
|
||
return wsCache.get(TenantIdKey)
|
||
}
|
||
|
||
export const setTenantId = (username: string) => {
|
||
wsCache.set(TenantIdKey, username)
|
||
}
|
||
|
||
export const removeTenantId = () => {
|
||
wsCache.delete(TenantIdKey)
|
||
}
|