2024-07-05 23:15:46 +08:00
|
|
|
import type { MenuRecordRaw, UserInfo } from '@vben/types';
|
|
|
|
import type { LoginAndRegisterParams } from '@vben/universal-ui';
|
|
|
|
import type { RouteRecordRaw } from 'vue-router';
|
|
|
|
|
|
|
|
import { computed, ref } from 'vue';
|
|
|
|
import { useRouter } from 'vue-router';
|
|
|
|
|
2024-07-11 20:11:11 +08:00
|
|
|
import { DEFAULT_HOME_PATH, LOGIN_PATH } from '@vben/constants';
|
2024-07-05 23:15:46 +08:00
|
|
|
import { useCoreAccessStore } from '@vben-core/stores';
|
|
|
|
|
|
|
|
import { defineStore } from 'pinia';
|
|
|
|
|
2024-07-06 13:28:08 +08:00
|
|
|
import { getAccessCodes, getUserInfo, userLogin } from '#/apis';
|
2024-07-05 23:15:46 +08:00
|
|
|
|
|
|
|
export const useAccessStore = defineStore('access', () => {
|
|
|
|
const coreStoreAccess = useCoreAccessStore();
|
|
|
|
const router = useRouter();
|
|
|
|
const loading = ref(false);
|
|
|
|
|
2024-07-11 20:11:11 +08:00
|
|
|
const openLoginExpiredModal = ref(false);
|
2024-07-11 11:05:01 +08:00
|
|
|
|
2024-07-10 00:50:41 +08:00
|
|
|
const accessToken = computed(() => coreStoreAccess.accessToken);
|
2024-07-11 20:11:11 +08:00
|
|
|
const refreshToken = computed(() => coreStoreAccess.refreshToken);
|
2024-07-10 00:50:41 +08:00
|
|
|
const userRoles = computed(() => coreStoreAccess.userRoles);
|
|
|
|
const userInfo = computed(() => coreStoreAccess.userInfo);
|
|
|
|
const accessRoutes = computed(() => coreStoreAccess.accessRoutes);
|
2024-07-05 23:15:46 +08:00
|
|
|
|
|
|
|
function setAccessMenus(menus: MenuRecordRaw[]) {
|
|
|
|
coreStoreAccess.setAccessMenus(menus);
|
|
|
|
}
|
|
|
|
|
2024-07-11 20:11:11 +08:00
|
|
|
function setAccessToken(token: null | string) {
|
|
|
|
coreStoreAccess.setAccessToken(token);
|
|
|
|
}
|
|
|
|
|
2024-07-05 23:15:46 +08:00
|
|
|
function setAccessRoutes(routes: RouteRecordRaw[]) {
|
|
|
|
coreStoreAccess.setAccessRoutes(routes);
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* 异步处理登录操作
|
|
|
|
* Asynchronously handle the login process
|
|
|
|
* @param params 登录表单数据
|
|
|
|
*/
|
|
|
|
async function authLogin(
|
|
|
|
params: LoginAndRegisterParams,
|
|
|
|
onSuccess?: () => Promise<void>,
|
|
|
|
) {
|
|
|
|
// 异步处理用户登录操作并获取 accessToken
|
|
|
|
let userInfo: UserInfo | null = null;
|
|
|
|
try {
|
|
|
|
loading.value = true;
|
|
|
|
const { accessToken, refreshToken } = await userLogin(params);
|
|
|
|
|
|
|
|
// 如果成功获取到 accessToken
|
|
|
|
// If accessToken is successfully obtained
|
|
|
|
if (accessToken) {
|
|
|
|
// 将 accessToken 存储到 accessStore 中
|
|
|
|
// Store the accessToken in accessStore
|
|
|
|
coreStoreAccess.setAccessToken(accessToken);
|
|
|
|
coreStoreAccess.setRefreshToken(refreshToken);
|
|
|
|
|
|
|
|
// 获取用户信息并存储到 accessStore 中
|
|
|
|
// Get user information and store it in accessStore
|
2024-07-06 13:28:08 +08:00
|
|
|
const [fetchUserInfoResult, accessCodes] = await Promise.all([
|
|
|
|
fetchUserInfo(),
|
|
|
|
getAccessCodes(),
|
|
|
|
]);
|
|
|
|
|
|
|
|
userInfo = fetchUserInfoResult;
|
2024-07-05 23:15:46 +08:00
|
|
|
|
|
|
|
coreStoreAccess.setUserInfo(userInfo);
|
2024-07-06 13:28:08 +08:00
|
|
|
coreStoreAccess.setAccessCodes(accessCodes);
|
2024-07-05 23:15:46 +08:00
|
|
|
|
2024-07-11 20:11:11 +08:00
|
|
|
openLoginExpiredModal.value = false;
|
2024-07-05 23:15:46 +08:00
|
|
|
onSuccess
|
|
|
|
? await onSuccess?.()
|
|
|
|
: await router.push(userInfo.homePath || DEFAULT_HOME_PATH);
|
|
|
|
}
|
|
|
|
} finally {
|
|
|
|
loading.value = false;
|
|
|
|
}
|
|
|
|
|
|
|
|
return {
|
|
|
|
accessToken,
|
|
|
|
userInfo,
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2024-07-11 20:11:11 +08:00
|
|
|
async function logout() {
|
|
|
|
coreStoreAccess.$reset();
|
|
|
|
openLoginExpiredModal.value = false;
|
|
|
|
|
|
|
|
// 回登陆页带上当前路由地址
|
|
|
|
await router.replace({
|
|
|
|
path: LOGIN_PATH,
|
|
|
|
query: {
|
|
|
|
redirect: encodeURIComponent(router.currentRoute.value.fullPath),
|
|
|
|
},
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
2024-07-05 23:15:46 +08:00
|
|
|
async function fetchUserInfo() {
|
|
|
|
let userInfo: UserInfo | null = null;
|
|
|
|
userInfo = await getUserInfo();
|
|
|
|
coreStoreAccess.setUserInfo(userInfo);
|
|
|
|
return userInfo;
|
|
|
|
}
|
|
|
|
|
|
|
|
function reset() {
|
|
|
|
coreStoreAccess.$reset();
|
|
|
|
}
|
|
|
|
|
|
|
|
return {
|
|
|
|
accessRoutes,
|
|
|
|
accessToken,
|
|
|
|
authLogin,
|
|
|
|
fetchUserInfo,
|
|
|
|
loading,
|
2024-07-11 20:11:11 +08:00
|
|
|
logout,
|
|
|
|
openLoginExpiredModal,
|
|
|
|
refreshToken,
|
2024-07-05 23:15:46 +08:00
|
|
|
reset,
|
|
|
|
setAccessMenus,
|
|
|
|
setAccessRoutes,
|
2024-07-11 20:11:11 +08:00
|
|
|
setAccessToken,
|
2024-07-05 23:15:46 +08:00
|
|
|
userInfo,
|
|
|
|
userRoles,
|
|
|
|
};
|
|
|
|
});
|