refactor: Migrating to trpc instead of next's route handers

4 files changed, 191 insertions, 0 deletions

diff --git a/packages/web/server/api/client.ts b/packages/web/server/api/client.ts
new file mode 100644
index 00000000..7008e10d
--- /dev/null
+++ b/packages/web/server/api/client.ts
@@ -0,0 +1,14 @@
+import { appRouter } from "./routers/_app";
+import { getServerAuthSession } from "@/server/auth";
+import { Context, createCallerFactory } from "./trpc";
+
+export const createContext = async (): Promise<Context> => {
+  const session = await getServerAuthSession();
+  return {
+    session,
+  };
+};
+
+const createCaller = createCallerFactory(appRouter);
+
+export const api = createCaller(createContext);
diff --git a/packages/web/server/api/routers/_app.ts b/packages/web/server/api/routers/_app.ts
new file mode 100644
index 00000000..a4f9c629
--- /dev/null
+++ b/packages/web/server/api/routers/_app.ts
@@ -0,0 +1,7 @@
+import { router } from "../trpc";
+import { bookmarksAppRouter } from "./bookmarks";
+export const appRouter = router({
+  bookmarks: bookmarksAppRouter,
+});
+// export type definition of API
+export type AppRouter = typeof appRouter;
diff --git a/packages/web/server/api/routers/bookmarks.ts b/packages/web/server/api/routers/bookmarks.ts
new file mode 100644
index 00000000..0b97563f
--- /dev/null
+++ b/packages/web/server/api/routers/bookmarks.ts
@@ -0,0 +1,139 @@
+import { z } from "zod";
+import { authedProcedure, router } from "../trpc";
+import {
+  ZBookmark,
+  ZBookmarkContent,
+  zBookmarkSchema,
+  zGetBookmarksRequestSchema,
+  zGetBookmarksResponseSchema,
+  zNewBookmarkRequestSchema,
+  zUpdateBookmarksRequestSchema,
+} from "@/lib/types/api/bookmarks";
+import { prisma } from "@remember/db";
+import { LinkCrawlerQueue } from "@remember/shared/queues";
+
+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 const bookmarksAppRouter = router({
+  bookmarkLink: authedProcedure
+    .input(zNewBookmarkRequestSchema)
+    .output(zBookmarkSchema)
+    .mutation(async ({ input, ctx }) => {
+      const { url } = input;
+      const userId = ctx.user.id;
+
+      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);
+    }),
+
+  updateBookmark: authedProcedure
+    .input(zUpdateBookmarksRequestSchema)
+    .output(zBookmarkSchema)
+    .mutation(async ({ input, ctx }) => {
+      const bookmark = await prisma.bookmark.update({
+        where: {
+          id: input.bookmarkId,
+          userId: ctx.user.id,
+        },
+        data: {
+          archived: input.archived,
+          favourited: input.favourited,
+        },
+        select: defaultBookmarkFields,
+      });
+      return toZodSchema(bookmark);
+    }),
+
+  deleteBookmark: authedProcedure
+    .input(z.object({ bookmarkId: z.string() }))
+    .mutation(async ({ input, ctx }) => {
+      await prisma.bookmark.delete({
+        where: {
+          id: input.bookmarkId,
+          userId: ctx.user.id,
+        },
+      });
+    }),
+  getBookmarks: authedProcedure
+    .input(zGetBookmarksRequestSchema)
+    .output(zGetBookmarksResponseSchema)
+    .query(async ({ input, ctx }) => {
+      const bookmarks = (
+        await prisma.bookmark.findMany({
+          where: {
+            userId: ctx.user.id,
+            archived: input.archived,
+            favourited: input.favourited,
+          },
+          orderBy: {
+            createdAt: "desc",
+          },
+          select: defaultBookmarkFields,
+        })
+      ).map(toZodSchema);
+
+      return { bookmarks };
+    }),
+});
diff --git a/packages/web/server/api/trpc.ts b/packages/web/server/api/trpc.ts
new file mode 100644
index 00000000..82aa2d18
--- /dev/null
+++ b/packages/web/server/api/trpc.ts
@@ -0,0 +1,31 @@
+import { TRPCError, initTRPC } from "@trpc/server";
+import { Session } from "next-auth";
+
+export type Context = {
+  session: Session | null;
+};
+
+// Avoid exporting the entire t-object
+// since it's not very descriptive.
+// For instance, the use of a t variable
+// is common in i18n libraries.
+const t = initTRPC.context<Context>().create();
+export const createCallerFactory = t.createCallerFactory;
+// Base router and procedure helpers
+export const router = t.router;
+export const procedure = t.procedure;
+export const publicProcedure = t.procedure;
+
+export const authedProcedure = t.procedure.use(function isAuthed(opts) {
+  const user = opts.ctx.session?.user;
+
+  if (!user) {
+    throw new TRPCError({ code: "UNAUTHORIZED" });
+  }
+
+  return opts.next({
+    ctx: {
+      user,
+    },
+  });
+});