Work

mui · Aug 14, 2024 · e7d40dd · e7d40dd
1 parent 0e3e43b
commit e7d40dd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/...ree-view-pro/src/internals/plugins/useTreeViewVirtualization/useTreeViewVirtualization.ts b/...ree-view-pro/src/internals/plugins/useTreeViewVirtualization/useTreeViewVirtualization.ts
@@ -159,7 +159,7 @@ export const useTreeViewVirtualization: TreeViewPlugin<UseTreeViewVirtualization
 
     const itemsToRenderWithVisibleContent = new Set(itemsToRender);
 
-    const addItem = (itemId: TreeViewItemId | null) => {
+    const addItem = (itemId: TreeViewItemId | null | undefined) => {
       if (itemId == null || itemsToRenderWithVisibleContent.has(itemId)) {
         return;
       }
@@ -168,6 +168,7 @@ export const useTreeViewVirtualization: TreeViewPlugin<UseTreeViewVirtualization
     };
 
     addItem(state.focusedItemId);
+    addItem(state.itemsReordering?.draggedItemId);
 
     const getPropsFromItemId = (
       id: TreeViewItemId,

diff --git a/...ew-pro/src/internals/plugins/useTreeViewVirtualization/useTreeViewVirtualization.types.ts b/...ew-pro/src/internals/plugins/useTreeViewVirtualization/useTreeViewVirtualization.types.ts
@@ -7,6 +7,7 @@ import {
   UseTreeViewExpansionSignature,
   UseTreeViewFocusSignature,
 } from '@mui/x-tree-view/internals';
+import { UseTreeViewItemsReorderingSignature } from '../useTreeViewItemsReordering';
 
 export interface UseTreeViewVirtualizationInstance {
   getDimensions: () => UseTreeViewVirtualizationState['virtualization'];
@@ -96,4 +97,5 @@ export type UseTreeViewVirtualizationSignature = TreeViewPluginSignature<{
     UseTreeViewExpansionSignature,
     UseTreeViewFocusSignature,
   ];
+  optionalDependencies: [UseTreeViewItemsReorderingSignature];
 }>;