yudao-admin-yiqun/yudao-ui-admin-vue3/src/store/modules/user.ts

import { store } from '@/store'
import { defineStore } from 'pinia'
import { getAccessToken, removeToken } from '@/utils/auth'
import { CACHE_KEY, useCache, deleteUserCache } from '@/hooks/web/useCache'
import { getInfo, loginOut } from '@/api/login'
import * as LoginApi from '@/api/login'
const { wsCache } = useCache()
import * as authUtil from '@/utils/auth'

interface UserVO {
  id: string
  avatar: string
  nickname: string
  deptId: number | string
  appointmentDept: string
  employeeNo?: string
  mobile?: string
}

interface UserInfoVO {
  // USER 缓存
  permissions: string[]
  roles: string[]
  isSetUser: boolean
  user: UserVO
}

export const useUserStore = defineStore('admin-user', {
  state: (): UserInfoVO => ({
    permissions: [],
    roles: [],
    isSetUser: false,
    user: {
      id: '',
      avatar: '',
      appointmentDept: '',
      nickname: '',
      deptId: '',
      mobile: ''
    }
  }),
  getters: {
    getPermissions(): string[] {
      return this.permissions
    },
    getRoles(): string[] {
      return this.roles
    },
    getIsSetUser(): boolean {
      return this.isSetUser
    },
    getUser(): UserVO {
      return this.user
    }
  },
  actions: {
    async setUserInfoAction() {
      if (!getAccessToken()) {
        this.resetState()
        return null
      }
      let userInfo = wsCache.get(CACHE_KEY.USER)
      if (!userInfo) {
        userInfo = await getInfo()
      }
      this.permissions = userInfo.permissions
      this.roles = userInfo.roles
      this.user = userInfo.user
      this.isSetUser = true
      wsCache.set(CACHE_KEY.USER, userInfo)
      wsCache.set(CACHE_KEY.ROLE_ROUTERS, userInfo.menus)
    },
    async setUserAppointmentDept(appointmentDept: string) {
      this.user = { ...this.user, appointmentDept }
      const cachedUserInfo = wsCache.get(CACHE_KEY.USER)
      if (cachedUserInfo) {
        cachedUserInfo.user.appointmentDept = appointmentDept
        wsCache.set(CACHE_KEY.USER, cachedUserInfo) // 重新写入缓存
      }
    },
    async setUserAvatarAction(avatar: string) {
      const userInfo = wsCache.get(CACHE_KEY.USER)
      // NOTE: 是否需要像`setUserInfoAction`一样判断`userInfo != null`
      this.user.avatar = avatar
      userInfo.user.avatar = avatar
      wsCache.set(CACHE_KEY.USER, userInfo)
    },
    async setUserNicknameAction(nickname: string) {
      const userInfo = wsCache.get(CACHE_KEY.USER)
      // NOTE: 是否需要像`setUserInfoAction`一样判断`userInfo != null`
      this.user.nickname = nickname
      userInfo.user.nickname = nickname
      wsCache.set(CACHE_KEY.USER, userInfo)
    },
    async ssoLogin(code) {
      try {
        const res = await LoginApi.loginCasMineInfoFn({code})
        if (res && res.accessToken) {
          authUtil.setToken(res)
          return res
        }
        return null
      } catch (error) {
        console.error('SSO登录失败:', error)
        throw error
      }
    },
    async loginOut() {
      await loginOut()
      removeToken()
      deleteUserCache() // 删除用户缓存
      this.resetState()
    },
    resetState() {
      this.permissions = []
      this.roles = []
      this.isSetUser = false
      this.user = {
        id: '',
        appointmentDept: '',
        avatar: '',
        nickname: '',
        deptId: '',
        mobile: ''
      }
    }
  }
})

export const useUserStoreWithOut = () => {
  return useUserStore(store)
}