iot-web/src/store/modules/user.ts

import { to } from 'await-to-js'
import defAva from '@/assets/images/profile.jpg'
import store from '@/store'
import { getToken, removeToken, removeName,setToken, setName } from '@/utils/auth'
import { login as loginApi, logout as logoutApi, getInfo as getUserInfo, loginWithoutPass as loginWithoutPassApi } from '@/api/login'
import { LoginData } from '@/api/types'
import { listByIds } from '@/api/system/oss'

export const useUserStore = defineStore('user', () => {
  const token = ref(getToken())
  const name = ref('')
  const nickname = ref('')
  const userId = ref<string | number>('')
  const avatar = ref('')
  const roles = ref<Array<string>>([]) // 用户角色编码集合 → 判断路由权限
  const permissions = ref<Array<string>>([]) // 用户权限编码集合 → 判断按钮权限

  /**
   * 登录
   * @param userInfo
   * @returns
   */
  const login = async (userInfo: LoginData): Promise<void> => {
    const [err, res] = await to(loginApi(userInfo))
    if (res) {
      const data = res.data
      setToken(data.token)
      token.value = data.token
      return Promise.resolve()
    }
    return Promise.reject(err)
  }
  const loginNoPass = async (userInfo: LoginData): Promise<void> => {
    const [err, res] = await to(loginWithoutPassApi(userInfo))
    if (res) {
      debugger
      const data = res.data
      setToken(data.token)
      token.value = data.token
      return Promise.resolve()
    }
    return Promise.reject(err)
  }

  // 获取用户信息
  const getInfo = async (): Promise<void> => {
    const [err, res] = await to(getUserInfo())
    if (res) {
      const data = res.data
      const user = data.user
      const profile = user.avatar

      if (data.roles && data.roles.length > 0) {
        // 验证返回的roles是否是一个非空数组
        roles.value = data.roles
        permissions.value = data.permissions
        console.log('##########################permissions', JSON.stringify(data.permissions))
      } else {
        roles.value = ['ROLE_DEFAULT']
      }
      name.value = user.userName
      setName(user.userName)
      nickname.value = user.nickName
      if (profile) {
        const ossObj = await listByIds(profile)
        console.log('ossObj', ossObj)
        if (ossObj.data) avatar.value = ossObj.data[0].url
        console.log(avatar.value)
      } else {
        avatar.value = defAva
      }
      userId.value = user.id
      return Promise.resolve()
    }
    return Promise.reject(err)
  }

  // 注销
  const logout = async (): Promise<void> => {
    await logoutApi()
    token.value = ''
    roles.value = []
    permissions.value = []
    removeToken()
    removeName()
  }

  const clearUserInfo = async (): Promise<void> => {
    token.value = ''
    roles.value = []
    permissions.value = []
    removeToken()
    removeName()
  }

  return {
    userId,
    token,
    nickname,
    avatar,
    roles,
    permissions,
    login,
    getInfo,
    logout,
    loginNoPass,
    clearUserInfo
  }
})

export default useUserStore
// 非setup
export function useUserStoreHook() {
  return useUserStore(store)
}