index.ts 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. import { useUserStore } from '/admin/stores/modules/user'
  2. import { getAuthToken, removeAuthToken, setPageTitle } from '/admin/support/helper'
  3. import progress from '/admin/support/progress'
  4. import { WhiteListPage } from '/admin/enum/app'
  5. import { Router, RouteRecordRaw } from 'vue-router'
  6. import { usePermissionsStore } from '/admin/stores/modules/user/permissions'
  7. import { Menu } from '/admin/types/Menu'
  8. import { toRaw } from 'vue'
  9. const guard = (router: Router) => {
  10. // white list
  11. const whiteList: string[] = [WhiteListPage.LOGIN_PATH, WhiteListPage.NOT_FOUND_PATH]
  12. router.beforeEach(async (to, from, next) => {
  13. // set page title
  14. setPageTitle(to.meta.title as unknown as string)
  15. // page start
  16. progress.start()
  17. // 获取用户的 token
  18. const authToken = getAuthToken()
  19. // 如果 token 存在
  20. if (authToken) {
  21. // 如果进入 /login 页面,重定向到首页
  22. if (to.path === WhiteListPage.LOGIN_PATH) {
  23. next({ path: '/' })
  24. } else {
  25. const userStore = useUserStore()
  26. // 获取用户ID
  27. if (userStore.getId) {
  28. next()
  29. } else {
  30. try {
  31. // 阻塞获取用户信息
  32. await userStore.getUserInfo()
  33. // 如果后端没有返回 permissions,前台则只使用静态路由
  34. if (userStore.getPermissions !== undefined) {
  35. // 挂载路由(实际是从后端获取用户的权限)
  36. const permissionStore = usePermissionsStore()
  37. // 动态路由挂载
  38. const asyncRoutes = permissionStore.getAsyncMenusFrom(toRaw(userStore.getPermissions))
  39. asyncRoutes.forEach((route: Menu) => {
  40. router.addRoute(route as unknown as RouteRecordRaw)
  41. })
  42. // 在动态路由之后挂载匹配 404 路由
  43. router.addRoute({
  44. path: '/:pathMatch(.*)*',
  45. redirect: '/404',
  46. })
  47. }
  48. next({ ...to, replace: true })
  49. } catch (e) {
  50. removeAuthToken()
  51. next({ path: `${WhiteListPage.LOGIN_PATH}?redirect=/${to.path}` })
  52. }
  53. }
  54. }
  55. progress.done()
  56. } else {
  57. // 如果不在白名单
  58. if (whiteList.indexOf(to.path) !== -1) {
  59. next()
  60. } else {
  61. next({ path: WhiteListPage.LOGIN_PATH })
  62. }
  63. progress.done()
  64. }
  65. })
  66. router.afterEach(() => {
  67. progress.done()
  68. })
  69. }
  70. export default guard