TcAssetvue3/src/store/modules/permission.js

import { defineStore } from 'pinia'
import auth from '@/plugins/auth'
import router, { constantRoutes, dynamicRoutes } from '@/router'
import { getRouters } from '@/api/menu'
import Layout from '@/layout/index'
import ParentView from '@/components/ParentView'
import InnerLink from '@/layout/components/InnerLink'

// 匹配views里面所有的.vue文件
const modules = import.meta.glob('@/views/**/*.vue')

// 动态路由遍历，验证是否具备权限
const filterDynamicRoutes = (routes) => {
  const res = []
  routes.forEach(route => {
    if (route.permissions) {
      if (auth.hasPermiOr(route.permissions)) {
        res.push(route)
      }
    } else if (route.roles) {
      if (auth.hasRoleOr(route.roles)) {
        res.push(route)
      }
    }
  })
  return res
}

// 遍历后台传来的路由字符串，转换为组件对象
const filterAsyncRouter = (asyncRouterMap, lastRouter = false, type = false) => {
  return asyncRouterMap.filter(route => {
    if (type && route.children) {
      route.children = filterChildren(route.children)
    }
    if (route.component) {
      // Layout ParentView 组件特殊处理
      if (route.component === 'Layout') {
        route.component = Layout
      } else if (route.component === 'ParentView') {
        route.component = ParentView
      } else if (route.component === 'InnerLink') {
        route.component = InnerLink
      } else {
        route.component = loadView(route.component)
      }
    }
    if (route.children != null && route.children && route.children.length) {
      route.children = filterAsyncRouter(route.children, route, type)
    } else {
      delete route['children']
      delete route['redirect']
    }
    return true
  })
}

const filterChildren = (childrenMap, lastRouter = false) => {
  const children = []
  childrenMap.forEach(el => {
    el.path = lastRouter ? lastRouter.path + '/' + el.path : el.path
    if (el.children && el.children.length && el.component === 'ParentView') {
      children.push(...filterChildren(el.children, el))
    } else {
      children.push(el)
    }
  })
  return children
}

const loadView = (view) => {
  let res
  for (const path in modules) {
    const dir = path.split('views/')[1].split('.vue')[0]
    if (dir === view) {
      res = modules[path]
    }
  }
  return res
}

export const usePermissionStore = defineStore('permission', {
  state: () => ({
    routes: [],
    addRoutes: [],
    defaultRoutes: [],
    topbarRouters: [],
    sidebarRouters: []
  }),

  getters: {
    getRoutes: (state) => state.routes,
    getAddRoutes: (state) => state.addRoutes,
    getDefaultRoutes: (state) => state.defaultRoutes,
    getTopbarRouters: (state) => state.topbarRouters,
    getSidebarRouters: (state) => state.sidebarRouters
  },

  actions: {
    // 设置路由
    setRoutes(routes) {
      this.addRoutes = routes
      this.routes = constantRoutes.concat(routes)
    },

    // 设置默认路由
    setDefaultRoutes(routes) {
      this.defaultRoutes = constantRoutes.concat(routes)
    },

    // 设置顶部导航栏路由
    setTopbarRoutes(routes) {
      this.topbarRouters = routes
    },

    // 设置侧边栏路由
    setSidebarRouters(routes) {
      this.sidebarRouters = routes
    },

    // 生成路由
    async generateRoutes() {
      try {
        const res = await getRouters()
        console.log(res)
        const sdata = JSON.parse(JSON.stringify(res.data))
        const rdata = JSON.parse(JSON.stringify(res.data))
        const defaultData = JSON.parse(JSON.stringify(res.data))
        
        const sidebarRoutes = filterAsyncRouter(sdata)
        const rewriteRoutes = filterAsyncRouter(rdata, false, true)
        const defaultRoutes = filterAsyncRouter(defaultData)
        const asyncRoutes = filterDynamicRoutes(dynamicRoutes)
        
        // 添加动态路由
        asyncRoutes.forEach(route => {
          router.addRoute(route)
        })
        console.log(rewriteRoutes)
        this.setRoutes(rewriteRoutes)
        this.setSidebarRouters(constantRoutes.concat(sidebarRoutes))
        this.setDefaultRoutes(sidebarRoutes)
        this.setTopbarRoutes(defaultRoutes)

        return rewriteRoutes
      } catch (error) {
        console.error('生成路由出错:', error)
        return []
      }
    }
  }
})