6b1aab9c67
When children is undefined, use empty array as fallback to prevent potential runtime errors. This matches the behavior when hideChildrenInMenu is true.
91 lines
2.2 KiB
TypeScript
91 lines
2.2 KiB
TypeScript
import type { Router, RouteRecordRaw } from 'vue-router';
|
||
|
||
import type {
|
||
ExRouteRecordRaw,
|
||
MenuRecordRaw,
|
||
RouteMeta,
|
||
} from '@vben-core/typings';
|
||
|
||
import { filterTree, mapTree } from '@vben-core/shared/utils';
|
||
|
||
/**
|
||
* 根据 routes 生成菜单列表
|
||
* @param routes - 路由配置列表
|
||
* @param router - Vue Router 实例
|
||
* @returns 生成的菜单列表
|
||
*/
|
||
function generateMenus(
|
||
routes: RouteRecordRaw[],
|
||
router: Router,
|
||
): MenuRecordRaw[] {
|
||
// 将路由列表转换为一个以 name 为键的对象映射
|
||
const finalRoutesMap: { [key: string]: string } = Object.fromEntries(
|
||
router.getRoutes().map(({ name, path }) => [name, path]),
|
||
);
|
||
|
||
let menus = mapTree<ExRouteRecordRaw, MenuRecordRaw>(routes, (route) => {
|
||
// 获取最终的路由路径
|
||
const path = finalRoutesMap[route.name as string] ?? route.path ?? '';
|
||
|
||
const {
|
||
meta = {} as RouteMeta,
|
||
name: routeName,
|
||
redirect,
|
||
children = [],
|
||
} = route;
|
||
const {
|
||
activeIcon,
|
||
badge,
|
||
badgeType,
|
||
badgeVariants,
|
||
hideChildrenInMenu = false,
|
||
icon,
|
||
link,
|
||
order,
|
||
title = '',
|
||
} = meta;
|
||
|
||
// 确保菜单名称不为空
|
||
const name = (title || routeName || '') as string;
|
||
|
||
// 处理子菜单
|
||
const resultChildren = hideChildrenInMenu
|
||
? []
|
||
: ((children as MenuRecordRaw[]) ?? []);
|
||
|
||
// 设置子菜单的父子关系
|
||
if (resultChildren.length > 0) {
|
||
resultChildren.forEach((child) => {
|
||
child.parents = [...(route.parents ?? []), path];
|
||
child.parent = path;
|
||
});
|
||
}
|
||
|
||
// 确定最终路径
|
||
const resultPath = hideChildrenInMenu ? redirect || path : link || path;
|
||
|
||
return {
|
||
activeIcon,
|
||
badge,
|
||
badgeType,
|
||
badgeVariants,
|
||
icon,
|
||
name,
|
||
order,
|
||
parent: route.parent,
|
||
parents: route.parents,
|
||
path: resultPath,
|
||
show: !meta.hideInMenu,
|
||
children: resultChildren,
|
||
};
|
||
});
|
||
|
||
// 对菜单进行排序，避免order=0时被替换成999的问题
|
||
menus = menus.sort((a, b) => (a?.order ?? 999) - (b?.order ?? 999));
|
||
|
||
// 过滤掉隐藏的菜单项
|
||
return filterTree(menus, (menu) => !!menu.show);
|
||
}
|
||
|
||
export { generateMenus };
|