TIANHE
@@ -0,0 +1,410 @@
|
||||
import storage from 'store'
|
||||
import expirePlugin from 'store/plugins/expire'
|
||||
import { login, emailLogin, mobileLogin, logout, getUserInfo, apiLogout, updateUserInfo, oauthCallback } from '@/api/login'
|
||||
import { ACCESS_TOKEN, USER_INFO, USER_ROLES } from '@/store/mutation-types'
|
||||
import { welcome } from '@/utils/util'
|
||||
|
||||
storage.addPlugin(expirePlugin)
|
||||
|
||||
const DEFAULT_ROLE = { id: 'default', permissionList: [] }
|
||||
|
||||
function normalizeRoles (roles) {
|
||||
if (!roles) return []
|
||||
if (Array.isArray(roles)) return roles
|
||||
return [roles]
|
||||
}
|
||||
|
||||
function getStoredInfo () {
|
||||
const info = storage.get(USER_INFO) || {}
|
||||
return (info && typeof info === 'object') ? info : {}
|
||||
}
|
||||
|
||||
function getStoredRoles () {
|
||||
const roles = storage.get(USER_ROLES) || []
|
||||
return normalizeRoles(roles)
|
||||
}
|
||||
|
||||
function getStoredToken () {
|
||||
const token = storage.get(ACCESS_TOKEN)
|
||||
return typeof token === 'string' ? token : (token && token.token) ? token.token : token
|
||||
}
|
||||
|
||||
const initialInfo = getStoredInfo()
|
||||
const initialRoles = getStoredRoles()
|
||||
const initialToken = getStoredToken() || ''
|
||||
const initialName = initialInfo.nickname || initialInfo.username || ''
|
||||
const initialAvatar = initialInfo.avatar || ''
|
||||
const initialWelcome = initialName ? welcome() : ''
|
||||
const user = {
|
||||
state: {
|
||||
token: initialToken,
|
||||
name: initialName,
|
||||
welcome: initialWelcome,
|
||||
avatar: initialAvatar,
|
||||
roles: initialRoles,
|
||||
info: initialInfo
|
||||
},
|
||||
|
||||
mutations: {
|
||||
SET_TOKEN: (state, token) => {
|
||||
state.token = token
|
||||
},
|
||||
SET_NAME: (state, { name, welcome }) => {
|
||||
state.name = name
|
||||
state.welcome = welcome
|
||||
},
|
||||
SET_AVATAR: (state, avatar) => {
|
||||
state.avatar = avatar
|
||||
},
|
||||
SET_ROLES: (state, roles) => {
|
||||
state.roles = roles
|
||||
},
|
||||
SET_INFO: (state, info) => {
|
||||
state.info = info
|
||||
}
|
||||
},
|
||||
|
||||
actions: {
|
||||
// 登录
|
||||
Login ({ commit }, userInfo) {
|
||||
return new Promise((resolve, reject) => {
|
||||
login(userInfo).then(response => {
|
||||
// 适配 Python 后端响应格式
|
||||
if (response && response.code === 1 && response.data) {
|
||||
const result = response.data
|
||||
const token = result.token
|
||||
const info = result.userinfo || {}
|
||||
|
||||
const expiresAt = new Date().getTime() + 7 * 24 * 60 * 60 * 1000
|
||||
storage.set(ACCESS_TOKEN, token, expiresAt)
|
||||
commit('SET_TOKEN', token)
|
||||
commit('SET_INFO', info)
|
||||
storage.set(USER_INFO, info, expiresAt)
|
||||
|
||||
// 设置基本信息
|
||||
const name = info.nickname || info.username || 'User'
|
||||
commit('SET_NAME', { name: name, welcome: welcome() })
|
||||
commit('SET_AVATAR', info.avatar || '/avatar2.jpg')
|
||||
|
||||
// 设置默认角色，防止路由鉴权失败
|
||||
const roles = [{ id: 'admin', permissionList: ['dashboard', 'exception', 'account'] }]
|
||||
commit('SET_ROLES', roles)
|
||||
storage.set(USER_ROLES, roles, expiresAt)
|
||||
|
||||
resolve(response)
|
||||
} else {
|
||||
reject(new Error((response && response.msg) || 'Login failed'))
|
||||
}
|
||||
}).catch(error => {
|
||||
reject(error)
|
||||
})
|
||||
})
|
||||
},
|
||||
|
||||
// 邮箱验证码登录
|
||||
EmailLogin ({ commit }, userInfo) {
|
||||
return new Promise((resolve, reject) => {
|
||||
emailLogin(userInfo).then(response => {
|
||||
// 新API响应格式: { code: 1, msg: "登录成功", data: { token, userInfo } }
|
||||
if (response.code === 1 && response.data) {
|
||||
const token = response.data.token
|
||||
storage.set(ACCESS_TOKEN, token, new Date().getTime() + 7 * 24 * 60 * 60 * 1000)
|
||||
commit('SET_TOKEN', token)
|
||||
// 保存用户信息（从登录接口返回的 userInfo）
|
||||
if (response.data.userInfo) {
|
||||
const userInfoData = response.data.userInfo
|
||||
commit('SET_INFO', userInfoData)
|
||||
|
||||
// 设置用户名
|
||||
if (userInfoData.nickname) {
|
||||
commit('SET_NAME', { name: userInfoData.nickname, welcome: welcome() })
|
||||
} else if (userInfoData.username) {
|
||||
commit('SET_NAME', { name: userInfoData.username, welcome: welcome() })
|
||||
}
|
||||
|
||||
// 设置头像
|
||||
if (userInfoData.avatar) {
|
||||
commit('SET_AVATAR', userInfoData.avatar)
|
||||
}
|
||||
|
||||
// 设置角色（如果有）
|
||||
if (userInfoData.role) {
|
||||
commit('SET_ROLES', userInfoData.role)
|
||||
} else if (userInfoData.roles) {
|
||||
commit('SET_ROLES', userInfoData.roles)
|
||||
} else {
|
||||
// 如果没有角色信息，设置一个默认角色对象，避免路由守卫卡住
|
||||
// 设置一个标记，表示已经初始化过用户信息
|
||||
commit('SET_ROLES', [{ id: 'default', permissionList: [] }])
|
||||
}
|
||||
}
|
||||
resolve(response)
|
||||
} else {
|
||||
reject(new Error(response.msg || '登录失败'))
|
||||
}
|
||||
}).catch(error => {
|
||||
reject(error)
|
||||
})
|
||||
})
|
||||
},
|
||||
|
||||
// 手机号验证码登录
|
||||
MobileLogin ({ commit }, userInfo) {
|
||||
return new Promise((resolve, reject) => {
|
||||
mobileLogin(userInfo).then(response => {
|
||||
// 新API响应格式: { code: 1, msg: "登录成功", data: { token, userInfo } }
|
||||
if (response.code === 1 && response.data) {
|
||||
const token = response.data.token
|
||||
storage.set(ACCESS_TOKEN, token, new Date().getTime() + 7 * 24 * 60 * 60 * 1000)
|
||||
commit('SET_TOKEN', token)
|
||||
// 保存用户信息（从登录接口返回的 userInfo）
|
||||
if (response.data.userInfo) {
|
||||
const userInfoData = response.data.userInfo
|
||||
commit('SET_INFO', userInfoData)
|
||||
|
||||
// 设置用户名
|
||||
if (userInfoData.nickname) {
|
||||
commit('SET_NAME', { name: userInfoData.nickname, welcome: welcome() })
|
||||
} else if (userInfoData.username) {
|
||||
commit('SET_NAME', { name: userInfoData.username, welcome: welcome() })
|
||||
}
|
||||
|
||||
// 设置头像
|
||||
if (userInfoData.avatar) {
|
||||
commit('SET_AVATAR', userInfoData.avatar)
|
||||
}
|
||||
|
||||
// 设置角色（如果有）
|
||||
if (userInfoData.role) {
|
||||
commit('SET_ROLES', userInfoData.role)
|
||||
} else if (userInfoData.roles) {
|
||||
commit('SET_ROLES', userInfoData.roles)
|
||||
} else {
|
||||
// 如果没有角色信息，设置一个默认角色对象，避免路由守卫卡住
|
||||
// 设置一个标记，表示已经初始化过用户信息
|
||||
commit('SET_ROLES', [{ id: 'default', permissionList: [] }])
|
||||
}
|
||||
}
|
||||
resolve(response)
|
||||
} else {
|
||||
reject(new Error(response.msg || '登录失败'))
|
||||
}
|
||||
}).catch(error => {
|
||||
reject(error)
|
||||
})
|
||||
})
|
||||
},
|
||||
|
||||
// Web3 登录完成后的统一处理
|
||||
Web3LoginFinalize ({ commit }, payload) {
|
||||
return new Promise((resolve, reject) => {
|
||||
try {
|
||||
const { token, userInfo } = payload
|
||||
if (!token || !userInfo) {
|
||||
reject(new Error('登录数据异常'))
|
||||
return
|
||||
}
|
||||
storage.set(ACCESS_TOKEN, token, new Date().getTime() + 7 * 24 * 60 * 60 * 1000)
|
||||
commit('SET_TOKEN', token)
|
||||
commit('SET_INFO', userInfo)
|
||||
|
||||
if (userInfo.nickname) {
|
||||
commit('SET_NAME', { name: userInfo.nickname, welcome: welcome() })
|
||||
} else if (userInfo.username) {
|
||||
commit('SET_NAME', { name: userInfo.username, welcome: welcome() })
|
||||
}
|
||||
|
||||
if (userInfo.avatar) {
|
||||
commit('SET_AVATAR', userInfo.avatar)
|
||||
}
|
||||
|
||||
if (userInfo.role) {
|
||||
commit('SET_ROLES', userInfo.role)
|
||||
} else if (userInfo.roles) {
|
||||
commit('SET_ROLES', userInfo.roles)
|
||||
} else {
|
||||
commit('SET_ROLES', [{ id: 'default', permissionList: [] }])
|
||||
}
|
||||
|
||||
resolve()
|
||||
} catch (e) {
|
||||
reject(e)
|
||||
}
|
||||
})
|
||||
},
|
||||
|
||||
// 刷新用户信息
|
||||
FetchUserInfo ({ commit }) {
|
||||
return new Promise((resolve, reject) => {
|
||||
getUserInfo().then(res => {
|
||||
if (res && res.code === 1 && res.data) {
|
||||
const info = res.data
|
||||
commit('SET_INFO', info)
|
||||
if (info.nickname) {
|
||||
commit('SET_NAME', { name: info.nickname, welcome: welcome() })
|
||||
} else if (info.username) {
|
||||
commit('SET_NAME', { name: info.username, welcome: welcome() })
|
||||
}
|
||||
if (info.avatar) {
|
||||
commit('SET_AVATAR', info.avatar)
|
||||
}
|
||||
resolve(info)
|
||||
} else {
|
||||
reject(new Error((res && res.msg) || '获取用户信息失败'))
|
||||
}
|
||||
}).catch(err => reject(err))
|
||||
})
|
||||
},
|
||||
|
||||
// 更新用户基本信息
|
||||
UpdateUserInfo ({ commit, state }, payload) {
|
||||
return new Promise((resolve, reject) => {
|
||||
updateUserInfo(payload).then(res => {
|
||||
if (res && res.code === 1) {
|
||||
// 合并本地 store 的 info
|
||||
const newInfo = { ...state.info, ...payload }
|
||||
commit('SET_INFO', newInfo)
|
||||
if (newInfo.nickname) {
|
||||
commit('SET_NAME', { name: newInfo.nickname, welcome: welcome() })
|
||||
}
|
||||
resolve(res)
|
||||
} else {
|
||||
reject(new Error((res && res.msg) || '修改失败'))
|
||||
}
|
||||
}).catch(err => reject(err))
|
||||
})
|
||||
},
|
||||
|
||||
// 获取用户信息（从 store 中获取，不再请求接口）
|
||||
GetInfo ({ commit, state }) {
|
||||
return new Promise((resolve, reject) => {
|
||||
// 用户信息已经在登录时保存到 store 中，直接返回
|
||||
if (state.info && Object.keys(state.info).length > 0) {
|
||||
// 补全 Roles
|
||||
const info = state.info
|
||||
if (info.role) {
|
||||
const roles = normalizeRoles(info.role)
|
||||
commit('SET_ROLES', roles)
|
||||
storage.set(USER_ROLES, roles, new Date().getTime() + 7 * 24 * 60 * 60 * 1000)
|
||||
} else if (info.roles) {
|
||||
const roles = normalizeRoles(info.roles)
|
||||
commit('SET_ROLES', roles)
|
||||
storage.set(USER_ROLES, roles, new Date().getTime() + 7 * 24 * 60 * 60 * 1000)
|
||||
} else {
|
||||
commit('SET_ROLES', [DEFAULT_ROLE])
|
||||
storage.set(USER_ROLES, [DEFAULT_ROLE], new Date().getTime() + 7 * 24 * 60 * 60 * 1000)
|
||||
}
|
||||
resolve(state.info)
|
||||
} else {
|
||||
// 尝试主动拉取一次
|
||||
getUserInfo().then(res => {
|
||||
if (res && res.code === 1 && res.data) {
|
||||
const info = res.data
|
||||
commit('SET_INFO', info)
|
||||
storage.set(USER_INFO, info, new Date().getTime() + 7 * 24 * 60 * 60 * 1000)
|
||||
if (info.nickname) {
|
||||
commit('SET_NAME', { name: info.nickname, welcome: welcome() })
|
||||
} else if (info.username) {
|
||||
commit('SET_NAME', { name: info.username, welcome: welcome() })
|
||||
}
|
||||
if (info.avatar) {
|
||||
commit('SET_AVATAR', info.avatar)
|
||||
}
|
||||
// 关键修复：设置角色，防止路由守卫死循环
|
||||
if (info.role) {
|
||||
const roles = normalizeRoles(info.role)
|
||||
commit('SET_ROLES', roles)
|
||||
storage.set(USER_ROLES, roles, new Date().getTime() + 7 * 24 * 60 * 60 * 1000)
|
||||
} else if (info.roles) {
|
||||
const roles = normalizeRoles(info.roles)
|
||||
commit('SET_ROLES', roles)
|
||||
storage.set(USER_ROLES, roles, new Date().getTime() + 7 * 24 * 60 * 60 * 1000)
|
||||
} else {
|
||||
commit('SET_ROLES', [DEFAULT_ROLE])
|
||||
storage.set(USER_ROLES, [DEFAULT_ROLE], new Date().getTime() + 7 * 24 * 60 * 60 * 1000)
|
||||
}
|
||||
resolve(info)
|
||||
} else {
|
||||
reject(new Error((res && res.msg) || '用户信息不存在'))
|
||||
}
|
||||
}).catch(err => reject(err))
|
||||
}
|
||||
})
|
||||
},
|
||||
|
||||
// OAuth 登录（Google/GitHub）
|
||||
OAuthLogin ({ commit }, userInfo) {
|
||||
return new Promise((resolve, reject) => {
|
||||
oauthCallback(userInfo).then(response => {
|
||||
// 新API响应格式: { code: 1, msg: "登录成功", data: { token, userInfo } }
|
||||
if (response.code === 1 && response.data) {
|
||||
const token = response.data.token
|
||||
storage.set(ACCESS_TOKEN, token, new Date().getTime() + 7 * 24 * 60 * 60 * 1000)
|
||||
commit('SET_TOKEN', token)
|
||||
// 保存用户信息（从登录接口返回的 userInfo）
|
||||
if (response.data.userInfo) {
|
||||
const userInfoData = response.data.userInfo
|
||||
commit('SET_INFO', userInfoData)
|
||||
|
||||
// 设置用户名
|
||||
if (userInfoData.nickname) {
|
||||
commit('SET_NAME', { name: userInfoData.nickname, welcome: welcome() })
|
||||
} else if (userInfoData.username) {
|
||||
commit('SET_NAME', { name: userInfoData.username, welcome: welcome() })
|
||||
}
|
||||
|
||||
// 设置头像
|
||||
if (userInfoData.avatar) {
|
||||
commit('SET_AVATAR', userInfoData.avatar)
|
||||
}
|
||||
|
||||
// 设置角色（如果有）
|
||||
if (userInfoData.role) {
|
||||
commit('SET_ROLES', userInfoData.role)
|
||||
} else if (userInfoData.roles) {
|
||||
commit('SET_ROLES', userInfoData.roles)
|
||||
} else {
|
||||
// 如果没有角色信息，设置一个默认角色对象，避免路由守卫卡住
|
||||
commit('SET_ROLES', [{ id: 'default', permissionList: [] }])
|
||||
}
|
||||
}
|
||||
resolve(response)
|
||||
} else {
|
||||
reject(new Error(response.msg || '登录失败'))
|
||||
}
|
||||
}).catch(error => {
|
||||
reject(error)
|
||||
})
|
||||
})
|
||||
},
|
||||
|
||||
// 登出
|
||||
Logout ({ commit, state }) {
|
||||
return new Promise((resolve) => {
|
||||
// 兼容旧登出与新后端登出
|
||||
const req = typeof apiLogout === 'function' ? apiLogout() : logout(state.token)
|
||||
req.then(() => {
|
||||
commit('SET_TOKEN', '')
|
||||
commit('SET_ROLES', [])
|
||||
commit('SET_INFO', {})
|
||||
commit('SET_NAME', { name: '', welcome: '' })
|
||||
commit('SET_AVATAR', '')
|
||||
storage.remove(ACCESS_TOKEN)
|
||||
storage.remove(USER_INFO)
|
||||
storage.remove(USER_ROLES)
|
||||
resolve()
|
||||
}).catch(() => {
|
||||
// 登出失败时也继续执行，确保清理本地状态
|
||||
storage.remove(ACCESS_TOKEN)
|
||||
storage.remove(USER_INFO)
|
||||
storage.remove(USER_ROLES)
|
||||
resolve()
|
||||
}).finally(() => {
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
export default user
|
||||
Reference in New Issue
Block a user