@mui/x-data-grid-pro/modern/hooks/features/treeData/useGridTreeData.js

The Pro plan edition of the Data Grid components (MUI X).

import * as React from 'react';
import { useGridApiEventHandler } from '@mui/x-data-grid';
import { GRID_TREE_DATA_GROUPING_FIELD } from './gridTreeDataGroupColDef';
export const useGridTreeData = (apiRef, props) => {
  /**
   * EVENTS
   */
  const handleCellKeyDown = React.useCallback((params, event) => {
    const cellParams = apiRef.current.getCellParams(params.id, params.field);
    if (cellParams.colDef.field === GRID_TREE_DATA_GROUPING_FIELD && event.key === ' ' && !event.shiftKey) {
      if (params.rowNode.type !== 'group') {
        return;
      }
      if (props.unstable_dataSource && !params.rowNode.childrenExpanded) {
        apiRef.current.unstable_dataSource.fetchRows(params.id);
        return;
      }
      apiRef.current.setRowChildrenExpansion(params.id, !params.rowNode.childrenExpanded);
    }
  }, [apiRef, props.unstable_dataSource]);
  useGridApiEventHandler(apiRef, 'cellKeyDown', handleCellKeyDown);
};