separate out save entity hook

hubmapconsortium · Feb 11, 2025 · 29e3a05 · 29e3a05
1 parent 93c492c
commit 29e3a05
Showing 1 changed file with 29 additions and 24 deletions.
diff --git a/context/app/static/js/components/savedLists/hooks.ts b/context/app/static/js/components/savedLists/hooks.ts
@@ -352,37 +352,41 @@ function useSavedEntitiesActions({ savedEntities }: { savedEntities: SavedEntiti
     setSuccessAlert(SavedListsSuccessAlertType.DeletedEntity);
   }
 
-  function useHandleSaveEntity({ entityUUID }: { entityUUID: string }) {
-    const {
-      entity: { entity_type },
-    } = useFlaskDataContext();
-    const [entityData] = useEntitiesData([entityUUID], ['hubmap_id']);
-
-    return useCallback(() => {
-      handleSaveEntities({ entityUUIDs: new Set([entityUUID]) }).catch((error) => {
-        console.error(error);
-      });
-
-      if (!entityData?.length) {
-        return;
-      }
-
-      trackEvent({
-        category: SavedListsEventCategories.EntityDetailPage(entity_type),
-        action: 'Save Entity to Items',
-        label: entityData[0].hubmap_id,
-      });
-    }, [entityUUID, entityData, entity_type]);
-  }
-
   return {
     handleSaveEntities,
     handleDeleteEntities,
-    useHandleSaveEntity,
     isUpdating,
   };
 }
 
+function useHandleSaveEntity({ entityUUID }: { entityUUID: string }) {
+  const { savedEntities } = useListSavedListsAndEntities();
+  const { handleSaveEntities } = useSavedEntitiesActions({ savedEntities });
+
+  const {
+    entity: { entity_type },
+  } = useFlaskDataContext();
+  const [entityData] = useEntitiesData([entityUUID], ['hubmap_id']);
+
+  const handleSaveEntity = useEventCallback(() => {
+    handleSaveEntities({ entityUUIDs: new Set([entityUUID]) }).catch((error) => {
+      console.error(error);
+    });
+
+    if (!entityData?.length) {
+      return;
+    }
+
+    trackEvent({
+      category: SavedListsEventCategories.EntityDetailPage(entity_type),
+      action: 'Save Entity to Items',
+      label: entityData[0].hubmap_id,
+    });
+  });
+
+  return handleSaveEntity;
+}
+
 export default function useSavedLists() {
   const { isLoading, savedLists, savedEntities, savedListsAndEntities } = useListSavedListsAndEntities();
 
@@ -391,6 +395,7 @@ export default function useSavedLists() {
     savedLists,
     savedEntities,
     savedListsAndEntities,
+    useHandleSaveEntity,
     ...useSavedListsActions({ savedListsAndEntities }),
     ...useSavedEntitiesActions({ savedEntities }),
   };