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, TMALLAB_TOKEN_KEY } from '/@/enums/cacheEnum'; import { getAuthCache, setAuthCache } from '/@/utils/auth'; import { GetUserInfoModel, LoginParams } from '/@/api/sys/model/userModel'; import { loginApi, tmallabLoginApi } 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'; interface UserState { userInfo: GetUserInfoModel; token?: string; roleList: []; sessionTimeout?: boolean; lastUpdateTime: number; } export const useUserStore = defineStore({ id: 'app-user', state: (): UserState => ({ // user info userInfo: { userId: '', userName: '', account: '', roles: [], menuCodes: [], perms: [], sAccount: {}, }, // token token: undefined, // roleList roleList: [], // Whether the login expired sessionTimeout: false, // Last fetch time lastUpdateTime: 0, }), getters: { getUserInfo(): GetUserInfoModel { return getAuthCache(USER_INFO_KEY) || {}; }, getToken(): string { return this.token || getAuthCache(TOKEN_KEY); }, getTmallabToken(): string { return getAuthCache(TMALLAB_TOKEN_KEY); }, getRoleList(): [] { return this.roleList.length > 0 ? this.roleList : getAuthCache<[]>(ROLES_KEY); }, getSessionTimeout(): boolean { return !!this.sessionTimeout; }, getLastUpdateTime(): number { return this.lastUpdateTime; }, }, actions: { setToken(info: string | undefined) { this.token = info ? info : ''; // for null or undefined value setAuthCache(TOKEN_KEY, info); }, setTmallabToken(info: string | undefined) { setAuthCache(TMALLAB_TOKEN_KEY, info); }, setRoleList(roleList: []) { this.roleList = roleList; setAuthCache(ROLES_KEY, roleList); }, setUserInfo(info: GetUserInfoModel) { this.userInfo = info; this.lastUpdateTime = new Date().getTime(); setAuthCache(USER_INFO_KEY, info); }, setSessionTimeout(flag: boolean) { this.sessionTimeout = flag; }, resetState() { this.userInfo = {}; this.token = ''; this.roleList = []; this.sessionTimeout = false; }, /** * @description: login */ async login( params: LoginParams & { goHome?: boolean; mode?: ErrorMessageMode; }, ): Promise { try { const { goHome = true, mode, ...loginParams } = params; const data = await loginApi(loginParams, mode); // const { token } = data; console.log(data); // save token this.setToken('112233'); // this.setToken(token); // @ts-ignore return this.afterLoginAction(goHome); } catch (error) { return Promise.reject(error); } }, //数据对接模拟登录 async tmallabSimulationLogin(params: { loginName: string; password: string }): Promise { try { const data = await tmallabLoginApi(params); this.setTmallabToken(data.data.token); this.setToken(data.data.token); return this.afterLoginAction(true, data.data); } catch (error) { return Promise.reject(error); } }, async afterLoginAction(goHome?: boolean, data?: any): Promise { if (!this.getTmallabToken) return null; // get user info const userInfo = await this.getUserInfoAction(data); const sessionTimeout = this.sessionTimeout; if (sessionTimeout) { this.setSessionTimeout(false); } else { const permissionStore = usePermissionStore(); if (!permissionStore.isDynamicAddedRoute) { const routes = await permissionStore.buildRoutesAction(); routes.forEach((route) => { router.addRoute(route as unknown as RouteRecordRaw); }); router.addRoute(PAGE_NOT_FOUND_ROUTE as unknown as RouteRecordRaw); permissionStore.setDynamicAddedRoute(true); } goHome && (await router.replace(userInfo?.homePath || PageEnum.BASE_HOME)); } // @ts-ignore this.setUserInfo(userInfo); return userInfo; }, async getUserInfoAction(data): Promise { if (!this.getTmallabToken) return null; const userInfo = data; // if (!userInfo.userId) { // userInfo = await getUserInfo(); // } const { userRoles } = userInfo; if (isArray(userRoles)) { const role = userInfo.userRoles[0].roleNum; //取第一个为默认角色 const roleList = Array.from(RoleEnum[role]); this.setRoleList(roleList as []); } else { userInfo.userRoles = []; this.setRoleList([]); } this.setUserInfo(userInfo); return userInfo; }, /** * @description: logout */ async logout(goLogin = false) { // if (this.getToken) { // try { // await doLogout(); // } catch { // console.log('注销Token失败'); // } // } this.setToken(undefined); this.setTmallabToken(undefined); this.setSessionTimeout(false); this.setUserInfo({}); goLogin && router.push(PageEnum.BASE_LOGIN); }, /** * @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 () => { await this.logout(true); }, }); }, }, }); // Need to be used outside the setup export function useUserStoreWithOut() { return useUserStore(store); }