fix(tree): fix tree export treeItem (#2157)

Co-authored-by: vermilionAnd <zhujiahong@growingio.com>

src/index.ts

@@ -43,7 +43,7 @@ export { default as Tag, TagProps } from './tag';
 export { default as Toast, ToastConfigOptions, ToastType } from './toast';
 export { default as Toggle, ToggleProps } from './toggle';
 export { default as Tooltip, TooltipProps } from './tooltip';
-export { default as Tree, TreeProps } from './tree';
+export { default as Tree, TreeItem, TreeProps, TreeItemProps } from './tree';
 export { default as Typography } from './typography';
 export { default as Upload, UploadProps, UploadType, UploadStatus } from './upload';
 export { default as Select, SelectProps, OptionProps } from './select';

src/tree/Tree.tsx

@@ -166,7 +166,9 @@ export const InnerTree = WithRef<HTMLDivElement, TreeProps>((props, ref?) => {
   );
 });
 
-const Tree: React.ForwardRefExoticComponent<TreeProps & React.RefAttributes<HTMLDivElement>> & { isTree?: boolean } =
-  InnerTree;
+const Tree: React.ForwardRefExoticComponent<TreeProps & React.RefAttributes<HTMLDivElement>> & {
+  isTree?: boolean;
+  TreeItem?: React.FC<TreeItemProps>;
+} = InnerTree;
 Tree.isTree = true;
 export default Tree;

src/tree/demos/Tree.stories.tsx

@@ -1,7 +1,7 @@
 import React, { useState } from 'react';
 import { Story, Meta } from '@storybook/react/types-6-0';
 import { CheckOutlined, CopyOutlined, EventOutlined } from '@gio-design/icons';
-import { Tree, TreeProps, TreeNode } from '../index';
+import { Tree, TreeProps, TreeItem as TreeNode } from '../index';
 import Docs from './TreePage';
 
 export default {
@@ -209,12 +209,12 @@ export const Options = () => (
 
 export const FilterTreeNode = () => (
   <Tree filterTreeNode={(e) => e.value === '0-0'}>
-    <TreeNode value="0-0" label="产品团队" prefix={<EventOutlined />}>
+    <Tree.TreeItem value="0-0" label="产品团队" prefix={<EventOutlined />}>
       <TreeNode value="0-0-0" label="产品经理团队" prefix={<EventOutlined />} disabled>
         <TreeNode value="0-0-0-0" label="产品一组" prefix={<CheckOutlined />} disabled />
         <TreeNode value="0-0-0-1" label="产品二组" prefix={<CheckOutlined />} />
       </TreeNode>
       <TreeNode value="0-0-1" label="设计师团队" prefix={<EventOutlined />} />
-    </TreeNode>
+    </Tree.TreeItem>
   </Tree>
 );

src/tree/index.ts

@@ -1,7 +1,9 @@
-import { TreeProps, TreeItem as TreeNodeProps } from './interface';
+import WithSubComponent from '../utils/withSubComponent';
+import { TreeProps, TreeItemProps } from './interface';
 import Tree from './Tree';
-import TreeNode from './TreeItem';
+import TreeItem from './TreeItem';
 
-export type { TreeProps, TreeNodeProps };
-export { Tree, TreeNode };
-export default Tree;
+export type { TreeProps, TreeItemProps };
+export { Tree, TreeItem };
+
+export default WithSubComponent(Tree, { TreeItem });