UNPKG

@mui/x-tree-view

Version:

The community edition of the MUI X Tree View components.

69 lines (68 loc) 2.6 kB
'use client'; import _extends from "@babel/runtime/helpers/esm/extends"; import _objectWithoutPropertiesLoose from "@babel/runtime/helpers/esm/objectWithoutPropertiesLoose"; const _excluded = ["ownerState"]; import * as React from 'react'; import useSlotProps from '@mui/utils/useSlotProps'; import { fastObjectShallowCompare } from '@mui/x-internals/fastObjectShallowCompare'; import { TreeItem } from "../../TreeItem/index.js"; import { useSelector } from "../hooks/useSelector.js"; import { selectorItemMeta, selectorItemOrderedChildrenIds } from "../plugins/useTreeViewItems/useTreeViewItems.selectors.js"; import { useTreeViewContext } from "../TreeViewProvider/index.js"; import { jsx as _jsx } from "react/jsx-runtime"; const RichTreeViewItemsContext = /*#__PURE__*/React.createContext(null); if (process.env.NODE_ENV !== "production") RichTreeViewItemsContext.displayName = "RichTreeViewItemsContext"; const WrappedTreeItem = /*#__PURE__*/React.memo(function WrappedTreeItem({ itemSlot, itemSlotProps, itemId }) { const renderItemForRichTreeView = React.useContext(RichTreeViewItemsContext); const { store } = useTreeViewContext(); const itemMeta = useSelector(store, selectorItemMeta, itemId); const children = useSelector(store, selectorItemOrderedChildrenIds, itemId); const Item = itemSlot ?? TreeItem; const _useSlotProps = useSlotProps({ elementType: Item, externalSlotProps: itemSlotProps, additionalProps: { label: itemMeta?.label, id: itemMeta?.idAttribute, itemId }, ownerState: { itemId, label: itemMeta?.label } }), itemProps = _objectWithoutPropertiesLoose(_useSlotProps, _excluded); return /*#__PURE__*/_jsx(Item, _extends({}, itemProps, { children: children?.map(renderItemForRichTreeView) })); }, fastObjectShallowCompare); if (process.env.NODE_ENV !== "production") WrappedTreeItem.displayName = "WrappedTreeItem"; export function RichTreeViewItems(props) { const { slots, slotProps } = props; const { store } = useTreeViewContext(); const itemSlot = slots?.item; const itemSlotProps = slotProps?.item; const items = useSelector(store, selectorItemOrderedChildrenIds, null); const renderItem = React.useCallback(itemId => { return /*#__PURE__*/_jsx(WrappedTreeItem, { itemSlot: itemSlot, itemSlotProps: itemSlotProps, itemId: itemId }, itemId); }, [itemSlot, itemSlotProps]); return /*#__PURE__*/_jsx(RichTreeViewItemsContext.Provider, { value: renderItem, children: items.map(renderItem) }); }