143 lines
3.5 KiB
TypeScript
143 lines
3.5 KiB
TypeScript
import type { LoginAndRegisterParams } from '@vben/common-ui'
|
|
import type { UserInfo } from '@vben/types'
|
|
|
|
import { ref } from 'vue'
|
|
import { useRouter } from 'vue-router'
|
|
|
|
import { LOGIN_PATH } from '@vben/constants'
|
|
import { preferences } from '@vben/preferences'
|
|
import { resetAllStores, useAccessStore, useUserStore } from '@vben/stores'
|
|
|
|
import { notification } from 'ant-design-vue'
|
|
import { defineStore } from 'pinia'
|
|
|
|
import { doLogout, getUserInfoApi, loginApi, seeConnectionClose } from '#/api'
|
|
import { $t } from '#/locales'
|
|
|
|
import { useDictStore } from './dict'
|
|
|
|
export const useAuthStore = defineStore('auth', () => {
|
|
const accessStore = useAccessStore()
|
|
const userStore = useUserStore()
|
|
const router = useRouter()
|
|
|
|
const loginLoading = ref(false)
|
|
|
|
/**
|
|
* 异步处理登录操作
|
|
* Asynchronously handle the login process
|
|
* @param params 登录表单数据
|
|
*/
|
|
async function authLogin(
|
|
params: LoginAndRegisterParams,
|
|
onSuccess?: () => Promise<void> | void,
|
|
) {
|
|
// 异步处理用户登录操作并获取 accessToken
|
|
let userInfo: null | UserInfo = null
|
|
try {
|
|
loginLoading.value = true
|
|
const { access_token } = await loginApi(params)
|
|
|
|
// 将 accessToken 存储到 accessStore 中
|
|
accessStore.setAccessToken(access_token)
|
|
accessStore.setRefreshToken(access_token)
|
|
|
|
// 获取用户信息并存储到 accessStore 中
|
|
userInfo = await fetchUserInfo()
|
|
/**
|
|
* 设置用户信息
|
|
*/
|
|
userStore.setUserInfo(userInfo)
|
|
/**
|
|
* 在这里设置权限
|
|
*/
|
|
accessStore.setAccessCodes(userInfo.permissions)
|
|
|
|
if (accessStore.loginExpired) {
|
|
accessStore.setLoginExpired(false)
|
|
} else {
|
|
onSuccess ? await onSuccess?.() : await router.push('/analytics')
|
|
}
|
|
|
|
if (userInfo?.realName) {
|
|
notification.success({
|
|
description: `${$t('authentication.loginSuccessDesc')}:${userInfo?.realName}`,
|
|
duration: 3,
|
|
message: $t('authentication.loginSuccess'),
|
|
})
|
|
}
|
|
} finally {
|
|
loginLoading.value = false
|
|
}
|
|
|
|
return {
|
|
userInfo,
|
|
}
|
|
}
|
|
|
|
async function logout(redirect: boolean = true) {
|
|
try {
|
|
await seeConnectionClose()
|
|
await doLogout()
|
|
} catch (error) {
|
|
console.error(error)
|
|
} finally {
|
|
resetAllStores()
|
|
accessStore.setLoginExpired(false)
|
|
|
|
// 回登陆页带上当前路由地址
|
|
await router.replace({
|
|
path: LOGIN_PATH,
|
|
query: redirect
|
|
? {
|
|
redirect: encodeURIComponent(router.currentRoute.value.fullPath),
|
|
}
|
|
: {},
|
|
})
|
|
}
|
|
}
|
|
|
|
async function fetchUserInfo() {
|
|
const backUserInfo = await getUserInfoApi()
|
|
/**
|
|
* 登录超时的情况
|
|
*/
|
|
if (!backUserInfo) {
|
|
throw new Error('获取用户信息失败.')
|
|
}
|
|
const { permissions = [], roles = [], user } = backUserInfo
|
|
/**
|
|
* 从后台user -> vben user转换
|
|
*/
|
|
const userInfo: UserInfo = {
|
|
avatar: user.avatar ?? '',
|
|
permissions,
|
|
realName: user.nickName,
|
|
roles,
|
|
userId: user.userId,
|
|
username: user.userName,
|
|
}
|
|
userStore.setUserInfo(userInfo)
|
|
|
|
/**
|
|
* 需要重新加载字典
|
|
* 比如退出登录切换到其他租户
|
|
*/
|
|
const dictStore = useDictStore()
|
|
dictStore.resetCache()
|
|
return userInfo
|
|
}
|
|
|
|
function $reset() {
|
|
loginLoading.value = false
|
|
}
|
|
|
|
return {
|
|
$reset,
|
|
authLogin,
|
|
fetchUserInfo,
|
|
loginLoading,
|
|
logout,
|
|
}
|
|
})
|