@fe-free/core 2.2.0 → 2.2.1

package/CHANGELOG.md

@@ -1,5 +1,12 @@
 # @fe-free/core
 
+## 2.2.1
+
+### Patch Changes
+
+- tree
+  - @fe-free/tool@2.2.1
+
 ## 2.2.0
 
 ### Minor Changes

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fe-free/core",
-  "version": "2.2.0",
+  "version": "2.2.1",
   "description": "",
   "main": "./src/index.ts",
   "author": "",
@@ -39,7 +39,7 @@
     "remark-gfm": "^4.0.1",
     "vanilla-jsoneditor": "^0.23.1",
     "zustand": "^4.5.4",
-    "@fe-free/tool": "2.2.0"
+    "@fe-free/tool": "2.2.1"
   },
   "peerDependencies": {
     "@ant-design/pro-components": "^2.8.7",

package/src/index.ts

@@ -32,6 +32,7 @@ export { Markdown } from './markdown';
 export { PageLayout } from './page_layout';
 export { Table } from './table';
 export type { TableProps } from './table';
-export { Tree } from './tree';
+export { Tree, flatToTreeData } from './tree';
+export type { TreeProps } from './tree';
 export { useLocalforageState } from './use_localforage_state';
 export { CustomValueTypeEnum, customValueTypeMap } from './value_type_map';

package/src/tree/helper.tsx

@@ -0,0 +1,53 @@
+import type { DataNode } from 'antd/es/tree';
+
+interface TreeData<T extends Record<string, any>> extends DataNode {
+  originData: T;
+}
+
+function flatToTreeData<T extends Record<string, any>>({
+  data,
+  titleField,
+  keyField,
+  parentField,
+  rootValue,
+}: {
+  data: T[];
+  titleField: keyof T;
+  keyField: keyof T;
+  parentField: keyof T;
+  rootValue: T[keyof T];
+}): {
+  treeData: TreeData<T>[];
+  treeMap: Record<string, TreeData<T>>;
+} {
+  const treeMap: Record<string, TreeData<T>> = {};
+  const treeData: TreeData<T>[] = [];
+
+  // 构造 map
+  data.forEach((item) => {
+    treeMap[item[keyField]] = {
+      title: item[titleField],
+      // 保证 string
+      key: item[keyField] + '',
+      children: [],
+      originData: item,
+    };
+  });
+
+  // 构造父子关系
+  data.forEach((item) => {
+    // root 节点
+    if (item[parentField] === rootValue) {
+      treeData.push(treeMap[item[keyField]]);
+    }
+
+    treeMap[item[parentField]]?.children?.push(treeMap[item[keyField]]);
+  });
+
+  return {
+    treeData,
+    treeMap,
+  };
+}
+
+export { flatToTreeData };

package/src/tree/index.tsx

@@ -4,6 +4,7 @@ import type { TreeProps as AntdTreeProps } from 'antd';
 import { Tree as AntdTree, Input } from 'antd';
 import type { DataNode } from 'antd/es/tree';
 import { useMemo, useState } from 'react';
+import { flatToTreeData } from './helper';
 
 interface TreeProps<T extends DataNode> extends AntdTreeProps<T> {
   enableSearch?: boolean;
@@ -127,4 +128,5 @@ function Tree<T extends DataNode>(props: TreeProps<T>) {
   );
 }
 
-export { Tree };
+export { flatToTreeData, Tree };
+export type { TreeProps };