@mui/x-tree-view 8.0.0-alpha.0 → 8.0.0-alpha.1

package/node/internals/plugins/useTreeViewLabel/useTreeViewLabel.itemPlugin.js

@@ -7,30 +7,38 @@ Object.defineProperty(exports, "__esModule", {
 exports.useTreeViewLabelItemPlugin = void 0;
 var React = _interopRequireWildcard(require("react"));
 var _TreeViewProvider = require("../../TreeViewProvider");
+var _useSelector = require("../../hooks/useSelector");
+var _useTreeViewLabel = require("./useTreeViewLabel.selectors");
 const useTreeViewLabelItemPlugin = ({
   props
 }) => {
   const {
-    instance
+    store,
+    label: {
+      isItemEditable
+    }
   } = (0, _TreeViewProvider.useTreeViewContext)();
   const {
     label,
     itemId
   } = props;
   const [labelInputValue, setLabelInputValue] = React.useState(label);
-  const isItemBeingEdited = instance.isItemBeingEdited(itemId);
+  const editable = (0, _useSelector.useSelector)(store, _useTreeViewLabel.selectorIsItemEditable, {
+    itemId,
+    isItemEditable
+  });
+  const editing = (0, _useSelector.useSelector)(store, _useTreeViewLabel.selectorIsItemBeingEdited, itemId);
   React.useEffect(() => {
-    if (!isItemBeingEdited) {
+    if (!editing) {
       setLabelInputValue(label);
     }
-  }, [isItemBeingEdited, label]);
+  }, [editing, label]);
   return {
     propsEnhancers: {
       labelInput: ({
         externalEventHandlers,
         interactions
       }) => {
-        const editable = instance.isItemEditable(itemId);
         if (!editable) {
           return {};
         }

package/node/internals/plugins/useTreeViewLabel/useTreeViewLabel.js

@@ -11,41 +11,26 @@ var React = _interopRequireWildcard(require("react"));
 var _warning = require("@mui/x-internals/warning");
 var _useTreeViewLabel = require("./useTreeViewLabel.itemPlugin");
 const useTreeViewLabel = ({
-  instance,
-  state,
-  setState,
+  store,
   params
 }) => {
-  const editedItemRef = React.useRef(state.editedItemId);
-  const isItemBeingEditedRef = itemId => editedItemRef.current === itemId;
   const setEditedItemId = editedItemId => {
-    setState(prevState => (0, _extends2.default)({}, prevState, {
-      editedItemId
+    store.update(prevState => (0, _extends2.default)({}, prevState, {
+      label: {
+        editedItemId
+      }
     }));
-    editedItemRef.current = editedItemId;
-  };
-  const isItemBeingEdited = itemId => itemId === state.editedItemId;
-  const isTreeViewEditable = Boolean(params.isItemEditable);
-  const isItemEditable = itemId => {
-    if (itemId == null || !isTreeViewEditable) {
-      return false;
-    }
-    const item = instance.getItem(itemId);
-    if (!item) {
-      return false;
-    }
-    return typeof params.isItemEditable === 'function' ? params.isItemEditable(item) : Boolean(params.isItemEditable);
   };
   const updateItemLabel = (itemId, label) => {
     if (!label) {
       throw new Error(['MUI X: The Tree View component requires all items to have a `label` property.', 'The label of an item cannot be empty.', itemId].join('\n'));
     }
-    setState(prevState => {
-      const item = prevState.items.itemMetaMap[itemId];
+    store.update(prevState => {
+      const item = prevState.items.itemMetaLookup[itemId];
       if (item.label !== label) {
         return (0, _extends2.default)({}, prevState, {
           items: (0, _extends2.default)({}, prevState.items, {
-            itemMetaMap: (0, _extends2.default)({}, prevState.items.itemMetaMap, {
+            itemMetaLookup: (0, _extends2.default)({}, prevState.items.itemMetaLookup, {
               [itemId]: (0, _extends2.default)({}, item, {
                 label
               })
@@ -59,18 +44,20 @@ const useTreeViewLabel = ({
       params.onItemLabelChange(itemId, label);
     }
   };
+  const pluginContextValue = React.useMemo(() => ({
+    label: {
+      isItemEditable: params.isItemEditable
+    }
+  }), [params.isItemEditable]);
   return {
     instance: {
       setEditedItemId,
-      isItemBeingEdited,
-      updateItemLabel,
-      isItemEditable,
-      isTreeViewEditable,
-      isItemBeingEditedRef
+      updateItemLabel
     },
     publicAPI: {
       updateItemLabel
-    }
+    },
+    contextValue: pluginContextValue
   };
 };
 exports.useTreeViewLabel = useTreeViewLabel;
@@ -90,7 +77,9 @@ useTreeViewLabel.getDefaultizedParams = ({
   });
 };
 useTreeViewLabel.getInitialState = () => ({
-  editedItemId: null
+  label: {
+    editedItemId: null
+  }
 });
 useTreeViewLabel.params = {
   onItemLabelChange: true,

package/node/internals/plugins/useTreeViewLabel/useTreeViewLabel.selectors.js

@@ -0,0 +1,32 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.selectorIsItemEditable = exports.selectorIsItemBeingEdited = void 0;
+var _selectors = require("../../utils/selectors");
+var _useTreeViewItems = require("../useTreeViewItems/useTreeViewItems.selectors");
+const selectorTreeViewLabelState = state => state.label;
+
+/**
+ * Check if an item is editable.
+ * @param {TreeViewState<[UseTreeViewItemsSignature]>} state The state of the tree view.
+ * @param {object} params The parameters.
+ * @param {TreeViewItemId} params.itemId The id of the item to check.
+ * @param {((item: any) => boolean) | boolean} params.isItemEditable The function to determine if an item is editable.
+ * @returns {boolean} `true` if the item is editable, `false` otherwise.
+ */
+const selectorIsItemEditable = exports.selectorIsItemEditable = (0, _selectors.createSelector)([(_, args) => args, (state, args) => (0, _useTreeViewItems.selectorItemModel)(state, args.itemId)], (args, itemModel) => {
+  if (!itemModel || !args.isItemEditable) {
+    return false;
+  }
+  return typeof args.isItemEditable === 'function' ? args.isItemEditable(itemModel) : true;
+});
+
+/**
+ * Check if an item is being edited.
+ * @param {TreeViewState<[UseTreeViewLabelSignature]>} state The state of the tree view.
+ * @param {TreeViewItemId} itemId The id of the item to check.
+ * @returns {boolean} `true` if the item is being edited, `false` otherwise.
+ */
+const selectorIsItemBeingEdited = exports.selectorIsItemBeingEdited = (0, _selectors.createSelector)([selectorTreeViewLabelState, (_, itemId) => itemId], (labelState, itemId) => labelState.editedItemId === itemId);

package/node/internals/plugins/useTreeViewSelection/useTreeViewSelection.itemPlugin.js

@@ -7,9 +7,11 @@ Object.defineProperty(exports, "__esModule", {
 exports.useTreeViewSelectionItemPlugin = void 0;
 var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
 var _TreeViewProvider = require("../../TreeViewProvider");
+var _useTreeViewItems = require("../useTreeViewItems/useTreeViewItems.selectors");
+var _useTreeViewSelection = require("./useTreeViewSelection.selectors");
 function getCheckboxStatus({
   itemId,
-  instance,
+  store,
   selectionPropagation,
   selected
 }) {
@@ -19,7 +21,7 @@ function getCheckboxStatus({
       checked: true
     };
   }
-  const children = instance.getItemOrderedChildrenIds(itemId);
+  const children = (0, _useTreeViewItems.selectorItemOrderedChildrenIds)(store.value, itemId);
   if (children.length === 0) {
     return {
       indeterminate: false,
@@ -30,13 +32,13 @@ function getCheckboxStatus({
   let hasUnSelectedDescendant = false;
   const traverseDescendants = itemToTraverseId => {
     if (itemToTraverseId !== itemId) {
-      if (instance.isItemSelected(itemToTraverseId)) {
+      if ((0, _useTreeViewSelection.selectorIsItemSelected)(store.value, itemToTraverseId)) {
         hasSelectedDescendant = true;
       } else {
         hasUnSelectedDescendant = true;
       }
     }
-    instance.getItemOrderedChildrenIds(itemToTraverseId).forEach(traverseDescendants);
+    (0, _useTreeViewItems.selectorItemOrderedChildrenIds)(store.value, itemToTraverseId).forEach(traverseDescendants);
   };
   traverseDescendants(itemId);
   return {
@@ -51,7 +53,7 @@ const useTreeViewSelectionItemPlugin = ({
     itemId
   } = props;
   const {
-    instance,
+    store,
     selection: {
       disableSelection,
       checkboxSelection,
@@ -76,7 +78,7 @@ const useTreeViewSelectionItemPlugin = ({
           interactions.handleCheckboxSelection(event);
         };
         const checkboxStatus = getCheckboxStatus({
-          instance,
+          store,
           itemId,
           selectionPropagation,
           selected: status.selected

package/node/internals/plugins/useTreeViewSelection/useTreeViewSelection.js

@@ -8,32 +8,30 @@ Object.defineProperty(exports, "__esModule", {
 exports.useTreeViewSelection = void 0;
 var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
 var React = _interopRequireWildcard(require("react"));
+var _useEnhancedEffect = _interopRequireDefault(require("@mui/utils/useEnhancedEffect"));
 var _tree = require("../../utils/tree");
 var _useTreeViewSelection = require("./useTreeViewSelection.utils");
-var _useTreeViewSelection2 = require("./useTreeViewSelection.itemPlugin");
+var _useTreeViewSelection2 = require("./useTreeViewSelection.selectors");
+var _useTreeViewSelection3 = require("./useTreeViewSelection.itemPlugin");
 const useTreeViewSelection = ({
-  instance,
+  store,
   params,
   models
 }) => {
   const lastSelectedItem = React.useRef(null);
   const lastSelectedRange = React.useRef({});
-  const selectedItemsMap = React.useMemo(() => {
-    const temp = new Map();
-    if (Array.isArray(models.selectedItems.value)) {
-      models.selectedItems.value.forEach(id => {
-        temp.set(id, true);
-      });
-    } else if (models.selectedItems.value != null) {
-      temp.set(models.selectedItems.value, true);
-    }
-    return temp;
-  }, [models.selectedItems.value]);
+  (0, _useEnhancedEffect.default)(() => {
+    store.update(prevState => (0, _extends2.default)({}, prevState, {
+      selection: {
+        selectedItemsMap: (0, _useTreeViewSelection.createSelectedItemsMap)(models.selectedItems.value)
+      }
+    }));
+  }, [store, models.selectedItems.value]);
   const setSelectedItems = (event, newModel, additionalItemsToPropagate) => {
     let cleanModel;
     if (params.multiSelect && (params.selectionPropagation.descendants || params.selectionPropagation.parents)) {
       cleanModel = (0, _useTreeViewSelection.propagateSelection)({
-        instance,
+        store,
         selectionPropagation: params.selectionPropagation,
         newModel: newModel,
         oldModel: models.selectedItems.value,
@@ -45,7 +43,7 @@ const useTreeViewSelection = ({
     if (params.onItemSelectionToggle) {
       if (params.multiSelect) {
         const changes = (0, _useTreeViewSelection.getAddedAndRemovedItems)({
-          instance,
+          store,
           newModel: cleanModel,
           oldModel: models.selectedItems.value
         });
@@ -71,7 +69,6 @@ const useTreeViewSelection = ({
     }
     models.selectedItems.setControlledValue(cleanModel);
   };
-  const isItemSelected = itemId => selectedItemsMap.has(itemId);
   const selectItem = ({
     event,
     itemId,
@@ -84,7 +81,7 @@ const useTreeViewSelection = ({
     let newSelected;
     if (keepExistingSelection) {
       const cleanSelectedItems = (0, _useTreeViewSelection.convertSelectedItemsToArray)(models.selectedItems.value);
-      const isSelectedBefore = instance.isItemSelected(itemId);
+      const isSelectedBefore = (0, _useTreeViewSelection2.selectorIsItemSelected)(store.value, itemId);
       if (isSelectedBefore && (shouldBeSelected === false || shouldBeSelected == null)) {
         newSelected = cleanSelectedItems.filter(id => id !== itemId);
       } else if (!isSelectedBefore && (shouldBeSelected === true || shouldBeSelected == null)) {
@@ -94,14 +91,14 @@ const useTreeViewSelection = ({
       }
     } else {
       // eslint-disable-next-line no-lonely-if
-      if (shouldBeSelected === false || shouldBeSelected == null && instance.isItemSelected(itemId)) {
+      if (shouldBeSelected === false || shouldBeSelected == null && (0, _useTreeViewSelection2.selectorIsItemSelected)(store.value, itemId)) {
         newSelected = params.multiSelect ? [] : null;
       } else {
         newSelected = params.multiSelect ? [itemId] : itemId;
       }
     }
     setSelectedItems(event, newSelected,
-    // If shouldBeSelected === instance.isItemSelect(itemId), we still want to propagate the select.
+    // If shouldBeSelected === selectorIsItemSelected(store, itemId), we still want to propagate the select.
     // This is useful when the element is in an indeterminate state.
     [itemId]);
     lastSelectedItem.current = itemId;
@@ -121,7 +118,7 @@ const useTreeViewSelection = ({
 
     // Add to the model the items that are part of the new range and not already part of the model.
     const selectedItemsLookup = (0, _useTreeViewSelection.getLookupFromArray)(newSelectedItems);
-    const range = (0, _tree.getNonDisabledItemsInRange)(instance, start, end);
+    const range = (0, _tree.getNonDisabledItemsInRange)(store.value, start, end);
     const itemsToAddToModel = range.filter(id => !selectedItemsLookup[id]);
     newSelectedItems = newSelectedItems.concat(itemsToAddToModel);
     setSelectedItems(event, newSelectedItems);
@@ -129,21 +126,21 @@ const useTreeViewSelection = ({
   };
   const expandSelectionRange = (event, itemId) => {
     if (lastSelectedItem.current != null) {
-      const [start, end] = (0, _tree.findOrderInTremauxTree)(instance, itemId, lastSelectedItem.current);
+      const [start, end] = (0, _tree.findOrderInTremauxTree)(store.value, itemId, lastSelectedItem.current);
       selectRange(event, [start, end]);
     }
   };
   const selectRangeFromStartToItem = (event, itemId) => {
-    selectRange(event, [(0, _tree.getFirstNavigableItem)(instance), itemId]);
+    selectRange(event, [(0, _tree.getFirstNavigableItem)(store.value), itemId]);
   };
   const selectRangeFromItemToEnd = (event, itemId) => {
-    selectRange(event, [itemId, (0, _tree.getLastNavigableItem)(instance)]);
+    selectRange(event, [itemId, (0, _tree.getLastNavigableItem)(store.value)]);
   };
   const selectAllNavigableItems = event => {
     if (params.disableSelection || !params.multiSelect) {
       return;
     }
-    const navigableItems = (0, _tree.getAllNavigableItems)(instance);
+    const navigableItems = (0, _tree.getAllNavigableItems)(store.value);
     setSelectedItems(event, navigableItems);
     lastSelectedRange.current = (0, _useTreeViewSelection.getLookupFromArray)(navigableItems);
   };
@@ -172,6 +169,17 @@ const useTreeViewSelection = ({
     }
     setSelectedItems(event, newSelectedItems);
   };
+  const pluginContextValue = React.useMemo(() => ({
+    selection: {
+      multiSelect: params.multiSelect,
+      checkboxSelection: params.checkboxSelection,
+      disableSelection: params.disableSelection,
+      selectionPropagation: {
+        descendants: params.selectionPropagation.descendants,
+        parents: params.selectionPropagation.parents
+      }
+    }
+  }), [params.multiSelect, params.checkboxSelection, params.disableSelection, params.selectionPropagation.descendants, params.selectionPropagation.parents]);
   return {
     getRootProps: () => ({
       'aria-multiselectable': params.multiSelect
@@ -180,7 +188,6 @@ const useTreeViewSelection = ({
       selectItem
     },
     instance: {
-      isItemSelected,
       selectItem,
       selectAllNavigableItems,
       expandSelectionRange,
@@ -188,24 +195,18 @@ const useTreeViewSelection = ({
       selectRangeFromItemToEnd,
       selectItemFromArrowNavigation
     },
-    contextValue: {
-      selection: {
-        multiSelect: params.multiSelect,
-        checkboxSelection: params.checkboxSelection,
-        disableSelection: params.disableSelection,
-        selectionPropagation: params.selectionPropagation
-      }
-    }
+    contextValue: pluginContextValue
   };
 };
 exports.useTreeViewSelection = useTreeViewSelection;
-useTreeViewSelection.itemPlugin = _useTreeViewSelection2.useTreeViewSelectionItemPlugin;
+useTreeViewSelection.itemPlugin = _useTreeViewSelection3.useTreeViewSelectionItemPlugin;
 useTreeViewSelection.models = {
   selectedItems: {
     getDefaultValue: params => params.defaultSelectedItems
   }
 };
 const DEFAULT_SELECTED_ITEMS = [];
+const EMPTY_SELECTION_PROPAGATION = {};
 useTreeViewSelection.getDefaultizedParams = ({
   params
 }) => (0, _extends2.default)({}, params, {
@@ -213,7 +214,17 @@ useTreeViewSelection.getDefaultizedParams = ({
   multiSelect: params.multiSelect ?? false,
   checkboxSelection: params.checkboxSelection ?? false,
   defaultSelectedItems: params.defaultSelectedItems ?? (params.multiSelect ? DEFAULT_SELECTED_ITEMS : null),
-  selectionPropagation: params.selectionPropagation ?? {}
+  selectionPropagation: params.selectionPropagation ?? EMPTY_SELECTION_PROPAGATION
+});
+useTreeViewSelection.getInitialState = params => ({
+  selection: {
+    selectedItemsMap: (0, _useTreeViewSelection.createSelectedItemsMap)(params.selectedItems === undefined ? params.defaultSelectedItems : params.selectedItems)
+  }
+});
+useTreeViewSelection.getInitialState = params => ({
+  selection: {
+    selectedItemsMap: (0, _useTreeViewSelection.createSelectedItemsMap)(params.selectedItems === undefined ? params.defaultSelectedItems : params.selectedItems)
+  }
 });
 useTreeViewSelection.params = {
   disableSelection: true,

package/node/internals/plugins/useTreeViewSelection/useTreeViewSelection.selectors.js

@@ -0,0 +1,15 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.selectorIsItemSelected = void 0;
+var _selectors = require("../../utils/selectors");
+const selectorTreeViewSelectionState = state => state.selection;
+
+/**
+ * Check if an item is selected.
+ * @param {TreeViewState<[UseTreeViewSelectionSignature]>} state The state of the tree view.
+ * @returns {boolean} `true` if the item is selected, `false` otherwise.
+ */
+const selectorIsItemSelected = exports.selectorIsItemSelected = (0, _selectors.createSelector)([selectorTreeViewSelectionState, (_, itemId) => itemId], (selectionState, itemId) => selectionState.selectedItemsMap.has(itemId));

package/node/internals/plugins/useTreeViewSelection/useTreeViewSelection.utils.js

@@ -3,7 +3,9 @@
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.propagateSelection = exports.getLookupFromArray = exports.getAddedAndRemovedItems = exports.convertSelectedItemsToArray = void 0;
+exports.propagateSelection = exports.getLookupFromArray = exports.getAddedAndRemovedItems = exports.createSelectedItemsMap = exports.convertSelectedItemsToArray = void 0;
+var _useTreeViewSelection = require("./useTreeViewSelection.selectors");
+var _useTreeViewItems = require("../useTreeViewItems/useTreeViewItems.selectors");
 /**
  * Transform the `selectedItems` model to be an array if it was a string or null.
  * @param {string[] | string | null} model The raw model.
@@ -19,6 +21,14 @@ const convertSelectedItemsToArray = model => {
   return [];
 };
 exports.convertSelectedItemsToArray = convertSelectedItemsToArray;
+const createSelectedItemsMap = selectedItems => {
+  const selectedItemsMap = new Map();
+  convertSelectedItemsToArray(selectedItems).forEach(id => {
+    selectedItemsMap.set(id, true);
+  });
+  return selectedItemsMap;
+};
+exports.createSelectedItemsMap = createSelectedItemsMap;
 const getLookupFromArray = array => {
   const lookup = {};
   array.forEach(itemId => {
@@ -28,19 +38,19 @@ const getLookupFromArray = array => {
 };
 exports.getLookupFromArray = getLookupFromArray;
 const getAddedAndRemovedItems = ({
-  instance,
+  store,
   oldModel,
   newModel
 }) => {
-  const newModelLookup = getLookupFromArray(newModel);
+  const newModelLookup = createSelectedItemsMap(newModel);
   return {
-    added: newModel.filter(itemId => !instance.isItemSelected(itemId)),
-    removed: oldModel.filter(itemId => !newModelLookup[itemId])
+    added: newModel.filter(itemId => !(0, _useTreeViewSelection.selectorIsItemSelected)(store.value, itemId)),
+    removed: oldModel.filter(itemId => !newModelLookup.has(itemId))
   };
 };
 exports.getAddedAndRemovedItems = getAddedAndRemovedItems;
 const propagateSelection = ({
-  instance,
+  store,
   selectionPropagation,
   newModel,
   oldModel,
@@ -52,7 +62,7 @@ const propagateSelection = ({
   let shouldRegenerateModel = false;
   const newModelLookup = getLookupFromArray(newModel);
   const changes = getAddedAndRemovedItems({
-    instance,
+    store,
     newModel,
     oldModel
   });
@@ -72,7 +82,7 @@ const propagateSelection = ({
           shouldRegenerateModel = true;
           newModelLookup[itemId] = true;
         }
-        instance.getItemOrderedChildrenIds(itemId).forEach(selectDescendants);
+        (0, _useTreeViewItems.selectorItemOrderedChildrenIds)(store.value, itemId).forEach(selectDescendants);
       };
       selectDescendants(addedItemId);
     }
@@ -81,15 +91,15 @@ const propagateSelection = ({
         if (!newModelLookup[itemId]) {
           return false;
         }
-        const children = instance.getItemOrderedChildrenIds(itemId);
+        const children = (0, _useTreeViewItems.selectorItemOrderedChildrenIds)(store.value, itemId);
         return children.every(checkAllDescendantsSelected);
       };
       const selectParents = itemId => {
-        const parentId = instance.getItemMeta(itemId).parentId;
+        const parentId = (0, _useTreeViewItems.selectorItemParentId)(store.value, itemId);
         if (parentId == null) {
           return;
         }
-        const siblings = instance.getItemOrderedChildrenIds(parentId);
+        const siblings = (0, _useTreeViewItems.selectorItemOrderedChildrenIds)(store.value, parentId);
         if (siblings.every(checkAllDescendantsSelected)) {
           shouldRegenerateModel = true;
           newModelLookup[parentId] = true;
@@ -101,13 +111,13 @@ const propagateSelection = ({
   });
   changes.removed.forEach(removedItemId => {
     if (selectionPropagation.parents) {
-      let parentId = instance.getItemMeta(removedItemId).parentId;
+      let parentId = (0, _useTreeViewItems.selectorItemParentId)(store.value, removedItemId);
       while (parentId != null) {
         if (newModelLookup[parentId]) {
           shouldRegenerateModel = true;
           delete newModelLookup[parentId];
         }
-        parentId = instance.getItemMeta(parentId).parentId;
+        parentId = (0, _useTreeViewItems.selectorItemParentId)(store.value, parentId);
       }
     }
     if (selectionPropagation.descendants) {
@@ -116,7 +126,7 @@ const propagateSelection = ({
           shouldRegenerateModel = true;
           delete newModelLookup[itemId];
         }
-        instance.getItemOrderedChildrenIds(itemId).forEach(deSelectDescendants);
+        (0, _useTreeViewItems.selectorItemOrderedChildrenIds)(store.value, itemId).forEach(deSelectDescendants);
       };
       deSelectDescendants(removedItemId);
     }

package/node/internals/useTreeView/useTreeView.js

@@ -14,6 +14,7 @@ var _useTreeViewModels = require("./useTreeViewModels");
 var _corePlugins = require("../corePlugins");
 var _extractPluginParamsFromProps = require("./extractPluginParamsFromProps");
 var _useTreeViewBuildContext = require("./useTreeViewBuildContext");
+var _TreeViewStore = require("../utils/TreeViewStore");
 function useTreeViewApiInitialization(inputApiRef) {
   const fallbackPublicApiRef = React.useRef({});
   if (inputApiRef) {
@@ -24,12 +25,13 @@ function useTreeViewApiInitialization(inputApiRef) {
   }
   return fallbackPublicApiRef.current;
 }
+let globalId = 0;
 const useTreeView = ({
   plugins: inPlugins,
   rootRef,
   props
 }) => {
-  const plugins = [..._corePlugins.TREE_VIEW_CORE_PLUGINS, ...inPlugins];
+  const plugins = React.useMemo(() => [..._corePlugins.TREE_VIEW_CORE_PLUGINS, ...inPlugins], [inPlugins]);
   const {
     pluginParams,
     forwardedProps,
@@ -47,22 +49,30 @@ const useTreeView = ({
   const publicAPI = useTreeViewApiInitialization(apiRef);
   const innerRootRef = React.useRef(null);
   const handleRootRef = (0, _useForkRef.default)(innerRootRef, rootRef);
-  const contextValue = (0, _useTreeViewBuildContext.useTreeViewBuildContext)({
-    plugins,
-    instance,
-    publicAPI,
-    rootRef: innerRootRef
-  });
-  const [state, setState] = React.useState(() => {
-    const temp = {};
+  const storeRef = React.useRef(null);
+  if (storeRef.current == null) {
+    globalId += 1;
+    const initialState = {
+      cacheKey: {
+        id: globalId
+      }
+    };
     plugins.forEach(plugin => {
       if (plugin.getInitialState) {
-        Object.assign(temp, plugin.getInitialState(pluginParams));
+        Object.assign(initialState, plugin.getInitialState(pluginParams));
       }
     });
-    return temp;
+    storeRef.current = new _TreeViewStore.TreeViewStore(initialState);
+  }
+  const baseContextValue = (0, _useTreeViewBuildContext.useTreeViewBuildContext)({
+    plugins,
+    instance,
+    publicAPI,
+    store: storeRef.current,
+    rootRef: innerRootRef
   });
   const rootPropsGetters = [];
+  const pluginContextValues = [];
   const runPlugin = plugin => {
     const pluginResponse = plugin({
       instance,
@@ -70,11 +80,10 @@ const useTreeView = ({
       slots,
       slotProps,
       experimentalFeatures,
-      state,
-      setState,
       rootRef: innerRootRef,
       models,
-      plugins
+      plugins,
+      store: storeRef.current
     });
     if (pluginResponse.getRootProps) {
       rootPropsGetters.push(pluginResponse.getRootProps);
@@ -86,7 +95,7 @@ const useTreeView = ({
       Object.assign(instance, pluginResponse.instance);
     }
     if (pluginResponse.contextValue) {
-      Object.assign(contextValue, pluginResponse.contextValue);
+      pluginContextValues.push(pluginResponse.contextValue);
     }
   };
   plugins.forEach(runPlugin);
@@ -101,11 +110,15 @@ const useTreeView = ({
     });
     return rootProps;
   };
+  const contextValue = React.useMemo(() => {
+    const copiedBaseContextValue = (0, _extends2.default)({}, baseContextValue);
+    return Object.assign(copiedBaseContextValue, ...pluginContextValues);
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [baseContextValue, ...pluginContextValues]);
   return {
     getRootProps,
     rootRef: handleRootRef,
-    contextValue,
-    instance
+    contextValue
   };
 };
 exports.useTreeView = useTreeView;