158 lines
3.9 KiB
TypeScript
158 lines
3.9 KiB
TypeScript
import { store } from "@/store";
|
|
|
|
import AuthAPI, { type LoginFormData } from "@/api/auth.api";
|
|
import UserAPI, { type UserInfo } from "@/api/system/user.api";
|
|
|
|
import { AuthStorage } from "@/utils/auth";
|
|
import { usePermissionStoreHook } from "@/store/modules/permission.store";
|
|
import { useDictStoreHook } from "@/store/modules/dict.store";
|
|
import { useTagsViewStore } from "@/store";
|
|
import { cleanupWebSocket } from "@/plugins/websocket";
|
|
|
|
export const useUserStore = defineStore("user", () => {
|
|
// 用户信息
|
|
const userInfo = ref<UserInfo>({} as UserInfo);
|
|
// 记住我状态
|
|
const rememberMe = ref(AuthStorage.getRememberMe());
|
|
|
|
/**
|
|
* 登录
|
|
*
|
|
* @param {LoginFormData}
|
|
* @returns
|
|
*/
|
|
function login(LoginFormData: LoginFormData) {
|
|
return new Promise<void>((resolve, reject) => {
|
|
AuthAPI.login(LoginFormData)
|
|
.then((data) => {
|
|
const { accessToken, refreshToken } = data;
|
|
// 保存记住我状态和token
|
|
rememberMe.value = LoginFormData.rememberMe;
|
|
AuthStorage.setTokens(accessToken, refreshToken, rememberMe.value);
|
|
resolve();
|
|
})
|
|
.catch((error) => {
|
|
reject(error);
|
|
});
|
|
});
|
|
}
|
|
|
|
/**
|
|
* 获取用户信息
|
|
*
|
|
* @returns {UserInfo} 用户信息
|
|
*/
|
|
function getUserInfo() {
|
|
return new Promise<UserInfo>((resolve, reject) => {
|
|
UserAPI.getInfo()
|
|
.then((data) => {
|
|
if (!data) {
|
|
reject("Verification failed, please Login again.");
|
|
return;
|
|
}
|
|
Object.assign(userInfo.value, { ...data });
|
|
resolve(data);
|
|
})
|
|
.catch((error) => {
|
|
reject(error);
|
|
});
|
|
});
|
|
}
|
|
|
|
/**
|
|
* 登出
|
|
*/
|
|
function logout() {
|
|
return new Promise<void>((resolve, reject) => {
|
|
AuthAPI.logout()
|
|
.then(() => {
|
|
// 重置所有系统状态
|
|
resetAllState();
|
|
resolve();
|
|
})
|
|
.catch((error) => {
|
|
reject(error);
|
|
});
|
|
});
|
|
}
|
|
|
|
/**
|
|
* 重置所有系统状态
|
|
* 统一处理所有清理工作，包括用户凭证、路由、缓存等
|
|
*/
|
|
function resetAllState() {
|
|
// 1. 重置用户状态
|
|
resetUserState();
|
|
|
|
// 2. 重置其他模块状态
|
|
// 重置路由
|
|
usePermissionStoreHook().resetRouter();
|
|
// 清除字典缓存
|
|
useDictStoreHook().clearDictCache();
|
|
// 清除标签视图
|
|
useTagsViewStore().delAllViews();
|
|
|
|
// 3. 清理 WebSocket 连接
|
|
cleanupWebSocket();
|
|
console.log("[UserStore] WebSocket connections cleaned up");
|
|
|
|
return Promise.resolve();
|
|
}
|
|
|
|
/**
|
|
* 重置用户状态
|
|
* 仅处理用户模块内的状态
|
|
*/
|
|
function resetUserState() {
|
|
// 清除用户凭证
|
|
AuthStorage.clearAuth();
|
|
// 重置用户信息
|
|
userInfo.value = {} as UserInfo;
|
|
}
|
|
|
|
/**
|
|
* 刷新 token
|
|
*/
|
|
function refreshToken() {
|
|
const refreshToken = AuthStorage.getRefreshToken();
|
|
|
|
if (!refreshToken) {
|
|
return Promise.reject(new Error("没有有效的刷新令牌"));
|
|
}
|
|
|
|
return new Promise<void>((resolve, reject) => {
|
|
AuthAPI.refreshToken(refreshToken)
|
|
.then((data) => {
|
|
const { accessToken, refreshToken: newRefreshToken } = data;
|
|
// 更新令牌，保持当前记住我状态
|
|
AuthStorage.setTokens(accessToken, newRefreshToken, AuthStorage.getRememberMe());
|
|
resolve();
|
|
})
|
|
.catch((error) => {
|
|
console.log(" refreshToken 刷新失败", error);
|
|
reject(error);
|
|
});
|
|
});
|
|
}
|
|
|
|
return {
|
|
userInfo,
|
|
rememberMe,
|
|
isLoggedIn: () => !!AuthStorage.getAccessToken(),
|
|
getUserInfo,
|
|
login,
|
|
logout,
|
|
resetAllState,
|
|
resetUserState,
|
|
refreshToken,
|
|
};
|
|
});
|
|
|
|
/**
|
|
* 在组件外部使用UserStore的钩子函数
|
|
* @see https://pinia.vuejs.org/core-concepts/outside-component-usage.html
|
|
*/
|
|
export function useUserStoreHook() {
|
|
return useUserStore(store);
|
|
}
|