front/front-vue/src/store/modules/user.js

import { login, logout, getInfo, refreshToken } from '@/api/login'
import { wxLogin } from '@/api/system/weChat'
import { login as authorizeLogin, loginByUser } from '@/api/system/authorizeLogin'
import { getToken, setToken, setExpiresIn, removeToken } from '@/utils/auth'

const user = {
  state: {
    token: getToken(),
    name: '',
    avatar: '',
    roles: [],
    permissions: []
  },

  mutations: {
    SET_TOKEN: (state, token) => {
      state.token = token
    },
    SET_EXPIRES_IN: (state, time) => {
      state.expires_in = time
    },
    SET_NAME: (state, name) => {
      state.name = name
    },
    SET_AVATAR: (state, avatar) => {
      state.avatar = avatar
    },
    SET_ROLES: (state, roles) => {
      state.roles = roles
    },
    SET_PERMISSIONS: (state, permissions) => {
      state.permissions = permissions
    }
  },

  actions: {
    // 登录
    Login({ commit }, userInfo) {
      const username = userInfo.username.trim()
      const password = userInfo.password
      const code = userInfo.code
      const uuid = userInfo.uuid
      const type = userInfo.type
      const shortMessageCode = userInfo.shortMessageCode
      const unionId = userInfo.unionId

      return new Promise((resolve, reject) => {
        login(username, password, code, uuid, type, shortMessageCode, unionId).then(res => {
          let data = res.data

          setToken(data.tokenInfo.access_token);
          commit('SET_TOKEN', data.tokenInfo.access_token)
          setExpiresIn(data.tokenInfo.expires_in)
          commit('SET_EXPIRES_IN', data.tokenInfo.expires_in)
          // 未认证
          if (data.code == "0") {
            console.log("未绑定企业")
          // 认证单个企业
          } else if (data.code == "1") {

          // 多加企业
          } else {
            console.log("绑定了多家企业!")
          }
          resolve(data)
        }).catch(error => {
          reject(error)
        })
      })
    },
    // 扫码登录
    WxLogin({ commit }, query) {
      return new Promise((resolve, reject) => {
        wxLogin(query).then(res => {
          let data = res.data


          // 未认证
          if (data.code == "0") {
            console.log("未绑定企业")
            setToken(data.tokenInfo.access_token);
            commit('SET_TOKEN', data.tokenInfo.access_token)
            setExpiresIn(data.tokenInfo.expires_in)
            commit('SET_EXPIRES_IN', data.tokenInfo.expires_in)
          // 认证单个企业
          } else if (data.code == "1") {
            setToken(data.tokenInfo.access_token);
            commit('SET_TOKEN', data.tokenInfo.access_token)
            setExpiresIn(data.tokenInfo.expires_in)
            commit('SET_EXPIRES_IN', data.tokenInfo.expires_in)
          // 多加企业
          } else if (data.code == "2") {
            setToken(data.tokenInfo.access_token);
            commit('SET_TOKEN', data.tokenInfo.access_token)
            setExpiresIn(data.tokenInfo.expires_in)
            commit('SET_EXPIRES_IN', data.tokenInfo.expires_in)
          // 未注册用户
          } else {

          }
          resolve(res)
        }).catch(error => {
          reject(error)
        })
      })
    },
    // 免登录
    AuthorizeLogin({ commit }, query) {
      return new Promise((resolve, reject) => {
        authorizeLogin(query).then(res => {
          let data = res.data
          if(data.userList){
            resolve(data)
          }else{
            setToken(data.access_token)
            commit('SET_TOKEN', data.access_token)
            setExpiresIn(data.expires_in)
            commit('SET_EXPIRES_IN', data.expires_in)
            resolve()
          }
        }).catch(error => {
          reject(error)
        })
      })
    },
    // 选择用户登录
    LoginByUser({ commit }, query) {
      return new Promise((resolve, reject) => {
        loginByUser(query).then(res => {
          let data = res.data
          setToken(data.access_token)
          commit('SET_TOKEN', data.access_token)
          setExpiresIn(data.expires_in)
          commit('SET_EXPIRES_IN', data.expires_in)
          resolve()
        }).catch(error => {
          reject(error)
        })
      })
    },
    // 获取用户信息
    GetInfo({ commit, state }) {
      return new Promise((resolve, reject) => {
        getInfo(state.token).then(res => {
          if (res.code == 501 || res.code == 502 || res.code == 503) {
            resolve(res)
          }
          const user = res.user
          const avatar = user.avatar == "" ? require("@/assets/images/profile.jpg") : user.avatar;
          if (res.roles && res.roles.length > 0) { // 验证返回的roles是否是一个非空数组
            commit('SET_ROLES', res.roles)
            commit('SET_PERMISSIONS', res.permissions)
          } else {
            // 管理员
            if (user.companyId == "000000") {
              commit('SET_ROLES', ['ROLE_DEFAULT'])
            } else {
              commit('SET_ROLES', [])
            }
          }
          commit('SET_NAME', user.userName)
          commit('SET_AVATAR', avatar)
          resolve(res)
        }).catch(error => {
          reject(error)
        })
      })
    },

    // 刷新token
    RefreshToken({commit, state}) {
      return new Promise((resolve, reject) => {
        refreshToken(state.token).then(res => {
          setExpiresIn(res.data)
          commit('SET_EXPIRES_IN', res.data)
          resolve()
        }).catch(error => {
          reject(error)
        })
      })
    },

    // 退出系统
    LogOut({ commit, state }) {
      return new Promise((resolve, reject) => {
        logout(state.token).then(() => {
          commit('SET_TOKEN', '')
          commit('SET_ROLES', [])
          commit('SET_PERMISSIONS', [])
          removeToken()
          resolve()
        }).catch(error => {
          reject(error)
        })
      })
    },

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

export default user