fix type

Author: afc163

src/TreeSelect.tsx

@@ -81,7 +81,7 @@ export interface BaseOptionType {
 
 export interface DefaultOptionType extends BaseOptionType {
   value?: SafeKey;
-  title?: React.ReactNode;
+  title?: React.ReactNode | ((data: DefaultOptionType) => React.ReactNode);
   label?: React.ReactNode;
   key?: SafeKey;
   children?: DefaultOptionType[];
@@ -96,6 +96,7 @@ export interface TreeSelectProps<
 > extends Omit<BaseSelectPropsWithoutPrivate, 'mode'> {
   prefixCls?: string;
   id?: string;
+  children?: React.ReactNode;
 
   // >>> Value
   value?: ValueType;
@@ -352,6 +353,7 @@ const TreeSelect = React.forwardRef<BaseSelectRef, TreeSelectProps>((props, ref)
         let rawDisabled: boolean | undefined;
 
         const entity = valueEntities.get(rawValue);
+        const a = entity.node;
 
         // Fill missing label & status
         if (entity) {

src/interface.ts

@@ -13,11 +13,11 @@ export interface LabeledValueType {
   halfChecked?: boolean;
 }
 
-export type DefaultValueType = SafeKey | SafeKey[] | LabeledValueType | LabeledValueType[];
+export type DefaultValueType = SafeKey | LabeledValueType | (SafeKey | LabeledValueType)[];
 
 export interface DataNode {
   value?: SafeKey;
-  title?: React.ReactNode;
+  title?: React.ReactNode | ((data: DataNode) => React.ReactNode);
   label?: React.ReactNode;
   key?: SafeKey;
   disabled?: boolean;
@@ -32,7 +32,7 @@ export interface DataNode {
 export interface InternalDataEntity {
   key: SafeKey;
   value: SafeKey;
-  title?: React.ReactNode;
+  title?: React.ReactNode | ((data: InternalDataEntity) => React.ReactNode);
   disableCheckbox?: boolean;
   disabled?: boolean;
   children?: InternalDataEntity[];

src/utils/legacyUtil.tsx

@@ -33,7 +33,7 @@ export function convertChildrenToData(nodes: React.ReactNode): DataNode[] {
     .filter(data => data);
 }
 
-export function fillLegacyProps(dataNode: DataNode): any {
+export function fillLegacyProps(dataNode: DataNode) {
   if (!dataNode) {
     return dataNode;
   }