import { computed, reactive, ref } from 'vue'; import { useRoute } from 'vue-router'; import { defineStore } from 'pinia'; import { useLoading } from '@sa/hooks'; import { SetupStoreId } from '@/enum'; import { useRouterPush } from '@/hooks/common/router'; import { fetchGetUserInfo, fetchLogin } from '@/service/api'; import { localStg } from '@/utils/storage'; // import { $t } from '@/locales'; import { useRouteStore } from '../route'; import { useTabStore } from '../tab'; import { clearAuthStorage, getToken } from './shared'; export const useAuthStore = defineStore(SetupStoreId.Auth, () => { const route = useRoute(); const routeStore = useRouteStore(); const tabStore = useTabStore(); const { toLogin, redirectFromLogin } = useRouterPush(false); const { loading: loginLoading, startLoading, endLoading } = useLoading(); const token = ref(getToken()); const userInfo: Api.Auth.UserInfo = reactive({ user: undefined, roles: [], permissions: [] }); /** is super role in static route */ const isStaticSuper = computed(() => { const { VITE_AUTH_ROUTE_MODE, VITE_STATIC_SUPER_ROLE } = import.meta.env; return VITE_AUTH_ROUTE_MODE === 'static' && userInfo.roles.includes(VITE_STATIC_SUPER_ROLE); }); /** Is login */ const isLogin = computed(() => Boolean(token.value)); /** Reset auth store */ async function resetStore() { const authStore = useAuthStore(); clearAuthStorage(); authStore.$reset(); if (!route.meta.constant) { await toLogin(); } tabStore.cacheTabs(); routeStore.resetStore(); } /** * Login * * @param [redirect=true] Whether to redirect after login. Default is `true` */ async function login(loginForm: Api.Auth.LoginForm, redirect = true) { startLoading(); const { VITE_APP_CLIENT_ID } = import.meta.env; const loginData: Api.Auth.LoginData = { ...loginForm, clientId: VITE_APP_CLIENT_ID!, grantType: 'password' }; const { data: loginToken, error } = await fetchLogin(loginData); if (!error) { const pass = await loginByToken(loginToken); if (pass) { await routeStore.initAuthRoute(); if (redirect) { await redirectFromLogin(); } if (routeStore.isInitAuthRoute) { // window.$notification?.success({ // title: $t('page.login.common.loginSuccess'), // content: $t('page.login.common.welcomeBack', { userName: userInfo.userName }), // duration: 4500 // }); } } } else { resetStore(); } endLoading(); return error ? Promise.reject(error) : Promise.resolve(); } async function loginByToken(loginToken: Api.Auth.LoginToken) { // 1. stored in the localStorage, the later requests need it in headers localStg.set('token', loginToken.access_token); localStg.set('refreshToken', loginToken.refresh_token); // 2. get user info const pass = await getUserInfo(); if (pass) { token.value = loginToken.access_token; return true; } return false; } async function getUserInfo() { const { data: info, error } = await fetchGetUserInfo(); if (!error) { // update store Object.assign(userInfo, info); return true; } return false; } async function initUserInfo() { const hasToken = getToken(); if (hasToken) { const pass = await getUserInfo(); if (!pass) { resetStore(); } } } return { token, userInfo, isStaticSuper, isLogin, loginLoading, resetStore, login, initUserInfo }; });