transitPlatForm-Vue3/src/store/modules/user.ts

import router from '@/router'
import { ElMessageBox } from 'element-plus'
import { login, logout, getInfo, loginAnonymous } from '@/api/login'
import { getToken, setToken, removeToken } from '@/utils/auth'
import { isHttp, isEmpty } from "@/utils/validate"
import defAva from '@/assets/images/profile.jpg'

interface RoleInfo {
  roleName: string
  [key: string]: any
}

interface UserState {
  token: string | undefined
  id: string | number
  name: string
  nickName: string
  roleName: string
  avatar: string
  roles: string[]
  permissions: string[]
  sysLoginParam: Object
}

const useUserStore = defineStore(
  'user',
  {
    state: (): UserState => ({
      token: getToken(),
      id: '',
      name: '',
      nickName: '',
      roleName: '',
      avatar: '',
      roles: [],
      permissions: [],
      sysLoginParam: {}, // 登录参数
    }),
    actions: {
      // 单点登录
      loginAnonymous(data: { name: string; yljg: string }) {
        return new Promise((resolve, reject) => {
          loginAnonymous({
            username: data.name,
            yljg: data.yljg
          }).then(res => {
            if (res.code == 200) {
              setToken(res.token)
              this.token = res.token
              resolve(true)
            }

          }).catch(error => {
            reject(error)
          })
        })
      },
      // 登录
      login(userInfo: { username: string; password: string; code: string; uuid: string; loginParam: Object }) {
        const username = userInfo.username.trim()
        const password = userInfo.password
        const code = userInfo.code
        const uuid = userInfo.uuid
        const loginParam = userInfo.loginParam
        return new Promise<void>((resolve, reject) => {
          login(username, password, code, uuid, loginParam).then(res => {
            setToken(res.token)
            this.token = res.token
            resolve()
          }).catch(error => {
            reject(error)
          })
        })
      },
      // 获取用户信息
      getInfo() {
        return new Promise((resolve, reject) => {
          getInfo().then(res => {
            const user = res.user
            let avatar = user.avatar || ''
            this.sysLoginParam = user.sysLoginParam
            if (!isHttp(avatar)) {
              avatar = (isEmpty(avatar)) ? defAva : import.meta.env.VITE_APP_BASE_API + avatar
            }
            const roles = res.user.roles as any[]
            if (roles && roles.length > 0) {
              this.roles = roles
              this.permissions = res.permissions
              this.roleName = (roles[0] as RoleInfo).roleName
            } else {
              this.roles = ['ROLE_DEFAULT']
            }
            this.id = user.userId || ''
            this.name = user.userName || ''
            this.nickName = user.nickName || ''
            this.avatar = avatar
            /* 初始密码提示 */
            if (res.isDefaultModifyPwd) {
              ElMessageBox.confirm('您的密码还是初始密码，请修改密码！', '安全提示', { confirmButtonText: '确定', cancelButtonText: '取消', type: 'warning' }).then(() => {
                router.push({ name: 'Profile', params: { activeTab: 'resetPwd' } })
              }).catch(() => { })
            }
            /* 过期密码提示 */
            if (!res.isDefaultModifyPwd && res.isPasswordExpired) {
              ElMessageBox.confirm('您的密码已过期，请尽快修改密码！', '安全提示', { confirmButtonText: '确定', cancelButtonText: '取消', type: 'warning' }).then(() => {
                router.push({ name: 'Profile', params: { activeTab: 'resetPwd' } })
              }).catch(() => { })
            }
            resolve(res)
          }).catch(error => {
            reject(error)
          })
        })
      },
      // 退出系统
      logOut() {
        return new Promise<void>((resolve, reject) => {
          logout().then(() => {
            this.token = ''
            this.roles = []
            this.permissions = []
            removeToken()
            resolve()
          }).catch((error: any) => {
            reject(error)
          })
        })
      }
    }
  })

export default useUserStore