rt-sim-training-client/src/store/modules/user.js

import { setSessionStorage } from '@/utils/auth';
import { login, logout, getInfo } from '@/api/login';
import { getToken, setToken, removeToken } from '@/utils/auth';
import { getUserConfigInfo } from '@/api/management/user';
import { LoginParams } from '@/utils/login';
import { creatSubscribe, clearSubscribe, perpetualTopic, commonTopic } from '@/utils/stomp';
import Cookies from 'js-cookie';

const user = {
    state: {
        token: getToken(),
        name: '',
        nickname: '',
        nationcode: '',
        roles: [],
        id: '',
        admin: false,
        wxId: '',
        wxUnionId: '',
        account: ''
    },

    mutations: {
        SET_TOKEN: (state, token) => {
            state.token = token;
        },
        SET_NAME: (state, name) => {
            state.name = name;
        },
        SET_NICKNAME: (state, nickname) => {
            state.nickname = nickname;
        },
        SET_ROLES: (state, roles) => {
            state.roles = roles;
        },
        SET_ID: (state, id) => {
            state.id = id;
        },
        SET_NATIONCODE: (state, nationcode) => {
            state.nationcode = nationcode;
        },
        SET_WXID: (state, wxId) => {
            state.wxId = wxId;
        },
        SET_WXUNIONID: (state, wxUnionId) => {
            state.wxUnionId = wxUnionId;
        },
        SET_ACCOUNT: (state, account) => {
            state.account = account;
        },
        CLEAR_AUTO_PARAMS: () => {
            Cookies.remove('UserDesignName');
            Cookies.remove('UserDesignToken');
            Cookies.remove('UserName');
            Cookies.remove('UserToken');
        }
    },

    actions: {
        // 登录
        Login({ commit }, userInfo) {
            const username = userInfo.username.trim();
            const password = userInfo.password.trim();
            return new Promise((resolve, reject) => {
                let params = null;
                let execFunc = null;
                if (userInfo.type == 'design') {
                    params = Object.assign({ account: username, password }, LoginParams.Design);
                    execFunc = (header) => {
                        creatSubscribe(commonTopic, header);
                    };
                } else {
                    params = Object.assign({ account: username, password }, LoginParams.LianKeTang);
                    execFunc = (header) => {
                        creatSubscribe(perpetualTopic, header);
                        creatSubscribe(commonTopic, header);
                    };
                }

                // 登录系统
                login(params).then(resp => {
                    const token = resp.data;
                    commit('SET_TOKEN', token);
                    setToken(token);
                    execFunc({ group: '', 'X-Token': token });
                    resolve();
                }).catch(error => { reject(error); });
            });
        },

        // 扫码登陆设置
        QrLoginSetting({ dispatch, commit }, token) {
            return new Promise((resolve, reject) => {
                // 设置user域token值
                commit(token.key, token.value);
                const header = { group: '', 'X-Token': token.value };
                creatSubscribe(perpetualTopic, header);
                creatSubscribe(commonTopic, header);
                // 获取用户参数
                dispatch('GetUserConfigInfo').then(response => {
                    resolve(response);
                }).catch(error => {
                    reject(error);
                });
            });
        },

        // 获取用户信息
        GetInfo({ commit }, token) {
            return new Promise((resolve, reject) => {
                getInfo(token).then(response => {
                    const user = response.data;
                    if (user.roles && user.roles.length > 0) { // 验证返回的roles是否是一个非空数组
                        commit('SET_ROLES', user.roles);
                    } else {
                        reject('getInfo: roles must be a non-null array !');
                    }
                    commit('SET_NAME', user.name);
                    commit('SET_NICKNAME', user.nickname);
                    commit('SET_ID', user.id);
                    commit('SET_NATIONCODE', user.nationcode);
                    commit('SET_WXID', user.wxId);
                    commit('SET_WXUNIONID', user.wxUnionId);
                    resolve(user);
                }).catch(error => {
                    reject(error);
                });
            });
        },

        // 获取用户参数
        GetUserConfigInfo() {
            getUserConfigInfo().then(resp => {
                if (resp.data) {
                    resp.data.forEach(elem => {
                        setSessionStorage(elem.code, elem.val);
                    });
                }
            });
        },
        // 前端登出
        FedLogOut({ commit }) {
            clearSubscribe(perpetualTopic);
            clearSubscribe(commonTopic);
            commit('SET_TOKEN', '');
            commit('SET_ID', '');
            removeToken();
        },

        // 登出系统
        LogOut({ commit, dispatch }) {
            // 清楚自动登陆参数
            commit('CLEAR_AUTO_PARAMS');
            const token = getToken();
            return new Promise((resolve, reject) => {
                dispatch('FedLogOut').then(resp => {
                    logout(token).then(() => {
                        resolve();
                    }).catch(error => {
                        reject(error);
                    });
                });
            });
        },

        SetAccount({ commit }, account) {
            commit('SET_ACCOUNT', account);
        }
    }
};

export default user;