@fle-ui/next 3.0.0-alpha.3 → 3.0.0-beta.1

package/es/table/hooks/useSelection.js

@@ -1,162 +1,178 @@
-import _defineProperty from "@babel/runtime/helpers/esm/defineProperty";
-import _toArray from "@babel/runtime/helpers/esm/toArray";
-import _extends from "@babel/runtime/helpers/esm/extends";
-import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
+"use client";
+
+function _defineProperty(obj, key, value) { key = _toPropertyKey(key); if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
+function _toPropertyKey(arg) { var key = _toPrimitive(arg, "string"); return _typeof(key) === "symbol" ? key : String(key); }
+function _toPrimitive(input, hint) { if (_typeof(input) !== "object" || input === null) return input; var prim = input[Symbol.toPrimitive]; if (prim !== undefined) { var res = prim.call(input, hint || "default"); if (_typeof(res) !== "object") return res; throw new TypeError("@@toPrimitive must return a primitive value."); } return (hint === "string" ? String : Number)(input); }
+function _toArray(arr) { return _arrayWithHoles(arr) || _iterableToArray(arr) || _unsupportedIterableToArray(arr) || _nonIterableRest(); }
+function _iterableToArray(iter) { if (typeof Symbol !== "undefined" && iter[Symbol.iterator] != null || iter["@@iterator"] != null) return Array.from(iter); }
+function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
+function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
+function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
+function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
+function _iterableToArrayLimit(r, l) { var t = null == r ? null : "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (null != t) { var e, n, i, u, a = [], f = !0, o = !1; try { if (i = (t = t.call(r)).next, 0 === l) { if (Object(t) !== t) return; f = !1; } else for (; !(f = (e = i.call(t)).done) && (a.push(e.value), a.length !== l); f = !0); } catch (r) { o = !0, n = r; } finally { try { if (!f && null != t.return && (u = t.return(), Object(u) !== u)) return; } finally { if (o) throw n; } } return a; } }
+function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
+function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
 import _toConsumableArray from "@babel/runtime/helpers/esm/toConsumableArray";
-import _typeof from "@babel/runtime/helpers/esm/typeof";
+import * as React from 'react';
+import { useCallback, useMemo } from 'react';
 import DownOutlined from "@ant-design/icons/es/icons/DownOutlined";
+import classNames from 'classnames';
 import { INTERNAL_COL_DEFINE } from 'rc-table';
 import { arrAdd, arrDel } from "rc-tree/es/util";
 import { conductCheck } from "rc-tree/es/utils/conductUtil";
 import { convertDataToEntities } from "rc-tree/es/utils/treeUtil";
 import useMergedState from "rc-util/es/hooks/useMergedState";
-import * as React from 'react';
-import { useCallback, useMemo, useState } from 'react';
+import useMultipleSelect from "../../_util/hooks/useMultipleSelect";
+import { devUseWarning } from "../../_util/warning";
 import Checkbox from "../../checkbox";
 import Dropdown from "../../dropdown";
 import Radio from "../../radio";
-import warning from "../../_util/warning"; // TODO: warning if use ajax!!!
-
+// TODO: warning if use ajax!!!
 export var SELECTION_COLUMN = {};
 export var SELECTION_ALL = 'SELECT_ALL';
 export var SELECTION_INVERT = 'SELECT_INVERT';
 export var SELECTION_NONE = 'SELECT_NONE';
 var EMPTY_LIST = [];
-
-function flattenData(data, childrenColumnName) {
+var flattenData = function flattenData(childrenColumnName, data) {
   var list = [];
   (data || []).forEach(function (record) {
     list.push(record);
-
     if (record && _typeof(record) === 'object' && childrenColumnName in record) {
-      list = [].concat(_toConsumableArray(list), _toConsumableArray(flattenData(record[childrenColumnName], childrenColumnName)));
+      list = [].concat(_toConsumableArray(list), _toConsumableArray(flattenData(childrenColumnName, record[childrenColumnName])));
     }
   });
   return list;
-}
-
-export default function useSelection(rowSelection, config) {
-  var _ref = rowSelection || {},
-      preserveSelectedRowKeys = _ref.preserveSelectedRowKeys,
-      selectedRowKeys = _ref.selectedRowKeys,
-      defaultSelectedRowKeys = _ref.defaultSelectedRowKeys,
-      getCheckboxProps = _ref.getCheckboxProps,
-      onSelectionChange = _ref.onChange,
-      onSelect = _ref.onSelect,
-      onSelectAll = _ref.onSelectAll,
-      onSelectInvert = _ref.onSelectInvert,
-      onSelectNone = _ref.onSelectNone,
-      onSelectMultiple = _ref.onSelectMultiple,
-      selectionColWidth = _ref.columnWidth,
-      selectionType = _ref.type,
-      selections = _ref.selections,
-      fixed = _ref.fixed,
-      customizeRenderCell = _ref.renderCell,
-      hideSelectAll = _ref.hideSelectAll,
-      _ref$checkStrictly = _ref.checkStrictly,
-      checkStrictly = _ref$checkStrictly === void 0 ? true : _ref$checkStrictly;
-
+};
+var useSelection = function useSelection(config, rowSelection) {
+  var _ref6 = rowSelection || {},
+    preserveSelectedRowKeys = _ref6.preserveSelectedRowKeys,
+    selectedRowKeys = _ref6.selectedRowKeys,
+    defaultSelectedRowKeys = _ref6.defaultSelectedRowKeys,
+    getCheckboxProps = _ref6.getCheckboxProps,
+    onSelectionChange = _ref6.onChange,
+    onSelect = _ref6.onSelect,
+    onSelectAll = _ref6.onSelectAll,
+    onSelectInvert = _ref6.onSelectInvert,
+    onSelectNone = _ref6.onSelectNone,
+    onSelectMultiple = _ref6.onSelectMultiple,
+    selectionColWidth = _ref6.columnWidth,
+    selectionType = _ref6.type,
+    selections = _ref6.selections,
+    fixed = _ref6.fixed,
+    customizeRenderCell = _ref6.renderCell,
+    hideSelectAll = _ref6.hideSelectAll,
+    _ref6$checkStrictly = _ref6.checkStrictly,
+    checkStrictly = _ref6$checkStrictly === void 0 ? true : _ref6$checkStrictly;
   var prefixCls = config.prefixCls,
-      data = config.data,
-      pageData = config.pageData,
-      getRecordByKey = config.getRecordByKey,
-      getRowKey = config.getRowKey,
-      expandType = config.expandType,
-      childrenColumnName = config.childrenColumnName,
-      tableLocale = config.locale,
-      getPopupContainer = config.getPopupContainer; // ========================= Keys =========================
-
+    data = config.data,
+    pageData = config.pageData,
+    getRecordByKey = config.getRecordByKey,
+    getRowKey = config.getRowKey,
+    expandType = config.expandType,
+    childrenColumnName = config.childrenColumnName,
+    tableLocale = config.locale,
+    getPopupContainer = config.getPopupContainer;
+  var warning = devUseWarning('Table');
+  // ========================= MultipleSelect =========================
+  var _useMultipleSelect = useMultipleSelect(function (item) {
+      return item;
+    }),
+    _useMultipleSelect2 = _slicedToArray(_useMultipleSelect, 2),
+    multipleSelect = _useMultipleSelect2[0],
+    updatePrevSelectedIndex = _useMultipleSelect2[1];
+  // ========================= Keys =========================
   var _useMergedState = useMergedState(selectedRowKeys || defaultSelectedRowKeys || EMPTY_LIST, {
-    value: selectedRowKeys
-  }),
-      _useMergedState2 = _slicedToArray(_useMergedState, 2),
-      mergedSelectedKeys = _useMergedState2[0],
-      setMergedSelectedKeys = _useMergedState2[1]; // ======================== Caches ========================
-
-
+      value: selectedRowKeys
+    }),
+    _useMergedState2 = _slicedToArray(_useMergedState, 2),
+    mergedSelectedKeys = _useMergedState2[0],
+    setMergedSelectedKeys = _useMergedState2[1];
+  // ======================== Caches ========================
   var preserveRecordsRef = React.useRef(new Map());
   var updatePreserveRecordsCache = useCallback(function (keys) {
     if (preserveSelectedRowKeys) {
-      var newCache = new Map(); // Keep key if mark as preserveSelectedRowKeys
-
+      var newCache = new Map();
+      // Keep key if mark as preserveSelectedRowKeys
       keys.forEach(function (key) {
         var record = getRecordByKey(key);
-
         if (!record && preserveRecordsRef.current.has(key)) {
           record = preserveRecordsRef.current.get(key);
         }
-
         newCache.set(key, record);
-      }); // Refresh to new cache
-
+      });
+      // Refresh to new cache
       preserveRecordsRef.current = newCache;
     }
-  }, [getRecordByKey, preserveSelectedRowKeys]); // Update cache with selectedKeys
-
+  }, [getRecordByKey, preserveSelectedRowKeys]);
+  // Update cache with selectedKeys
   React.useEffect(function () {
     updatePreserveRecordsCache(mergedSelectedKeys);
   }, [mergedSelectedKeys]);
-
   var _useMemo = useMemo(function () {
-    return checkStrictly ? {
-      keyEntities: null
-    } : convertDataToEntities(data, {
-      externalGetKey: getRowKey,
-      childrenPropName: childrenColumnName
-    });
-  }, [data, getRowKey, checkStrictly, childrenColumnName]),
-      keyEntities = _useMemo.keyEntities; // Get flatten data
-
-
+      if (checkStrictly) {
+        return {
+          keyEntities: null
+        };
+      }
+      var convertData = data;
+      if (preserveSelectedRowKeys) {
+        var keysSet = new Set(data.map(function (record, index) {
+          return getRowKey(record, index);
+        }));
+        // remove preserveRecords that duplicate data
+        var preserveRecords = Array.from(preserveRecordsRef.current).reduce(function (total, _ref) {
+          var _ref7 = _slicedToArray(_ref, 2),
+            key = _ref7[0],
+            value = _ref7[1];
+          return keysSet.has(key) ? total : total.concat(value);
+        }, []);
+        convertData = [].concat(_toConsumableArray(convertData), _toConsumableArray(preserveRecords));
+      }
+      return convertDataToEntities(convertData, {
+        externalGetKey: getRowKey,
+        childrenPropName: childrenColumnName
+      });
+    }, [data, getRowKey, checkStrictly, childrenColumnName, preserveSelectedRowKeys]),
+    keyEntities = _useMemo.keyEntities;
+  // Get flatten data
   var flattedData = useMemo(function () {
-    return flattenData(pageData, childrenColumnName);
-  }, [pageData, childrenColumnName]); // Get all checkbox props
-
+    return flattenData(childrenColumnName, pageData);
+  }, [childrenColumnName, pageData]);
+  // Get all checkbox props
   var checkboxPropsMap = useMemo(function () {
     var map = new Map();
     flattedData.forEach(function (record, index) {
       var key = getRowKey(record, index);
       var checkboxProps = (getCheckboxProps ? getCheckboxProps(record) : null) || {};
       map.set(key, checkboxProps);
-      process.env.NODE_ENV !== "production" ? warning(!('checked' in checkboxProps || 'defaultChecked' in checkboxProps), 'Table', 'Do not set `checked` or `defaultChecked` in `getCheckboxProps`. Please use `selectedRowKeys` instead.') : void 0;
+      process.env.NODE_ENV !== "production" ? warning(!('checked' in checkboxProps || 'defaultChecked' in checkboxProps), 'usage', 'Do not set `checked` or `defaultChecked` in `getCheckboxProps`. Please use `selectedRowKeys` instead.') : void 0;
     });
     return map;
   }, [flattedData, getRowKey, getCheckboxProps]);
   var isCheckboxDisabled = useCallback(function (r) {
     var _a;
-
     return !!((_a = checkboxPropsMap.get(getRowKey(r))) === null || _a === void 0 ? void 0 : _a.disabled);
   }, [checkboxPropsMap, getRowKey]);
-
   var _useMemo2 = useMemo(function () {
-    if (checkStrictly) {
-      return [mergedSelectedKeys || [], []];
-    }
-
-    var _conductCheck = conductCheck(mergedSelectedKeys, true, keyEntities, isCheckboxDisabled),
+      if (checkStrictly) {
+        return [mergedSelectedKeys || [], []];
+      }
+      var _conductCheck = conductCheck(mergedSelectedKeys, true, keyEntities, isCheckboxDisabled),
         checkedKeys = _conductCheck.checkedKeys,
         halfCheckedKeys = _conductCheck.halfCheckedKeys;
-
-    return [checkedKeys || [], halfCheckedKeys];
-  }, [mergedSelectedKeys, checkStrictly, keyEntities, isCheckboxDisabled]),
-      _useMemo3 = _slicedToArray(_useMemo2, 2),
-      derivedSelectedKeys = _useMemo3[0],
-      derivedHalfSelectedKeys = _useMemo3[1];
-
+      return [checkedKeys || [], halfCheckedKeys];
+    }, [mergedSelectedKeys, checkStrictly, keyEntities, isCheckboxDisabled]),
+    _useMemo3 = _slicedToArray(_useMemo2, 2),
+    derivedSelectedKeys = _useMemo3[0],
+    derivedHalfSelectedKeys = _useMemo3[1];
   var derivedSelectedKeySet = useMemo(function () {
     var keys = selectionType === 'radio' ? derivedSelectedKeys.slice(0, 1) : derivedSelectedKeys;
     return new Set(keys);
   }, [derivedSelectedKeys, selectionType]);
   var derivedHalfSelectedKeySet = useMemo(function () {
     return selectionType === 'radio' ? new Set() : new Set(derivedHalfSelectedKeys);
-  }, [derivedHalfSelectedKeys, selectionType]); // Save last selected key to enable range selection
-
-  var _useState = useState(null),
-      _useState2 = _slicedToArray(_useState, 2),
-      lastSelectedKey = _useState2[0],
-      setLastSelectedKey = _useState2[1]; // Reset if rowSelection reset
-
-
+  }, [derivedHalfSelectedKeys, selectionType]);
+  // Reset if rowSelection reset
   React.useEffect(function () {
     if (!rowSelection) {
       setMergedSelectedKeys(EMPTY_LIST);
@@ -166,7 +182,6 @@ export default function useSelection(rowSelection, config) {
     var availableKeys;
     var records;
     updatePreserveRecordsCache(keys);
-
     if (preserveSelectedRowKeys) {
       availableKeys = keys;
       records = keys.map(function (key) {
@@ -178,21 +193,19 @@ export default function useSelection(rowSelection, config) {
       records = [];
       keys.forEach(function (key) {
         var record = getRecordByKey(key);
-
         if (record !== undefined) {
           availableKeys.push(key);
           records.push(record);
         }
       });
     }
-
     setMergedSelectedKeys(availableKeys);
     onSelectionChange === null || onSelectionChange === void 0 ? void 0 : onSelectionChange(availableKeys, records, {
       type: method
     });
-  }, [setMergedSelectedKeys, getRecordByKey, onSelectionChange, preserveSelectedRowKeys]); // ====================== Selections ======================
+  }, [setMergedSelectedKeys, getRecordByKey, onSelectionChange, preserveSelectedRowKeys]);
+  // ====================== Selections ======================
   // Trigger single `onSelect` event
-
   var triggerSingleSelection = useCallback(function (key, selected, keys, event) {
     if (onSelect) {
       var rows = keys.map(function (k) {
@@ -200,14 +213,12 @@ export default function useSelection(rowSelection, config) {
       });
       onSelect(getRecordByKey(key), selected, rows, event);
     }
-
     setSelectedKeys(keys, 'single');
   }, [onSelect, getRecordByKey, setSelectedKeys]);
   var mergedSelections = useMemo(function () {
     if (!selections || hideSelectAll) {
       return null;
     }
-
     var selectionList = selections === true ? [SELECTION_ALL, SELECTION_INVERT, SELECTION_NONE] : selections;
     return selectionList.map(function (selection) {
       if (selection === SELECTION_ALL) {
@@ -224,7 +235,6 @@ export default function useSelection(rowSelection, config) {
           }
         };
       }
-
       if (selection === SELECTION_INVERT) {
         return {
           key: 'invert',
@@ -234,27 +244,23 @@ export default function useSelection(rowSelection, config) {
             pageData.forEach(function (record, index) {
               var key = getRowKey(record, index);
               var checkProps = checkboxPropsMap.get(key);
-
               if (!(checkProps === null || checkProps === void 0 ? void 0 : checkProps.disabled)) {
                 if (keySet.has(key)) {
-                  keySet["delete"](key);
+                  keySet.delete(key);
                 } else {
                   keySet.add(key);
                 }
               }
             });
             var keys = Array.from(keySet);
-
             if (onSelectInvert) {
-              process.env.NODE_ENV !== "production" ? warning(false, 'Table', '`onSelectInvert` will be removed in future. Please use `onChange` instead.') : void 0;
+              warning.deprecated(false, 'onSelectInvert', 'onChange');
               onSelectInvert(keys);
             }
-
             setSelectedKeys(keys, 'invert');
           }
         };
       }
-
       if (selection === SELECTION_NONE) {
         return {
           key: 'none',
@@ -268,42 +274,35 @@ export default function useSelection(rowSelection, config) {
           }
         };
       }
-
       return selection;
     }).map(function (selection) {
-      return _extends(_extends({}, selection), {
+      return Object.assign(Object.assign({}, selection), {
         onSelect: function onSelect() {
           var _a2;
-
           var _a;
-
           for (var _len = arguments.length, rest = new Array(_len), _key = 0; _key < _len; _key++) {
             rest[_key] = arguments[_key];
           }
-
           (_a = selection.onSelect) === null || _a === void 0 ? void 0 : (_a2 = _a).call.apply(_a2, [selection].concat(rest));
-          setLastSelectedKey(null);
+          updatePrevSelectedIndex(null);
         }
       });
     });
-  }, [selections, derivedSelectedKeySet, pageData, getRowKey, onSelectInvert, setSelectedKeys]); // ======================= Columns ========================
-
+  }, [selections, derivedSelectedKeySet, pageData, getRowKey, onSelectInvert, setSelectedKeys]);
+  // ======================= Columns ========================
   var transformColumns = useCallback(function (columns) {
-    var _a; // >>>>>>>>>>> Skip if not exists `rowSelection`
-
-
+    var _a;
+    // >>>>>>>>>>> Skip if not exists `rowSelection`
     if (!rowSelection) {
-      process.env.NODE_ENV !== "production" ? warning(!columns.includes(SELECTION_COLUMN), 'Table', '`rowSelection` is not config but `SELECTION_COLUMN` exists in the `columns`.') : void 0;
+      process.env.NODE_ENV !== "production" ? warning(!columns.includes(SELECTION_COLUMN), 'usage', '`rowSelection` is not config but `SELECTION_COLUMN` exists in the `columns`.') : void 0;
       return columns.filter(function (col) {
         return col !== SELECTION_COLUMN;
       });
-    } // >>>>>>>>>>> Support selection
-
-
+    }
+    // >>>>>>>>>>> Support selection
     var cloneColumns = _toConsumableArray(columns);
-
-    var keySet = new Set(derivedSelectedKeySet); // Record key only need check with enabled
-
+    var keySet = new Set(derivedSelectedKeySet);
+    // Record key only need check with enabled
     var recordKeys = flattedData.map(getRowKey).filter(function (key) {
       return !checkboxPropsMap.get(key).disabled;
     });
@@ -313,13 +312,11 @@ export default function useSelection(rowSelection, config) {
     var checkedCurrentSome = recordKeys.some(function (key) {
       return keySet.has(key);
     });
-
     var onSelectAllChange = function onSelectAllChange() {
       var changeKeys = [];
-
       if (checkedCurrentAll) {
         recordKeys.forEach(function (key) {
-          keySet["delete"](key);
+          keySet.delete(key);
           changeKeys.push(key);
         });
       } else {
@@ -330,7 +327,6 @@ export default function useSelection(rowSelection, config) {
           }
         });
       }
-
       var keys = Array.from(keySet);
       onSelectAll === null || onSelectAll === void 0 ? void 0 : onSelectAll(!checkedCurrentAll, keys.map(function (k) {
         return getRecordByKey(k);
@@ -338,25 +334,23 @@ export default function useSelection(rowSelection, config) {
         return getRecordByKey(k);
       }));
       setSelectedKeys(keys, 'all');
-      setLastSelectedKey(null);
-    }; // ===================== Render =====================
+      updatePrevSelectedIndex(null);
+    };
+    // ===================== Render =====================
     // Title Cell
-
-
     var title;
-
+    var columnTitleCheckbox;
     if (selectionType !== 'radio') {
       var customizeSelections;
-
       if (mergedSelections) {
         var menu = {
           getPopupContainer: getPopupContainer,
           items: mergedSelections.map(function (selection, index) {
             var key = selection.key,
-                text = selection.text,
-                onSelectionClick = selection.onSelect;
+              text = selection.text,
+              onSelectionClick = selection.onSelect;
             return {
-              key: key || index,
+              key: key !== null && key !== void 0 ? key : index,
               onClick: function onClick() {
                 onSelectionClick === null || onSelectionClick === void 0 ? void 0 : onSelectionClick(recordKeys);
               },
@@ -365,17 +359,16 @@ export default function useSelection(rowSelection, config) {
           })
         };
         customizeSelections = /*#__PURE__*/React.createElement("div", {
-          className: prefixCls + "-selection-extra"
+          className: "".concat(prefixCls, "-selection-extra")
         }, /*#__PURE__*/React.createElement(Dropdown, {
           menu: menu,
           getPopupContainer: getPopupContainer
         }, /*#__PURE__*/React.createElement("span", null, /*#__PURE__*/React.createElement(DownOutlined, null))));
       }
-
       var allDisabledData = flattedData.map(function (record, index) {
         var key = getRowKey(record, index);
         var checkboxProps = checkboxPropsMap.get(key) || {};
-        return _extends({
+        return Object.assign({
           checked: keySet.has(key)
         }, checkboxProps);
       }).filter(function (_ref2) {
@@ -391,27 +384,26 @@ export default function useSelection(rowSelection, config) {
         var checked = _ref4.checked;
         return checked;
       });
-      title = !hideSelectAll && /*#__PURE__*/React.createElement("div", {
-        className: prefixCls + "-selection"
-      }, /*#__PURE__*/React.createElement(Checkbox, {
+      columnTitleCheckbox = /*#__PURE__*/React.createElement(Checkbox, {
         checked: !allDisabled ? !!flattedData.length && checkedCurrentAll : allDisabledAndChecked,
         indeterminate: !allDisabled ? !checkedCurrentAll && checkedCurrentSome : !allDisabledAndChecked && allDisabledSomeChecked,
         onChange: onSelectAllChange,
         disabled: flattedData.length === 0 || allDisabled,
         "aria-label": customizeSelections ? 'Custom selection' : 'Select all',
         skipGroup: true
-      }), customizeSelections);
-    } // Body Cell
-
-
+      });
+      title = !hideSelectAll && /*#__PURE__*/React.createElement("div", {
+        className: "".concat(prefixCls, "-selection")
+      }, columnTitleCheckbox, customizeSelections);
+    }
+    // Body Cell
     var renderCell;
-
     if (selectionType === 'radio') {
       renderCell = function renderCell(_, record, index) {
         var key = getRowKey(record, index);
         var checked = keySet.has(key);
         return {
-          node: /*#__PURE__*/React.createElement(Radio, _extends({}, checkboxPropsMap.get(key), {
+          node: /*#__PURE__*/React.createElement(Radio, Object.assign({}, checkboxPropsMap.get(key), {
             checked: checked,
             onClick: function onClick(e) {
               return e.stopPropagation();
@@ -428,23 +420,20 @@ export default function useSelection(rowSelection, config) {
     } else {
       renderCell = function renderCell(_, record, index) {
         var _a;
-
         var key = getRowKey(record, index);
         var checked = keySet.has(key);
         var indeterminate = derivedHalfSelectedKeySet.has(key);
         var checkboxProps = checkboxPropsMap.get(key);
         var mergedIndeterminate;
-
         if (expandType === 'nest') {
           mergedIndeterminate = indeterminate;
-          process.env.NODE_ENV !== "production" ? warning(typeof (checkboxProps === null || checkboxProps === void 0 ? void 0 : checkboxProps.indeterminate) !== 'boolean', 'Table', 'set `indeterminate` using `rowSelection.getCheckboxProps` is not allowed with tree structured dataSource.') : void 0;
+          process.env.NODE_ENV !== "production" ? warning(typeof (checkboxProps === null || checkboxProps === void 0 ? void 0 : checkboxProps.indeterminate) !== 'boolean', 'usage', 'set `indeterminate` using `rowSelection.getCheckboxProps` is not allowed with tree structured dataSource.') : void 0;
         } else {
           mergedIndeterminate = (_a = checkboxProps === null || checkboxProps === void 0 ? void 0 : checkboxProps.indeterminate) !== null && _a !== void 0 ? _a : indeterminate;
-        } // Record checked
-
-
+        }
+        // Record checked
         return {
-          node: /*#__PURE__*/React.createElement(Checkbox, _extends({}, checkboxProps, {
+          node: /*#__PURE__*/React.createElement(Checkbox, Object.assign({}, checkboxProps, {
             indeterminate: mergedIndeterminate,
             checked: checked,
             skipGroup: true,
@@ -454,46 +443,14 @@ export default function useSelection(rowSelection, config) {
             onChange: function onChange(_ref5) {
               var nativeEvent = _ref5.nativeEvent;
               var shiftKey = nativeEvent.shiftKey;
-              var startIndex = -1;
-              var endIndex = -1; // Get range of this
-
-              if (shiftKey && checkStrictly) {
-                var pointKeys = new Set([lastSelectedKey, key]);
-                recordKeys.some(function (recordKey, recordIndex) {
-                  if (pointKeys.has(recordKey)) {
-                    if (startIndex === -1) {
-                      startIndex = recordIndex;
-                    } else {
-                      endIndex = recordIndex;
-                      return true;
-                    }
-                  }
-
-                  return false;
-                });
-              }
-
-              if (endIndex !== -1 && startIndex !== endIndex && checkStrictly) {
-                // Batch update selections
-                var rangeKeys = recordKeys.slice(startIndex, endIndex + 1);
-                var changedKeys = [];
-
-                if (checked) {
-                  rangeKeys.forEach(function (recordKey) {
-                    if (keySet.has(recordKey)) {
-                      changedKeys.push(recordKey);
-                      keySet["delete"](recordKey);
-                    }
-                  });
-                } else {
-                  rangeKeys.forEach(function (recordKey) {
-                    if (!keySet.has(recordKey)) {
-                      changedKeys.push(recordKey);
-                      keySet.add(recordKey);
-                    }
-                  });
-                }
-
+              var currentSelectedIndex = recordKeys.findIndex(function (item) {
+                return item === key;
+              });
+              var isMultiple = derivedSelectedKeys.some(function (item) {
+                return recordKeys.includes(item);
+              });
+              if (shiftKey && checkStrictly && isMultiple) {
+                var changedKeys = multipleSelect(currentSelectedIndex, recordKeys, keySet);
                 var keys = Array.from(keySet);
                 onSelectMultiple === null || onSelectMultiple === void 0 ? void 0 : onSelectMultiple(!checked, keys.map(function (recordKey) {
                   return getRecordByKey(recordKey);
@@ -504,7 +461,6 @@ export default function useSelection(rowSelection, config) {
               } else {
                 // Single record selected
                 var originCheckedKeys = derivedSelectedKeys;
-
                 if (checkStrictly) {
                   var checkedKeys = checked ? arrDel(originCheckedKeys, key) : arrAdd(originCheckedKeys, key);
                   triggerSingleSelection(key, !checked, checkedKeys, nativeEvent);
@@ -512,26 +468,24 @@ export default function useSelection(rowSelection, config) {
                   // Always fill first
                   var result = conductCheck([].concat(_toConsumableArray(originCheckedKeys), [key]), true, keyEntities, isCheckboxDisabled);
                   var _checkedKeys = result.checkedKeys,
-                      halfCheckedKeys = result.halfCheckedKeys;
-                  var nextCheckedKeys = _checkedKeys; // If remove, we do it again to correction
-
+                    halfCheckedKeys = result.halfCheckedKeys;
+                  var nextCheckedKeys = _checkedKeys;
+                  // If remove, we do it again to correction
                   if (checked) {
                     var tempKeySet = new Set(_checkedKeys);
-                    tempKeySet["delete"](key);
+                    tempKeySet.delete(key);
                     nextCheckedKeys = conductCheck(Array.from(tempKeySet), {
                       checked: false,
                       halfCheckedKeys: halfCheckedKeys
                     }, keyEntities, isCheckboxDisabled).checkedKeys;
                   }
-
                   triggerSingleSelection(key, !checked, nextCheckedKeys, nativeEvent);
                 }
               }
-
               if (checked) {
-                setLastSelectedKey(null);
+                updatePrevSelectedIndex(null);
               } else {
-                setLastSelectedKey(key);
+                updatePrevSelectedIndex(currentSelectedIndex);
               }
             }
           })),
@@ -539,52 +493,44 @@ export default function useSelection(rowSelection, config) {
         };
       };
     }
-
     var renderSelectionCell = function renderSelectionCell(_, record, index) {
       var _renderCell = renderCell(_, record, index),
-          node = _renderCell.node,
-          checked = _renderCell.checked;
-
+        node = _renderCell.node,
+        checked = _renderCell.checked;
       if (customizeRenderCell) {
         return customizeRenderCell(checked, record, index, node);
       }
-
       return node;
-    }; // Insert selection column if not exist
-
-
+    };
+    // Insert selection column if not exist
     if (!cloneColumns.includes(SELECTION_COLUMN)) {
       // Always after expand icon
       if (cloneColumns.findIndex(function (col) {
         var _a;
-
         return ((_a = col[INTERNAL_COL_DEFINE]) === null || _a === void 0 ? void 0 : _a.columnType) === 'EXPAND_COLUMN';
       }) === 0) {
         var _cloneColumns = cloneColumns,
-            _cloneColumns2 = _toArray(_cloneColumns),
-            expandColumn = _cloneColumns2[0],
-            restColumns = _cloneColumns2.slice(1);
-
+          _cloneColumns2 = _toArray(_cloneColumns),
+          expandColumn = _cloneColumns2[0],
+          restColumns = _cloneColumns2.slice(1);
         cloneColumns = [expandColumn, SELECTION_COLUMN].concat(_toConsumableArray(restColumns));
       } else {
         // Normal insert at first column
         cloneColumns = [SELECTION_COLUMN].concat(_toConsumableArray(cloneColumns));
       }
-    } // Deduplicate selection column
-
-
+    }
+    // Deduplicate selection column
     var selectionColumnIndex = cloneColumns.indexOf(SELECTION_COLUMN);
     process.env.NODE_ENV !== "production" ? warning(cloneColumns.filter(function (col) {
       return col === SELECTION_COLUMN;
-    }).length <= 1, 'Table', 'Multiple `SELECTION_COLUMN` exist in `columns`.') : void 0;
+    }).length <= 1, 'usage', 'Multiple `SELECTION_COLUMN` exist in `columns`.') : void 0;
     cloneColumns = cloneColumns.filter(function (column, index) {
       return column !== SELECTION_COLUMN || index === selectionColumnIndex;
-    }); // Fixed column logic
-
+    });
+    // Fixed column logic
     var prevCol = cloneColumns[selectionColumnIndex - 1];
     var nextCol = cloneColumns[selectionColumnIndex + 1];
     var mergedFixed = fixed;
-
     if (mergedFixed === undefined) {
       if ((nextCol === null || nextCol === void 0 ? void 0 : nextCol.fixed) !== undefined) {
         mergedFixed = nextCol.fixed;
@@ -592,25 +538,34 @@ export default function useSelection(rowSelection, config) {
         mergedFixed = prevCol.fixed;
       }
     }
-
     if (mergedFixed && prevCol && ((_a = prevCol[INTERNAL_COL_DEFINE]) === null || _a === void 0 ? void 0 : _a.columnType) === 'EXPAND_COLUMN' && prevCol.fixed === undefined) {
       prevCol.fixed = mergedFixed;
-    } // Replace with real selection column
-
-
+    }
+    var columnCls = classNames("".concat(prefixCls, "-selection-col"), _defineProperty({}, "".concat(prefixCls, "-selection-col-with-dropdown"), selections && selectionType === 'checkbox'));
+    var renderColumnTitle = function renderColumnTitle() {
+      if (!(rowSelection === null || rowSelection === void 0 ? void 0 : rowSelection.columnTitle)) {
+        return title;
+      }
+      if (typeof rowSelection.columnTitle === 'function') {
+        return rowSelection.columnTitle(columnTitleCheckbox);
+      }
+      return rowSelection.columnTitle;
+    };
+    // Replace with real selection column
     var selectionColumn = _defineProperty({
       fixed: mergedFixed,
       width: selectionColWidth,
-      className: prefixCls + "-selection-column",
-      title: rowSelection.columnTitle || title,
-      render: renderSelectionCell
+      className: "".concat(prefixCls, "-selection-column"),
+      title: renderColumnTitle(),
+      render: renderSelectionCell,
+      onCell: rowSelection.onCell
     }, INTERNAL_COL_DEFINE, {
-      className: prefixCls + "-selection-col"
+      className: columnCls
     });
-
     return cloneColumns.map(function (col) {
       return col === SELECTION_COLUMN ? selectionColumn : col;
     });
-  }, [getRowKey, flattedData, rowSelection, derivedSelectedKeys, derivedSelectedKeySet, derivedHalfSelectedKeySet, selectionColWidth, mergedSelections, expandType, lastSelectedKey, checkboxPropsMap, onSelectMultiple, triggerSingleSelection, isCheckboxDisabled]);
+  }, [getRowKey, flattedData, rowSelection, derivedSelectedKeys, derivedSelectedKeySet, derivedHalfSelectedKeySet, selectionColWidth, mergedSelections, expandType, checkboxPropsMap, onSelectMultiple, triggerSingleSelection, isCheckboxDisabled]);
   return [transformColumns, derivedSelectedKeySet];
-}
+};
+export default useSelection;