nova-admin/src/store/auth.ts

import { router } from '@/router'
import { fetchLogin, fetchUserInfo } from '@/api'
import { local } from '@/utils'
import { useRouteStore } from './router'
import { useTabStore } from './tab'

interface AuthStatus {
  userInfo: Entity.User
  roles: string[]
  permissions: string[]
}
export const useAuthStore = defineStore('auth-store', {
  state: (): AuthStatus => {
    return {
      userInfo: {},
      roles: [],
      permissions: [],
    }
  },
  getters: {
    /** 是否登录 */
    isLogin(state) {
      return Boolean(state.userInfo)
    },
  },
  actions: {
    /* 登录退出，重置用户信息等 */
    async logout() {
      const route = unref(router.currentRoute)
      // 清除本地缓存
      this.clearAuthStorage()
      // 清空路由、菜单等数据
      const routeStore = useRouteStore()
      routeStore.resetRouteStore()
      // 清空标签栏数据
      const tabStore = useTabStore()
      tabStore.clearAllTabs()
      // 重置当前存储库
      this.$reset()
      // 重定向到登录页
      if (route.meta.requiresAuth) {
        router.push({
          name: 'login',
          query: {
            redirect: route.fullPath,
          },
        })
      }
    },
    clearAuthStorage() {
      local.remove('accessToken')
      local.remove('refreshToken')
    },

    /* 用户登录 */
    async login(userName: string, password: string, captchaId?: string, captcha?: string) {
      const loginData: any = { userName, password }

      // 如果提供了验证码相关参数，则添加到登录数据中
      if (captchaId && captcha) {
        loginData.captchaId = captchaId
        loginData.captcha = captcha
      }

      const { data } = await fetchLogin(loginData)

      // 处理登录信息
      try {
        await this.handleLoginInfo(data)
      }
      catch (error) {
        console.error('Failed to handle login info:', error)
        throw error
      }

      // 更新用户信息
      await this.updataUserInfo()
    },
    async updataUserInfo() {
      const { data } = await fetchUserInfo()
      this.userInfo = data
      this.roles = data.roles.map((role: Entity.Role) => role.roleKey)
      this.permissions = data.permissions
    },
    /* 处理登录返回的数据 */
    async handleLoginInfo(data: any) {
      // 将token保存下来
      local.set('accessToken', data.accessToken)
      if (data.refreshToken) {
        local.set('refreshToken', data.refreshToken)
      }

      // 添加路由和菜单
      const { initAuthRoute } = useRouteStore()
      await initAuthRoute()

      // 进行重定向跳转
      const route = unref(router.currentRoute)
      const query = route.query as { redirect: string }
      router.push({
        path: query.redirect || '/',
      })
    },
  },
  persist: {
    storage: localStorage,
  },
})