feat(menu): 支持配置字段别名

packages/ui/menu/src/Menu.tsx

@@ -21,7 +21,7 @@ import { uuid } from '@hi-ui/use-id'
 import { MenuDataItem, MenuFooterRenderProps } from './types'
 import { MenuItem } from './MenuItem'
 import MenuContext from './context'
-import { getAncestorIds } from './util'
+import { getAncestorIds, transformTreeData } from './util'
 
 const MENU_PREFIX = getPrefixCls('menu')
 
@@ -41,6 +41,7 @@ export const Menu = forwardRef<HTMLDivElement | null, MenuProps>(
       role = 'menu',
       className,
       data = NOOP_ARRAY,
+      fieldNames,
       placement = 'vertical',
       showCollapse = false,
       // 仅对垂直模式有效
@@ -70,6 +71,10 @@ export const Menu = forwardRef<HTMLDivElement | null, MenuProps>(
 
     const [activeParents, updateActiveParents] = useState(() => getAncestorIds(activeId, data))
 
+    data = useMemo(() => {
+      return transformTreeData(data, fieldNames)
+    }, [data, fieldNames])
+
     useEffect(() => {
       updateActiveParents(getAncestorIds(activeId, data))
     }, [activeId, data])
@@ -303,6 +308,10 @@ export interface MenuProps extends Omit<HiBaseHTMLProps<'div'>, 'onClick'> {
    * 菜单项数据列表
    */
   data: MenuDataItem[]
+  /**
+   * 设置 data 中 id, title, disabled, children 对应的 key
+   */
+  fieldNames?: Record<string, string>
   /**
    * 默认激活的菜单项 id
    */

packages/ui/menu/src/util.ts

@@ -55,3 +55,36 @@ export const filterTreeData = (
     }) ?? []
   )
 }
+
+export const transformTreeData = (
+  data: MenuDataItem[],
+  fieldNames: Record<string, string> | undefined
+) => {
+  /**
+   * 转换对象
+   */
+  const getKeyFields = (node: any, key: any) => {
+    if (fieldNames) {
+      return node[(fieldNames as any)[key] || key]
+    }
+    return node[key]
+  }
+
+  /**
+   * 递归处理树形数组
+   */
+  const traverseNode = (node: MenuDataItem): MenuDataItem => {
+    const newNode: MenuDataItem = { ...node }
+    newNode.id = getKeyFields(newNode, 'id')
+    newNode.title = getKeyFields(newNode, 'title')
+    newNode.icon = getKeyFields(newNode, 'icon')
+    newNode.disabled = getKeyFields(newNode, 'disabled') ?? false
+    newNode.children = getKeyFields(newNode, 'children')
+    if (newNode.children) {
+      newNode.children = newNode.children.map(traverseNode)
+    }
+    return newNode
+  }
+
+  return data.map(traverseNode)
+}