From 1342eb0ee66f842fa59e9c34f1a678c5ea5b87cd Mon Sep 17 00:00:00 2001
From: Jackson Harper <jacksonh@gmail.com>
Date: Fri, 6 Oct 2023 17:50:16 +0800
Subject: [PATCH] Better caching on the left menu

---
 .../templates/homeFeed/LibraryFilterMenu.tsx  | 58 ++++++++++++++-----
 1 file changed, 45 insertions(+), 13 deletions(-)

diff --git a/packages/web/components/templates/homeFeed/LibraryFilterMenu.tsx b/packages/web/components/templates/homeFeed/LibraryFilterMenu.tsx
index 71a6c0c4c..0324c8aa5 100644
--- a/packages/web/components/templates/homeFeed/LibraryFilterMenu.tsx
+++ b/packages/web/components/templates/homeFeed/LibraryFilterMenu.tsx
@@ -1,4 +1,4 @@
-import { ReactNode, useMemo } from 'react'
+import { ReactNode, useEffect, useMemo } from 'react'
 import { StyledText } from '../../elements/StyledText'
 import { Box, HStack, SpanBox, VStack } from '../../elements/LayoutPrimitives'
 import { Button } from '../../elements/Button'
@@ -33,10 +33,47 @@ type LibraryFilterMenuProps = {
 }
 
 export function LibraryFilterMenu(props: LibraryFilterMenuProps): JSX.Element {
-  const { labels, isLoading: labelsLoading } = useGetLabelsQuery()
-  const { savedSearches, isLoading: searchesLoading } = useGetSavedSearchQuery()
-  const { subscriptions, isLoading: subscriptionsLoading } =
-    useGetSubscriptionsQuery()
+  const [labels, setLabels] = usePersistedState<Label[]>({
+    key: 'menu-labels',
+    isSessionStorage: false,
+    initialValue: [],
+  })
+  const [savedSearches, setSavedSearches] = usePersistedState<SavedSearch[]>({
+    key: 'menu-searches',
+    isSessionStorage: false,
+    initialValue: [],
+  })
+  const [subscriptions, setSubscriptions] = usePersistedState<Subscription[]>({
+    key: 'menu-subscriptions',
+    isSessionStorage: false,
+    initialValue: [],
+  })
+  const { labels: networkLabels, isLoading: labelsLoading } =
+    useGetLabelsQuery()
+  const { savedSearches: networkSearches, isLoading: searchesLoading } =
+    useGetSavedSearchQuery()
+  const {
+    subscriptions: networkSubscriptions,
+    isLoading: subscriptionsLoading,
+  } = useGetSubscriptionsQuery()
+
+  useEffect(() => {
+    if (!labelsLoading) {
+      setLabels(networkLabels)
+    }
+  }, [setLabels, networkLabels, labelsLoading])
+
+  useEffect(() => {
+    if (!subscriptionsLoading) {
+      setSubscriptions(networkSubscriptions)
+    }
+  }, [setSubscriptions, networkLabels, subscriptionsLoading])
+
+  useEffect(() => {
+    if (!searchesLoading) {
+      setSavedSearches(networkSearches ?? [])
+    }
+  }, [setSavedSearches, networkSearches, searchesLoading])
 
   return (
     <>
@@ -72,14 +109,9 @@ export function LibraryFilterMenu(props: LibraryFilterMenuProps): JSX.Element {
         >
           <LogoBox />
         </Box>
-
-        {!labelsLoading && !searchesLoading && !subscriptionsLoading && (
-          <>
-            <SavedSearches {...props} savedSearches={savedSearches} />
-            <Subscriptions {...props} subscriptions={subscriptions} />
-            <Labels {...props} labels={labels} />
-          </>
-        )}
+        <SavedSearches {...props} savedSearches={savedSearches} />
+        <Subscriptions {...props} subscriptions={subscriptions} />
+        <Labels {...props} labels={labels} />
         <Box css={{ height: '250px ' }} />
       </Box>
       {/* This spacer pushes library content to the right of