From 731d2dfbea39aa140ccb6d2d2cabd49186320299 Mon Sep 17 00:00:00 2001
From: Mohamed Bassem <me@mbassem.com>
Date: Sun, 27 Oct 2024 00:12:11 +0000
Subject: feature: Add a summarize with AI button for links

---
 packages/shared-react/hooks/bookmarks.ts | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'packages/shared-react/hooks/bookmarks.ts')

diff --git a/packages/shared-react/hooks/bookmarks.ts b/packages/shared-react/hooks/bookmarks.ts
index cba4d107..f4dd203c 100644
--- a/packages/shared-react/hooks/bookmarks.ts
+++ b/packages/shared-react/hooks/bookmarks.ts
@@ -96,6 +96,21 @@ export function useUpdateBookmarkText(
   });
 }
 
+export function useSummarizeBookmark(
+  ...opts: Parameters<typeof api.bookmarks.summarizeBookmark.useMutation>
+) {
+  const apiUtils = api.useUtils();
+  return api.bookmarks.summarizeBookmark.useMutation({
+    ...opts[0],
+    onSuccess: (res, req, meta) => {
+      apiUtils.bookmarks.getBookmarks.invalidate();
+      apiUtils.bookmarks.searchBookmarks.invalidate();
+      apiUtils.bookmarks.getBookmark.invalidate({ bookmarkId: req.bookmarkId });
+      return opts[0]?.onSuccess?.(res, req, meta);
+    },
+  });
+}
+
 export function useRecrawlBookmark(
   ...opts: Parameters<typeof api.bookmarks.recrawlBookmark.useMutation>
 ) {
-- 
cgit v1.2.3-70-g09d2