fix: treeCheckable should support ReactNode

react-component · Sep 27, 2019 · 90dd558 · 90dd558
1 parent 72ec97f
commit 90dd558
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/Context.tsx b/src/Context.tsx
@@ -3,7 +3,7 @@ import { IconType } from 'rc-tree/lib/interface';
 import { Key, LegacyDataNode } from './interface';
 
 interface ContextProps {
-  checkable: boolean;
+  checkable: boolean | React.ReactNode;
   checkedKeys: Key[];
   halfCheckedKeys: Key[];
   treeExpandedKeys: Key[];

diff --git a/src/TreeSelect.tsx b/src/TreeSelect.tsx
@@ -191,7 +191,7 @@ const RefTreeSelect = React.forwardRef<RefSelectProps, TreeSelectProps>((props,
     onSelect,
     onDeselect,
   } = props;
-  const mergedCheckable = !!(treeCheckable || treeCheckStrictly);
+  const mergedCheckable: React.ReactNode | boolean = treeCheckable || treeCheckStrictly;
   const mergedMultiple = multiple || mergedCheckable;
   const treeConduction = treeCheckable && !treeCheckStrictly;
   const mergedLabelInValue = treeCheckStrictly || labelInValue;