@mui/x-data-grid-pro 5.5.0 → 5.6.1

package/node/hooks/features/rowGrouping/gridRowGroupingUtils.js

@@ -0,0 +1,153 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.mergeStateWithRowGroupingModel = exports.isGroupingColumn = exports.getRowGroupingFieldFromGroupingCriteria = exports.getRowGroupingCriteriaFromGroupingField = exports.getColDefOverrides = exports.filterRowTreeFromGroupingColumns = exports.GROUPING_COLUMNS_FEATURE_NAME = exports.GRID_ROW_GROUPING_SINGLE_GROUPING_FIELD = void 0;
+
+var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
+
+const GRID_ROW_GROUPING_SINGLE_GROUPING_FIELD = '__row_group_by_columns_group__';
+exports.GRID_ROW_GROUPING_SINGLE_GROUPING_FIELD = GRID_ROW_GROUPING_SINGLE_GROUPING_FIELD;
+const GROUPING_COLUMNS_FEATURE_NAME = 'grouping-columns';
+exports.GROUPING_COLUMNS_FEATURE_NAME = GROUPING_COLUMNS_FEATURE_NAME;
+
+const getRowGroupingFieldFromGroupingCriteria = groupingCriteria => {
+  if (groupingCriteria === null) {
+    return GRID_ROW_GROUPING_SINGLE_GROUPING_FIELD;
+  }
+
+  return `__row_group_by_columns_group_${groupingCriteria}__`;
+};
+
+exports.getRowGroupingFieldFromGroupingCriteria = getRowGroupingFieldFromGroupingCriteria;
+
+const getRowGroupingCriteriaFromGroupingField = groupingColDefField => {
+  const match = groupingColDefField.match(/^__row_group_by_columns_group_(.*)__$/);
+
+  if (!match) {
+    return null;
+  }
+
+  return match[1];
+};
+
+exports.getRowGroupingCriteriaFromGroupingField = getRowGroupingCriteriaFromGroupingField;
+
+const isGroupingColumn = field => field === GRID_ROW_GROUPING_SINGLE_GROUPING_FIELD || getRowGroupingCriteriaFromGroupingField(field) !== null;
+
+exports.isGroupingColumn = isGroupingColumn;
+
+/**
+ * When filtering a group, we only want to filter according to the items related to this grouping column.
+ */
+const shouldApplyFilterItemOnGroup = (item, node) => {
+  if (item.columnField === GRID_ROW_GROUPING_SINGLE_GROUPING_FIELD) {
+    return true;
+  }
+
+  const groupingCriteriaField = getRowGroupingCriteriaFromGroupingField(item.columnField);
+  return groupingCriteriaField === node.groupingField;
+};
+/**
+ * A leaf is visible if it passed the filter
+ * A group is visible if all the following criteria are met:
+ * - One of its children is passing the filter
+ * - It is passing the filter
+ */
+
+
+const filterRowTreeFromGroupingColumns = params => {
+  const {
+    rowTree,
+    isRowMatchingFilters
+  } = params;
+  const visibleRowsLookup = {};
+  const filteredRowsLookup = {};
+  const filteredDescendantCountLookup = {};
+
+  const filterTreeNode = (node, areAncestorsPassingChildren, areAncestorsExpanded) => {
+    var _node$children, _node$children2;
+
+    let isMatchingFilters;
+
+    if (!isRowMatchingFilters) {
+      isMatchingFilters = true;
+    } else {
+      const shouldApplyItem = node.isAutoGenerated ? item => shouldApplyFilterItemOnGroup(item, node) : undefined;
+      isMatchingFilters = isRowMatchingFilters(node.id, shouldApplyItem);
+    }
+
+    let filteredDescendantCount = 0;
+    (_node$children = node.children) == null ? void 0 : _node$children.forEach(childId => {
+      const childNode = rowTree[childId];
+      const childSubTreeSize = filterTreeNode(childNode, areAncestorsPassingChildren && isMatchingFilters, areAncestorsExpanded && !!node.childrenExpanded);
+      filteredDescendantCount += childSubTreeSize;
+    });
+    let shouldPassFilters;
+
+    if (!areAncestorsPassingChildren) {
+      shouldPassFilters = false;
+    } else if ((_node$children2 = node.children) != null && _node$children2.length) {
+      shouldPassFilters = isMatchingFilters && filteredDescendantCount > 0;
+    } else {
+      shouldPassFilters = isMatchingFilters;
+    }
+
+    visibleRowsLookup[node.id] = shouldPassFilters && areAncestorsExpanded;
+    filteredRowsLookup[node.id] = shouldPassFilters;
+
+    if (!shouldPassFilters) {
+      return 0;
+    }
+
+    filteredDescendantCountLookup[node.id] = filteredDescendantCount;
+
+    if (!node.children) {
+      return filteredDescendantCount + 1;
+    }
+
+    return filteredDescendantCount;
+  };
+
+  const nodes = Object.values(rowTree);
+
+  for (let i = 0; i < nodes.length; i += 1) {
+    const node = nodes[i];
+
+    if (node.depth === 0) {
+      filterTreeNode(node, true, true);
+    }
+  }
+
+  return {
+    visibleRowsLookup,
+    filteredRowsLookup,
+    filteredDescendantCountLookup
+  };
+};
+
+exports.filterRowTreeFromGroupingColumns = filterRowTreeFromGroupingColumns;
+
+const getColDefOverrides = (groupingColDefProp, fields) => {
+  if (typeof groupingColDefProp === 'function') {
+    return groupingColDefProp({
+      groupingName: GROUPING_COLUMNS_FEATURE_NAME,
+      fields
+    });
+  }
+
+  return groupingColDefProp;
+};
+
+exports.getColDefOverrides = getColDefOverrides;
+
+const mergeStateWithRowGroupingModel = rowGroupingModel => state => (0, _extends2.default)({}, state, {
+  rowGrouping: (0, _extends2.default)({}, state.rowGrouping, {
+    model: rowGroupingModel
+  })
+});
+
+exports.mergeStateWithRowGroupingModel = mergeStateWithRowGroupingModel;

package/node/hooks/features/rowGrouping/index.js

@@ -0,0 +1,51 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+var _exportNames = {
+  GRID_ROW_GROUPING_SINGLE_GROUPING_FIELD: true,
+  getRowGroupingFieldFromGroupingCriteria: true
+};
+Object.defineProperty(exports, "GRID_ROW_GROUPING_SINGLE_GROUPING_FIELD", {
+  enumerable: true,
+  get: function () {
+    return _gridRowGroupingUtils.GRID_ROW_GROUPING_SINGLE_GROUPING_FIELD;
+  }
+});
+Object.defineProperty(exports, "getRowGroupingFieldFromGroupingCriteria", {
+  enumerable: true,
+  get: function () {
+    return _gridRowGroupingUtils.getRowGroupingFieldFromGroupingCriteria;
+  }
+});
+
+var _gridRowGroupingSelector = require("./gridRowGroupingSelector");
+
+Object.keys(_gridRowGroupingSelector).forEach(function (key) {
+  if (key === "default" || key === "__esModule") return;
+  if (Object.prototype.hasOwnProperty.call(_exportNames, key)) return;
+  if (key in exports && exports[key] === _gridRowGroupingSelector[key]) return;
+  Object.defineProperty(exports, key, {
+    enumerable: true,
+    get: function () {
+      return _gridRowGroupingSelector[key];
+    }
+  });
+});
+
+var _gridRowGroupingInterfaces = require("./gridRowGroupingInterfaces");
+
+Object.keys(_gridRowGroupingInterfaces).forEach(function (key) {
+  if (key === "default" || key === "__esModule") return;
+  if (Object.prototype.hasOwnProperty.call(_exportNames, key)) return;
+  if (key in exports && exports[key] === _gridRowGroupingInterfaces[key]) return;
+  Object.defineProperty(exports, key, {
+    enumerable: true,
+    get: function () {
+      return _gridRowGroupingInterfaces[key];
+    }
+  });
+});
+
+var _gridRowGroupingUtils = require("./gridRowGroupingUtils");

package/node/hooks/features/rowGrouping/useGridRowGrouping.js

@@ -0,0 +1,355 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.useGridRowGrouping = exports.rowGroupingStateInitializer = void 0;
+
+var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
+
+var React = _interopRequireWildcard(require("react"));
+
+var _Divider = _interopRequireDefault(require("@mui/material/Divider"));
+
+var _xDataGrid = require("@mui/x-data-grid");
+
+var _internals = require("@mui/x-data-grid/internals");
+
+var _buildRowTree = require("../../../utils/tree/buildRowTree");
+
+var _gridRowGroupingSelector = require("./gridRowGroupingSelector");
+
+var _gridRowGroupingUtils = require("./gridRowGroupingUtils");
+
+var _sortRowTree = require("../../../utils/tree/sortRowTree");
+
+var _GridRowGroupableColumnMenuItems = require("../../../components/GridRowGroupableColumnMenuItems");
+
+var _GridRowGroupingColumnMenuItems = require("../../../components/GridRowGroupingColumnMenuItems");
+
+var _jsxRuntime = require("react/jsx-runtime");
+
+function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function (nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
+
+function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
+
+const rowGroupingStateInitializer = (state, props) => {
+  var _ref, _props$rowGroupingMod, _props$initialState, _props$initialState$r;
+
+  return (0, _extends2.default)({}, state, {
+    rowGrouping: {
+      model: (_ref = (_props$rowGroupingMod = props.rowGroupingModel) != null ? _props$rowGroupingMod : (_props$initialState = props.initialState) == null ? void 0 : (_props$initialState$r = _props$initialState.rowGrouping) == null ? void 0 : _props$initialState$r.model) != null ? _ref : []
+    }
+  });
+};
+/**
+ * Only available in DataGridPro
+ * @requires useGridColumns (state, method) - can be after, async only
+ * @requires useGridRows (state, method) - can be after, async only
+ * @requires useGridParamsApi (method) - can be after, async only
+ * TODO: Move the the Premium plan once available and remove the `experimentalFeatures.rowGrouping` flag
+ */
+
+
+exports.rowGroupingStateInitializer = rowGroupingStateInitializer;
+
+const useGridRowGrouping = (apiRef, props) => {
+  apiRef.current.unstable_updateControlState({
+    stateId: 'rowGrouping',
+    propModel: props.rowGroupingModel,
+    propOnChange: props.onRowGroupingModelChange,
+    stateSelector: _gridRowGroupingSelector.gridRowGroupingModelSelector,
+    changeEvent: _xDataGrid.GridEvents.rowGroupingModelChange
+  });
+  /**
+   * ROW GROUPING
+   */
+  // Tracks the model on the last pre-processing to check if we need to re-build the grouping columns when the grid upserts a column.
+
+  const sanitizedModelOnLastRowPreProcessing = React.useRef([]);
+  const updateRowGrouping = React.useCallback(() => {
+    const groupRows = params => {
+      const rowGroupingModel = (0, _gridRowGroupingSelector.gridRowGroupingSanitizedModelSelector)(apiRef);
+      const columnsLookup = (0, _xDataGrid.gridColumnLookupSelector)(apiRef);
+      sanitizedModelOnLastRowPreProcessing.current = rowGroupingModel;
+
+      if (props.disableRowGrouping || rowGroupingModel.length === 0) {
+        return null;
+      }
+
+      const distinctValues = Object.fromEntries(rowGroupingModel.map(groupingField => [groupingField, {
+        lookup: {},
+        list: []
+      }]));
+
+      const getCellGroupingCriteria = ({
+        row,
+        id,
+        colDef
+      }) => {
+        let key;
+
+        if (colDef.groupingValueGetter) {
+          const groupingValueGetterParams = {
+            colDef,
+            field: colDef.field,
+            value: row[colDef.field],
+            id,
+            row,
+            rowNode: {
+              isAutoGenerated: false,
+              id
+            }
+          };
+          key = colDef.groupingValueGetter(groupingValueGetterParams);
+        } else {
+          key = row[colDef.field];
+        }
+
+        return {
+          key,
+          field: colDef.field
+        };
+      };
+
+      params.ids.forEach(rowId => {
+        const row = params.idRowsLookup[rowId];
+        rowGroupingModel.forEach(groupingCriteria => {
+          const {
+            key
+          } = getCellGroupingCriteria({
+            row,
+            id: rowId,
+            colDef: columnsLookup[groupingCriteria]
+          });
+          const groupingFieldsDistinctKeys = distinctValues[groupingCriteria];
+
+          if (key != null && !groupingFieldsDistinctKeys.lookup[key.toString()]) {
+            groupingFieldsDistinctKeys.lookup[key.toString()] = true;
+            groupingFieldsDistinctKeys.list.push(key);
+          }
+        });
+      });
+      const rows = params.ids.map(rowId => {
+        const row = params.idRowsLookup[rowId];
+        const parentPath = rowGroupingModel.map(groupingField => getCellGroupingCriteria({
+          row,
+          id: rowId,
+          colDef: columnsLookup[groupingField]
+        })).filter(cell => cell.key != null);
+        const leafGroupingCriteria = {
+          key: rowId.toString(),
+          field: null
+        };
+        return {
+          path: [...parentPath, leafGroupingCriteria],
+          id: rowId
+        };
+      });
+      return (0, _buildRowTree.buildRowTree)((0, _extends2.default)({}, params, {
+        rows,
+        defaultGroupingExpansionDepth: props.defaultGroupingExpansionDepth,
+        isGroupExpandedByDefault: props.isGroupExpandedByDefault,
+        groupingName: _gridRowGroupingUtils.GROUPING_COLUMNS_FEATURE_NAME
+      }));
+    };
+
+    return apiRef.current.unstable_registerRowGroupsBuilder('rowGrouping', groupRows);
+  }, [apiRef, props.defaultGroupingExpansionDepth, props.isGroupExpandedByDefault, props.disableRowGrouping]);
+  (0, _xDataGrid.useFirstRender)(() => {
+    updateRowGrouping();
+  });
+  const isFirstRender = React.useRef(true);
+  React.useEffect(() => {
+    if (isFirstRender.current) {
+      isFirstRender.current = false;
+      return () => {};
+    }
+
+    return updateRowGrouping();
+  }, [updateRowGrouping]);
+  /**
+   * PRE-PROCESSING
+   */
+
+  const addColumnMenuButtons = React.useCallback((initialValue, columns) => {
+    if (props.disableRowGrouping) {
+      return initialValue;
+    }
+
+    let menuItems;
+
+    if ((0, _gridRowGroupingUtils.isGroupingColumn)(columns.field)) {
+      menuItems = /*#__PURE__*/(0, _jsxRuntime.jsx)(_GridRowGroupingColumnMenuItems.GridRowGroupingColumnMenuItems, {});
+    } else if (columns.groupable) {
+      menuItems = /*#__PURE__*/(0, _jsxRuntime.jsx)(_GridRowGroupableColumnMenuItems.GridRowGroupableColumnMenuItems, {});
+    } else {
+      menuItems = null;
+    }
+
+    if (menuItems == null) {
+      return initialValue;
+    }
+
+    return [...initialValue, /*#__PURE__*/(0, _jsxRuntime.jsx)(_Divider.default, {}), menuItems];
+  }, [props.disableRowGrouping]);
+  const filteringMethod = React.useCallback(params => {
+    const rowTree = (0, _xDataGrid.gridRowTreeSelector)(apiRef);
+    return (0, _gridRowGroupingUtils.filterRowTreeFromGroupingColumns)({
+      rowTree,
+      isRowMatchingFilters: params.isRowMatchingFilters
+    });
+  }, [apiRef]);
+  const sortingMethod = React.useCallback(params => {
+    const rowTree = (0, _xDataGrid.gridRowTreeSelector)(apiRef);
+    const rowIds = (0, _xDataGrid.gridRowIdsSelector)(apiRef);
+    return (0, _sortRowTree.sortRowTree)({
+      rowTree,
+      rowIds,
+      sortRowList: params.sortRowList,
+      disableChildrenSorting: false
+    });
+  }, [apiRef]);
+  (0, _internals.useGridRegisterPreProcessor)(apiRef, 'columnMenu', addColumnMenuButtons);
+  (0, _internals.useGridRegisterFilteringMethod)(apiRef, _gridRowGroupingUtils.GROUPING_COLUMNS_FEATURE_NAME, filteringMethod);
+  (0, _internals.useGridRegisterSortingMethod)(apiRef, _gridRowGroupingUtils.GROUPING_COLUMNS_FEATURE_NAME, sortingMethod);
+  /**
+   * API METHODS
+   */
+
+  const setRowGroupingModel = React.useCallback(model => {
+    const currentModel = (0, _gridRowGroupingSelector.gridRowGroupingModelSelector)(apiRef);
+
+    if (currentModel !== model) {
+      apiRef.current.setState((0, _gridRowGroupingUtils.mergeStateWithRowGroupingModel)(model));
+      updateRowGrouping();
+      apiRef.current.forceUpdate();
+    }
+  }, [apiRef, updateRowGrouping]);
+  const addRowGroupingCriteria = React.useCallback((field, groupingIndex) => {
+    const currentModel = (0, _gridRowGroupingSelector.gridRowGroupingModelSelector)(apiRef);
+
+    if (currentModel.includes(field)) {
+      return;
+    }
+
+    const cleanGroupingIndex = groupingIndex != null ? groupingIndex : currentModel.length;
+    const updatedModel = [...currentModel.slice(0, cleanGroupingIndex), field, ...currentModel.slice(cleanGroupingIndex)];
+    apiRef.current.setRowGroupingModel(updatedModel);
+  }, [apiRef]);
+  const removeRowGroupingCriteria = React.useCallback(field => {
+    const currentModel = (0, _gridRowGroupingSelector.gridRowGroupingModelSelector)(apiRef);
+
+    if (!currentModel.includes(field)) {
+      return;
+    }
+
+    apiRef.current.setRowGroupingModel(currentModel.filter(el => el !== field));
+  }, [apiRef]);
+  const setRowGroupingCriteriaIndex = React.useCallback((field, targetIndex) => {
+    const currentModel = (0, _gridRowGroupingSelector.gridRowGroupingModelSelector)(apiRef);
+    const currentTargetIndex = currentModel.indexOf(field);
+
+    if (currentTargetIndex === -1) {
+      return;
+    }
+
+    const updatedModel = [...currentModel];
+    updatedModel.splice(targetIndex, 0, updatedModel.splice(currentTargetIndex, 1)[0]);
+    apiRef.current.setRowGroupingModel(updatedModel);
+  }, [apiRef]);
+  const rowGroupingApi = {
+    setRowGroupingModel,
+    addRowGroupingCriteria,
+    removeRowGroupingCriteria,
+    setRowGroupingCriteriaIndex
+  };
+  (0, _xDataGrid.useGridApiMethod)(apiRef, rowGroupingApi, 'GridRowGroupingApi');
+  /**
+   * PRE-PROCESSING
+   */
+
+  const stateExportPreProcessing = React.useCallback(prevState => {
+    if (props.disableRowGrouping) {
+      return prevState;
+    }
+
+    const rowGroupingModelToExport = (0, _gridRowGroupingSelector.gridRowGroupingModelSelector)(apiRef);
+
+    if (rowGroupingModelToExport.length === 0) {
+      return prevState;
+    }
+
+    return (0, _extends2.default)({}, prevState, {
+      rowGrouping: {
+        model: rowGroupingModelToExport
+      }
+    });
+  }, [apiRef, props.disableRowGrouping]);
+  const stateRestorePreProcessing = React.useCallback((params, context) => {
+    var _context$stateToResto;
+
+    if (props.disableRowGrouping) {
+      return params;
+    }
+
+    const rowGroupingModel = (_context$stateToResto = context.stateToRestore.rowGrouping) == null ? void 0 : _context$stateToResto.model;
+
+    if (rowGroupingModel != null) {
+      apiRef.current.setState((0, _gridRowGroupingUtils.mergeStateWithRowGroupingModel)(rowGroupingModel));
+    }
+
+    return params;
+  }, [apiRef, props.disableRowGrouping]);
+  (0, _internals.useGridRegisterPreProcessor)(apiRef, 'exportState', stateExportPreProcessing);
+  (0, _internals.useGridRegisterPreProcessor)(apiRef, 'restoreState', stateRestorePreProcessing);
+  /**
+   * EVENTS
+   */
+
+  const handleCellKeyDown = React.useCallback((params, event) => {
+    const cellParams = apiRef.current.getCellParams(params.id, params.field);
+
+    if ((0, _gridRowGroupingUtils.isGroupingColumn)(cellParams.field) && event.key === ' ' && !event.shiftKey) {
+      var _gridFilteredDescenda;
+
+      event.stopPropagation();
+      event.preventDefault();
+      const filteredDescendantCount = (_gridFilteredDescenda = (0, _xDataGrid.gridFilteredDescendantCountLookupSelector)(apiRef)[params.id]) != null ? _gridFilteredDescenda : 0;
+      const isOnGroupingCell = props.rowGroupingColumnMode === 'single' || (0, _gridRowGroupingUtils.getRowGroupingFieldFromGroupingCriteria)(params.rowNode.groupingField) === params.field;
+
+      if (!isOnGroupingCell || filteredDescendantCount === 0) {
+        return;
+      }
+
+      apiRef.current.setRowChildrenExpansion(params.id, !params.rowNode.childrenExpanded);
+    }
+  }, [apiRef, props.rowGroupingColumnMode]);
+  const checkGroupingColumnsModelDiff = React.useCallback(() => {
+    const rowGroupingModel = (0, _gridRowGroupingSelector.gridRowGroupingSanitizedModelSelector)(apiRef);
+    const lastGroupingColumnsModelApplied = sanitizedModelOnLastRowPreProcessing.current;
+
+    if (!(0, _internals.isDeepEqual)(lastGroupingColumnsModelApplied, rowGroupingModel)) {
+      sanitizedModelOnLastRowPreProcessing.current = rowGroupingModel; // Refresh the column pre-processing
+
+      apiRef.current.updateColumns([]);
+      updateRowGrouping();
+    }
+  }, [apiRef, updateRowGrouping]);
+  (0, _xDataGrid.useGridApiEventHandler)(apiRef, _xDataGrid.GridEvents.cellKeyDown, handleCellKeyDown);
+  (0, _xDataGrid.useGridApiEventHandler)(apiRef, _xDataGrid.GridEvents.columnsChange, checkGroupingColumnsModelDiff);
+  (0, _xDataGrid.useGridApiEventHandler)(apiRef, _xDataGrid.GridEvents.rowGroupingModelChange, checkGroupingColumnsModelDiff);
+  /**
+   * EFFECTS
+   */
+
+  React.useEffect(() => {
+    if (props.rowGroupingModel !== undefined) {
+      apiRef.current.setRowGroupingModel(props.rowGroupingModel);
+    }
+  }, [apiRef, props.rowGroupingModel]);
+};
+
+exports.useGridRowGrouping = useGridRowGrouping;

package/node/hooks/features/rowGrouping/useGridRowGroupingPreProcessors.js

@@ -0,0 +1,94 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.useGridRowGroupingPreProcessors = void 0;
+
+var React = _interopRequireWildcard(require("react"));
+
+var _internals = require("@mui/x-data-grid/internals");
+
+var _gridRowGroupingSelector = require("./gridRowGroupingSelector");
+
+var _createGroupingColDef = require("./createGroupingColDef");
+
+var _gridRowGroupingUtils = require("./gridRowGroupingUtils");
+
+function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function (nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
+
+function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
+
+const useGridRowGroupingPreProcessors = (apiRef, props) => {
+  const getGroupingColDefs = React.useCallback(columnsState => {
+    if (props.disableRowGrouping) {
+      return [];
+    }
+
+    const groupingColDefProp = props.groupingColDef; // We can't use `gridGroupingRowsSanitizedModelSelector` here because the new columns are not in the state yet
+
+    const rowGroupingModel = (0, _gridRowGroupingSelector.gridRowGroupingModelSelector)(apiRef).filter(field => !!columnsState.lookup[field]);
+
+    if (rowGroupingModel.length === 0) {
+      return [];
+    }
+
+    switch (props.rowGroupingColumnMode) {
+      case 'single':
+        {
+          return [(0, _createGroupingColDef.createGroupingColDefForAllGroupingCriteria)({
+            apiRef,
+            rowGroupingModel,
+            colDefOverride: (0, _gridRowGroupingUtils.getColDefOverrides)(groupingColDefProp, rowGroupingModel),
+            columnsLookup: columnsState.lookup
+          })];
+        }
+
+      case 'multiple':
+        {
+          return rowGroupingModel.map(groupingCriteria => (0, _createGroupingColDef.createGroupingColDefForOneGroupingCriteria)({
+            groupingCriteria,
+            colDefOverride: (0, _gridRowGroupingUtils.getColDefOverrides)(groupingColDefProp, [groupingCriteria]),
+            groupedByColDef: columnsState.lookup[groupingCriteria],
+            columnsLookup: columnsState.lookup
+          }));
+        }
+
+      default:
+        {
+          return [];
+        }
+    }
+  }, [apiRef, props.groupingColDef, props.rowGroupingColumnMode, props.disableRowGrouping]);
+  const updateGroupingColumn = React.useCallback(columnsState => {
+    const groupingColDefs = getGroupingColDefs(columnsState);
+    let newColumnFields = [];
+    const newColumnsLookup = {}; // We only keep the non-grouping columns
+
+    columnsState.all.forEach(field => {
+      if (!(0, _gridRowGroupingUtils.isGroupingColumn)(field)) {
+        newColumnFields.push(field);
+        newColumnsLookup[field] = columnsState.lookup[field];
+      }
+    }); // We add the grouping column
+
+    groupingColDefs.forEach(groupingColDef => {
+      const matchingGroupingColDef = columnsState.lookup[groupingColDef.field];
+
+      if (matchingGroupingColDef) {
+        groupingColDef.width = matchingGroupingColDef.width;
+        groupingColDef.flex = matchingGroupingColDef.flex;
+      }
+
+      newColumnsLookup[groupingColDef.field] = groupingColDef;
+    });
+    const startIndex = newColumnFields[0] === '__check__' ? 1 : 0;
+    newColumnFields = [...newColumnFields.slice(0, startIndex), ...groupingColDefs.map(colDef => colDef.field), ...newColumnFields.slice(startIndex)];
+    columnsState.all = newColumnFields;
+    columnsState.lookup = newColumnsLookup;
+    return columnsState;
+  }, [getGroupingColDefs]);
+  (0, _internals.useGridRegisterPreProcessor)(apiRef, 'hydrateColumns', updateGroupingColumn);
+};
+
+exports.useGridRowGroupingPreProcessors = useGridRowGroupingPreProcessors;

package/node/hooks/features/treeData/gridTreeDataGroupColDef.js

@@ -0,0 +1,35 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.GRID_TREE_DATA_GROUPING_FIELD = exports.GRID_TREE_DATA_GROUPING_COL_DEF_FORCED_PROPERTIES = exports.GRID_TREE_DATA_GROUPING_COL_DEF = void 0;
+
+var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
+
+var _xDataGrid = require("@mui/x-data-grid");
+
+/**
+ * TODO: Add sorting and filtering on the value and the filteredDescendantCount
+ */
+const GRID_TREE_DATA_GROUPING_COL_DEF = (0, _extends2.default)({}, _xDataGrid.GRID_STRING_COL_DEF, {
+  type: 'treeDataGroup',
+  sortable: false,
+  filterable: false,
+  disableColumnMenu: true,
+  disableReorder: true,
+  align: 'left',
+  width: 200,
+  valueGetter: params => params.rowNode.groupingKey
+});
+exports.GRID_TREE_DATA_GROUPING_COL_DEF = GRID_TREE_DATA_GROUPING_COL_DEF;
+const GRID_TREE_DATA_GROUPING_FIELD = '__tree_data_group__';
+exports.GRID_TREE_DATA_GROUPING_FIELD = GRID_TREE_DATA_GROUPING_FIELD;
+const GRID_TREE_DATA_GROUPING_COL_DEF_FORCED_PROPERTIES = {
+  field: GRID_TREE_DATA_GROUPING_FIELD,
+  editable: false,
+  groupable: false
+};
+exports.GRID_TREE_DATA_GROUPING_COL_DEF_FORCED_PROPERTIES = GRID_TREE_DATA_GROUPING_COL_DEF_FORCED_PROPERTIES;