rsuite 5.60.2 → 5.61.0

package/cjs/{CheckTreePicker → CheckTree}/utils.js

@@ -5,56 +5,89 @@ var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefau
 exports.__esModule = true;
 exports.isEveryChildChecked = isEveryChildChecked;
 exports.isSomeChildChecked = isSomeChildChecked;
-exports.isSomeNodeHasChildren = isSomeNodeHasChildren;
+exports.hasGrandchild = hasGrandchild;
 exports.isAllSiblingNodeUncheckable = isAllSiblingNodeUncheckable;
 exports.isEveryFirstLevelNodeUncheckable = isEveryFirstLevelNodeUncheckable;
 exports.isNodeUncheckable = isNodeUncheckable;
 exports.getFormattedTree = getFormattedTree;
 exports.getDisabledState = getDisabledState;
-exports.getCheckTreePickerDefaultValue = getCheckTreePickerDefaultValue;
+exports.getCheckTreeDefaultValue = getCheckTreeDefaultValue;
 exports.getSelectedItems = getSelectedItems;
 exports.getNodeCheckState = getNodeCheckState;
 var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
 var _isUndefined2 = _interopRequireDefault(require("lodash/isUndefined"));
 var _isNil2 = _interopRequireDefault(require("lodash/isNil"));
 var _utils = require("../utils");
-var _treeUtils = require("../utils/treeUtils");
+var _utils2 = require("../Tree/utils");
 var _attachParent = require("../utils/attachParent");
-function isEveryChildChecked(nodes, parent) {
+/**
+ * Retrieves the children of a given parent node from a flattened node map.
+ */
+function getChildrenByFlattenNodes(nodes, parent) {
+  if (!(0, _isNil2.default)(parent.refKey) && (0, _isNil2.default)(nodes[parent.refKey])) {
+    return [];
+  }
+  return Object.values(nodes).filter(function (item) {
+    var _item$parent;
+    return (item === null || item === void 0 ? void 0 : (_item$parent = item.parent) === null || _item$parent === void 0 ? void 0 : _item$parent.refKey) === parent.refKey && item.refKey && !nodes[item.refKey].uncheckable;
+  });
+}
+
+/**
+ * Checks if every child of a given parent node is checked.
+ */
+function isEveryChildChecked(parent, options) {
+  var nodes = options.nodes,
+    childrenKey = options.childrenKey;
   if ((0, _isNil2.default)(parent.refKey) || (0, _isNil2.default)(nodes[parent.refKey])) {
     return false;
   }
-  var children = (0, _treeUtils.getChildrenByFlattenNodes)(nodes, parent);
+  var children = getChildrenByFlattenNodes(nodes, parent);
   if (!children.length) {
     var _nodes$parent$refKey$;
     return (_nodes$parent$refKey$ = nodes[parent.refKey].check) !== null && _nodes$parent$refKey$ !== void 0 ? _nodes$parent$refKey$ : false;
   }
   return children.every(function (child) {
+    var _child$childrenKey;
+    if ((child === null || child === void 0 ? void 0 : (_child$childrenKey = child[childrenKey]) === null || _child$childrenKey === void 0 ? void 0 : _child$childrenKey.length) > 0) {
+      // fix: #3559
+      return isEveryChildChecked(child, {
+        nodes: nodes,
+        childrenKey: childrenKey
+      });
+    }
     return !(0, _isNil2.default)(child.refKey) && nodes[child.refKey].check;
   });
 }
+
+/**
+ * Checks if any child node is checked.
+ */
 function isSomeChildChecked(nodes, parent, childrenKey) {
   if (!(0, _isNil2.default)(parent.refKey) && (0, _isNil2.default)(nodes[parent.refKey])) {
     return false;
   }
-  var children = (0, _treeUtils.getChildrenByFlattenNodes)(nodes, parent);
+  var children = getChildrenByFlattenNodes(nodes, parent);
   return children.some(function (child) {
-    var _child$childrenKey;
-    if ((child === null || child === void 0 ? void 0 : (_child$childrenKey = child[childrenKey]) === null || _child$childrenKey === void 0 ? void 0 : _child$childrenKey.length) > 0) {
+    var _child$childrenKey2;
+    if ((child === null || child === void 0 ? void 0 : (_child$childrenKey2 = child[childrenKey]) === null || _child$childrenKey2 === void 0 ? void 0 : _child$childrenKey2.length) > 0) {
       return isSomeChildChecked(nodes, child, childrenKey);
     }
     return !(0, _isNil2.default)(child.refKey) && nodes[child.refKey].check;
   });
 }
-function isSomeNodeHasChildren(data, childrenKey) {
+
+/**
+ * Checks if any node in the data has a grandchild.
+ */
+function hasGrandchild(data, childrenKey) {
   return data.some(function (node) {
     return Array.isArray(node[childrenKey]);
   });
 }
 
 /**
- * is all siblings nodes is uncheckable
- * @param {*} node
+ * Checks if all sibling nodes of a given node are uncheckable.
  */
 function isAllSiblingNodeUncheckable(node, nodes, uncheckableItemValues, valueKey) {
   var list = [];
@@ -77,7 +110,7 @@ function isAllSiblingNodeUncheckable(node, nodes, uncheckableItemValues, valueKe
 }
 
 /**
- * get each first level node uncheckable state
+ * Checks if every first-level node is uncheckable based on the provided criteria.
  */
 function isEveryFirstLevelNodeUncheckable(nodes, uncheckableItemValues, valueKey) {
   var list = [];
@@ -96,8 +129,7 @@ function isEveryFirstLevelNodeUncheckable(nodes, uncheckableItemValues, valueKey
 }
 
 /**
- * get node uncheckable state
- * @param {*} node
+ * Checks if a node is uncheckable.
  */
 function isNodeUncheckable(node, props) {
   var _props$uncheckableIte = props.uncheckableItemValues,
@@ -115,14 +147,12 @@ function getFormattedTree(nodes, data, props) {
     var curNode = nodes[node.refKey];
     if (curNode) {
       var _node$childrenKey;
-      var checkState = !(0, _isUndefined2.default)(cascade) ? getNodeCheckState({
-        node: curNode,
+      var checkState = !(0, _isUndefined2.default)(cascade) ? getNodeCheckState(curNode, {
         cascade: cascade,
         nodes: nodes,
         childrenKey: childrenKey
       }) : undefined;
       formatted.check = curNode.check;
-      formatted.expand = curNode.expand;
       formatted.uncheckable = curNode.uncheckable;
       (0, _attachParent.attachParent)(formatted, curNode.parent);
       formatted.checkState = checkState;
@@ -133,6 +163,10 @@ function getFormattedTree(nodes, data, props) {
     return formatted;
   });
 }
+
+/**
+ * Determines the disabled state of a tree node.
+ */
 function getDisabledState(nodes, node, props) {
   var _props$disabledItemVa = props.disabledItemValues,
     disabledItemValues = _props$disabledItemVa === void 0 ? [] : _props$disabledItemVa,
@@ -144,18 +178,26 @@ function getDisabledState(nodes, node, props) {
     return node.refKey && nodes[node.refKey][valueKey] === value;
   });
 }
-function getCheckTreePickerDefaultValue(value, uncheckableItemValues) {
-  if (Array.isArray(value)) {
+
+/**
+ * Returns the default value for the check tree.
+ */
+function getCheckTreeDefaultValue(value, uncheckableItemValues) {
+  if (Array.isArray(value) && Array.isArray(uncheckableItemValues)) {
     return value.filter(function (v) {
       return !uncheckableItemValues.includes(v);
     });
   }
   return [];
 }
+
+/**
+ * Retrieves the selected items from the given nodes.
+ */
 function getSelectedItems(nodes, values) {
   var checkedItems = [];
   values.forEach(function (value) {
-    var refKey = (0, _treeUtils.getNodeFormattedRefKey)(value);
+    var refKey = (0, _utils2.formatNodeRefKey)(value);
     var node = nodes[refKey];
     if (!(0, _isNil2.default)(node)) {
       checkedItems.push(node);
@@ -163,11 +205,16 @@ function getSelectedItems(nodes, values) {
   });
   return checkedItems;
 }
-function getNodeCheckState(_ref) {
-  var nodes = _ref.nodes,
-    node = _ref.node,
-    cascade = _ref.cascade,
-    childrenKey = _ref.childrenKey;
+/**
+ * Calculates the check state of a node in a check tree.
+ */
+function getNodeCheckState(node, options) {
+  var nodes = options.nodes,
+    cascade = options.cascade,
+    childrenKey = options.childrenKey;
+  if (node.refKey === undefined) {
+    return _utils.CHECK_STATE.UNCHECK;
+  }
   if ((0, _isNil2.default)(nodes[node.refKey])) {
     return _utils.CHECK_STATE.UNCHECK;
   }
@@ -175,7 +222,10 @@ function getNodeCheckState(_ref) {
     nodes[node.refKey].checkAll = false;
     return node.check ? _utils.CHECK_STATE.CHECK : _utils.CHECK_STATE.UNCHECK;
   }
-  if (isEveryChildChecked(nodes, node)) {
+  if (isEveryChildChecked(node, {
+    nodes: nodes,
+    childrenKey: childrenKey
+  })) {
     nodes[node.refKey].checkAll = true;
     nodes[node.refKey].check = true;
     return _utils.CHECK_STATE.CHECK;

package/cjs/CheckTreePicker/CheckTreePicker.d.ts

@@ -1,20 +1,38 @@
 import React from 'react';
 import { PickerLocale } from '../locales';
 import { PickerComponent, PickerToggleProps } from '../internals/Picker';
-import { TreeBaseProps } from '../Tree/Tree';
-import { FormControlPickerProps, ItemDataType } from '../@types/common';
+import { type CheckTreeViewProps } from '../CheckTree/CheckTreeView';
+import { TreeNode } from '../Tree/types';
+import type { FormControlPickerProps, ItemDataType, DeprecatedPickerProps } from '../@types/common';
+import type { TreeExtraProps } from '../Tree/types';
 export declare type ValueType = (string | number)[];
-export interface CheckTreePickerProps<T = ValueType> extends TreeBaseProps<T, ItemDataType>, FormControlPickerProps<T, PickerLocale, ItemDataType>, Pick<PickerToggleProps, 'caretAs' | 'loading'> {
-    /** Tree node cascade */
-    cascade?: boolean;
-    /** A picker that can be counted */
+export interface CheckTreePickerProps<V = ValueType> extends Omit<CheckTreeViewProps<V>, 'value' | 'onChange' | 'data'>, TreeExtraProps, DeprecatedPickerProps, FormControlPickerProps<V, PickerLocale, ItemDataType>, Pick<PickerToggleProps, 'caretAs' | 'loading'> {
+    /**
+     * A picker that can be counted
+     */
     countable?: boolean;
-    /** Set the option value for the check box not to be rendered */
-    uncheckableItemValues?: T;
-    /** Custom render selected items */
-    renderValue?: (value: any[], selectedItems: any[], selectedElement: React.ReactNode) => React.ReactNode;
-    /** Called when scrolling */
-    onScroll?: (event: React.SyntheticEvent) => void;
+    /**
+     * Custom popup style
+     */
+    popupClassName?: string;
+    /**
+     * Custom popup style
+     */
+    popupStyle?: React.CSSProperties;
+    /**
+     * The height of the tree
+     */
+    treeHeight?: number;
+    /**
+     * Popup auto width
+     *
+     * @default true
+     */
+    popupAutoWidth?: boolean;
+    /**
+     * Custom render selected items
+     */
+    renderValue?: (value: V, selectedNodes: TreeNode[], selectedElement: React.ReactNode) => React.ReactNode;
 }
 /**
  * The `CheckTreePicker` component is used for selecting multiple options which are organized in a tree structure.