SQLBot/frontend/src/router/dynamic.ts

import LayoutDsl from '@/components/layout/LayoutDsl.vue'
import { i18n } from '@/i18n'
import { useUserStore } from '@/stores/user'
import type { Router } from 'vue-router'
const userStore = useUserStore()
const t = i18n.global.t

const dynamicRouterList = [
  {
    path: '/ds',
    component: LayoutDsl,
    name: 'ds-menu',
    redirect: '/ds/index',
    children: [
      {
        path: 'index',
        name: 'ds',
        component: import('@/views/ds/index.vue'),
        meta: { title: t('menu.Data Connections'), iconActive: 'ds', iconDeActive: 'noDs' },
      },
    ],
  },
] as any[]

const reduceRouters = (router: Router, invalid_router_name_list: string[]) => {
  const tree = router.getRoutes()
  const invalid_name_set = [] as any
  invalid_router_name_list.forEach((router_name: string) => {
    router.removeRoute(router_name)
    invalid_name_set.push(router_name)
  })

  const pathsSet = new Set(invalid_router_name_list)

  function processNode(node: any): boolean {
    if (node.name && pathsSet.has(node.name)) {
      return true
    }
    if (node.children) {
      const newChildren: any[] = []

      for (const child of node.children) {
        const shouldRemove = processNode(child)
        if (!shouldRemove) {
          newChildren.push(child)
        }
      }

      if (newChildren.length > 0) {
        node.children = newChildren
        return false
      } else {
        node.children = undefined
      }
    }

    return false
  }

  let i = 0
  while (i < tree.length) {
    if (processNode(tree[i])) {
      tree.splice(i, 1)
    } else {
      i++
    }
  }
}

export const generateDynamicRouters = (router: Router) => {
  if (userStore.isAdmin || userStore.isSpaceAdmin) {
    dynamicRouterList.forEach((item: any) => router.addRoute(item))
  } else {
    const router_name_list = [] as string[]
    const stack = [...dynamicRouterList]
    while (stack.length) {
      const item = stack.pop()
      if (item.name) {
        router_name_list.push(item.name)
      }
      if (item.children?.length) {
        item.children.forEach((child: any) => {
          stack.push(child)
        })
      }
    }
    reduceRouters(router, router_name_list)
  }
}