Use truncateAndInvalidate strategy

src/state/queries/bookmarks/useBookmarkMutation.ts

@@ -3,7 +3,7 @@ import {useMutation, useQueryClient} from '@tanstack/react-query'
 import {isNetworkError} from '#/lib/strings/errors'
 import {logger} from '#/logger'
 import {updatePostShadow} from '#/state/cache/post-shadow'
-import {bookmarksQueryKeyRoot} from '#/state/queries/bookmarks/useBookmarksQuery'
+import {truncateAndInvalidate as invalidateBookmarksQuery} from '#/state/queries/bookmarks/useBookmarksQuery'
 import {useAgent} from '#/state/session'
 
 type MutationArgs =
@@ -30,7 +30,7 @@ export function useBookmarkMutation() {
       }
     },
     onSuccess() {
-      qc.invalidateQueries({queryKey: [bookmarksQueryKeyRoot]})
+      invalidateBookmarksQuery(qc)
     },
     onError(e, args) {
       if (args.action === 'create') {

src/state/queries/bookmarks/useBookmarksQuery.ts

@@ -1,6 +1,7 @@
 import {type AppBskyBookmarkGetBookmarks} from '@atproto/api'
 import {
   type InfiniteData,
+  type QueryClient,
   type QueryKey,
   useInfiniteQuery,
 } from '@tanstack/react-query'
@@ -31,3 +32,19 @@ export function useBookmarksQuery() {
     getNextPageParam: lastPage => lastPage.cursor,
   })
 }
+
+export async function truncateAndInvalidate(qc: QueryClient) {
+  qc.setQueriesData<InfiniteData<AppBskyBookmarkGetBookmarks.OutputSchema>>(
+    {queryKey: [bookmarksQueryKeyRoot]},
+    data => {
+      if (data) {
+        return {
+          pageParams: data.pageParams.slice(0, 1),
+          pages: data.pages.slice(0, 1),
+        }
+      }
+      return data
+    },
+  )
+  return qc.invalidateQueries({queryKey: [bookmarksQueryKeyRoot]})
+}