@mui/x-tree-view 7.0.0-alpha.7 → 7.0.0-alpha.9

package/internals/useTreeView/useTreeView.types.d.ts

@@ -2,14 +2,14 @@ import * as React from 'react';
 import { EventHandlers } from '@mui/base/utils';
 import type { TreeViewContextValue } from '../TreeViewProvider';
 import { TreeViewAnyPluginSignature, TreeViewPlugin, ConvertPluginsIntoSignatures, MergePluginsProperty, TreeViewInstance } from '../models';
-export type UseTreeViewParameters<TPlugins extends readonly TreeViewPlugin<TreeViewAnyPluginSignature>[]> = {
+export type UseTreeViewParameters<TPlugins extends readonly TreeViewPlugin<TreeViewAnyPluginSignature>[]> = UseTreeViewBaseParameters<TPlugins> & MergePluginsProperty<ConvertPluginsIntoSignatures<TPlugins>, 'params'>;
+export interface UseTreeViewBaseParameters<TPlugins extends readonly TreeViewPlugin<TreeViewAnyPluginSignature>[]> {
     rootRef?: React.Ref<HTMLUListElement> | undefined;
     plugins: TPlugins;
-} & MergePluginsProperty<ConvertPluginsIntoSignatures<TPlugins>, 'params'>;
-export type UseTreeViewDefaultizedParameters<TPlugins extends readonly TreeViewPlugin<TreeViewAnyPluginSignature>[]> = {
-    rootRef?: React.Ref<HTMLUListElement> | undefined;
-    plugins: TPlugins;
-} & MergePluginsProperty<ConvertPluginsIntoSignatures<TPlugins>, 'defaultizedParams'>;
+    slots: MergePluginsProperty<ConvertPluginsIntoSignatures<TPlugins>, 'slots'>;
+    slotProps: MergePluginsProperty<ConvertPluginsIntoSignatures<TPlugins>, 'slotProps'>;
+}
+export type UseTreeViewDefaultizedParameters<TPlugins extends readonly TreeViewPlugin<TreeViewAnyPluginSignature>[]> = UseTreeViewBaseParameters<TPlugins> & MergePluginsProperty<ConvertPluginsIntoSignatures<TPlugins>, 'defaultizedParams'>;
 export interface UseTreeViewRootSlotProps extends Pick<React.HTMLAttributes<HTMLUListElement>, 'onFocus' | 'onBlur' | 'onKeyDown' | 'id' | 'aria-activedescendant' | 'aria-multiselectable' | 'role' | 'tabIndex'> {
     ref: React.Ref<HTMLUListElement>;
 }

package/internals/useTreeView/useTreeViewModels.js

@@ -10,23 +10,22 @@ export const useTreeViewModels = (plugins, props) => {
     const initialState = {};
     plugins.forEach(plugin => {
       if (plugin.models) {
-        Object.entries(plugin.models).forEach(([modelName, model]) => {
+        Object.entries(plugin.models).forEach(([modelName, modelInitializer]) => {
           modelsRef.current[modelName] = {
-            controlledProp: model.controlledProp,
-            defaultProp: model.defaultProp,
-            isControlled: props[model.controlledProp] !== undefined
+            isControlled: props[modelName] !== undefined,
+            getDefaultValue: modelInitializer.getDefaultValue
           };
-          initialState[modelName] = props[model.defaultProp];
+          initialState[modelName] = modelInitializer.getDefaultValue(props);
         });
       }
     });
     return initialState;
   });
   const models = Object.fromEntries(Object.entries(modelsRef.current).map(([modelName, model]) => {
-    const value = model.isControlled ? props[model.controlledProp] : modelsState[modelName];
+    const value = model.isControlled ? props[modelName] : modelsState[modelName];
     return [modelName, {
       value,
-      setValue: newValue => {
+      setControlledValue: newValue => {
         if (!model.isControlled) {
           setModelsState(prevState => _extends({}, prevState, {
             [modelName]: newValue
@@ -40,19 +39,19 @@ export const useTreeViewModels = (plugins, props) => {
   /* eslint-disable react-hooks/rules-of-hooks, react-hooks/exhaustive-deps */
   if (process.env.NODE_ENV !== 'production') {
     Object.entries(modelsRef.current).forEach(([modelName, model]) => {
-      const controlled = props[model.controlledProp];
-      const defaultProp = props[model.defaultProp];
+      const controlled = props[modelName];
+      const newDefaultValue = model.getDefaultValue(props);
       React.useEffect(() => {
         if (model.isControlled !== (controlled !== undefined)) {
-          console.error([`MUI: A component is changing the ${model.isControlled ? '' : 'un'}controlled ${modelName} state of TreeView to be ${model.isControlled ? 'un' : ''}controlled.`, 'Elements should not switch from uncontrolled to controlled (or vice versa).', `Decide between using a controlled or uncontrolled ${modelName} ` + 'element for the lifetime of the component.', "The nature of the state is determined during the first render. It's considered controlled if the value is not `undefined`.", 'More info: https://fb.me/react-controlled-components'].join('\n'));
+          console.error([`MUI X: A component is changing the ${model.isControlled ? '' : 'un'}controlled ${modelName} state of TreeView to be ${model.isControlled ? 'un' : ''}controlled.`, 'Elements should not switch from uncontrolled to controlled (or vice versa).', `Decide between using a controlled or uncontrolled ${modelName} ` + 'element for the lifetime of the component.', "The nature of the state is determined during the first render. It's considered controlled if the value is not `undefined`.", 'More info: https://fb.me/react-controlled-components'].join('\n'));
         }
       }, [controlled]);
       const {
         current: defaultValue
-      } = React.useRef(defaultProp);
+      } = React.useRef(newDefaultValue);
       React.useEffect(() => {
-        if (!model.isControlled && defaultValue !== defaultProp) {
-          console.error([`MUI: A component is changing the default ${modelName} state of an uncontrolled TreeView after being initialized. ` + `To suppress this warning opt to use a controlled TreeView.`].join('\n'));
+        if (!model.isControlled && defaultValue !== newDefaultValue) {
+          console.error([`MUI X: A component is changing the default ${modelName} state of an uncontrolled TreeView after being initialized. ` + `To suppress this warning opt to use a controlled TreeView.`].join('\n'));
         }
       }, [JSON.stringify(defaultValue)]);
     });

package/internals/utils/extractPluginParamsFromProps.d.ts

@@ -0,0 +1,16 @@
+import * as React from 'react';
+import { ConvertPluginsIntoSignatures, MergePluginsProperty, TreeViewPlugin } from '../models';
+import { UseTreeViewBaseParameters } from '../useTreeView/useTreeView.types';
+export declare const extractPluginParamsFromProps: <TPlugins extends readonly TreeViewPlugin<any>[], TSlots extends MergePluginsProperty<TPlugins, "slots">, TSlotProps extends MergePluginsProperty<TPlugins, "slotProps">, TProps extends {
+    slots?: TSlots | undefined;
+    slotProps?: TSlotProps | undefined;
+}>({ props: { slots, slotProps, ...props }, plugins, rootRef, }: {
+    props: TProps;
+    plugins: TPlugins;
+    rootRef?: React.Ref<HTMLUListElement> | undefined;
+}) => {
+    pluginParams: UseTreeViewBaseParameters<TPlugins> & MergePluginsProperty<ConvertPluginsIntoSignatures<TPlugins>, "params">;
+    slots: TSlots | undefined;
+    slotProps: TSlotProps | undefined;
+    otherProps: Omit<TProps, keyof MergePluginsProperty<ConvertPluginsIntoSignatures<TPlugins>, "params">>;
+};

package/internals/utils/extractPluginParamsFromProps.js

@@ -0,0 +1,38 @@
+import _objectWithoutPropertiesLoose from "@babel/runtime/helpers/esm/objectWithoutPropertiesLoose";
+const _excluded = ["slots", "slotProps"];
+export const extractPluginParamsFromProps = _ref => {
+  let {
+      props: {
+        slots,
+        slotProps
+      },
+      plugins,
+      rootRef
+    } = _ref,
+    props = _objectWithoutPropertiesLoose(_ref.props, _excluded);
+  const paramsLookup = {};
+  plugins.forEach(plugin => {
+    Object.assign(paramsLookup, plugin.params);
+  });
+  const pluginParams = {
+    plugins,
+    rootRef,
+    slots: slots != null ? slots : {},
+    slotProps: slotProps != null ? slotProps : {}
+  };
+  const otherProps = {};
+  Object.keys(props).forEach(propName => {
+    const prop = props[propName];
+    if (paramsLookup[propName]) {
+      pluginParams[propName] = prop;
+    } else {
+      otherProps[propName] = prop;
+    }
+  });
+  return {
+    pluginParams,
+    slots,
+    slotProps,
+    otherProps
+  };
+};

package/legacy/RichTreeView/RichTreeView.js

@@ -1,6 +1,4 @@
-import _objectWithoutProperties from "@babel/runtime/helpers/esm/objectWithoutProperties";
 import _extends from "@babel/runtime/helpers/esm/extends";
-var _excluded = ["disabledItemsFocusable", "expandedNodes", "defaultExpandedNodes", "onExpandedNodesChange", "onNodeExpansionToggle", "onNodeFocus", "disableSelection", "defaultSelectedNodes", "selectedNodes", "multiSelect", "onSelectedNodesChange", "onNodeSelectionToggle", "id", "defaultCollapseIcon", "defaultEndIcon", "defaultExpandIcon", "defaultParentIcon", "items", "getItemId", "getItemLabel", "isItemDisabled", "slots", "slotProps"];
 import * as React from 'react';
 import PropTypes from 'prop-types';
 import { styled, useThemeProps } from '@mui/material/styles';
@@ -12,6 +10,7 @@ import { TreeViewProvider } from '../internals/TreeViewProvider';
 import { DEFAULT_TREE_VIEW_PLUGINS } from '../internals/plugins';
 import { TreeItem } from '../TreeItem';
 import { buildWarning } from '../internals/utils/warning';
+import { extractPluginParamsFromProps } from '../internals/utils/extractPluginParamsFromProps';
 import { jsx as _jsx } from "react/jsx-runtime";
 var useUtilityClasses = function useUtilityClasses(ownerState) {
   var classes = ownerState.classes;
@@ -58,7 +57,7 @@ function WrappedTreeItem(_ref) {
     children: children
   }));
 }
-var childrenWarning = buildWarning(['MUI: The `RichTreeView` component does not support JSX children.', 'If you want to add items, you need to use the `items` prop', 'Check the documentation for more details: https://next.mui.com/x/react-tree-view/rich-tree-view/items/']);
+var childrenWarning = buildWarning(['MUI X: The `RichTreeView` component does not support JSX children.', 'If you want to add items, you need to use the `items` prop', 'Check the documentation for more details: https://next.mui.com/x/react-tree-view/rich-tree-view/items/']);
 
 /**
  *
@@ -76,61 +75,21 @@ var RichTreeView = /*#__PURE__*/React.forwardRef(function RichTreeView(inProps,
     props: inProps,
     name: 'MuiRichTreeView'
   });
-  var _ref2 = props,
-    disabledItemsFocusable = _ref2.disabledItemsFocusable,
-    expandedNodes = _ref2.expandedNodes,
-    defaultExpandedNodes = _ref2.defaultExpandedNodes,
-    onExpandedNodesChange = _ref2.onExpandedNodesChange,
-    onNodeExpansionToggle = _ref2.onNodeExpansionToggle,
-    onNodeFocus = _ref2.onNodeFocus,
-    disableSelection = _ref2.disableSelection,
-    defaultSelectedNodes = _ref2.defaultSelectedNodes,
-    selectedNodes = _ref2.selectedNodes,
-    multiSelect = _ref2.multiSelect,
-    onSelectedNodesChange = _ref2.onSelectedNodesChange,
-    onNodeSelectionToggle = _ref2.onNodeSelectionToggle,
-    treeId = _ref2.id,
-    defaultCollapseIcon = _ref2.defaultCollapseIcon,
-    defaultEndIcon = _ref2.defaultEndIcon,
-    defaultExpandIcon = _ref2.defaultExpandIcon,
-    defaultParentIcon = _ref2.defaultParentIcon,
-    items = _ref2.items,
-    getItemId = _ref2.getItemId,
-    getItemLabel = _ref2.getItemLabel,
-    isItemDisabled = _ref2.isItemDisabled,
-    slots = _ref2.slots,
-    slotProps = _ref2.slotProps,
-    other = _objectWithoutProperties(_ref2, _excluded);
   if (process.env.NODE_ENV !== 'production') {
     if (props.children != null) {
       childrenWarning();
     }
   }
-  var _useTreeView = useTreeView({
-      disabledItemsFocusable: disabledItemsFocusable,
-      expandedNodes: expandedNodes,
-      defaultExpandedNodes: defaultExpandedNodes,
-      onExpandedNodesChange: onExpandedNodesChange,
-      onNodeExpansionToggle: onNodeExpansionToggle,
-      onNodeFocus: onNodeFocus,
-      disableSelection: disableSelection,
-      defaultSelectedNodes: defaultSelectedNodes,
-      selectedNodes: selectedNodes,
-      multiSelect: multiSelect,
-      onSelectedNodesChange: onSelectedNodesChange,
-      onNodeSelectionToggle: onNodeSelectionToggle,
-      id: treeId,
-      defaultCollapseIcon: defaultCollapseIcon,
-      defaultEndIcon: defaultEndIcon,
-      defaultExpandIcon: defaultExpandIcon,
-      defaultParentIcon: defaultParentIcon,
-      items: items,
-      getItemId: getItemId,
-      getItemLabel: getItemLabel,
-      isItemDisabled: isItemDisabled,
+  var _extractPluginParamsF = extractPluginParamsFromProps({
+      props: props,
       plugins: DEFAULT_TREE_VIEW_PLUGINS,
       rootRef: ref
     }),
+    pluginParams = _extractPluginParamsF.pluginParams,
+    slots = _extractPluginParamsF.slots,
+    slotProps = _extractPluginParamsF.slotProps,
+    otherProps = _extractPluginParamsF.otherProps;
+  var _useTreeView = useTreeView(pluginParams),
     getRootProps = _useTreeView.getRootProps,
     contextValue = _useTreeView.contextValue,
     instance = _useTreeView.instance;
@@ -139,17 +98,17 @@ var RichTreeView = /*#__PURE__*/React.forwardRef(function RichTreeView(inProps,
   var rootProps = useSlotProps({
     elementType: Root,
     externalSlotProps: slotProps == null ? void 0 : slotProps.root,
-    externalForwardedProps: other,
+    externalForwardedProps: otherProps,
     className: classes.root,
     getSlotProps: getRootProps,
     ownerState: props
   });
   var nodesToRender = instance.getNodesToRender();
-  var renderNode = function renderNode(_ref3) {
-    var label = _ref3.label,
-      nodeId = _ref3.nodeId,
-      id = _ref3.id,
-      children = _ref3.children;
+  var renderNode = function renderNode(_ref2) {
+    var label = _ref2.label,
+      nodeId = _ref2.nodeId,
+      id = _ref2.id,
+      children = _ref2.children;
     return /*#__PURE__*/_jsx(WrappedTreeItem, {
       slots: slots,
       slotProps: slotProps,
@@ -175,34 +134,13 @@ process.env.NODE_ENV !== "production" ? RichTreeView.propTypes = {
    * Override or extend the styles applied to the component.
    */
   classes: PropTypes.object,
-  /**
-   * className applied to the root element.
-   */
   className: PropTypes.string,
-  /**
-   * The default icon used to collapse the node.
-   */
-  defaultCollapseIcon: PropTypes.node,
-  /**
-   * The default icon displayed next to a end node. This is applied to all
-   * tree nodes and can be overridden by the TreeItem `icon` prop.
-   */
-  defaultEndIcon: PropTypes.node,
   /**
    * Expanded node ids.
    * Used when the item's expansion is not controlled.
    * @default []
    */
   defaultExpandedNodes: PropTypes.arrayOf(PropTypes.string),
-  /**
-   * The default icon used to expand the node.
-   */
-  defaultExpandIcon: PropTypes.node,
-  /**
-   * The default icon displayed next to a parent node. This is applied to all
-   * parent nodes and can be overridden by the TreeItem `icon` prop.
-   */
-  defaultParentIcon: PropTypes.node,
   /**
    * Selected node ids. (Uncontrolled)
    * When `multiSelect` is true this takes an array of strings; when false (default) a string.

package/legacy/SimpleTreeView/SimpleTreeView.js

@@ -1,6 +1,4 @@
 import _extends from "@babel/runtime/helpers/esm/extends";
-import _objectWithoutProperties from "@babel/runtime/helpers/esm/objectWithoutProperties";
-var _excluded = ["disabledItemsFocusable", "expandedNodes", "defaultExpandedNodes", "onExpandedNodesChange", "onNodeExpansionToggle", "onNodeFocus", "disableSelection", "defaultSelectedNodes", "selectedNodes", "multiSelect", "onSelectedNodesChange", "onNodeSelectionToggle", "id", "defaultCollapseIcon", "defaultEndIcon", "defaultExpandIcon", "defaultParentIcon", "children", "slots", "slotProps"];
 import * as React from 'react';
 import PropTypes from 'prop-types';
 import { styled, useThemeProps } from '@mui/material/styles';
@@ -11,6 +9,7 @@ import { useTreeView } from '../internals/useTreeView';
 import { TreeViewProvider } from '../internals/TreeViewProvider';
 import { SIMPLE_TREE_VIEW_PLUGINS } from './SimpleTreeView.plugins';
 import { buildWarning } from '../internals/utils/warning';
+import { extractPluginParamsFromProps } from '../internals/utils/extractPluginParamsFromProps';
 import { jsx as _jsx } from "react/jsx-runtime";
 var useUtilityClasses = function useUtilityClasses(ownerState) {
   var classes = ownerState.classes;
@@ -32,7 +31,7 @@ export var SimpleTreeViewRoot = styled('ul', {
   outline: 0
 });
 var EMPTY_ITEMS = [];
-var itemsPropWarning = buildWarning(['MUI: The `SimpleTreeView` component does not support the `items` prop.', 'If you want to add items, you need to pass them as JSX children.', 'Check the documentation for more details: https://next.mui.com/x/react-tree-view/simple-tree-view/items/']);
+var itemsPropWarning = buildWarning(['MUI X: The `SimpleTreeView` component does not support the `items` prop.', 'If you want to add items, you need to pass them as JSX children.', 'Check the documentation for more details: https://next.mui.com/x/react-tree-view/simple-tree-view/items/']);
 
 /**
  *
@@ -51,72 +50,38 @@ var SimpleTreeView = /*#__PURE__*/React.forwardRef(function SimpleTreeView(inPro
     name: 'MuiSimpleTreeView'
   });
   var ownerState = props;
-  var _ref = props,
-    disabledItemsFocusable = _ref.disabledItemsFocusable,
-    expandedNodes = _ref.expandedNodes,
-    defaultExpandedNodes = _ref.defaultExpandedNodes,
-    onExpandedNodesChange = _ref.onExpandedNodesChange,
-    onNodeExpansionToggle = _ref.onNodeExpansionToggle,
-    onNodeFocus = _ref.onNodeFocus,
-    disableSelection = _ref.disableSelection,
-    defaultSelectedNodes = _ref.defaultSelectedNodes,
-    selectedNodes = _ref.selectedNodes,
-    multiSelect = _ref.multiSelect,
-    onSelectedNodesChange = _ref.onSelectedNodesChange,
-    onNodeSelectionToggle = _ref.onNodeSelectionToggle,
-    id = _ref.id,
-    defaultCollapseIcon = _ref.defaultCollapseIcon,
-    defaultEndIcon = _ref.defaultEndIcon,
-    defaultExpandIcon = _ref.defaultExpandIcon,
-    defaultParentIcon = _ref.defaultParentIcon,
-    children = _ref.children,
-    slots = _ref.slots,
-    slotProps = _ref.slotProps,
-    other = _objectWithoutProperties(_ref, _excluded);
   if (process.env.NODE_ENV !== 'production') {
     if (props.items != null) {
       itemsPropWarning();
     }
   }
-  var _useTreeView = useTreeView({
-      disabledItemsFocusable: disabledItemsFocusable,
-      expandedNodes: expandedNodes,
-      defaultExpandedNodes: defaultExpandedNodes,
-      onExpandedNodesChange: onExpandedNodesChange,
-      onNodeExpansionToggle: onNodeExpansionToggle,
-      onNodeFocus: onNodeFocus,
-      disableSelection: disableSelection,
-      defaultSelectedNodes: defaultSelectedNodes,
-      selectedNodes: selectedNodes,
-      multiSelect: multiSelect,
-      onSelectedNodesChange: onSelectedNodesChange,
-      onNodeSelectionToggle: onNodeSelectionToggle,
-      id: id,
-      defaultCollapseIcon: defaultCollapseIcon,
-      defaultEndIcon: defaultEndIcon,
-      defaultExpandIcon: defaultExpandIcon,
-      defaultParentIcon: defaultParentIcon,
-      items: EMPTY_ITEMS,
+  var _extractPluginParamsF = extractPluginParamsFromProps({
+      props: _extends({}, props, {
+        items: EMPTY_ITEMS
+      }),
       plugins: SIMPLE_TREE_VIEW_PLUGINS,
       rootRef: ref
     }),
+    pluginParams = _extractPluginParamsF.pluginParams,
+    slots = _extractPluginParamsF.slots,
+    slotProps = _extractPluginParamsF.slotProps,
+    otherProps = _extractPluginParamsF.otherProps;
+  var _useTreeView = useTreeView(pluginParams),
     getRootProps = _useTreeView.getRootProps,
     contextValue = _useTreeView.contextValue;
   var classes = useUtilityClasses(props);
   var Root = (_slots$root = slots == null ? void 0 : slots.root) != null ? _slots$root : SimpleTreeViewRoot;
   var rootProps = useSlotProps({
     elementType: Root,
-    externalSlotProps: {},
-    externalForwardedProps: other,
+    externalSlotProps: slotProps == null ? void 0 : slotProps.root,
+    externalForwardedProps: otherProps,
     className: classes.root,
     getSlotProps: getRootProps,
     ownerState: ownerState
   });
   return /*#__PURE__*/_jsx(TreeViewProvider, {
     value: contextValue,
-    children: /*#__PURE__*/_jsx(Root, _extends({}, rootProps, {
-      children: children
-    }))
+    children: /*#__PURE__*/_jsx(Root, _extends({}, rootProps))
   });
 });
 process.env.NODE_ENV !== "production" ? SimpleTreeView.propTypes = {
@@ -132,34 +97,13 @@ process.env.NODE_ENV !== "production" ? SimpleTreeView.propTypes = {
    * Override or extend the styles applied to the component.
    */
   classes: PropTypes.object,
-  /**
-   * className applied to the root element.
-   */
   className: PropTypes.string,
-  /**
-   * The default icon used to collapse the node.
-   */
-  defaultCollapseIcon: PropTypes.node,
-  /**
-   * The default icon displayed next to a end node. This is applied to all
-   * tree nodes and can be overridden by the TreeItem `icon` prop.
-   */
-  defaultEndIcon: PropTypes.node,
   /**
    * Expanded node ids.
    * Used when the item's expansion is not controlled.
    * @default []
    */
   defaultExpandedNodes: PropTypes.arrayOf(PropTypes.string),
-  /**
-   * The default icon used to expand the node.
-   */
-  defaultExpandIcon: PropTypes.node,
-  /**
-   * The default icon displayed next to a parent node. This is applied to all
-   * parent nodes and can be overridden by the TreeItem `icon` prop.
-   */
-  defaultParentIcon: PropTypes.node,
   /**
    * Selected node ids. (Uncontrolled)
    * When `multiSelect` is true this takes an array of strings; when false (default) a string.

package/legacy/TreeItem/TreeItem.js

@@ -1,11 +1,15 @@
 import _objectWithoutProperties from "@babel/runtime/helpers/esm/objectWithoutProperties";
 import _extends from "@babel/runtime/helpers/esm/extends";
 import _defineProperty from "@babel/runtime/helpers/esm/defineProperty";
-var _excluded = ["children", "className", "collapseIcon", "ContentComponent", "ContentProps", "endIcon", "expandIcon", "icon", "nodeId", "id", "label", "onClick", "onMouseDown", "TransitionComponent", "TransitionProps"];
+var _excluded = ["children", "className", "slots", "slotProps", "ContentComponent", "ContentProps", "nodeId", "id", "label", "onClick", "onMouseDown", "TransitionComponent", "TransitionProps"],
+  _excluded2 = ["ownerState"],
+  _excluded3 = ["ownerState"],
+  _excluded4 = ["ownerState"];
 import * as React from 'react';
 import PropTypes from 'prop-types';
 import clsx from 'clsx';
 import Collapse from '@mui/material/Collapse';
+import { resolveComponentProps, useSlotProps } from '@mui/base/utils';
 import { alpha, styled, useThemeProps } from '@mui/material/styles';
 import unsupportedProp from '@mui/utils/unsupportedProp';
 import elementTypeAcceptingRef from '@mui/utils/elementTypeAcceptingRef';
@@ -13,6 +17,7 @@ import { unstable_composeClasses as composeClasses } from '@mui/base';
 import { TreeItemContent } from './TreeItemContent';
 import { treeItemClasses, getTreeItemUtilityClass } from './treeItemClasses';
 import { useTreeViewContext } from '../internals/TreeViewProvider/useTreeViewContext';
+import { TreeViewCollapseIcon, TreeViewExpandIcon } from '../icons';
 import { jsx as _jsx } from "react/jsx-runtime";
 import { jsxs as _jsxs } from "react/jsx-runtime";
 var useUtilityClasses = function useUtilityClasses(ownerState) {
@@ -124,10 +129,11 @@ var TreeItemGroup = styled(Collapse, {
  * - [TreeItem API](https://mui.com/x/api/tree-view/tree-item/)
  */
 export var TreeItem = /*#__PURE__*/React.forwardRef(function TreeItem(inProps, inRef) {
+  var _ref5, _inSlots$expandIcon, _ref6, _inSlots$collapseIcon, _inSlots$endIcon;
   var _useTreeViewContext = useTreeViewContext(),
     contextIcons = _useTreeViewContext.icons,
     runItemPlugins = _useTreeViewContext.runItemPlugins,
-    multiSelect = _useTreeViewContext.multiSelect,
+    multiSelect = _useTreeViewContext.selection.multiSelect,
     disabledItemsFocusable = _useTreeViewContext.disabledItemsFocusable,
     instance = _useTreeViewContext.instance;
   var inPropsWithTheme = useThemeProps({
@@ -143,13 +149,11 @@ export var TreeItem = /*#__PURE__*/React.forwardRef(function TreeItem(inProps, i
     wrapItem = _runItemPlugins.wrapItem;
   var children = props.children,
     className = props.className,
-    collapseIcon = props.collapseIcon,
+    inSlots = props.slots,
+    inSlotProps = props.slotProps,
     _props$ContentCompone = props.ContentComponent,
     ContentComponent = _props$ContentCompone === void 0 ? TreeItemContent : _props$ContentCompone,
     ContentProps = props.ContentProps,
-    endIcon = props.endIcon,
-    expandIcon = props.expandIcon,
-    icon = props.icon,
     nodeId = props.nodeId,
     id = props.id,
     label = props.label,
@@ -159,6 +163,12 @@ export var TreeItem = /*#__PURE__*/React.forwardRef(function TreeItem(inProps, i
     TransitionComponent = _props$TransitionComp === void 0 ? Collapse : _props$TransitionComp,
     TransitionProps = props.TransitionProps,
     other = _objectWithoutProperties(props, _excluded);
+  var slots = {
+    expandIcon: (_ref5 = (_inSlots$expandIcon = inSlots == null ? void 0 : inSlots.expandIcon) != null ? _inSlots$expandIcon : contextIcons.slots.expandIcon) != null ? _ref5 : TreeViewExpandIcon,
+    collapseIcon: (_ref6 = (_inSlots$collapseIcon = inSlots == null ? void 0 : inSlots.collapseIcon) != null ? _inSlots$collapseIcon : contextIcons.slots.collapseIcon) != null ? _ref6 : TreeViewCollapseIcon,
+    endIcon: (_inSlots$endIcon = inSlots == null ? void 0 : inSlots.endIcon) != null ? _inSlots$endIcon : contextIcons.slots.endIcon,
+    icon: inSlots == null ? void 0 : inSlots.icon
+  };
   var expandable = Boolean(Array.isArray(children) ? children.length : children);
   var expanded = instance.isNodeExpanded(nodeId);
   var focused = instance.isNodeFocused(nodeId);
@@ -171,20 +181,43 @@ export var TreeItem = /*#__PURE__*/React.forwardRef(function TreeItem(inProps, i
     disabled: disabled
   });
   var classes = useUtilityClasses(ownerState);
-  var displayIcon;
-  var expansionIcon;
-  if (expandable) {
-    if (!expanded) {
-      expansionIcon = expandIcon || contextIcons.defaultExpandIcon;
-    } else {
-      expansionIcon = collapseIcon || contextIcons.defaultCollapseIcon;
-    }
-  }
-  if (expandable) {
-    displayIcon = contextIcons.defaultParentIcon;
-  } else {
-    displayIcon = endIcon || contextIcons.defaultEndIcon;
-  }
+  var ExpansionIcon = expanded ? slots.collapseIcon : slots.expandIcon;
+  var _useSlotProps = useSlotProps({
+      elementType: ExpansionIcon,
+      ownerState: {},
+      externalSlotProps: function externalSlotProps(tempOwnerState) {
+        if (expanded) {
+          return _extends({}, resolveComponentProps(contextIcons.slotProps.collapseIcon, tempOwnerState), resolveComponentProps(inSlotProps == null ? void 0 : inSlotProps.collapseIcon, tempOwnerState));
+        }
+        return _extends({}, resolveComponentProps(contextIcons.slotProps.expandIcon, tempOwnerState), resolveComponentProps(inSlotProps == null ? void 0 : inSlotProps.expandIcon, tempOwnerState));
+      }
+    }),
+    expansionIconOwnerState = _useSlotProps.ownerState,
+    expansionIconProps = _objectWithoutProperties(_useSlotProps, _excluded2);
+  var expansionIcon = expandable && !!ExpansionIcon ? /*#__PURE__*/_jsx(ExpansionIcon, _extends({}, expansionIconProps)) : null;
+  var DisplayIcon = expandable ? undefined : slots.endIcon;
+  var _useSlotProps2 = useSlotProps({
+      elementType: DisplayIcon,
+      ownerState: {},
+      externalSlotProps: function externalSlotProps(tempOwnerState) {
+        if (expandable) {
+          return {};
+        }
+        return _extends({}, resolveComponentProps(contextIcons.slotProps.endIcon, tempOwnerState), resolveComponentProps(inSlotProps == null ? void 0 : inSlotProps.endIcon, tempOwnerState));
+      }
+    }),
+    displayIconOwnerState = _useSlotProps2.ownerState,
+    displayIconProps = _objectWithoutProperties(_useSlotProps2, _excluded3);
+  var displayIcon = DisplayIcon ? /*#__PURE__*/_jsx(DisplayIcon, _extends({}, displayIconProps)) : null;
+  var Icon = slots.icon;
+  var _useSlotProps3 = useSlotProps({
+      elementType: Icon,
+      ownerState: {},
+      externalSlotProps: inSlotProps == null ? void 0 : inSlotProps.icon
+    }),
+    iconOwnerState = _useSlotProps3.ownerState,
+    iconProps = _objectWithoutProperties(_useSlotProps3, _excluded4);
+  var icon = Icon ? /*#__PURE__*/_jsx(Icon, _extends({}, iconProps)) : null;
   var ariaSelected;
   if (multiSelect) {
     ariaSelected = selected;
@@ -265,14 +298,7 @@ TreeItem.propTypes = {
    * Override or extend the styles applied to the component.
    */
   classes: PropTypes.object,
-  /**
-   * className applied to the root element.
-   */
   className: PropTypes.string,
-  /**
-   * The icon used to collapse the node.
-   */
-  collapseIcon: PropTypes.node,
   /**
    * The component used for the content node.
    * @default TreeItemContent
@@ -287,18 +313,6 @@ TreeItem.propTypes = {
    * @default false
    */
   disabled: PropTypes.bool,
-  /**
-   * The icon displayed next to an end node.
-   */
-  endIcon: PropTypes.node,
-  /**
-   * The icon used to expand the node.
-   */
-  expandIcon: PropTypes.node,
-  /**
-   * The icon to display next to the tree node's label.
-   */
-  icon: PropTypes.node,
   /**
    * The tree node label.
    */
@@ -312,6 +326,16 @@ TreeItem.propTypes = {
    * Use the `onNodeFocus` callback on the tree if you need to monitor a node's focus.
    */
   onFocus: unsupportedProp,
+  /**
+   * The props used for each component slot.
+   * @default {}
+   */
+  slotProps: PropTypes.object,
+  /**
+   * Overridable component slots.
+   * @default {}
+   */
+  slots: PropTypes.object,
   /**
    * The system prop that allows defining system overrides as well as additional CSS styles.
    */

package/legacy/TreeItem/TreeItemContent.js

@@ -4,7 +4,7 @@ var _excluded = ["classes", "className", "displayIcon", "expansionIcon", "icon",
 import * as React from 'react';
 import PropTypes from 'prop-types';
 import clsx from 'clsx';
-import { useTreeItem } from './useTreeItem';
+import { useTreeItemState } from './useTreeItemState';
 import { jsx as _jsx } from "react/jsx-runtime";
 import { jsxs as _jsxs } from "react/jsx-runtime";
 /**
@@ -21,14 +21,14 @@ var TreeItemContent = /*#__PURE__*/React.forwardRef(function TreeItemContent(pro
     onClick = props.onClick,
     onMouseDown = props.onMouseDown,
     other = _objectWithoutProperties(props, _excluded);
-  var _useTreeItem = useTreeItem(nodeId),
-    disabled = _useTreeItem.disabled,
-    expanded = _useTreeItem.expanded,
-    selected = _useTreeItem.selected,
-    focused = _useTreeItem.focused,
-    handleExpansion = _useTreeItem.handleExpansion,
-    handleSelection = _useTreeItem.handleSelection,
-    preventSelection = _useTreeItem.preventSelection;
+  var _useTreeItemState = useTreeItemState(nodeId),
+    disabled = _useTreeItemState.disabled,
+    expanded = _useTreeItemState.expanded,
+    selected = _useTreeItemState.selected,
+    focused = _useTreeItemState.focused,
+    handleExpansion = _useTreeItemState.handleExpansion,
+    handleSelection = _useTreeItemState.handleSelection,
+    preventSelection = _useTreeItemState.preventSelection;
   var icon = iconProp || expansionIcon || displayIcon;
   var handleMouseDown = function handleMouseDown(event) {
     preventSelection(event);
@@ -70,9 +70,6 @@ process.env.NODE_ENV !== "production" ? TreeItemContent.propTypes = {
    * Override or extend the styles applied to the component.
    */
   classes: PropTypes.object.isRequired,
-  /**
-   * className applied to the root element.
-   */
   className: PropTypes.string,
   /**
    * The icon to display next to the tree node's label. Either a parent or end icon.

package/legacy/TreeItem/index.js

@@ -1,4 +1,4 @@
 export { TreeItem } from './TreeItem';
 export * from './treeItemClasses';
-export * from './useTreeItem';
+export * from './useTreeItemState';
 export { TreeItemContent } from './TreeItemContent';