diff --git a/src/webapp/features/cards/lib/mutations/useRemoveCardFromLibrary.tsx b/src/webapp/features/cards/lib/mutations/useRemoveCardFromLibrary.tsx index 3a955745..507b1c61 100644 --- a/src/webapp/features/cards/lib/mutations/useRemoveCardFromLibrary.tsx +++ b/src/webapp/features/cards/lib/mutations/useRemoveCardFromLibrary.tsx @@ -2,6 +2,7 @@ import { useMutation, useQueryClient } from '@tanstack/react-query'; import { removeCardFromLibrary } from '../dal'; import { cardKeys } from '../cardKeys'; import { collectionKeys } from '@/features/collections/lib/collectionKeys'; +import { noteKeys } from '@/features/notes/lib/noteKeys'; export default function useRemoveCardFromLibrary() { const queryClient = useQueryClient(); @@ -13,6 +14,7 @@ export default function useRemoveCardFromLibrary() { onSuccess: () => { queryClient.invalidateQueries({ queryKey: cardKeys.all() }); + queryClient.invalidateQueries({ queryKey: noteKeys.all() }); queryClient.invalidateQueries({ queryKey: collectionKeys.all() }); }, }); diff --git a/src/webapp/features/notes/lib/mutations/useUpdateNote.tsx b/src/webapp/features/notes/lib/mutations/useUpdateNote.tsx index ca8444df..c25a9808 100644 --- a/src/webapp/features/notes/lib/mutations/useUpdateNote.tsx +++ b/src/webapp/features/notes/lib/mutations/useUpdateNote.tsx @@ -3,6 +3,7 @@ import { updateNoteCard } from '../dal'; import { cardKeys } from '@/features/cards/lib/cardKeys'; import { collectionKeys } from '@/features/collections/lib/collectionKeys'; import { feedKeys } from '@/features/feeds/lib/feedKeys'; +import { noteKeys } from '../noteKeys'; export default function useUpdateNote() { const queryClient = useQueryClient(); @@ -15,7 +16,11 @@ export default function useUpdateNote() { onSuccess: (data) => { queryClient.invalidateQueries({ queryKey: cardKeys.card(data.cardId) }); queryClient.invalidateQueries({ queryKey: cardKeys.infinite() }); - queryClient.invalidateQueries({ queryKey: cardKeys.infinite() }); + queryClient.invalidateQueries({ + queryKey: cardKeys.infinite(data.cardId), + }); + queryClient.invalidateQueries({ queryKey: cardKeys.all() }); + queryClient.invalidateQueries({ queryKey: noteKeys.all() }); queryClient.invalidateQueries({ queryKey: feedKeys.all() }); queryClient.invalidateQueries({ queryKey: collectionKeys.all() }); },