78 lines
2.5 KiB
TypeScript
78 lines
2.5 KiB
TypeScript
import { defineStore } from 'pinia'
|
||
import { login } from '@/api/auth'
|
||
import { ref } from 'vue'
|
||
|
||
export const useUserStore = defineStore('user', () => {
|
||
// 1. State: 初始化时优先从 localStorage 获取，防止刷新丢失
|
||
const token = ref(localStorage.getItem('token') || '')
|
||
const role = ref(localStorage.getItem('role') || '')
|
||
const username = ref(localStorage.getItem('username') || '')
|
||
|
||
// 2. Actions
|
||
// 登录逻辑
|
||
const handleLogin = async (loginForm: any) => {
|
||
const res = await login(loginForm)
|
||
|
||
// [调试日志] 查看实际返回的数据结构
|
||
console.log('Login API Response:', res)
|
||
|
||
// ============================================================
|
||
// [关键修复] 兼容 Axios 拦截器的不同处理方式
|
||
// 如果拦截器已经返回了 response.data，那么 res 本身就是数据对象
|
||
// ============================================================
|
||
const data = res.data || res
|
||
|
||
// 安全检查：确保 data 存在且包含 access_token
|
||
if (!data || !data.access_token) {
|
||
console.error('Login Error: 响应数据中缺少 access_token', data)
|
||
throw new Error('登录失败: 响应数据异常')
|
||
}
|
||
|
||
// 更新 Pinia 状态 (内存)
|
||
token.value = data.access_token
|
||
|
||
// 处理用户信息 (确保后端返回结构中有 user 字段)
|
||
if (data.user) {
|
||
role.value = data.user.role || 'user' // 默认给个 user 角色防止空
|
||
username.value = data.user.username || '用户'
|
||
|
||
// 持久化存储用户信息
|
||
localStorage.setItem('role', role.value)
|
||
localStorage.setItem('username', username.value)
|
||
}
|
||
|
||
// 持久化存储 Token
|
||
localStorage.setItem('token', data.access_token)
|
||
|
||
return true // 返回 true 表示登录成功
|
||
}
|
||
|
||
// 退出逻辑
|
||
const logout = () => {
|
||
// 1. 清空 Pinia 状态 (内存)
|
||
token.value = ''
|
||
role.value = ''
|
||
username.value = ''
|
||
|
||
// 2. 清空 LocalStorage (硬盘)
|
||
localStorage.removeItem('token')
|
||
localStorage.removeItem('role')
|
||
localStorage.removeItem('username')
|
||
}
|
||
|
||
// 3. Getters / Helpers
|
||
// 判断当前用户是否拥有某些角色
|
||
const hasRole = (roles: string[]) => {
|
||
return roles.includes(role.value)
|
||
}
|
||
|
||
return {
|
||
token,
|
||
role,
|
||
username,
|
||
handleLogin,
|
||
logout,
|
||
hasRole
|
||
}
|
||
})
|