import type { UserInfo } from '#/store';
|
import type { ErrorMessageMode } from '#/axios';
|
import { defineStore } from 'pinia';
|
import { store } from '@/store';
|
import { RoleEnum } from '@/enums/roleEnum';
|
import { PageEnum } from '@/enums/pageEnum';
|
import { ROLES_KEY, TOKEN_KEY, USER_INFO_KEY } from '@/enums/cacheEnum';
|
import { getAuthCache, setAuthCache } from '@/utils/auth';
|
import { GetUserInfoModel, LoginParams } from '@/api/sys/model/userModel';
|
import { doLogout, getTgUserInfo, getUserInfo, loginApi } from '@/api/sys/user';
|
import { useI18n } from '@/hooks/web/useI18n';
|
import { useMessage } from '@/hooks/web/useMessage';
|
import { router } from '@/router';
|
import { usePermissionStore } from '@/store/modules/permission';
|
import { RouteRecordRaw } from 'vue-router';
|
import { PAGE_NOT_FOUND_ROUTE } from '@/router/routes/basic';
|
import { isArray } from '@/utils/is';
|
import { h } from 'vue';
|
import { AccountInfo } from '/@/api/tigerapi/model/systemModel';
|
import { getMenuAuthList } from '/@/api/sys/menu';
|
import { iV_USER_MENU } from '/@/api/tigerapi/model/vModel';
|
|
interface UserState {
|
userInfo: Nullable<UserInfo>;
|
token?: string;
|
roleList: RoleEnum[];
|
sessionTimeout?: boolean;
|
lastUpdateTime: number;
|
}
|
|
export const useUserStore = defineStore({
|
id: 'app-user',
|
state: (): UserState => ({
|
// user info
|
userInfo: null,
|
// token
|
token: undefined,
|
// roleList
|
roleList: [],
|
// Whether the login expired
|
sessionTimeout: false,
|
// Last fetch time
|
lastUpdateTime: 0,
|
}),
|
getters: {
|
getUserInfo(state): UserInfo {
|
return state.userInfo || getAuthCache<UserInfo>(USER_INFO_KEY) || {};
|
},
|
getToken(state): string {
|
return state.token || getAuthCache<string>(TOKEN_KEY);
|
},
|
getRoleList(state): RoleEnum[] {
|
return state.roleList.length > 0 ? state.roleList : getAuthCache<RoleEnum[]>(ROLES_KEY);
|
},
|
getSessionTimeout(state): boolean {
|
return !!state.sessionTimeout;
|
},
|
getLastUpdateTime(state): number {
|
return state.lastUpdateTime;
|
},
|
},
|
actions: {
|
setToken(info: string | undefined) {
|
this.token = info ? info : ''; // for null or undefined value
|
setAuthCache(TOKEN_KEY, info);
|
},
|
setRoleList(roleList: RoleEnum[]) {
|
this.roleList = roleList;
|
setAuthCache(ROLES_KEY, roleList);
|
},
|
setUserInfo(info: UserInfo | null) {
|
this.userInfo = info;
|
this.lastUpdateTime = new Date().getTime();
|
setAuthCache(USER_INFO_KEY, info);
|
},
|
setSessionTimeout(flag: boolean) {
|
this.sessionTimeout = flag;
|
},
|
resetState() {
|
this.userInfo = null;
|
this.token = '';
|
this.roleList = [];
|
this.sessionTimeout = false;
|
},
|
/**
|
* @description: login
|
*/
|
async login(
|
params: AccountInfo & {
|
goHome?: boolean;
|
mode?: ErrorMessageMode;
|
},
|
): Promise<GetUserInfoModel | null> {
|
try {
|
const { goHome = true, mode, ...loginParams } = params;
|
const data = await loginApi(loginParams, mode);
|
// const { token } = data;
|
if (!data.IsSuccessed) {
|
return Promise.reject(data.LocaleMsg);
|
}
|
|
// save token
|
this.setToken(data.Data?.TOKEN);
|
return this.afterLoginAction(goHome, loginParams.data.password === '000000');
|
} catch (error) {
|
return Promise.reject(error);
|
}
|
},
|
async afterLoginAction(
|
goHome?: boolean,
|
changepwd?: boolean,): Promise<GetUserInfoModel | null> {
|
if (!this.getToken) return null;
|
// get user info
|
const userInfo = await this.getTgUserInfoAction(changepwd);
|
|
const sessionTimeout = this.sessionTimeout;
|
if (sessionTimeout) {
|
this.setSessionTimeout(false);
|
} else {
|
// const permissionStore = usePermissionStore();
|
|
// // 动态路由加载（首次）
|
// if (!permissionStore.isDynamicAddedRoute) {
|
// const routes = await permissionStore.buildRoutesAction();
|
// [...routes, PAGE_NOT_FOUND_ROUTE].forEach((route) => {
|
// router.addRoute(route as unknown as RouteRecordRaw);
|
// });
|
// // 记录动态路由加载完成
|
// permissionStore.setDynamicAddedRoute(true);
|
// }
|
|
goHome && (await router.replace(userInfo?.homePath || PageEnum.BASE_HOME));
|
}
|
return userInfo;
|
},
|
async getUserInfoAction(changepwd?: boolean): Promise<UserInfo | null> {
|
if (!this.getToken) return null;
|
const data = await getTgUserInfo();
|
const vUserMenu = await getMenuAuthList(data.USER_ID);
|
const userInfo: UserInfo = {
|
userId: data.USER_ID,
|
changepwd: changepwd,
|
username: data.USER_NAME,
|
realName: data.USER_NAME,
|
orgCode: '', //globSetting.useOrg == 'Y' ? data.ORG_CODE : '',
|
prodCode: data.FTY_CODE, //globSetting.useOrg == 'N' ? data.ORG_CODE : '',
|
avatar: '',
|
roles: vUserMenu.Data.Items as unknown as iV_USER_MENU[],
|
};
|
const { roles = [] } = userInfo;
|
if (isArray(roles)) {
|
const roleList = roles.map((item) => item.value) as RoleEnum[];
|
this.setRoleList(roleList);
|
} else {
|
userInfo.roles = [];
|
this.setRoleList([]);
|
}
|
this.setUserInfo(userInfo);
|
return userInfo;
|
},
|
/**
|
* @description: logout
|
*/
|
async logout(goLogin = false) {
|
if (this.getToken) {
|
try {
|
await doLogout({ Data: this.getToken });
|
} catch {
|
console.log('注销Token失败');
|
}
|
}
|
this.setToken(undefined);
|
this.setSessionTimeout(false);
|
this.setUserInfo(null);
|
if (goLogin) {
|
// 直接回登陆页
|
router.replace(PageEnum.BASE_LOGIN);
|
} else {
|
// 回登陆页带上当前路由地址
|
router.replace({
|
path: PageEnum.BASE_LOGIN,
|
query: {
|
redirect: encodeURIComponent(router.currentRoute.value.fullPath),
|
},
|
});
|
}
|
},
|
async getTgUserInfoAction(changepwd?: boolean, orgCode?: string): Promise<UserInfo | null> {
|
if (!this.getToken) return null;
|
const data = await getTgUserInfo();
|
const vUserMenu = await getMenuAuthList(data.USER_ID);
|
const userInfo: UserInfo = {
|
userId: data.USER_ID,
|
username: data.USER_NAME,
|
changepwd: changepwd,
|
realName: data.USER_NAME,
|
orgCode: '', //globSetting.useOrg == 'Y' ? data.ORG_CODE : '',
|
prodCode: data.FTY_CODE, //globSetting.useOrg == 'N' ? data.FTY_CODE : '',
|
avatar: '',
|
roles: vUserMenu.Data.Items as unknown as iV_USER_MENU[],
|
};
|
this.setUserInfo(userInfo);
|
//console.log(data[0].LOGINNAME);
|
return userInfo;
|
},
|
/**
|
* @description: Confirm before logging out
|
*/
|
confirmLoginOut() {
|
const { createConfirm } = useMessage();
|
const { t } = useI18n();
|
createConfirm({
|
iconType: 'warning',
|
title: () => h('span', t('sys.app.logoutTip')),
|
content: () => h('span', t('sys.app.logoutMessage')),
|
onOk: async () => {
|
// 主动登出，不带redirect地址
|
await this.logout(true);
|
},
|
});
|
},
|
},
|
});
|
|
// Need to be used outside the setup
|
export function useUserStoreWithOut() {
|
return useUserStore(store);
|
}
|
