@mui/x-data-grid 8.0.0-alpha.12 → 8.0.0-alpha.13

package/hooks/features/rowSelection/useGridRowSelection.js

@@ -8,6 +8,7 @@ Object.defineProperty(exports, "__esModule", {
 exports.useGridRowSelection = exports.rowSelectionStateInitializer = void 0;
 var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
 var React = _interopRequireWildcard(require("react"));
+var _useEventCallback = _interopRequireDefault(require("@mui/utils/useEventCallback"));
 var _signature = require("../../../constants/signature");
 var _useGridApiEventHandler = require("../../utils/useGridApiEventHandler");
 var _useGridApiMethod = require("../../utils/useGridApiMethod");
@@ -15,7 +16,6 @@ var _useGridLogger = require("../../utils/useGridLogger");
 var _useGridSelector = require("../../utils/useGridSelector");
 var _gridRowsSelector = require("../rows/gridRowsSelector");
 var _gridRowSelectionSelector = require("./gridRowSelectionSelector");
-var _pagination = require("../pagination");
 var _gridFocusStateSelector = require("../focus/gridFocusStateSelector");
 var _gridFilterSelector = require("../filter/gridFilterSelector");
 var _colDef = require("../../../colDef");
@@ -26,20 +26,14 @@ var _constants = require("../../../internals/constants");
 var _gridClasses = require("../../../constants/gridClasses");
 var _domUtils = require("../../../utils/domUtils");
 var _utils = require("./utils");
-const getSelectionModelPropValue = (selectionModelProp, prevSelectionModel) => {
-  if (selectionModelProp == null) {
-    return selectionModelProp;
-  }
-  if (Array.isArray(selectionModelProp)) {
-    return selectionModelProp;
-  }
-  if (prevSelectionModel && prevSelectionModel[0] === selectionModelProp) {
-    return prevSelectionModel;
-  }
-  return [selectionModelProp];
+var _gridRowSelectionManager = require("../../../models/gridRowSelectionManager");
+var _pagination = require("../pagination");
+const emptyModel = {
+  type: 'include',
+  ids: new Set()
 };
 const rowSelectionStateInitializer = (state, props) => (0, _extends2.default)({}, state, {
-  rowSelection: props.rowSelection ? getSelectionModelPropValue(props.rowSelectionModel) ?? [] : []
+  rowSelection: props.rowSelection ? props.rowSelectionModel ?? emptyModel : emptyModel
 });
 
 /**
@@ -58,8 +52,8 @@ const useGridRowSelection = (apiRef, props) => {
   }, [props.rowSelection]);
   const applyAutoSelection = props.signature !== _signature.GridSignature.DataGrid && (props.rowSelectionPropagation?.parents || props.rowSelectionPropagation?.descendants);
   const propRowSelectionModel = React.useMemo(() => {
-    return getSelectionModelPropValue(props.rowSelectionModel, (0, _gridRowSelectionSelector.gridRowSelectionStateSelector)(apiRef));
-  }, [apiRef, props.rowSelectionModel]);
+    return props.rowSelectionModel;
+  }, [props.rowSelectionModel]);
   const lastRowToggled = React.useRef(null);
   apiRef.current.registerControlState({
     stateId: 'rowSelection',
@@ -99,23 +93,30 @@ const useGridRowSelection = (apiRef, props) => {
       endId
     }, !isSelected);
   }, [apiRef]);
+  const getRowsToBeSelected = (0, _useEventCallback.default)(() => {
+    const rowsToBeSelected = props.pagination && props.checkboxSelectionVisibleOnly && props.paginationMode === 'client' ? (0, _pagination.gridPaginatedVisibleSortedGridRowIdsSelector)(apiRef) : (0, _gridFilterSelector.gridExpandedSortedRowIdsSelector)(apiRef);
+    return rowsToBeSelected;
+  });
 
   /*
    * API METHODS
    */
   const setRowSelectionModel = React.useCallback(model => {
-    if (props.signature === _signature.GridSignature.DataGrid && !canHaveMultipleSelection && Array.isArray(model) && model.length > 1) {
+    if (props.signature === _signature.GridSignature.DataGrid && !canHaveMultipleSelection && (model.type !== 'include' || model.ids.size > 1)) {
       throw new Error(['MUI X: `rowSelectionModel` can only contain 1 item in DataGrid.', 'You need to upgrade to DataGridPro or DataGridPremium component to unlock multiple selection.'].join('\n'));
     }
     const currentModel = (0, _gridRowSelectionSelector.gridRowSelectionStateSelector)(apiRef);
     if (currentModel !== model) {
       logger.debug(`Setting selection model`);
       apiRef.current.setState(state => (0, _extends2.default)({}, state, {
-        rowSelection: props.rowSelection ? model : []
+        rowSelection: props.rowSelection ? model : emptyModel
       }));
     }
   }, [apiRef, logger, props.rowSelection, props.signature, canHaveMultipleSelection]);
-  const isRowSelected = React.useCallback(id => (0, _gridRowSelectionSelector.gridRowSelectionStateSelector)(apiRef).includes(id), [apiRef]);
+  const isRowSelected = React.useCallback(id => {
+    const selectionManager = (0, _gridRowSelectionSelector.gridRowSelectionManagerSelector)(apiRef);
+    return selectionManager.has(id);
+  }, [apiRef]);
   const isRowSelectable = React.useCallback(id => {
     if (props.rowSelection === false) {
       return false;
@@ -129,7 +130,7 @@ const useGridRowSelection = (apiRef, props) => {
     }
     return true;
   }, [apiRef, props.rowSelection, propIsRowSelectable]);
-  const getSelectedRows = React.useCallback(() => (0, _gridRowSelectionSelector.selectedGridRowsSelector)(apiRef), [apiRef]);
+  const getSelectedRows = React.useCallback(() => (0, _gridRowSelectionSelector.gridRowSelectionIdsSelector)(apiRef), [apiRef]);
   const selectRow = React.useCallback((id, isSelected = true, resetSelection = false) => {
     if (!apiRef.current.isRowSelectable(id)) {
       return;
@@ -137,9 +138,12 @@ const useGridRowSelection = (apiRef, props) => {
     lastRowToggled.current = id;
     if (resetSelection) {
       logger.debug(`Setting selection for row ${id}`);
-      const newSelection = [];
+      const newSelectionModel = {
+        type: 'include',
+        ids: new Set()
+      };
       const addRow = rowId => {
-        newSelection.push(rowId);
+        newSelectionModel.ids.add(rowId);
       };
       if (isSelected) {
         addRow(id);
@@ -147,17 +151,21 @@ const useGridRowSelection = (apiRef, props) => {
           (0, _utils.findRowsToSelect)(apiRef, tree, id, props.rowSelectionPropagation?.descendants ?? false, props.rowSelectionPropagation?.parents ?? false, addRow);
         }
       }
-      apiRef.current.setRowSelectionModel(newSelection);
+      apiRef.current.setRowSelectionModel(newSelectionModel);
     } else {
       logger.debug(`Toggling selection for row ${id}`);
-      const selection = (0, _gridRowSelectionSelector.gridRowSelectionStateSelector)(apiRef);
-      const newSelection = new Set(selection);
-      newSelection.delete(id);
+      const selectionModel = (0, _gridRowSelectionSelector.gridRowSelectionStateSelector)(apiRef);
+      const newSelectionModel = {
+        type: selectionModel.type,
+        ids: new Set(selectionModel.ids)
+      };
+      const selectionManager = (0, _gridRowSelectionManager.createRowSelectionManager)(newSelectionModel);
+      selectionManager.unselect(id);
       const addRow = rowId => {
-        newSelection.add(rowId);
+        selectionManager.select(rowId);
       };
       const removeRow = rowId => {
-        newSelection.delete(rowId);
+        selectionManager.unselect(rowId);
       };
       if (isSelected) {
         addRow(id);
@@ -167,45 +175,62 @@ const useGridRowSelection = (apiRef, props) => {
       } else if (applyAutoSelection) {
         (0, _utils.findRowsToDeselect)(apiRef, tree, id, props.rowSelectionPropagation?.descendants ?? false, props.rowSelectionPropagation?.parents ?? false, removeRow);
       }
-      const isSelectionValid = newSelection.size < 2 || canHaveMultipleSelection;
+      const isSelectionValid = newSelectionModel.type === 'include' && newSelectionModel.ids.size < 2 || canHaveMultipleSelection;
       if (isSelectionValid) {
-        apiRef.current.setRowSelectionModel(Array.from(newSelection));
+        apiRef.current.setRowSelectionModel(newSelectionModel);
       }
     }
   }, [apiRef, logger, applyAutoSelection, tree, props.rowSelectionPropagation?.descendants, props.rowSelectionPropagation?.parents, canHaveMultipleSelection]);
   const selectRows = React.useCallback((ids, isSelected = true, resetSelection = false) => {
     logger.debug(`Setting selection for several rows`);
-    const selectableIds = ids.filter(id => apiRef.current.isRowSelectable(id));
-    let newSelection;
+    if (props.rowSelection === false) {
+      return;
+    }
+    const selectableIds = new Set();
+    for (let i = 0; i < ids.length; i += 1) {
+      const id = ids[i];
+      if (apiRef.current.isRowSelectable(id)) {
+        selectableIds.add(id);
+      }
+    }
+    const currentSelectionModel = (0, _gridRowSelectionSelector.gridRowSelectionStateSelector)(apiRef);
+    let newSelectionModel;
     if (resetSelection) {
+      newSelectionModel = {
+        type: 'include',
+        ids: selectableIds
+      };
       if (isSelected) {
-        newSelection = new Set(selectableIds);
+        const selectionManager = (0, _gridRowSelectionManager.createRowSelectionManager)(newSelectionModel);
         if (applyAutoSelection) {
           const addRow = rowId => {
-            newSelection.add(rowId);
+            selectionManager.select(rowId);
           };
-          selectableIds.forEach(id => {
+          for (const id of selectableIds) {
             (0, _utils.findRowsToSelect)(apiRef, tree, id, props.rowSelectionPropagation?.descendants ?? false, props.rowSelectionPropagation?.parents ?? false, addRow);
-          });
+          }
         }
       } else {
-        newSelection = new Set();
+        newSelectionModel.ids = new Set();
       }
-      const currentLookup = (0, _gridRowSelectionSelector.selectedIdsLookupSelector)(apiRef);
-      if (newSelection.size === Object.keys(currentLookup).length && Array.from(newSelection).every(id => currentLookup[id] === id)) {
+      if (currentSelectionModel.type === newSelectionModel.type && newSelectionModel.ids.size === currentSelectionModel.ids.size && Array.from(newSelectionModel.ids).every(id => currentSelectionModel.ids.has(id))) {
         return;
       }
     } else {
-      newSelection = new Set(Object.values((0, _gridRowSelectionSelector.selectedIdsLookupSelector)(apiRef)));
+      newSelectionModel = {
+        type: currentSelectionModel.type,
+        ids: new Set(currentSelectionModel.ids)
+      };
+      const selectionManager = (0, _gridRowSelectionManager.createRowSelectionManager)(newSelectionModel);
       const addRow = rowId => {
-        newSelection.add(rowId);
+        selectionManager.select(rowId);
       };
       const removeRow = rowId => {
-        newSelection.delete(rowId);
+        selectionManager.unselect(rowId);
       };
-      selectableIds.forEach(id => {
+      for (const id of selectableIds) {
         if (isSelected) {
-          newSelection.add(id);
+          selectionManager.select(id);
           if (applyAutoSelection) {
             (0, _utils.findRowsToSelect)(apiRef, tree, id, props.rowSelectionPropagation?.descendants ?? false, props.rowSelectionPropagation?.parents ?? false, addRow);
           }
@@ -215,13 +240,13 @@ const useGridRowSelection = (apiRef, props) => {
             (0, _utils.findRowsToDeselect)(apiRef, tree, id, props.rowSelectionPropagation?.descendants ?? false, props.rowSelectionPropagation?.parents ?? false, removeRow);
           }
         }
-      });
+      }
     }
-    const isSelectionValid = newSelection.size < 2 || canHaveMultipleSelection;
+    const isSelectionValid = newSelectionModel.type === 'include' && newSelectionModel.ids.size < 2 || canHaveMultipleSelection;
     if (isSelectionValid) {
-      apiRef.current.setRowSelectionModel(Array.from(newSelection));
+      apiRef.current.setRowSelectionModel(newSelectionModel);
     }
-  }, [logger, applyAutoSelection, canHaveMultipleSelection, apiRef, tree, props.rowSelectionPropagation?.descendants, props.rowSelectionPropagation?.parents]);
+  }, [logger, applyAutoSelection, canHaveMultipleSelection, apiRef, tree, props.rowSelectionPropagation?.descendants, props.rowSelectionPropagation?.parents, props.rowSelection]);
   const selectRowRange = React.useCallback(({
     startId,
     endId
@@ -260,57 +285,72 @@ const useGridRowSelection = (apiRef, props) => {
     const currentSelection = (0, _gridRowSelectionSelector.gridRowSelectionStateSelector)(apiRef);
     const rowsLookup = (0, _gridRowsSelector.gridRowsLookupSelector)(apiRef);
     const filteredRowsLookup = (0, _gridFilterSelector.gridFilteredRowsLookupSelector)(apiRef);
-
-    // We clone the existing object to avoid mutating the same object returned by the selector to others part of the project
-    const selectionLookup = (0, _extends2.default)({}, (0, _gridRowSelectionSelector.selectedIdsLookupSelector)(apiRef));
     const isNonExistent = id => {
       if (props.filterMode === 'server') {
         return !rowsLookup[id];
       }
       return !rowsLookup[id] || filteredRowsLookup[id] === false;
     };
+    const newSelectionModel = {
+      type: currentSelection.type,
+      ids: new Set(currentSelection.ids)
+    };
+    const selectionManager = (0, _gridRowSelectionManager.createRowSelectionManager)(newSelectionModel);
     let hasChanged = false;
-    currentSelection.forEach(id => {
+    for (const id of currentSelection.ids) {
       if (isNonExistent(id)) {
         if (props.keepNonExistentRowsSelected) {
-          return;
+          continue;
         }
-        delete selectionLookup[id];
+        selectionManager.unselect(id);
         hasChanged = true;
-        return;
+        continue;
       }
       if (!props.rowSelectionPropagation?.parents) {
-        return;
+        continue;
       }
       const node = tree[id];
-      if (node.type === 'group') {
+      if (node?.type === 'group') {
         const isAutoGenerated = node.isAutoGenerated;
         if (isAutoGenerated) {
-          delete selectionLookup[id];
+          selectionManager.unselect(id);
           hasChanged = true;
-          return;
+          continue;
         }
         // Keep previously selected tree data parents selected if all their children are filtered out
         if (!node.children.every(childId => filteredRowsLookup[childId] === false)) {
-          delete selectionLookup[id];
+          selectionManager.unselect(id);
           hasChanged = true;
         }
       }
-    });
+    }
 
     // For nested data, on row tree updation (filtering, adding rows, etc.) when the selection is
     // not empty, we need to re-run scanning of the tree to propagate the selection changes
     // Example: A parent whose de-selected children are filtered out should now be selected
-    const shouldReapplyPropagation = isNestedData && props.rowSelectionPropagation?.parents && Object.keys(selectionLookup).length > 0;
+    const shouldReapplyPropagation = isNestedData && props.rowSelectionPropagation?.parents && (newSelectionModel.ids.size > 0 ||
+    // In case of exclude selection, newSelectionModel.ids.size === 0 means all rows are selected
+    newSelectionModel.type === 'exclude');
     if (hasChanged || shouldReapplyPropagation && !sortModelUpdated) {
-      const newSelection = Object.values(selectionLookup);
       if (shouldReapplyPropagation) {
-        apiRef.current.selectRows(newSelection, true, true);
+        if (newSelectionModel.type === 'exclude') {
+          const unfilteredSelectedRowIds = getRowsToBeSelected();
+          const selectedRowIds = [];
+          for (let i = 0; i < unfilteredSelectedRowIds.length; i += 1) {
+            const rowId = unfilteredSelectedRowIds[i];
+            if ((props.keepNonExistentRowsSelected || !isNonExistent(rowId)) && selectionManager.has(rowId)) {
+              selectedRowIds.push(rowId);
+            }
+          }
+          apiRef.current.selectRows(selectedRowIds, true, true);
+        } else {
+          apiRef.current.selectRows(Array.from(newSelectionModel.ids), true, true);
+        }
       } else {
-        apiRef.current.setRowSelectionModel(newSelection);
+        apiRef.current.setRowSelectionModel(newSelectionModel);
       }
     }
-  }, [apiRef, isNestedData, props.rowSelectionPropagation?.parents, props.keepNonExistentRowsSelected, props.filterMode, tree]);
+  }, [apiRef, isNestedData, props.rowSelectionPropagation?.parents, props.keepNonExistentRowsSelected, props.filterMode, tree, getRowsToBeSelected]);
   const handleSingleRowSelection = React.useCallback((id, event) => {
     const hasCtrlKey = event.metaKey || event.ctrlKey;
 
@@ -321,7 +361,7 @@ const useGridRowSelection = (apiRef, props) => {
 
     const isMultipleSelectionDisabled = !checkboxSelection && !hasCtrlKey && !(0, _keyboardUtils.isKeyboardEvent)(event);
     const resetSelection = !canHaveMultipleSelection || isMultipleSelectionDisabled;
-    const selectedRowsCount = apiRef.current.getSelectedRows().size;
+    const selectedRowsCount = (0, _gridRowSelectionSelector.gridRowSelectionCountSelector)(apiRef);
     if (canHaveMultipleSelection && selectedRowsCount > 1 && !hasCtrlKey) {
       apiRef.current.selectRow(id, true, resetSelection);
     } else {
@@ -370,10 +410,22 @@ const useGridRowSelection = (apiRef, props) => {
       apiRef.current.selectRow(params.id, params.value, !canHaveMultipleSelection);
     }
   }, [apiRef, expandMouseRowRangeSelection, canHaveMultipleSelection]);
+  const toggleAllRows = React.useCallback(value => {
+    const filterModel = (0, _gridFilterSelector.gridFilterModelSelector)(apiRef);
+    const quickFilterModel = (0, _gridFilterSelector.gridQuickFilterValuesSelector)(apiRef);
+    const hasFilters = filterModel.items.length > 0 || (quickFilterModel?.length || 0) > 0;
+    if (!props.isRowSelectable && !props.checkboxSelectionVisibleOnly && applyAutoSelection && !hasFilters) {
+      apiRef.current.setRowSelectionModel({
+        type: value ? 'exclude' : 'include',
+        ids: new Set()
+      });
+    } else {
+      apiRef.current.selectRows(getRowsToBeSelected(), value);
+    }
+  }, [apiRef, applyAutoSelection, getRowsToBeSelected, props.checkboxSelectionVisibleOnly, props.isRowSelectable]);
   const handleHeaderSelectionCheckboxChange = React.useCallback(params => {
-    const rowsToBeSelected = props.pagination && props.checkboxSelectionVisibleOnly && props.paginationMode === 'client' ? (0, _pagination.gridPaginatedVisibleSortedGridRowIdsSelector)(apiRef) : (0, _gridFilterSelector.gridExpandedSortedRowIdsSelector)(apiRef);
-    apiRef.current.selectRows(rowsToBeSelected, params.value);
-  }, [apiRef, props.checkboxSelectionVisibleOnly, props.pagination, props.paginationMode]);
+    toggleAllRows(params.value);
+  }, [toggleAllRows]);
   const handleCellKeyDown = React.useCallback((params, event) => {
     // Get the most recent cell mode because it may have been changed by another listener
     if (apiRef.current.getCellMode(params.id, params.field) === _gridEditRowModel.GridCellModes.Edit) {
@@ -422,7 +474,10 @@ const useGridRowSelection = (apiRef, props) => {
           }
         }
         const visibleRows = (0, _useGridVisibleRows.getVisibleRows)(apiRef);
-        const rowsBetweenStartAndEnd = visibleRows.rows.slice(start, end + 1).map(row => row.id);
+        const rowsBetweenStartAndEnd = [];
+        for (let i = start; i <= end; i += 1) {
+          rowsBetweenStartAndEnd.push(visibleRows.rows[i].id);
+        }
         apiRef.current.selectRows(rowsBetweenStartAndEnd, !isNextRowSelected);
         return;
       }
@@ -434,9 +489,9 @@ const useGridRowSelection = (apiRef, props) => {
     }
     if (String.fromCharCode(event.keyCode) === 'A' && (event.ctrlKey || event.metaKey)) {
       event.preventDefault();
-      selectRows(apiRef.current.getAllRowIds(), true);
+      toggleAllRows(true);
     }
-  }, [apiRef, handleSingleRowSelection, selectRows, canHaveMultipleSelection]);
+  }, [apiRef, canHaveMultipleSelection, handleSingleRowSelection, toggleAllRows]);
   (0, _useGridApiEventHandler.useGridApiEventHandler)(apiRef, 'sortedRowsSet', runIfRowSelectionIsEnabled(() => removeOutdatedSelection(true)));
   (0, _useGridApiEventHandler.useGridApiEventHandler)(apiRef, 'filteredRowsSet', runIfRowSelectionIsEnabled(removeOutdatedSelection));
   (0, _useGridApiEventHandler.useGridApiEventHandler)(apiRef, 'rowClick', runIfRowSelectionIsEnabled(handleRowClick));
@@ -455,7 +510,7 @@ const useGridRowSelection = (apiRef, props) => {
   }, [apiRef, propRowSelectionModel, props.rowSelection]);
   React.useEffect(() => {
     if (!props.rowSelection) {
-      apiRef.current.setRowSelectionModel([]);
+      apiRef.current.setRowSelectionModel(emptyModel);
     }
   }, [apiRef, props.rowSelection]);
   const isStateControlled = propRowSelectionModel != null;
@@ -466,10 +521,22 @@ const useGridRowSelection = (apiRef, props) => {
 
     // props.isRowSelectable changed
     const currentSelection = (0, _gridRowSelectionSelector.gridRowSelectionStateSelector)(apiRef);
-    if (isRowSelectable) {
-      const newSelection = currentSelection.filter(id => isRowSelectable(id));
-      if (newSelection.length < currentSelection.length) {
-        apiRef.current.setRowSelectionModel(newSelection);
+    if (typeof isRowSelectable === 'function') {
+      let selectableIds = new Set();
+      if (currentSelection.type === 'include') {
+        for (const id of currentSelection.ids) {
+          if (isRowSelectable(id)) {
+            selectableIds.add(id);
+          }
+        }
+      } else {
+        selectableIds = new Set(currentSelection.ids);
+      }
+      if (currentSelection.type === 'include' && selectableIds.size < currentSelection.ids.size) {
+        apiRef.current.setRowSelectionModel({
+          type: currentSelection.type,
+          ids: selectableIds
+        });
       }
     }
   }, [apiRef, isRowSelectable, isStateControlled, props.rowSelection]);
@@ -478,9 +545,9 @@ const useGridRowSelection = (apiRef, props) => {
       return;
     }
     const currentSelection = (0, _gridRowSelectionSelector.gridRowSelectionStateSelector)(apiRef);
-    if (!canHaveMultipleSelection && currentSelection.length > 1) {
+    if (!canHaveMultipleSelection && (currentSelection.type === 'include' && currentSelection.ids.size > 1 || currentSelection.type === 'exclude')) {
       // See https://github.com/mui/mui-x/issues/8455
-      apiRef.current.setRowSelectionModel([]);
+      apiRef.current.setRowSelectionModel(emptyModel);
     }
   }, [apiRef, canHaveMultipleSelection, checkboxSelection, isStateControlled, props.rowSelection]);
   React.useEffect(() => {

package/hooks/features/rowSelection/utils.d.ts

@@ -2,9 +2,9 @@ import { RefObject } from '@mui/x-internals/types';
 import type { GridRowId, GridRowTreeConfig } from '../../../models/gridRows';
 import type { DataGridProcessedProps } from '../../../models/props/DataGridProps';
 import type { GridPrivateApiCommunity } from '../../../models/api/gridApiCommunity';
-import type { GridRowSelectionPropagation } from '../../../models/gridRowSelectionModel';
+import { type GridRowSelectionPropagation } from "../../../models/gridRowSelectionModel.js";
 export declare const ROW_SELECTION_PROPAGATION_DEFAULT: GridRowSelectionPropagation;
-export declare function getCheckboxPropsSelector(groupId: GridRowId, autoSelectParents: boolean): import("../../..").OutputSelector<import("../../../models/gridStateCommunity").GridStateCommunity, Record<GridRowId, GridRowId>, {
+export declare function getCheckboxPropsSelector(groupId: GridRowId, autoSelectParents: boolean): import("../../..").OutputSelector<import("../../../models/gridStateCommunity").GridStateCommunity, import("../../../models/gridRowSelectionManager").RowSelectionManager, {
   isIndeterminate: boolean;
   isChecked: boolean;
 }>;

package/hooks/features/rowSelection/utils.js

@@ -38,15 +38,15 @@ function getGridRowGroupSelectableDescendants(apiRef, groupId) {
 
 // TODO v8: Use `createSelectorV8`
 function getCheckboxPropsSelector(groupId, autoSelectParents) {
-  return (0, _createSelector.createSelector)(_gridRowsSelector.gridRowTreeSelector, _gridSortingSelector.gridSortedRowIdsSelector, _gridFilterSelector.gridFilteredRowsLookupSelector, _gridRowSelectionSelector.selectedIdsLookupSelector, (rowTree, sortedRowIds, filteredRowsLookup, rowSelectionLookup) => {
+  return (0, _createSelector.createSelector)(_gridRowsSelector.gridRowTreeSelector, _gridSortingSelector.gridSortedRowIdsSelector, _gridFilterSelector.gridFilteredRowsLookupSelector, _gridRowSelectionSelector.gridRowSelectionManagerSelector, (rowTree, sortedRowIds, filteredRowsLookup, rowSelectionManager) => {
     const groupNode = rowTree[groupId];
     if (!groupNode || groupNode.type !== 'group') {
       return {
         isIndeterminate: false,
-        isChecked: rowSelectionLookup[groupId] === groupId
+        isChecked: rowSelectionManager.has(groupId)
       };
     }
-    if (rowSelectionLookup[groupId] === groupId) {
+    if (rowSelectionManager.has(groupId)) {
       return {
         isIndeterminate: false,
         isChecked: true
@@ -59,14 +59,14 @@ function getCheckboxPropsSelector(groupId, autoSelectParents) {
       const id = sortedRowIds[index];
       if (filteredRowsLookup[id] !== false) {
         selectableDescendantsCount += 1;
-        if (rowSelectionLookup[id] !== undefined) {
+        if (rowSelectionManager.has(id)) {
           selectedDescendantsCount += 1;
         }
       }
     }
     return {
-      isIndeterminate: selectedDescendantsCount > 0 && (selectedDescendantsCount < selectableDescendantsCount || rowSelectionLookup[groupId] === undefined),
-      isChecked: autoSelectParents ? selectedDescendantsCount > 0 : rowSelectionLookup[groupId] === groupId
+      isIndeterminate: selectedDescendantsCount > 0 && (selectedDescendantsCount < selectableDescendantsCount || !rowSelectionManager.has(groupId)),
+      isChecked: autoSelectParents ? selectedDescendantsCount > 0 : rowSelectionManager.has(groupId)
     };
   });
 }
@@ -104,7 +104,7 @@ const getFilteredRowNodeSiblings = (tree, filteredRows, id) => {
 };
 const findRowsToSelect = (apiRef, tree, selectedRow, autoSelectDescendants, autoSelectParents, addRow) => {
   const filteredRows = (0, _gridFilterSelector.gridFilteredRowsLookupSelector)(apiRef);
-  const selectedIdsLookup = (0, _gridRowSelectionSelector.selectedIdsLookupSelector)(apiRef);
+  const rowSelectionManager = (0, _gridRowSelectionSelector.gridRowSelectionManagerSelector)(apiRef);
   const selectedDescendants = new Set([]);
   if (!autoSelectDescendants && !autoSelectParents) {
     return;
@@ -121,7 +121,7 @@ const findRowsToSelect = (apiRef, tree, selectedRow, autoSelectDescendants, auto
   }
   if (autoSelectParents) {
     const checkAllDescendantsSelected = rowId => {
-      if (selectedIdsLookup[rowId] !== rowId && !selectedDescendants.has(rowId)) {
+      if (!rowSelectionManager.has(rowId) && !selectedDescendants.has(rowId)) {
         return false;
       }
       const node = tree[rowId];
@@ -147,14 +147,14 @@ const findRowsToSelect = (apiRef, tree, selectedRow, autoSelectDescendants, auto
 };
 exports.findRowsToSelect = findRowsToSelect;
 const findRowsToDeselect = (apiRef, tree, deselectedRow, autoSelectDescendants, autoSelectParents, removeRow) => {
-  const selectedIdsLookup = (0, _gridRowSelectionSelector.selectedIdsLookupSelector)(apiRef);
+  const rowSelectionManager = (0, _gridRowSelectionSelector.gridRowSelectionManagerSelector)(apiRef);
   if (!autoSelectParents && !autoSelectDescendants) {
     return;
   }
   if (autoSelectParents) {
     const allParents = getRowNodeParents(tree, deselectedRow);
     allParents.forEach(parent => {
-      const isSelected = selectedIdsLookup[parent] === parent;
+      const isSelected = rowSelectionManager.has(parent);
       if (isSelected) {
         removeRow(parent);
       }

package/hooks/features/rows/gridRowsInterfaces.d.ts

@@ -61,7 +61,7 @@ export interface GridRowsState {
   /**
    * Contains some values of type `GridRowId` that have been requested to be fetched
    * either by `defaultGroupingExpansionDepth` or `isGroupExpandedByDefault` props.
-   * Applicable with server-side grouped data and `unstable_dataSource` only.
+   * Applicable with server-side grouped data and `dataSource` only.
    */
   groupsToFetch?: GridRowId[];
 }

package/hooks/features/rows/gridRowsUtils.d.ts

@@ -1,5 +1,5 @@
 import { RefObject } from '@mui/x-internals/types';
-import { GridAutoGeneratedGroupNode, GridAutoGeneratedPinnedRowNode, GridFooterNode, GridGroupNode, GridRowId, GridRowIdGetter, GridRowModel, GridRowModelUpdate, GridRowTreeConfig, GridSkeletonRowNode, GridTreeNode } from "../../../models/index.js";
+import { GridAutoGeneratedGroupNode, GridAutoGeneratedPinnedRowNode, GridColDef, GridFooterNode, GridGroupNode, GridRowId, GridRowIdGetter, GridRowModel, GridRowModelUpdate, GridRowTreeConfig, GridSkeletonRowNode, GridTreeNode, GridValidRowModel } from "../../../models/index.js";
 import { DataGridProcessedProps } from "../../../models/props/DataGridProps.js";
 import { GridApiCommunity, GridPrivateApiCommunity } from "../../../models/api/gridApiCommunity.js";
 import { GridRowsInternalCache, GridRowsState, GridRowTreeCreationParams } from "./gridRowsInterfaces.js";
@@ -14,6 +14,7 @@ export declare const buildRootGroup: () => GridGroupNode;
  */
 export declare function checkGridRowIdIsValid(id: GridRowId, row: GridRowModel | Partial<GridRowModel>, detailErrorMessage?: string): void;
 export declare const getRowIdFromRowModel: (rowModel: GridRowModel, getRowId?: GridRowIdGetter, detailErrorMessage?: string) => GridRowId;
+export declare const getRowValue: (row: GridValidRowModel, colDef: GridColDef, apiRef: RefObject<GridApiCommunity>) => any;
 export declare const createRowsInternalCache: ({
   rows,
   getRowId,

package/hooks/features/rows/gridRowsUtils.js

@@ -7,7 +7,7 @@ Object.defineProperty(exports, "__esModule", {
 exports.buildRootGroup = exports.GRID_ROOT_GROUP_ID = exports.GRID_ID_AUTOGENERATED = void 0;
 exports.checkGridRowIdIsValid = checkGridRowIdIsValid;
 exports.computeRowsUpdates = computeRowsUpdates;
-exports.updateCacheWithNewRows = exports.rowHeightWarning = exports.minimalContentHeight = exports.isAutogeneratedRowNode = exports.isAutogeneratedRow = exports.getValidRowHeight = exports.getTreeNodeDescendants = exports.getTopLevelRowCount = exports.getRowsStateFromCache = exports.getRowIdFromRowModel = exports.getRowHeightWarning = exports.createRowsInternalCache = void 0;
+exports.updateCacheWithNewRows = exports.rowHeightWarning = exports.minimalContentHeight = exports.isAutogeneratedRowNode = exports.isAutogeneratedRow = exports.getValidRowHeight = exports.getTreeNodeDescendants = exports.getTopLevelRowCount = exports.getRowsStateFromCache = exports.getRowValue = exports.getRowIdFromRowModel = exports.getRowHeightWarning = exports.createRowsInternalCache = void 0;
 var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
 var _gridRowsSelector = require("./gridRowsSelector");
 const GRID_ROOT_GROUP_ID = exports.GRID_ROOT_GROUP_ID = `auto-generated-group-node-root`;
@@ -43,6 +43,15 @@ const getRowIdFromRowModel = (rowModel, getRowId, detailErrorMessage) => {
   return id;
 };
 exports.getRowIdFromRowModel = getRowIdFromRowModel;
+const getRowValue = (row, colDef, apiRef) => {
+  const field = colDef.field;
+  if (!colDef || !colDef.valueGetter) {
+    return row[field];
+  }
+  const value = row[colDef.field];
+  return colDef.valueGetter(value, row, colDef, apiRef);
+};
+exports.getRowValue = getRowValue;
 const createRowsInternalCache = ({
   rows,
   getRowId,

package/hooks/features/rows/useGridParamsApi.js

@@ -12,6 +12,7 @@ var _useGridApiMethod = require("../../utils/useGridApiMethod");
 var _gridFocusStateSelector = require("../focus/gridFocusStateSelector");
 var _gridListViewSelectors = require("../listView/gridListViewSelectors");
 var _gridRowsSelector = require("./gridRowsSelector");
+var _gridRowsUtils = require("./gridRowsUtils");
 class MissingRowIdError extends Error {}
 
 /**
@@ -97,14 +98,7 @@ function useGridParamsApi(apiRef, props) {
     }
     return colDef.valueGetter(row[colDef.field], row, colDef, apiRef);
   }, [apiRef]);
-  const getRowValue = React.useCallback((row, colDef) => {
-    const field = colDef.field;
-    if (!colDef || !colDef.valueGetter) {
-      return row[field];
-    }
-    const value = row[colDef.field];
-    return colDef.valueGetter(value, row, colDef, apiRef);
-  }, [apiRef]);
+  const getRowValue = React.useCallback((row, colDef) => (0, _gridRowsUtils.getRowValue)(row, colDef, apiRef), [apiRef]);
   const getRowFormattedValue = React.useCallback((row, colDef) => {
     const value = getRowValue(row, colDef);
     if (!colDef || !colDef.valueFormatter) {

package/hooks/features/rows/useGridRowAriaAttributes.js

@@ -6,22 +6,20 @@ Object.defineProperty(exports, "__esModule", {
 });
 exports.useGridRowAriaAttributes = void 0;
 var React = _interopRequireWildcard(require("react"));
-var _rowSelection = require("../rowSelection");
 var _useGridSelector = require("../../utils/useGridSelector");
 var _gridColumnGroupsSelector = require("../columnGrouping/gridColumnGroupsSelector");
 var _useGridPrivateApiContext = require("../../utils/useGridPrivateApiContext");
 const useGridRowAriaAttributes = () => {
   const apiRef = (0, _useGridPrivateApiContext.useGridPrivateApiContext)();
-  const selectedIdsLookup = (0, _useGridSelector.useGridSelector)(apiRef, _rowSelection.selectedIdsLookupSelector);
   const headerGroupingMaxDepth = (0, _useGridSelector.useGridSelector)(apiRef, _gridColumnGroupsSelector.gridColumnGroupsHeaderMaxDepthSelector);
   return React.useCallback((rowNode, index) => {
     const ariaAttributes = {};
     const ariaRowIndex = index + headerGroupingMaxDepth + 2; // 1 for the header row and 1 as it's 1-based
     ariaAttributes['aria-rowindex'] = ariaRowIndex;
     if (apiRef.current.isRowSelectable(rowNode.id)) {
-      ariaAttributes['aria-selected'] = selectedIdsLookup[rowNode.id] !== undefined;
+      ariaAttributes['aria-selected'] = apiRef.current.isRowSelected(rowNode.id);
     }
     return ariaAttributes;
-  }, [apiRef, selectedIdsLookup, headerGroupingMaxDepth]);
+  }, [apiRef, headerGroupingMaxDepth]);
 };
 exports.useGridRowAriaAttributes = useGridRowAriaAttributes;

package/hooks/features/rows/useGridRows.d.ts

@@ -2,5 +2,5 @@ import { RefObject } from '@mui/x-internals/types';
 import { DataGridProcessedProps } from "../../../models/props/DataGridProps.js";
 import { GridPrivateApiCommunity } from "../../../models/api/gridApiCommunity.js";
 import { GridStateInitializer } from "../../utils/useGridInitializeState.js";
-export declare const rowsStateInitializer: GridStateInitializer<Pick<DataGridProcessedProps, 'unstable_dataSource' | 'rows' | 'rowCount' | 'getRowId' | 'loading'>>;
-export declare const useGridRows: (apiRef: RefObject<GridPrivateApiCommunity>, props: Pick<DataGridProcessedProps, "rows" | "getRowId" | "rowCount" | "throttleRowsMs" | "signature" | "pagination" | "paginationMode" | "loading" | "unstable_dataSource">) => void;
+export declare const rowsStateInitializer: GridStateInitializer<Pick<DataGridProcessedProps, 'dataSource' | 'rows' | 'rowCount' | 'getRowId' | 'loading'>>;
+export declare const useGridRows: (apiRef: RefObject<GridPrivateApiCommunity>, props: Pick<DataGridProcessedProps, "rows" | "getRowId" | "rowCount" | "throttleRowsMs" | "signature" | "pagination" | "paginationMode" | "loading" | "dataSource">) => void;