From 1cb0a1d72aac55bc936c1974b59b962665ffd8c9 Mon Sep 17 00:00:00 2001
From: jeffvli <jeffvictorli@gmail.com>
Date: Thu, 20 Jul 2023 00:39:06 -0700
Subject: [PATCH] Add initial data fetch function to grid

---
 .../components/virtual-grid/virtual-infinite-grid.tsx         | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/renderer/components/virtual-grid/virtual-infinite-grid.tsx b/src/renderer/components/virtual-grid/virtual-infinite-grid.tsx
index 9663367f..4a8d9eae 100644
--- a/src/renderer/components/virtual-grid/virtual-infinite-grid.tsx
+++ b/src/renderer/components/virtual-grid/virtual-infinite-grid.tsx
@@ -26,6 +26,7 @@ interface VirtualGridProps
     cardRows: CardRow<any>[];
     display?: ListDisplayType;
     fetchFn: (options: { columnCount: number; skip: number; take: number }) => Promise<any>;
+    fetchInitialData?: () => any;
     handleFavorite?: (options: {
         id: string[];
         isFavorite: boolean;
@@ -56,6 +57,7 @@ export const VirtualInfiniteGrid = forwardRef(
             handlePlayQueueAdd,
             minimumBatchSize,
             fetchFn,
+            fetchInitialData,
             loading,
             initialScrollOffset,
             handleFavorite,
@@ -67,7 +69,7 @@ export const VirtualInfiniteGrid = forwardRef(
         const listRef = useRef<any>(null);
         const loader = useRef<InfiniteLoader>(null);
 
-        const [itemData, setItemData] = useState<any[]>([]);
+        const [itemData, setItemData] = useState<any[]>(fetchInitialData?.() || []);
 
         const { itemHeight, rowCount, columnCount } = useMemo(() => {
             const itemsPerRow = itemSize;