From b27c1972b1b7605abc535f0b25c40b502600f3f6 Mon Sep 17 00:00:00 2001
From: Eliezer Steinbock <3090527+elie222@users.noreply.github.com>
Date: Mon, 30 Dec 2024 19:17:51 +0200
Subject: [PATCH] Allow loading longer email history

---
 apps/web/app/(app)/stats/LoadStatsButton.tsx |  4 ++--
 apps/web/providers/StatLoaderProvider.tsx    | 21 +++++++++++++++++++-
 2 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/apps/web/app/(app)/stats/LoadStatsButton.tsx b/apps/web/app/(app)/stats/LoadStatsButton.tsx
index 09a44edec..f04274dc5 100644
--- a/apps/web/app/(app)/stats/LoadStatsButton.tsx
+++ b/apps/web/app/(app)/stats/LoadStatsButton.tsx
@@ -6,13 +6,13 @@ import { ButtonLoader } from "@/components/Loading";
 import { useStatLoader } from "@/providers/StatLoaderProvider";
 
 export function LoadStatsButton() {
-  const { isLoading, onLoad } = useStatLoader();
+  const { isLoading, onLoadBatch } = useStatLoader();
 
   return (
     <div>
       <Button
         variant="outline"
-        onClick={() => onLoad({ loadBefore: true, showToast: true })}
+        onClick={() => onLoadBatch({ loadBefore: true, showToast: true })}
         disabled={isLoading}
       >
         {isLoading ? (
diff --git a/apps/web/providers/StatLoaderProvider.tsx b/apps/web/providers/StatLoaderProvider.tsx
index ea20a2ce3..8751913fe 100644
--- a/apps/web/providers/StatLoaderProvider.tsx
+++ b/apps/web/providers/StatLoaderProvider.tsx
@@ -19,11 +19,16 @@ type Context = {
     loadBefore: boolean;
     showToast: boolean;
   }) => Promise<void>;
+  onLoadBatch: (options: {
+    loadBefore: boolean;
+    showToast: boolean;
+  }) => Promise<void>;
 };
 
 const StatLoaderContext = createContext<Context>({
   isLoading: false,
   onLoad: async () => {},
+  onLoadBatch: async () => {},
 });
 
 export const useStatLoader = () => useContext(StatLoaderContext);
@@ -69,8 +74,22 @@ export function StatLoaderProvider(props: { children: React.ReactNode }) {
     [],
   );
 
+  const onLoadBatch = useCallback(
+    async (options: { loadBefore: boolean; showToast: boolean }) => {
+      const batchSize = 50;
+      for (let i = 0; i < batchSize; i++) {
+        console.log("Loading batch", i);
+        await onLoad({
+          ...options,
+          showToast: options.showToast && i === batchSize - 1,
+        });
+      }
+    },
+    [onLoad],
+  );
+
   return (
-    <StatLoaderContext.Provider value={{ isLoading, onLoad }}>
+    <StatLoaderContext.Provider value={{ isLoading, onLoad, onLoadBatch }}>
       {props.children}
     </StatLoaderContext.Provider>
   );