fix: fix

react-component · Dec 30, 2023 · 7c4f99c · 7c4f99c
1 parent faa4b41
commit 7c4f99c
Showing 1 changed file with 15 additions and 17 deletions.
diff --git a/src/hooks/useMaxCount.ts b/src/hooks/useMaxCount.ts
@@ -3,23 +3,21 @@ import SelectContext from '../SelectContext';
 import type { SelectContextProps } from '../SelectContext';
 
 const useMaxCount = (multiple: boolean) => {
-  const { maxCount, rawValues } = React.useContext<SelectContextProps>(SelectContext);
-  return {
-    shouldTruncate: React.useCallback(
-      (overCount = true) => {
-        if (!multiple || typeof maxCount === 'undefined') {
-          return false;
-        }
-        const overMaxCount = rawValues?.size >= maxCount;
-        return overCount ? overMaxCount : !overMaxCount;
-      },
-      [multiple, maxCount, rawValues?.size],
-    ),
-    truncateLength: React.useMemo<number>(
-      () => maxCount - rawValues?.size,
-      [maxCount, rawValues?.size],
-    ),
-  };
+  const { maxCount, rawValues } = React.useContext<SelectContextProps>(SelectContext) || {};
+  const truncateLength = React.useMemo<number>(
+    () => maxCount - rawValues?.size,
+    [maxCount, rawValues?.size],
+  );
+  const shouldTruncate = React.useCallback(
+    (overCount = true) => {
+      if (!multiple || typeof maxCount === 'undefined') {
+        return false;
+      }
+      return overCount ? truncateLength < 0 : truncateLength >= 0;
+    },
+    [multiple, maxCount, truncateLength],
+  );
+  return { shouldTruncate, truncateLength };
 };
 
 export default useMaxCount;