CN_Tool_client/frontend/src/routers/modules/dynamicRouter.ts

import router from '@/routers'
import { LOGIN_URL } from '@/config'
import { type RouteRecordRaw } from 'vue-router'
import { ElNotification } from 'element-plus'
import { useUserStore } from '@/stores/modules/user'
import { useAuthStore } from '@/stores/modules/auth'

// 引入 views 文件夹下所有 vue 文件
const modules = import.meta.glob('@/views/**/*.vue')
const STATIC_ROUTE_NAMES = new Set(['layout', 'login', 'home', 'tools', 'toolWaveform', 'toolMmsMapping', '403', '404', '500'])

let isInitializing = false

/**
 * 清除已有的动态路由
 */
const clearDynamicRoutes = () => {
    const routes = router.getRoutes()
    routes.forEach(route => {
        if (route.name && !STATIC_ROUTE_NAMES.has(String(route.name))) {
            router.removeRoute(route.name)
        }
    })
}

/**
 * 根据菜单 component 路径查找对应的页面模块。
 * 兼容两种仓库写法：
 * 1. /foo/bar.vue
 * 2. /foo/bar/index.vue
 */
const resolveComponentModule = (path: string) => {
    let normalizedPath = path.trim()
    if (!normalizedPath.startsWith('/')) {
        normalizedPath = '/' + normalizedPath
    }
    if (normalizedPath.endsWith('.vue')) {
        normalizedPath = normalizedPath.slice(0, -4)
    }
    if (normalizedPath.length > 1 && normalizedPath.endsWith('/')) {
        normalizedPath = normalizedPath.slice(0, -1)
    }

    const candidatePaths = [`/src/views${normalizedPath}.vue`, `/src/views${normalizedPath}/index.vue`]

    for (const candidatePath of candidatePaths) {
        const moduleLoader = modules[candidatePath]
        if (moduleLoader) {
            return {
                moduleLoader,
                resolvedPath: candidatePath
            }
        }
    }

    return {
        moduleLoader: undefined,
        resolvedPath: candidatePaths
    }
}

/**
 * @description 初始化动态路由
 */
export const initDynamicRouter = async () => {
    if (isInitializing) return Promise.reject(new Error('Dynamic router initialization in progress'))

    isInitializing = true
    const userStore = useUserStore()
    const authStore = useAuthStore()
    const unresolvedRoutes: Array<{ name?: string; path?: string; component?: string; candidates: string[] }> = []

    try {
        // 1. 获取菜单列表 && 按钮权限列表
        await authStore.getAuthMenuList()
        await authStore.getAuthButtonList()

        // 2. 判断当前用户有没有菜单权限
        if (!authStore.authMenuListGet.length) {
            ElNotification({
                title: '无权限访问',
                message: '当前账号无任何菜单权限，请联系系统管理员！',
                type: 'warning',
                duration: 3000
            })
            userStore.setAccessToken('')
            userStore.setRefreshToken('')
            userStore.setExp(0)
            await router.replace(LOGIN_URL)
            return Promise.reject('No permission')
        }

        // 3. 清理之前的动态路由
        clearDynamicRoutes()

        // 4. 添加动态路由
        for (const item of authStore.flatMenuListGet) {
            // 删除 children 避免冗余嵌套
            if (item.children) delete item.children

            // 处理组件映射
            if (item.component && typeof item.component === 'string') {
                const { moduleLoader, resolvedPath } = resolveComponentModule(item.component)
                if (moduleLoader) {
                    item.component = moduleLoader
                } else {
                    // 动态路由组件一旦解析失败，对应菜单会落入 404，这里必须打印清楚候选路径。
                    unresolvedRoutes.push({
                        name: item.name,
                        path: item.path,
                        component: item.component,
                        candidates: resolvedPath
                    })
                    continue
                }
            }

            // 类型守卫：确保满足 RouteRecordRaw 接口要求
            if (
                typeof item.path === 'string' &&
                (typeof item.component === 'function' || typeof item.redirect === 'string')
            ) {
                const routeItem = item as unknown as RouteRecordRaw
                if (item.meta?.isFull) {
                    router.addRoute(routeItem)
                } else {
                    router.addRoute('layout', routeItem)
                }
            } else {
                console.warn('Invalid route item:', item)
            }
        }

        if (unresolvedRoutes.length) {
            console.error('[dynamic-router] unresolved route components', unresolvedRoutes)
        }
    } catch (error) {
        // 当按钮 || 菜单请求出错时，重定向到登陆页
        userStore.setAccessToken('')
        userStore.setRefreshToken('')
        userStore.setExp(0)
        await router.replace(LOGIN_URL)
        return Promise.reject(error)
    } finally {
        isInitializing = false
    }
}