mec-cloud_-intelligent-manufacturing_crm/zkqy-ui/src/store/modules/user.js

import {login, logout, getInfo, adminLoginApi} from '@/api/login'
import {changeDatasource} from '@/api/dataEngine/index'
import {getToken, setToken, removeToken} from '@/utils/auth'

const user = {
  state: {
    token: getToken(),
    name: '',
    userId: '',
    avatar: '',
    roles: [],
    permissions: [],
    tenant: {},
    dataSource: {},
    nickName: "",
  },

  mutations: {
    SET_TOKEN: (state, token) => {
      state.token = token
    },
    SET_NAME: (state, name) => {
      state.name = name
    },
    SET_NICKNAME: (state, nickName) => {
      state.nickName = nickName
    },
    SET_USER_ID: (state, userId) => {
      state.userId = userId
    },
    SET_AVATAR: (state, avatar) => {
      state.avatar = avatar
    },
    SET_ROLES: (state, roles) => {
      state.roles = roles
    },
    SET_PERMISSIONS: (state, permissions) => {
      state.permissions = permissions
    },
    SET_TENANT: (state, tenant) => {
      state.tenant = tenant
    },
    SET_DATASOURCE: (state, dataSource) => {
      state.dataSource = dataSource
    }
  },

  actions: {
    // 登录
    Login({commit}, userInfo) {
      const username = userInfo.username.trim()
      const password = userInfo.password
      const tenantID = userInfo.tenantID
      const code = userInfo.code
      const uuid = userInfo.uuid
      return new Promise((resolve, reject) => {
        login(username, password, code, tenantID, uuid).then(res => {
          commit('SET_NAME', username)
          setToken(res.token)
          commit('SET_TOKEN', res.token)
          resolve()
        }).catch(error => {
          reject(error)
        })
      })
    },
    // admin登录
    adminLogin({commit}, userInfo) {
      const username = userInfo.username.trim()
      const password = userInfo.password

      const code = userInfo.code
      const uuid = userInfo.uuid
      return new Promise((resolve, reject) => {
        adminLoginApi(username, password, code, uuid).then(res => {
          commit('SET_NAME', username)

          setToken(res.token)
          commit('SET_TOKEN', res.token)
          resolve()
        }).catch(error => {
          reject(error)
        })
      })
    },

    // 获取用户信息
    GetInfo({commit, state}) {
      return new Promise((resolve, reject) => {
        getInfo().then(res => {
          const user = res.user
          const nickName = res.user?.nickName
          commit('SET_USER_ID', res.user.userId)
          commit('SET_NICKNAME', nickName)
          const avatar = (user.avatar == "" || user.avatar == null) ? require("@/assets/images/profile.jpg") : process.env.VUE_APP_BASE_API + user.avatar;
          if (res.roles && res.roles.length > 0) { // 验证返回的roles是否是一个非空数组
            commit('SET_ROLES', res.roles)
            commit('SET_PERMISSIONS', res.permissions)
          } else {
            commit('SET_ROLES', ['ROLE_DEFAULT'])
          }
          if (res.tenant != null) {
            commit('SET_TENANT', res.tenant)
            commit('SET_DATASOURCE', res.dataSource)

          }
          commit('SET_NAME', user.userName)
          commit('SET_AVATAR', avatar)
          resolve(res)

        }).catch(error => {
          reject(error)
        })
      })
    },

    // 退出系统
    LogOut({commit, state}) {
      return new Promise((resolve, reject) => {
        logout(state.token).then(() => {
          if (state.userId == 1) {  // 如果当前用户是admin的话
            location.href = "/adminLogin"
          } else {
            location.href = state.tenant.tenantToolLoginUrl
          }
        }).catch(error => {
          reject(error)
        }).finally(() => {
          commit('SET_TOKEN', '')
          commit('SET_ROLES', [])
          commit('SET_PERMISSIONS', [])
          commit('SET_TENANT', {})
          removeToken()
          localStorage.clear();
          sessionStorage.clear();
          resolve()
        })
      })
    },

    // 前端 登出
    FedLogOut({commit}) {
      return new Promise(resolve => {
        commit('SET_TOKEN', '')
        removeToken()
        resolve()
      })
    }
  }
}

export default user