@mui/x-tree-view-pro 8.0.0-alpha.8 → 8.0.0-beta.0

package/{node → esm}/internals/plugins/useTreeViewItemsReordering/useTreeViewItemsReordering.utils.js

@@ -1,17 +1,10 @@
-"use strict";
-
-var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-exports.moveItemInTree = exports.isAncestor = exports.chooseActionToApply = void 0;
-var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
-var _internals = require("@mui/x-tree-view/internals");
+import _extends from "@babel/runtime/helpers/esm/extends";
+import { buildSiblingIndexes, TREE_VIEW_ROOT_PARENT_ID, selectorItemMeta } from '@mui/x-tree-view/internals';
 /**
  * Checks if the item with the id itemIdB is an ancestor of the item with the id itemIdA.
  */
-const isAncestor = (store, itemIdA, itemIdB) => {
-  const itemMetaA = (0, _internals.selectorItemMeta)(store.value, itemIdA);
+export const isAncestor = (store, itemIdA, itemIdB) => {
+  const itemMetaA = selectorItemMeta(store.value, itemIdA);
   if (itemMetaA.parentId === itemIdB) {
     return true;
   }
@@ -26,7 +19,6 @@ const isAncestor = (store, itemIdA, itemIdB) => {
  * @param {string | null} itemChildrenIndentation The indentation as passed to the `itemChildrenIndentation` prop.
  * @param {HTMLElement} contentElement The DOM element to which the indentation will be applied.
  */
-exports.isAncestor = isAncestor;
 const parseItemChildrenIndentation = (itemChildrenIndentation, contentElement) => {
   if (typeof itemChildrenIndentation === 'number') {
     return itemChildrenIndentation;
@@ -45,7 +37,7 @@ const parseItemChildrenIndentation = (itemChildrenIndentation, contentElement) =
   contentElement.removeChild(tempElement);
   return value;
 };
-const chooseActionToApply = ({
+export const chooseActionToApply = ({
   itemChildrenIndentation,
   validActions,
   targetHeight,
@@ -90,24 +82,23 @@ const chooseActionToApply = ({
   }
   return action;
 };
-exports.chooseActionToApply = chooseActionToApply;
-const moveItemInTree = ({
+export const moveItemInTree = ({
   itemToMoveId,
   oldPosition,
   newPosition,
   prevState
 }) => {
   const itemToMoveMeta = prevState.itemMetaLookup[itemToMoveId];
-  const oldParentId = oldPosition.parentId ?? _internals.TREE_VIEW_ROOT_PARENT_ID;
-  const newParentId = newPosition.parentId ?? _internals.TREE_VIEW_ROOT_PARENT_ID;
+  const oldParentId = oldPosition.parentId ?? TREE_VIEW_ROOT_PARENT_ID;
+  const newParentId = newPosition.parentId ?? TREE_VIEW_ROOT_PARENT_ID;
 
   // 1. Update the `itemOrderedChildrenIds`.
-  const itemOrderedChildrenIds = (0, _extends2.default)({}, prevState.itemOrderedChildrenIdsLookup);
+  const itemOrderedChildrenIds = _extends({}, prevState.itemOrderedChildrenIdsLookup);
   if (oldParentId === newParentId) {
     const updatedChildren = [...itemOrderedChildrenIds[oldParentId]];
     updatedChildren.splice(oldPosition.index, 1);
     updatedChildren.splice(newPosition.index, 0, itemToMoveId);
-    itemOrderedChildrenIds[itemToMoveMeta.parentId ?? _internals.TREE_VIEW_ROOT_PARENT_ID] = updatedChildren;
+    itemOrderedChildrenIds[itemToMoveMeta.parentId ?? TREE_VIEW_ROOT_PARENT_ID] = updatedChildren;
   } else {
     const updatedOldParentChildren = [...itemOrderedChildrenIds[oldParentId]];
     updatedOldParentChildren.splice(oldPosition.index, 1);
@@ -118,51 +109,50 @@ const moveItemInTree = ({
   }
 
   // 2. Update the `itemChildrenIndexes`
-  const itemChildrenIndexes = (0, _extends2.default)({}, prevState.itemChildrenIndexesLookup);
-  itemChildrenIndexes[oldParentId] = (0, _internals.buildSiblingIndexes)(itemOrderedChildrenIds[oldParentId]);
+  const itemChildrenIndexes = _extends({}, prevState.itemChildrenIndexesLookup);
+  itemChildrenIndexes[oldParentId] = buildSiblingIndexes(itemOrderedChildrenIds[oldParentId]);
   if (newParentId !== oldParentId) {
-    itemChildrenIndexes[newParentId] = (0, _internals.buildSiblingIndexes)(itemOrderedChildrenIds[newParentId]);
+    itemChildrenIndexes[newParentId] = buildSiblingIndexes(itemOrderedChildrenIds[newParentId]);
   }
 
   // 3. Update the `itemMetaLookup`
-  const itemMetaLookup = (0, _extends2.default)({}, prevState.itemMetaLookup);
+  const itemMetaLookup = _extends({}, prevState.itemMetaLookup);
 
   // 3.1 Update the `expandable` property of the old and the new parent
   function updateExpandable(itemId) {
     const isExpandable = itemOrderedChildrenIds[itemId].length > 0;
     if (itemMetaLookup[itemId].expandable !== isExpandable) {
-      itemMetaLookup[itemId] = (0, _extends2.default)({}, itemMetaLookup[itemId], {
+      itemMetaLookup[itemId] = _extends({}, itemMetaLookup[itemId], {
         expandable: isExpandable
       });
     }
   }
-  if (oldParentId !== _internals.TREE_VIEW_ROOT_PARENT_ID && oldParentId !== newParentId) {
+  if (oldParentId !== TREE_VIEW_ROOT_PARENT_ID && oldParentId !== newParentId) {
     updateExpandable(oldParentId);
   }
-  if (newParentId !== _internals.TREE_VIEW_ROOT_PARENT_ID && newParentId !== oldParentId) {
+  if (newParentId !== TREE_VIEW_ROOT_PARENT_ID && newParentId !== oldParentId) {
     updateExpandable(newParentId);
   }
 
   // 3.2 Update the `parentId` and `depth` properties of the item to move
   // The depth is always defined because drag&drop is only usable with Rich Tree View components.
   const itemToMoveDepth = newPosition.parentId == null ? 0 : itemMetaLookup[newParentId].depth + 1;
-  itemMetaLookup[itemToMoveId] = (0, _extends2.default)({}, itemToMoveMeta, {
+  itemMetaLookup[itemToMoveId] = _extends({}, itemToMoveMeta, {
     parentId: newPosition.parentId,
     depth: itemToMoveDepth
   });
 
   // 3.3 Update the depth of all the children of the item to move
   const updateItemDepth = (itemId, depth) => {
-    itemMetaLookup[itemId] = (0, _extends2.default)({}, itemMetaLookup[itemId], {
+    itemMetaLookup[itemId] = _extends({}, itemMetaLookup[itemId], {
       depth
     });
     itemOrderedChildrenIds[itemId]?.forEach(childId => updateItemDepth(childId, depth + 1));
   };
   itemOrderedChildrenIds[itemToMoveId]?.forEach(childId => updateItemDepth(childId, itemToMoveDepth + 1));
-  return (0, _extends2.default)({}, prevState, {
+  return _extends({}, prevState, {
     itemOrderedChildrenIdsLookup: itemOrderedChildrenIds,
     itemChildrenIndexesLookup: itemChildrenIndexes,
     itemMetaLookup
   });
-};
-exports.moveItemInTree = moveItemInTree;
+};

package/esm/internals/plugins/useTreeViewLazyLoading/index.d.ts

@@ -0,0 +1 @@
+export { useTreeViewLazyLoading } from "./useTreeViewLazyLoading.js";

package/esm/internals/plugins/useTreeViewLazyLoading/index.js

@@ -0,0 +1 @@
+export { useTreeViewLazyLoading } from "./useTreeViewLazyLoading.js";

package/esm/internals/plugins/useTreeViewLazyLoading/useTreeViewLazyLoading.d.ts

@@ -0,0 +1,3 @@
+import { TreeViewPlugin } from '@mui/x-tree-view/internals';
+import type { UseTreeViewLazyLoadingSignature } from '@mui/x-tree-view/internals';
+export declare const useTreeViewLazyLoading: TreeViewPlugin<UseTreeViewLazyLoadingSignature>;

package/esm/internals/plugins/useTreeViewLazyLoading/useTreeViewLazyLoading.js

@@ -0,0 +1,288 @@
+import _extends from "@babel/runtime/helpers/esm/extends";
+import * as React from 'react';
+import useLazyRef from '@mui/utils/useLazyRef';
+import { warnOnce } from '@mui/x-internals/warning';
+import useEventCallback from '@mui/utils/useEventCallback';
+import { selectorItemMeta, selectorIsItemSelected, useInstanceEventHandler, selectorDataSourceState, selectorGetTreeItemError } from '@mui/x-tree-view/internals';
+import { DataSourceCacheDefault } from '@mui/x-tree-view/utils';
+import { NestedDataManager } from "./utils.js";
+const INITIAL_STATE = {
+  loading: {},
+  errors: {}
+};
+const noopCache = {
+  clear: () => {},
+  get: () => undefined,
+  set: () => {}
+};
+function getCache(cacheProp) {
+  if (cacheProp === null) {
+    return noopCache;
+  }
+  return cacheProp ?? new DataSourceCacheDefault({});
+}
+export const useTreeViewLazyLoading = ({
+  instance,
+  params,
+  store
+}) => {
+  const isLazyLoadingEnabled = params.dataSource?.getChildrenCount !== undefined;
+  const firstRenderRef = React.useRef(true);
+  const nestedDataManager = useLazyRef(() => new NestedDataManager(instance)).current;
+  const cacheRef = useLazyRef(() => getCache(params.dataSourceCache));
+  const setDataSourceLoading = useEventCallback((itemId, isLoading) => {
+    if (!isLazyLoadingEnabled) {
+      return;
+    }
+    store.update(prevState => {
+      if (!prevState.lazyLoading.dataSource.loading[itemId] && !isLoading) {
+        return prevState;
+      }
+      const loading = _extends({}, prevState.lazyLoading.dataSource.loading);
+      if (isLoading === false) {
+        delete loading[itemId];
+      } else {
+        loading[itemId] = isLoading;
+      }
+      return _extends({}, prevState, {
+        lazyLoading: _extends({}, prevState.lazyLoading, {
+          dataSource: _extends({}, prevState.lazyLoading.dataSource, {
+            loading
+          })
+        })
+      });
+    });
+  });
+  const setDataSourceError = (itemId, error) => {
+    if (!isLazyLoadingEnabled) {
+      return;
+    }
+    store.update(prevState => {
+      const errors = _extends({}, prevState.lazyLoading.dataSource.errors);
+      if (error === null && errors[itemId] !== undefined) {
+        delete errors[itemId];
+      } else {
+        errors[itemId] = error;
+      }
+      errors[itemId] = error;
+      return _extends({}, prevState, {
+        lazyLoading: _extends({}, prevState.lazyLoading, {
+          dataSource: _extends({}, prevState.lazyLoading.dataSource, {
+            errors
+          })
+        })
+      });
+    });
+  };
+  const resetDataSourceState = useEventCallback(() => {
+    if (!isLazyLoadingEnabled) {
+      return;
+    }
+    store.update(prevState => _extends({}, prevState, {
+      lazyLoading: _extends({}, prevState.lazyLoading, {
+        dataSource: INITIAL_STATE
+      })
+    }));
+  });
+  const fetchItems = useEventCallback(async parentIds => {
+    if (!isLazyLoadingEnabled) {
+      return;
+    }
+    const getChildrenCount = params.dataSource?.getChildrenCount || (() => 0);
+    const getTreeItems = params.dataSource?.getTreeItems;
+    if (!getTreeItems) {
+      return;
+    }
+    if (parentIds) {
+      await nestedDataManager.queue(parentIds);
+      return;
+    }
+    nestedDataManager.clear();
+
+    // reset the state if we are refetching the first visible items
+    if (selectorDataSourceState(store.value) !== INITIAL_STATE) {
+      resetDataSourceState();
+    }
+    // handle caching here
+    const cachedData = cacheRef.current.get('root');
+    if (cachedData !== undefined) {
+      return;
+    }
+
+    // handle loading here
+    instance.setTreeViewLoading(true);
+    try {
+      const getTreeItemsResponse = await getTreeItems();
+
+      // set caching
+      cacheRef.current.set('root', getTreeItemsResponse);
+
+      // update the items in the state
+      instance.addItems({
+        items: getTreeItemsResponse,
+        depth: 0,
+        getChildrenCount
+      });
+    } catch (error) {
+      // set the items to empty
+      instance.addItems({
+        items: [],
+        depth: 0,
+        getChildrenCount
+      });
+      // set error state
+      instance.setTreeViewError(error);
+    } finally {
+      // set loading state
+      instance.setTreeViewLoading(false);
+    }
+  });
+  const fetchItemChildren = useEventCallback(async id => {
+    if (!isLazyLoadingEnabled) {
+      return;
+    }
+    const getChildrenCount = params.dataSource?.getChildrenCount || (() => 0);
+    const getTreeItems = params.dataSource?.getTreeItems;
+    if (!getTreeItems) {
+      nestedDataManager.clearPendingRequest(id);
+      return;
+    }
+    const parent = selectorItemMeta(store.value, id);
+    if (!parent) {
+      nestedDataManager.clearPendingRequest(id);
+      return;
+    }
+    const depth = parent.depth ? parent.depth + 1 : 1;
+
+    // handle loading here
+    instance.setDataSourceLoading(id, true);
+
+    // handle caching here
+    const cachedData = cacheRef.current.get(id);
+    if (cachedData !== undefined && cachedData !== -1) {
+      nestedDataManager.setRequestSettled(id);
+      instance.addItems({
+        items: cachedData,
+        depth,
+        parentId: id,
+        getChildrenCount
+      });
+      instance.setDataSourceLoading(id, false);
+      return;
+    }
+    if (cachedData === -1) {
+      instance.removeChildren(id);
+    }
+    const existingError = selectorGetTreeItemError(store.value, id) ?? null;
+    if (existingError) {
+      instance.setDataSourceError(id, null);
+    }
+    try {
+      const getTreeItemsResponse = await getTreeItems(id);
+      nestedDataManager.setRequestSettled(id);
+
+      // set caching
+      cacheRef.current.set(id, getTreeItemsResponse);
+      // update the items in the state
+      instance.addItems({
+        items: getTreeItemsResponse,
+        depth,
+        parentId: id,
+        getChildrenCount
+      });
+    } catch (error) {
+      const childrenFetchError = error;
+      // handle errors here
+      instance.setDataSourceError(id, childrenFetchError);
+      instance.removeChildren(id);
+    } finally {
+      // unset loading
+      instance.setDataSourceLoading(id, false);
+      nestedDataManager.setRequestSettled(id);
+    }
+  });
+  useInstanceEventHandler(instance, 'beforeItemToggleExpansion', async eventParameters => {
+    if (!isLazyLoadingEnabled || !eventParameters.shouldBeExpanded) {
+      return;
+    }
+    eventParameters.isExpansionPrevented = true;
+    await instance.fetchItems([eventParameters.itemId]);
+    const fetchErrors = Boolean(selectorGetTreeItemError(store.value, eventParameters.itemId));
+    if (!fetchErrors) {
+      instance.applyItemExpansion({
+        itemId: eventParameters.itemId,
+        shouldBeExpanded: true,
+        event: eventParameters.event
+      });
+      if (selectorIsItemSelected(store.value, eventParameters.itemId)) {
+        // make sure selection propagation works correctly
+        instance.setItemSelection({
+          event: eventParameters.event,
+          itemId: eventParameters.itemId,
+          keepExistingSelection: true,
+          shouldBeSelected: true
+        });
+      }
+    }
+  });
+  React.useEffect(() => {
+    if (isLazyLoadingEnabled && firstRenderRef.current) {
+      store.update(prevState => _extends({}, prevState, {
+        lazyLoading: _extends({}, prevState.lazyLoading, {
+          enabled: true
+        })
+      }));
+      if (params.items.length) {
+        const getChildrenCount = params.dataSource?.getChildrenCount || (() => 0);
+        instance.addItems({
+          items: params.items,
+          depth: 0,
+          getChildrenCount
+        });
+      } else {
+        instance.fetchItems();
+      }
+      firstRenderRef.current = false;
+    }
+  }, [instance, params.items, params.dataSource, isLazyLoadingEnabled, store]);
+  if (isLazyLoadingEnabled) {
+    instance.preventItemUpdates();
+  }
+  return {
+    instance: {
+      fetchItemChildren,
+      fetchItems,
+      setDataSourceLoading,
+      setDataSourceError
+    },
+    publicAPI: {}
+  };
+};
+useTreeViewLazyLoading.getDefaultizedParams = ({
+  params,
+  experimentalFeatures
+}) => {
+  const canUseFeature = experimentalFeatures?.lazyLoading;
+  if (process.env.NODE_ENV !== 'production') {
+    if (params.dataSource && !canUseFeature) {
+      warnOnce(['MUI X: The label editing feature requires the `lazyLoading` experimental feature to be enabled.', 'You can do it by passing `experimentalFeatures={{ lazyLoading: true}}` to the Rich Tree View Pro component.', 'Check the documentation for more details: https://mui.com/x/react-tree-view/rich-tree-view/lazy-loading/']);
+    }
+  }
+  const defaultDataSource = params?.dataSource ?? {
+    getChildrenCount: () => 0,
+    getTreeItems: () => Promise.resolve([])
+  };
+  return _extends({}, params, {
+    dataSource: canUseFeature ? defaultDataSource : {}
+  });
+};
+useTreeViewLazyLoading.getInitialState = () => ({
+  lazyLoading: {
+    enabled: false,
+    dataSource: INITIAL_STATE
+  }
+});
+useTreeViewLazyLoading.params = {
+  dataSource: true,
+  dataSourceCache: true
+};

package/esm/internals/plugins/useTreeViewLazyLoading/utils.d.ts

@@ -0,0 +1,32 @@
+import { TreeViewInstance, UseTreeViewLazyLoadingSignature } from '@mui/x-tree-view/internals';
+import { TreeViewItemId } from '@mui/x-tree-view/models';
+export declare enum RequestStatus {
+  QUEUED = 0,
+  PENDING = 1,
+  SETTLED = 2,
+  UNKNOWN = 3,
+}
+/**
+ * Plugins that the `NestedDataManager` class can use if they are present, but are not required.
+ */
+export type NestedDataManagerOptionalPlugins = readonly [];
+/**
+ * Fetches row children from the server with option to limit the number of concurrent requests
+ * Determines the status of a request based on the enum `RequestStatus`
+ * Uses `ParentId` to uniquely identify a request
+ */
+export declare class NestedDataManager {
+  private pendingRequests;
+  private queuedRequests;
+  private settledRequests;
+  private instance;
+  private maxConcurrentRequests;
+  constructor(instance: TreeViewInstance<[UseTreeViewLazyLoadingSignature]>, maxConcurrentRequests?: number);
+  private processQueue;
+  queue: (ids: TreeViewItemId[]) => Promise<void>;
+  setRequestSettled: (id: TreeViewItemId) => Promise<void>;
+  clear: () => void;
+  clearPendingRequest: (id: TreeViewItemId) => Promise<void>;
+  getRequestStatus: (id: TreeViewItemId) => RequestStatus;
+  getActiveRequestsCount: () => number;
+}

package/esm/internals/plugins/useTreeViewLazyLoading/utils.js

@@ -0,0 +1,85 @@
+const MAX_CONCURRENT_REQUESTS = Infinity;
+export let RequestStatus = /*#__PURE__*/function (RequestStatus) {
+  RequestStatus[RequestStatus["QUEUED"] = 0] = "QUEUED";
+  RequestStatus[RequestStatus["PENDING"] = 1] = "PENDING";
+  RequestStatus[RequestStatus["SETTLED"] = 2] = "SETTLED";
+  RequestStatus[RequestStatus["UNKNOWN"] = 3] = "UNKNOWN";
+  return RequestStatus;
+}({});
+
+/**
+ * Plugins that need to be present in the Tree View in order for the `NestedDataManager` class to work correctly.
+ */
+
+/**
+ * Plugins that the `NestedDataManager` class can use if they are present, but are not required.
+ */
+
+/**
+ * Fetches row children from the server with option to limit the number of concurrent requests
+ * Determines the status of a request based on the enum `RequestStatus`
+ * Uses `ParentId` to uniquely identify a request
+ */
+export class NestedDataManager {
+  constructor(instance, maxConcurrentRequests = MAX_CONCURRENT_REQUESTS) {
+    this.pendingRequests = new Set();
+    this.queuedRequests = new Set();
+    this.settledRequests = new Set();
+    this.instance = void 0;
+    this.maxConcurrentRequests = void 0;
+    this.processQueue = async () => {
+      if (this.queuedRequests.size === 0 || this.pendingRequests.size >= this.maxConcurrentRequests) {
+        return;
+      }
+      const loopLength = Math.min(this.maxConcurrentRequests - this.pendingRequests.size, this.queuedRequests.size);
+      if (loopLength === 0) {
+        return;
+      }
+      const fetchQueue = Array.from(this.queuedRequests);
+      const fetchPromises = [];
+      for (let i = 0; i < loopLength; i += 1) {
+        const id = fetchQueue[i];
+        this.queuedRequests.delete(id);
+        this.pendingRequests.add(id);
+        fetchPromises.push(this.instance.fetchItemChildren(id));
+      }
+      await Promise.all(fetchPromises);
+    };
+    this.queue = async ids => {
+      const loadingIds = {};
+      ids.forEach(id => {
+        this.queuedRequests.add(id);
+        loadingIds[id] = true;
+      });
+      await this.processQueue();
+    };
+    this.setRequestSettled = async id => {
+      this.pendingRequests.delete(id);
+      this.settledRequests.add(id);
+      await this.processQueue();
+    };
+    this.clear = () => {
+      this.queuedRequests.clear();
+      Array.from(this.pendingRequests).forEach(id => this.clearPendingRequest(id));
+    };
+    this.clearPendingRequest = async id => {
+      this.pendingRequests.delete(id);
+      await this.processQueue();
+    };
+    this.getRequestStatus = id => {
+      if (this.pendingRequests.has(id)) {
+        return RequestStatus.PENDING;
+      }
+      if (this.queuedRequests.has(id)) {
+        return RequestStatus.QUEUED;
+      }
+      if (this.settledRequests.has(id)) {
+        return RequestStatus.SETTLED;
+      }
+      return RequestStatus.UNKNOWN;
+    };
+    this.getActiveRequestsCount = () => this.pendingRequests.size + this.queuedRequests.size;
+    this.instance = instance;
+    this.maxConcurrentRequests = maxConcurrentRequests;
+  }
+}

package/esm/internals/utils/releaseInfo.d.ts

@@ -0,0 +1 @@
+export declare const getReleaseInfo: () => any;

package/esm/internals/utils/releaseInfo.js

@@ -0,0 +1,13 @@
+import ponyfillGlobal from '@mui/utils/ponyfillGlobal';
+export const getReleaseInfo = () => {
+  const releaseInfo = "MTc0MjI0ODgwMDAwMA==";
+  if (process.env.NODE_ENV !== 'production') {
+    // A simple hack to set the value in the test environment (has no build step).
+    // eslint-disable-next-line no-useless-concat
+    if (releaseInfo === '__RELEASE' + '_INFO__') {
+      // eslint-disable-next-line no-underscore-dangle
+      return ponyfillGlobal.__MUI_RELEASE_INFO__;
+    }
+  }
+  return releaseInfo;
+};

package/esm/internals/zero-styled/index.d.ts

@@ -0,0 +1,3 @@
+import { useThemeProps } from '@mui/material/styles';
+export { styled } from '@mui/material/styles';
+export declare function createUseThemeProps(name: string): typeof useThemeProps;

package/esm/internals/zero-styled/index.js

@@ -0,0 +1,7 @@
+import { useThemeProps } from '@mui/material/styles';
+export { styled } from '@mui/material/styles';
+
+// eslint-disable-next-line @typescript-eslint/no-unused-vars
+export function createUseThemeProps(name) {
+  return useThemeProps;
+}

package/esm/package.json

@@ -0,0 +1 @@
+{"type":"module","sideEffects":false}

package/esm/themeAugmentation/components.d.ts

@@ -0,0 +1,11 @@
+import { ComponentsProps, ComponentsOverrides, ComponentsVariants } from '@mui/material/styles';
+export interface TreeViewComponents<Theme = unknown> {
+  MuiRichTreeViewPro?: {
+    defaultProps?: ComponentsProps['MuiRichTreeViewPro'];
+    styleOverrides?: ComponentsOverrides<Theme>['MuiRichTreeViewPro'];
+    variants?: ComponentsVariants<Theme>['MuiRichTreeViewPro'];
+  };
+}
+declare module '@mui/material/styles' {
+  interface Components<Theme = unknown> extends TreeViewComponents<Theme> {}
+}

package/esm/themeAugmentation/index.d.ts

@@ -0,0 +1,3 @@
+export type * from './overrides';
+export type * from './props';
+export type * from './components';

package/esm/themeAugmentation/overrides.d.ts

@@ -0,0 +1,12 @@
+import { RichTreeViewProClassKey } from "../RichTreeViewPro/index.js";
+
+// prettier-ignore
+export interface TreeViewComponentNameToClassKey {
+  MuiRichTreeViewPro: RichTreeViewProClassKey;
+}
+declare module '@mui/material/styles' {
+  interface ComponentNameToClassKey extends TreeViewComponentNameToClassKey {}
+}
+
+// disable automatic export
+export {};

package/esm/themeAugmentation/props.d.ts

@@ -0,0 +1,10 @@
+import { RichTreeViewProProps } from "../RichTreeViewPro/index.js";
+export interface TreeViewComponentsPropsList {
+  MuiRichTreeViewPro: RichTreeViewProProps<any, any>;
+}
+declare module '@mui/material/styles' {
+  interface ComponentsPropsList extends TreeViewComponentsPropsList {}
+}
+
+// disable automatic export
+export {};

package/index.d.ts

@@ -1,5 +1,5 @@
 export * from '@mui/x-tree-view/SimpleTreeView';
-export * from './RichTreeViewPro';
+export * from "./RichTreeViewPro/index.js";
 export * from '@mui/x-tree-view/TreeItem';
 export * from '@mui/x-tree-view/useTreeItem';
 export * from '@mui/x-tree-view/TreeItemIcon';
@@ -9,4 +9,4 @@ export * from '@mui/x-tree-view/TreeItemLabelInput';
 export { unstable_resetCleanupTracking } from '@mui/x-tree-view/internals';
 export * from '@mui/x-tree-view/models';
 export * from '@mui/x-tree-view/icons';
-export * from '@mui/x-tree-view/hooks';
+export * from '@mui/x-tree-view/hooks';