46 lines
1.5 KiB
TypeScript
46 lines
1.5 KiB
TypeScript
|
import { defineStore } from 'pinia'
|
|||
|
|
import { login } from '@/api/auth'
|
|||
|
|
import { ref } from 'vue'
|
|||
|
|
|
|||
|
|
export const useUserStore = defineStore('user', () => {
|
|||
|
|
const token = ref(localStorage.getItem('token') || '')
|
|||
|
|
const role = ref(localStorage.getItem('role') || '') // 持久化角色
|
|||
|
|
const username = ref(localStorage.getItem('username') || '')
|
|||
|
|
|
|||
|
|
const handleLogin = async (loginForm: any) => {
|
|||
|
|
try {
|
|||
|
|
const res = await login(loginForm)
|
|||
|
|
// res.data 结构: { access_token, user: { role, username, ... } }
|
|||
|
|
const data = res.data
|
|||
|
|
|
|||
|
|
token.value = data.access_token
|
|||
|
|
role.value = data.user.role
|
|||
|
|
username.value = data.user.username
|
|||
|
|
|
|||
|
|
// 持久化存储 (简单处理，生产环境建议加密或仅存Token)
|
|||
|
|
localStorage.setItem('token', data.access_token)
|
|||
|
|
localStorage.setItem('role', data.user.role)
|
|||
|
|
localStorage.setItem('username', data.user.username)
|
|||
|
|
|
|||
|
|
return true
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error(error)
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const logout = () => {
|
|||
|
|
token.value = ''
|
|||
|
|
role.value = ''
|
|||
|
|
username.value = ''
|
|||
|
|
localStorage.clear()
|
|||
|
|
window.location.reload()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 辅助函数：判断当前用户是否拥有某些角色
|
|||
|
|
const hasRole = (roles: string[]) => {
|
|||
|
|
return roles.includes(role.value)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return { token, role, username, handleLogin, logout, hasRole }
|
|||
|
|
})
|