admin-vben5/apps/web-antd/src/store/auth.ts

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,
  }
})