1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
import { useCache, CACHE_KEY } 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'
const TenantId = 'tenantId'
 
// 获取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)
  wsCache.set(TenantId, import.meta.env.VITE_TENANT_ID)
}
 
// 删除token
export const removeToken = () => {
  wsCache.delete(AccessTokenKey)
  wsCache.delete(RefreshTokenKey)
}
 
/** 格式化token（jwt格式） */
export const formatToken = (token: string): string => {
  return 'Bearer ' + token
}
// ========== 账号相关 ==========
 
export type LoginFormType = {
  username: string
  password: string
  rememberMe: boolean
}
 
export const getLoginForm = () => {
  const loginForm: LoginFormType = wsCache.get(CACHE_KEY.LoginForm)
  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(CACHE_KEY.LoginForm, loginForm, { exp: 30 * 24 * 60 * 60 })
}
 
export const removeLoginForm = () => {
  wsCache.delete(CACHE_KEY.LoginForm)
}
 
// ========== 租户相关 ==========
 
export const getTenantId = () => {
  return wsCache.get(CACHE_KEY.TenantId)
}
 
export const setTenantId = (username: string) => {
  wsCache.set(CACHE_KEY.TenantId, username)
}