feature(web): Allow uploading directly into lists/tags. Fixes #69

1 files changed, 14 insertions, 1 deletions

diff --git a/packages/shared-react/hooks/lists.ts b/packages/shared-react/hooks/lists.ts
index 5cfcd194..f4b19c3c 100644
--- a/packages/shared-react/hooks/lists.ts
+++ b/packages/shared-react/hooks/lists.ts
@@ -1,5 +1,18 @@
 import { api } from "../trpc";
 
+export function useAddBookmarkToList(
+  ...opts: Parameters<typeof api.lists.removeFromList.useMutation>
+) {
+  const apiUtils = api.useUtils();
+  return api.lists.addToList.useMutation({
+    ...opts,
+    onSuccess: (res, req, meta) => {
+      apiUtils.bookmarks.getBookmarks.invalidate({ listId: req.listId });
+      return opts[0]?.onSuccess?.(res, req, meta);
+    },
+  });
+}
+
 export function useRemoveBookmarkFromList(
   ...opts: Parameters<typeof api.lists.removeFromList.useMutation>
 ) {
@@ -8,7 +21,7 @@ export function useRemoveBookmarkFromList(
     ...opts,
     onSuccess: (res, req, meta) => {
       apiUtils.bookmarks.getBookmarks.invalidate({ listId: req.listId });
-      opts[0]?.onSuccess?.(res, req, meta);
+      return opts[0]?.onSuccess?.(res, req, meta);
     },
   });
 }