60 lines
1.9 KiB
TypeScript
60 lines
1.9 KiB
TypeScript
import type { RouteRecordRaw } from "vue-router"
|
||
import { pinia } from "@/pinia"
|
||
import { constantRoutes, dynamicRoutes } from "@/router"
|
||
import { routerConfig } from "@/router/config"
|
||
import { flatMultiLevelRoutes } from "@/router/helper"
|
||
|
||
function hasPermission(roles: string[], route: RouteRecordRaw) {
|
||
const routeRoles = route.meta?.roles
|
||
return routeRoles ? roles.some(role => routeRoles.includes(role)) : true
|
||
}
|
||
|
||
function filterDynamicRoutes(routes: RouteRecordRaw[], roles: string[]) {
|
||
const res: RouteRecordRaw[] = []
|
||
routes.forEach((route) => {
|
||
const tempRoute = { ...route }
|
||
if (hasPermission(roles, tempRoute)) {
|
||
if (tempRoute.children) {
|
||
tempRoute.children = filterDynamicRoutes(tempRoute.children, roles)
|
||
}
|
||
res.push(tempRoute)
|
||
}
|
||
})
|
||
return res
|
||
}
|
||
|
||
export const usePermissionStore = defineStore("permission", () => {
|
||
// 可访问的路由
|
||
const routes = ref<RouteRecordRaw[]>([])
|
||
|
||
// 有访问权限的动态路由
|
||
const addRoutes = ref<RouteRecordRaw[]>([])
|
||
|
||
// 根据角色生成可访问的 Routes（可访问的路由 = 常驻路由 + 有访问权限的动态路由）
|
||
const setRoutes = (roles: string[]) => {
|
||
const accessedRoutes = filterDynamicRoutes(dynamicRoutes, roles)
|
||
set(accessedRoutes)
|
||
}
|
||
|
||
// 所有路由 = 所有常驻路由 + 所有动态路由
|
||
const setAllRoutes = () => {
|
||
set(dynamicRoutes)
|
||
}
|
||
|
||
// 统一设置
|
||
const set = (accessedRoutes: RouteRecordRaw[]) => {
|
||
routes.value = constantRoutes.concat(accessedRoutes)
|
||
addRoutes.value = routerConfig.thirdLevelRouteCache ? flatMultiLevelRoutes(accessedRoutes) : accessedRoutes
|
||
}
|
||
|
||
return { routes, addRoutes, setRoutes, setAllRoutes }
|
||
})
|
||
|
||
/**
|
||
* @description 在 SPA 应用中可用于在 pinia 实例被激活前使用 store
|
||
* @description 在 SSR 应用中可用于在 setup 外使用 store
|
||
*/
|
||
export function usePermissionStoreOutside() {
|
||
return usePermissionStore(pinia)
|
||
}
|