[refactor] Extract the bookmark model to be a high level model to support other type of bookmarks

1 files changed, 92 insertions, 0 deletions

diff --git a/packages/web/lib/services/bookmarks.ts b/packages/web/lib/services/bookmarks.ts
new file mode 100644
index 00000000..3c9929bc
--- /dev/null
+++ b/packages/web/lib/services/bookmarks.ts
@@ -0,0 +1,92 @@
+import { LinkCrawlerQueue } from "@remember/shared/queues";
+import prisma from "@remember/db";
+import { ZBookmark, ZBookmarkContent } from "@/lib/types/api/bookmarks";
+
+const defaultBookmarkFields = {
+  id: true,
+  favourited: true,
+  archived: true,
+  createdAt: true,
+  link: {
+    select: {
+      url: true,
+      title: true,
+      description: true,
+      imageUrl: true,
+      favicon: true,
+    },
+  },
+  tags: {
+    include: {
+      tag: true,
+    },
+  },
+};
+
+async function dummyPrismaReturnType() {
+  const x = await prisma.bookmark.findFirstOrThrow({
+    select: defaultBookmarkFields,
+  });
+  return x;
+}
+
+function toZodSchema(
+  bookmark: Awaited<ReturnType<typeof dummyPrismaReturnType>>,
+): ZBookmark {
+  const { tags, link, ...rest } = bookmark;
+
+  let content: ZBookmarkContent;
+  if (link) {
+    content = { type: "link", ...link };
+  } else {
+    throw new Error("Unknown content type");
+  }
+
+  return {
+    tags: tags.map((t) => t.tag),
+    content,
+    ...rest,
+  };
+}
+
+export async function deleteBookmark(bookmarkId: string, userId: string) {
+  await prisma.bookmark.delete({
+    where: {
+      id: bookmarkId,
+      userId,
+    },
+  });
+}
+
+export async function bookmarkLink(url: string, userId: string) {
+  const bookmark = await prisma.bookmark.create({
+    data: {
+      link: {
+        create: {
+          url,
+        },
+      },
+      userId,
+    },
+    select: defaultBookmarkFields,
+  });
+
+  // Enqueue crawling request
+  await LinkCrawlerQueue.add("crawl", {
+    bookmarkId: bookmark.id,
+    url: url,
+  });
+
+  return toZodSchema(bookmark);
+}
+
+export async function getBookmarks(userId: string) {
+  return (
+    await prisma.bookmark.findMany({
+      where: {
+        userId,
+      },
+      select: defaultBookmarkFields,
+    })
+  ).map(toZodSchema);
+}