feat: add disabledStrategy

Author: aojunhao123

src/OptionList.tsx

@@ -8,7 +8,7 @@ import useMemo from 'rc-util/lib/hooks/useMemo';
 import * as React from 'react';
 import LegacyContext from './LegacyContext';
 import TreeSelectContext from './TreeSelectContext';
-import type { Key, SafeKey } from './interface';
+import type { DataNode, Key, SafeKey } from './interface';
 import { getAllKeys, isCheckDisabled } from './utils/valueUtil';
 
 const HIDDEN_STYLE = {
@@ -156,6 +156,21 @@ const OptionList: React.ForwardRefRenderFunction<ReviseRefOptionListProps> = (_,
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }, [searchValue]);
 
+  const disabledStrategy = (node: DataNode) => {
+    if (node.disabled) {
+      return true;
+    }
+
+    if (isOverMaxCount) {
+      const selectedValues = displayValues?.map(v => v.value) || [];
+      if (!selectedValues.includes(node[fieldNames.value])) {
+        return true;
+      }
+    }
+
+    return undefined;
+  };
+
   // ========================== Get First Selectable Node ==========================
   const getFirstMatchingNode = (nodes: EventDataNode<any>[]): EventDataNode<any> | null => {
     for (const node of nodes) {
@@ -385,6 +400,7 @@ const OptionList: React.ForwardRefRenderFunction<ReviseRefOptionListProps> = (_,
         filterTreeNode={filterTreeNode}
         expandAction={treeExpandAction}
         onScroll={onPopupScroll}
+        disabledStrategy={disabledStrategy}
       />
     </div>
   );

src/TreeSelect.tsx

@@ -610,30 +610,14 @@ const TreeSelect = React.forwardRef<BaseSelectRef, TreeSelectProps>((props, ref)
   const isOverMaxCount =
     mergedMultiple && maxCount !== undefined && cachedDisplayValues?.length >= maxCount;
 
-  const traverse = (nodes: DataNode[]): DataNode[] => {
-    return nodes.map(node => ({
-      ...node,
-      disabled:
-        isOverMaxCount && !cachedDisplayValues?.some(v => v.value === node.value)
-          ? true
-          : node.disabled,
-      children: node.children ? traverse(node.children) : undefined,
-    }));
-  };
-
-  const processedTreeData = React.useMemo(
-    () => traverse(filteredTreeData),
-    [filteredTreeData, isOverMaxCount],
-  );
-
   const treeSelectContext = React.useMemo<TreeSelectContextProps>(() => {
     return {
       virtual,
       dropdownMatchSelectWidth,
       listHeight,
       listItemHeight,
       listItemScrollOffset,
-      treeData: processedTreeData,
+      treeData: filteredTreeData,
       fieldNames: mergedFieldNames,
       onSelect: onOptionSelect,
       treeExpandAction,
@@ -648,7 +632,7 @@ const TreeSelect = React.forwardRef<BaseSelectRef, TreeSelectProps>((props, ref)
     listHeight,
     listItemHeight,
     listItemScrollOffset,
-    processedTreeData,
+    filteredTreeData,
     mergedFieldNames,
     onOptionSelect,
     treeExpandAction,