From 2950b3b8c062f9b80dcc445c410ed35823bded16 Mon Sep 17 00:00:00 2001
From: Eric Bailey <git@esb.lol>
Date: Mon, 25 Aug 2025 14:32:32 -0500
Subject: [PATCH] Hook up shadow and mutation

---
 .../PostControls/BookmarkButton.tsx           | 52 ++++++++++++++++---
 .../PostControls/PostControlButton.tsx        | 12 ++++-
 src/components/PostControls/index.tsx         |  2 +-
 src/state/cache/post-shadow.ts                |  3 ++
 .../queries/bookmarks/useBookmarkMutation.ts  | 46 ++++++++++++++++
 5 files changed, 106 insertions(+), 9 deletions(-)
 create mode 100644 src/state/queries/bookmarks/useBookmarkMutation.ts

diff --git a/src/components/PostControls/BookmarkButton.tsx b/src/components/PostControls/BookmarkButton.tsx
index f018bb11db..558f4fcc6c 100644
--- a/src/components/PostControls/BookmarkButton.tsx
+++ b/src/components/PostControls/BookmarkButton.tsx
@@ -1,28 +1,68 @@
-import {memo} from 'react'
-import {type AppBskyFeedDefs, type AppBskyFeedPost} from '@atproto/api'
+import {memo, useCallback} from 'react'
+import {type AppBskyFeedDefs} from '@atproto/api'
 import {msg} from '@lingui/macro'
 import {useLingui} from '@lingui/react'
 import type React from 'react'
 
+import {useCleanError} from '#/lib/hooks/useCleanError'
 import {type Shadow} from '#/state/cache/post-shadow'
-import {Bookmark} from '#/components/icons/Bookmark'
+import {useBookmarkMutation} from '#/state/queries/bookmarks/useBookmarkMutation'
+import {useTheme} from '#/alf'
+import {Bookmark, BookmarkFilled} from '#/components/icons/Bookmark'
 import {PostControlButton, PostControlButtonIcon} from './PostControlButton'
 
 export const BookmarkButton = memo(function BookmarkButton({
+  post,
   big,
 }: {
   post: Shadow<AppBskyFeedDefs.PostView>
   big?: boolean
-  record: AppBskyFeedPost.Record
 }): React.ReactNode {
+  const t = useTheme()
   const {_} = useLingui()
+  const {mutateAsync: bookmark} = useBookmarkMutation()
+  const cleanError = useCleanError()
+
+  const {uri, cid, viewer} = post
+  const isBookmarked = !!viewer?.bookmarked
+
+  const onHandlePress = useCallback(async () => {
+    try {
+      if (isBookmarked) {
+        await bookmark({
+          action: 'delete',
+          uri,
+        })
+        // TODO toast
+      } else {
+        await bookmark({
+          action: 'create',
+          uri,
+          cid,
+        })
+        // TODO toast
+      }
+    } catch (e) {
+      const {raw, clean} = cleanError(e)
+      console.log(clean || raw || e)
+      // TODO toast
+    }
+  }, [uri, cid, isBookmarked, bookmark, cleanError])
 
   return (
     <PostControlButton
       testID="postBookmarkBtn"
       big={big}
-      label={_(msg`Bookmark this post`)}>
-      <PostControlButtonIcon icon={Bookmark} />
+      label={
+        isBookmarked
+          ? _(msg`Remove this post from your bookmarks`)
+          : _(msg`Save this post to your bookmarks`)
+      }
+      onPress={onHandlePress}>
+      <PostControlButtonIcon
+        fill={isBookmarked ? t.palette.primary_500 : undefined}
+        icon={isBookmarked ? BookmarkFilled : Bookmark}
+      />
     </PostControlButton>
   )
 })
diff --git a/src/components/PostControls/PostControlButton.tsx b/src/components/PostControls/PostControlButton.tsx
index ae69b1322d..807f24d2d3 100644
--- a/src/components/PostControls/PostControlButton.tsx
+++ b/src/components/PostControls/PostControlButton.tsx
@@ -102,12 +102,20 @@ export function PostControlButton({
 
 export function PostControlButtonIcon({
   icon: Comp,
-}: {
+  style,
+  ...rest
+}: SVGIconProps & {
   icon: React.ComponentType<SVGIconProps>
 }) {
   const {big, color} = useContext(PostControlContext)
 
-  return <Comp style={[color, a.pointer_events_none]} width={big ? 22 : 18} />
+  return (
+    <Comp
+      style={[color, a.pointer_events_none, style]}
+      {...rest}
+      width={big ? 22 : 18}
+    />
+  )
 }
 
 export function PostControlButtonText({style, ...props}: TextProps) {
diff --git a/src/components/PostControls/index.tsx b/src/components/PostControls/index.tsx
index 278f54e422..27f12f372c 100644
--- a/src/components/PostControls/index.tsx
+++ b/src/components/PostControls/index.tsx
@@ -284,7 +284,7 @@ let PostControls = ({
           a.flex_1,
           big ? a.gap_sm : a.gap_xs,
         ]}>
-        <BookmarkButton post={post} big={big} record={record} />
+        <BookmarkButton post={post} big={big} />
         <ShareMenuButton
           testID="postShareBtn"
           post={post}
diff --git a/src/state/cache/post-shadow.ts b/src/state/cache/post-shadow.ts
index 8cc3dca1a4..345ada79aa 100644
--- a/src/state/cache/post-shadow.ts
+++ b/src/state/cache/post-shadow.ts
@@ -25,6 +25,7 @@ export interface PostShadow {
   embed: AppBskyEmbedRecord.View | AppBskyEmbedRecordWithMedia.View | undefined
   pinned: boolean
   optimisticReplyCount: number | undefined
+  bookmarked: boolean | undefined
 }
 
 export const POST_TOMBSTONE = Symbol('PostTombstone')
@@ -132,6 +133,8 @@ function mergeShadow(
       like: 'likeUri' in shadow ? shadow.likeUri : post.viewer?.like,
       repost: 'repostUri' in shadow ? shadow.repostUri : post.viewer?.repost,
       pinned: 'pinned' in shadow ? shadow.pinned : post.viewer?.pinned,
+      bookmarked:
+        'bookmarked' in shadow ? shadow.bookmarked : post.viewer?.bookmarked,
     },
   })
 }
diff --git a/src/state/queries/bookmarks/useBookmarkMutation.ts b/src/state/queries/bookmarks/useBookmarkMutation.ts
new file mode 100644
index 0000000000..467e7f4dda
--- /dev/null
+++ b/src/state/queries/bookmarks/useBookmarkMutation.ts
@@ -0,0 +1,46 @@
+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 {useAgent} from '#/state/session'
+
+type MutationArgs =
+  | {action: 'create'; uri: string; cid: string}
+  | {action: 'delete'; uri: string}
+
+export function useBookmarkMutation() {
+  const qc = useQueryClient()
+  const agent = useAgent()
+
+  return useMutation({
+    async mutationFn(args: MutationArgs) {
+      if (args.action === 'create') {
+        updatePostShadow(qc, args.uri, {bookmarked: true})
+        await agent.app.bsky.bookmark.createBookmark({
+          uri: args.uri,
+          cid: args.cid,
+        })
+      } else if (args.action === 'delete') {
+        updatePostShadow(qc, args.uri, {bookmarked: false})
+        await agent.app.bsky.bookmark.deleteBookmark({
+          uri: args.uri,
+        })
+      }
+    },
+    onError(e, args) {
+      if (args.action === 'create') {
+        updatePostShadow(qc, args.uri, {bookmarked: false})
+      } else if (args.action === 'delete') {
+        updatePostShadow(qc, args.uri, {bookmarked: true})
+      }
+
+      if (!isNetworkError(e)) {
+        logger.error('bookmark mutation failed', {
+          bookmarkAction: args.action,
+          safeMessage: e,
+        })
+      }
+    },
+  })
+}