leromro-mini/store/modules/user.ts

import config from '@/config'
import storage from '@/utils/storage'
import constant from '@/utils/constant'
import { login, logout, getInfo } from '@/api/login'
import { getToken, setToken, removeToken } from '@/utils/auth'
import { UserState, UserForm } from '@/types/store'
import { Module } from 'vuex'
import WebSocketManager from '@/utils/WebSocketManager.js';
const baseUrl = config.baseUrl

const user: Module<UserState, UserState> = {
  state: {
    token: getToken(),
    name: storage.get(constant.name),
    avatar: storage.get(constant.avatar),
    roles: storage.get(constant.roles),
    permissions: storage.get(constant.permissions),
    userOrWorker: storage.get(constant.userOrWorker),
    nickName: storage.get(constant.nickName),
    userId: storage.get(constant.userId),
    messageCount: 0,
    wsManager: null
  },
  mutations: {
    SET_TOKEN: (state, token: string) => {
      state.token = token
    },
    SET_NAME: (state, name: string) => {
      state.name = name
      storage.set(constant.name, name)
    },
    SET_AVATAR: (state, avatar: string) => {
      state.avatar = avatar
      storage.set(constant.avatar, avatar)
    },
    SET_ROLES: (state, roles: Array<string>) => {
      state.roles = roles
      storage.set(constant.roles, roles)
    },
    SET_PERMISSIONS: (state, permissions: Array<string>) => {
      state.permissions = permissions
      storage.set(constant.permissions, permissions)
    },
    SET_USERORWORKER: (state, userOrWorker: Array<string>) => {
      state.userOrWorker = userOrWorker
      storage.set(constant.userOrWorker, userOrWorker)
    },
    SET_NICKNAME: (state, nickName: String) => {
      state.nickName = nickName
      storage.set(constant.nickName, nickName)
    },
    SET_USERID: (state, id: String) => {
      state.userId = id
    },
    SET_MESSAGECOUNT: (state, num: Number) => {
      state.messageCount = num
    },
    SET_MESSAGE: (state, ms:any) => {
      state.wsManager = ms
    },
  },

  actions: {
    // 登录
    Login({ commit }, userInfo: UserForm) {
      const username = userInfo.username
      const password = userInfo.password
      const code = userInfo.code
      const uuid = userInfo.uuid
      const referrerType = userInfo.referrerType
      const referrerId = userInfo.referrerId
      return new Promise((resolve, reject) => {
        login(username, password, code, uuid, referrerType, referrerId).then((res: any) => {
          setToken(res.token)
          commit('SET_TOKEN', res.token)
          resolve(res)
        }).catch(error => {
          reject(error)
        })
      })
    },

    // 获取用户信息
    GetInfo({ commit, state }) {
      return new Promise((resolve, reject) => {
        getInfo().then((res: any) => {
          const user = res.user
          const avatar = (user == null || user.avatar == "" || user.avatar == null) ? "/static/serverImg/mine/user.png" : user.avatar
          const username = (user == null || user.userName == "" || user.userName == null) ? "" : user.userName
          if (res.roles && res.roles.length > 0) {
            commit('SET_ROLES', res.roles)
            commit('SET_PERMISSIONS', res.permissions)
          } else {
            commit('SET_ROLES', ['ROLE_DEFAULT'])
          }

          console.log(res.user.userOrWorker, '>>>>>>res.user.userOrWorker');

          if (res.user) commit('SET_USERORWORKER', res.user.userOrWorker)

          console.log(state.userOrWorker, '>>>>>>');

          commit('SET_NAME', username)
          commit('SET_AVATAR', avatar)
          commit('SET_NICKNAME', res.user.nickName)
          commit('SET_USERID', res.user.userId)


          console.log("TCL: GetInfo -> wsManager1", this.wsManager)
          if (this.wsManager) {
            this.wsManager.closeConnection();
          }else{
            //获取账户时，连接soket 
            const ms = new WebSocketManager(res.user.userId);
            // 设置消息回调
            ms.onMessage(data => {
              // 处理消息逻辑
              if (data.type === 'msgUnreadCount') {
                commit('SET_MESSAGECOUNT', data.data);
              }
            });
            // 建立连接
            ms.connect();
            commit('SET_MESSAGE', ms)

          }
          

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

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

          uni.setStorageSync('userType', 1)
          // uni.setStorageSync('userId', null)
          uni.removeStorageSync('userId')
          storage.clean()
          resolve(res)
        }).catch(error => {
          reject(error)
        })
      })
    },
    handleMessageCount({ commit, state }, count) {
      commit('SET_MESSAGECOUNT', count);
    },
  }
}

export default user