43 lines
858 B
TypeScript
43 lines
858 B
TypeScript
import type { BasicUserInfo } from '@vben-core/typings';
|
|
|
|
import { acceptHMRUpdate, defineStore } from 'pinia';
|
|
|
|
interface AccessState {
|
|
/**
|
|
* 用户信息
|
|
*/
|
|
userInfo: BasicUserInfo | null;
|
|
/**
|
|
* 用户角色
|
|
*/
|
|
userRoles: string[];
|
|
}
|
|
|
|
/**
|
|
* @zh_CN 用户信息相关
|
|
*/
|
|
export const useUserStore = defineStore('core-user', {
|
|
actions: {
|
|
setUserInfo(userInfo: BasicUserInfo | null) {
|
|
// 设置用户信息
|
|
this.userInfo = userInfo;
|
|
// 设置角色信息
|
|
const roles = userInfo?.roles ?? [];
|
|
this.setUserRoles(roles);
|
|
},
|
|
setUserRoles(roles: string[]) {
|
|
this.userRoles = roles;
|
|
},
|
|
},
|
|
state: (): AccessState => ({
|
|
userInfo: null,
|
|
userRoles: [],
|
|
}),
|
|
});
|
|
|
|
// 解决热更新问题
|
|
const hot = import.meta.hot;
|
|
if (hot) {
|
|
hot.accept(acceptHMRUpdate(useUserStore, hot));
|
|
}
|