From e7d40dd94a99a79ec05c219a41097c4609b9c44a Mon Sep 17 00:00:00 2001
From: delangle <flaviendelangle@gmail.com>
Date: Wed, 14 Aug 2024 10:48:22 +0200
Subject: [PATCH] Work

---
 .../useTreeViewVirtualization/useTreeViewVirtualization.ts     | 3 ++-
 .../useTreeViewVirtualization.types.ts                         | 2 ++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/packages/x-tree-view-pro/src/internals/plugins/useTreeViewVirtualization/useTreeViewVirtualization.ts b/packages/x-tree-view-pro/src/internals/plugins/useTreeViewVirtualization/useTreeViewVirtualization.ts
index 2667c8e803f7c..54f8628590f5e 100644
--- a/packages/x-tree-view-pro/src/internals/plugins/useTreeViewVirtualization/useTreeViewVirtualization.ts
+++ b/packages/x-tree-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/packages/x-tree-view-pro/src/internals/plugins/useTreeViewVirtualization/useTreeViewVirtualization.types.ts b/packages/x-tree-view-pro/src/internals/plugins/useTreeViewVirtualization/useTreeViewVirtualization.types.ts
index f6be11f1d6437..b89e8d07d7d72 100644
--- a/packages/x-tree-view-pro/src/internals/plugins/useTreeViewVirtualization/useTreeViewVirtualization.types.ts
+++ b/packages/x-tree-view-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];
 }>;