🎨 精简完善系统

web-admin/src/lib/tree.ts

@@ -24,3 +24,13 @@ export function collectCheckedLeafMenus(menus: MenuNode[], checkedKeys: Set<numb
     return collectCheckedLeafMenus(children, checkedKeys)
   })
 }
+
+export function collectMenusByIds(menus: MenuNode[], checkedKeys: Set<number>): MenuNode[] {
+  return menus.flatMap((menu) => {
+    const matchedChildren = collectMenusByIds(menu.children || [], checkedKeys)
+    if (checkedKeys.has(menu.ID)) {
+      return [{ ...menu, children: undefined }, ...matchedChildren]
+    }
+    return matchedChildren
+  })
+}