import {
  login,
  logout,
  getMenu,
  getPermissionTreeAll
} from '@/api/user'
import {
  getToken,
  setToken,
  removeToken,
  getUserInfo,
  setUserInfo,
  removeUserInfo,
  setLocalMenu,
  getLocalMenu,
  removeLocalMenu
} from '@/utils/auth'
import {
  resetRouter
} from '@/router'

const getDefaultState = () => {
  return {
    token: getToken(),
    userInfo: getUserInfo(),
    menu: getLocalMenu(),
    name: '',
    avatar: '',
    permissionTree: []
  }
}

const state = getDefaultState()

const mutations = {
  RESET_STATE: (state) => {
    Object.assign(state, getDefaultState())
  },
  SET_TOKEN: (state, token) => {
    state.token = token
  },
  SET_NAME: (state, name) => {
    state.name = name
  },
  SET_AVATAR: (state, avatar) => {
    state.avatar = avatar
  },
  SET_USERINFO: (state, userInfo) => {
    state.userInfo = userInfo
  },
  SET_MENU: (state, menu) => {
    state.menu = menu
  },
  SET_PERMISSION_TREE_ALL: (state, permissionTree) => {
    state.permissionTree = permissionTree
  },
}

const actions = {
  // user login
  login({
    commit
  }, userInfo) {
    const {
      username,
      password,
      captcha
    } = userInfo
    return new Promise((resolve, reject) => {
      login({
        username: username.trim(),
        password: password,
        captcha: captcha
      }).then(response => {
        const {
          data
        } = response
        commit('SET_TOKEN', data.accessToken)
        commit('SET_NAME', data.username)
        commit('SET_USERINFO', data)
        setToken(data.accessToken)
        setUserInfo(data)
        resolve()
      }).catch(error => {
        reject(error)
      })
    })
  },

  // user logout
  logout({
    commit,
    state
  }) {
    return new Promise((resolve, reject) => {
      logout(state.token).then(() => {
        removeToken() // must remove  token  first
        removeUserInfo()
        removeLocalMenu()
        resetRouter()
        commit('RESET_STATE')
        window.location.reload() //刷新路由表
        resolve()
      }).catch(error => {
        console.log('err', error);
        // reject(error)
      })
    })
  },

  // remove token
  resetToken({
    commit
  }) {
    return new Promise(resolve => {
      removeToken() // must remove  token  first
      commit('RESET_STATE')
      resolve()
    })
  },

  getMenuList({
    commit
  }) {
    console.log('getMenuList');
    return new Promise((resolve, reject) => {
      getMenu().then((res) => {
        const {
          menus
        } = res.data
        let menuObj = getMenuObj(menus)

        function getMenuObj(menu) {
          let obj = {}
          traverseMenuList(menu || []);

          function traverseMenuList(arr) {
            if (arr.length > 0) {
              arr.map((item) => {
                if (item.menuName) {
                  obj[item.menuName] = true;
                }
                if (item.children) {
                  traverseMenuList(item.children);
                }
              });
            }
          }
          return obj
        }

        commit('SET_MENU', menuObj)
        setLocalMenu(menuObj)
        resolve()
      }).catch(error => {
        reject(error)
      })
    })
  },

  // 获取权限树所有
  getPermissionTreeAll({
    commit
  }) {
    return new Promise(resolve => {
      getPermissionTreeAll().then(res => {
        commit('SET_PERMISSION_TREE_ALL', res.data)
        resolve(res.data)
      }).catch(err => {
        reject(err)
      })
    })
  },


}

export default {
  namespaced: true,
  state,
  mutations,
  actions
}
