cyber-narrator-cn-ui/src/store/modules/user.js

import { post } from '@/utils/http'
import router from '@/router'
import { sortByOrderNum, getWelcomeMenu } from '@/permission'
import dayjs from 'dayjs'
import utc from 'dayjs/plugin/utc'
import { ElMessage } from 'element-plus' // dependent on utc plugin
dayjs.extend(utc)

const user = {
  state () {
    return {
      menuList: [],
      buttonList: [],
      roleList: []
    }
  },
  mutations: {
    setMenuList (state, menuList) {
      state.menuList = [...menuList]
    },
    setButtonList (state, buttonList) {
      state.buttonList = [...buttonList]
    },
    setRoleList (state, roleList) {
      state.roleList = [...roleList]
    },
    clean (state) {
      state.menuList = []
      state.buttonList = []
      state.roleList = []
    }
  },
  getters: {
    menuList (state) {
      return state.menuList
    },
    buttonList (state) {
      return state.buttonList
    },
    roleList (state) {
      return state.roleList
    }
  },
  actions: {
    loginSuccess (store, res) {
      console.info(res)
      window.$dayJs.tz.setDefault(res.data.timezone)
      localStorage.setItem('cn-token', res.data.token)
      localStorage.setItem('cn-sys-name', res.data.systemName)
      if (res.systemLogo) {
        localStorage.setItem('cn-sys-logo', res.data.systemLogo)
      }
      localStorage.setItem('cn-sys-timezone', res.data.timezone)
      localStorage.setItem('cn-timezone-offset', dayjs.tz().utcOffset() / 60)
      localStorage.setItem('cn-timezone-local-offset', dayjs().utcOffset() / 60)
      post('/sys/user/permissions', { token: res.data.token }).then(res2 => {
        const menuList = sortByOrderNum(res2.data.menus)
        store.commit('setMenuList', menuList)
        store.commit('setButtonList', res2.data.buttons)
        store.commit('setRoleList', res2.data.roles)

        if (res.loginSuccessPath) {
          let tempArr = res.loginSuccessPath.split('?')
          const path = tempArr[0]
          const query = {}
          if (tempArr[1]) {
            tempArr = tempArr[1].split('&')
            tempArr.forEach(t => {
              const kv = t.split('=')
              query[kv[0]] = kv[1]
            })
          }
          router.push({
            path: path,
            query: query
          })
        } else {
          const welcomeMenu = getWelcomeMenu(menuList)
          if (welcomeMenu) {
            router.push({
              path: welcomeMenu.route,
              query: {
                t: +new Date()
              }
            })
          } else {
            ElMessage.error('No menu')
          }
        }
      })
    },
    logoutSuccess (store, res) {
      localStorage.removeItem('cn-username')
      localStorage.removeItem('cn-username')
      localStorage.removeItem('cn-token')
    }
  }
}
export default user