mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-15 17:19:11 +08:00
44 lines
940 B
TypeScript
44 lines
940 B
TypeScript
|
import { useMemo } from 'react';
|
||
|
import type { BreadcrumbItemType, BreadcrumbSeparatorType, ItemType, Route } from './Breadcrumb';
|
||
|
|
||
|
type MergedType = BreadcrumbItemType & {
|
||
|
children?: Route['children'];
|
||
|
};
|
||
|
|
||
|
function route2item(route: Route): MergedType {
|
||
|
const { breadcrumbName, children, ...rest } = route;
|
||
|
|
||
|
const clone: MergedType = {
|
||
|
title: breadcrumbName,
|
||
|
...rest,
|
||
|
};
|
||
|
|
||
|
if (children) {
|
||
|
clone.menu = {
|
||
|
items: children.map(({ breadcrumbName: itemBreadcrumbName, ...itemProps }) => ({
|
||
|
...itemProps,
|
||
|
title: itemBreadcrumbName,
|
||
|
})),
|
||
|
};
|
||
|
}
|
||
|
|
||
|
return clone;
|
||
|
}
|
||
|
|
||
|
export default function useItems(
|
||
|
items?: ItemType[],
|
||
|
routes?: Route[],
|
||
|
): Partial<MergedType & BreadcrumbSeparatorType>[] | null {
|
||
|
return useMemo<ItemType[] | null>(() => {
|
||
|
if (items) {
|
||
|
return items;
|
||
|
}
|
||
|
|
||
|
if (routes) {
|
||
|
return routes.map(route2item);
|
||
|
}
|
||
|
|
||
|
return null;
|
||
|
}, [items, routes]);
|
||
|
}
|