Update useHatsTree hook to accept safeAddress parameter and adjust re…

…setStore conditional to just reset store when safeAddress changes
decentdao · Dec 11, 2024 · 5adc033 · 5adc033
1 parent d1f5bb0
commit 5adc033
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/hooks/DAO/loaders/useHatsTree.ts b/src/hooks/DAO/loaders/useHatsTree.ts
@@ -19,7 +19,7 @@ import { getValue, setValue } from '../../utils/cache/useLocalStorage';
 
 const hatsSubgraphClient = new HatsSubgraphClient({});
 
-const useHatsTree = () => {
+const useHatsTree = ({ safeAddress }: { safeAddress: Address | undefined }) => {
   const { t } = useTranslation('roles');
   const {
     governanceContracts: {
@@ -293,10 +293,10 @@ const useHatsTree = () => {
   }, [hatsTree, updateRolesWithStreams, getPaymentStreams, streamsFetched]);
 
   useEffect(() => {
-    if (!hatsTreeId && !!hatsTree) {
+    if (safeAddress === undefined) {
       resetHatsStore();
     }
-  }, [resetHatsStore, hatsTree, hatsTreeId]);
+  }, [resetHatsStore, safeAddress]);
 };
 
 export { useHatsTree };
diff --git a/src/pages/dao/SafeController.tsx b/src/pages/dao/SafeController.tsx
@@ -43,7 +43,7 @@ export function SafeController() {
   useAzoriusListeners();
 
   useKeyValuePairs();
-  useHatsTree();
+  useHatsTree({ safeAddress });
 
   // the order of the if blocks of these next three error states matters
   if (invalidQuery) {