diff --git a/packages/@react-spectrum/s2/src/TreeView.tsx b/packages/@react-spectrum/s2/src/TreeView.tsx index 5d1f0c26e5d..2855356f4e8 100644 --- a/packages/@react-spectrum/s2/src/TreeView.tsx +++ b/packages/@react-spectrum/s2/src/TreeView.tsx @@ -15,6 +15,7 @@ import {ActionMenuContext} from './ActionMenu'; import { Button, ButtonContext, + CheckboxContext, ListLayout, Provider, TreeItemProps as RACTreeItemProps, @@ -26,6 +27,7 @@ import { TreeLoadMoreItem, TreeLoadMoreItemProps, useContextProps, + useSlottedContext, Virtualizer } from 'react-aria-components'; import {centerBaseline} from './CenterBaseline'; @@ -333,7 +335,7 @@ export const TreeViewItemContent = (props: TreeViewItemContentProps): ReactNode let { children } = props; - let {isDetached, isEmphasized} = useContext(InternalTreeContext); + let {isDetached} = useContext(InternalTreeContext); let scale = useScale(); return ( @@ -349,12 +351,7 @@ export const TreeViewItemContent = (props: TreeViewItemContentProps): ReactNode return (
{selectionMode !== 'none' && selectionBehavior === 'toggle' && ( - // TODO: add transition? -
- -
+ )}
+
+ +
+
+ ); +} + const centeredWrapper = style({ display: 'flex', alignItems: 'center', diff --git a/packages/@react-spectrum/s2/stories/TreeView.stories.tsx b/packages/@react-spectrum/s2/stories/TreeView.stories.tsx index 8059d6cf7af..694512440ae 100644 --- a/packages/@react-spectrum/s2/stories/TreeView.stories.tsx +++ b/packages/@react-spectrum/s2/stories/TreeView.stories.tsx @@ -75,7 +75,7 @@ const TreeExampleStatic = (args: TreeViewProps): ReactElement => (