feat(notifications): added notification system w/ interwoven refactoring

server/api/v1/notifications/[id]/index.delete.ts

@@ -0,0 +1,28 @@
+import prisma from "~/server/internal/db/database";
+
+export default defineEventHandler(async (h3) => {
+  const userId = await h3.context.session.getUserId(h3);
+  if (!userId) throw createError({ statusCode: 403 });
+
+  const notificationId = getRouterParam(h3, "id");
+  if (!notificationId)
+    throw createError({
+      statusCode: 400,
+      statusMessage: "Missing notification ID",
+    });
+
+  const notification = await prisma.notification.delete({
+    where: {
+      id: notificationId,
+      userId,
+    },
+  });
+
+  if (!notification)
+    throw createError({
+      statusCode: 400,
+      statusMessage: "Invalid notification ID",
+    });
+
+  return {};
+});

server/api/v1/notifications/[id]/index.get.ts

@@ -0,0 +1,28 @@
+import prisma from "~/server/internal/db/database";
+
+export default defineEventHandler(async (h3) => {
+  const userId = await h3.context.session.getUserId(h3);
+  if (!userId) throw createError({ statusCode: 403 });
+
+  const notificationId = getRouterParam(h3, "id");
+  if (!notificationId)
+    throw createError({
+      statusCode: 400,
+      statusMessage: "Missing notification ID",
+    });
+
+  const notification = await prisma.notification.findFirst({
+    where: {
+      id: notificationId,
+      userId,
+    },
+  });
+
+  if (!notification)
+    throw createError({
+      statusCode: 400,
+      statusMessage: "Invalid notification ID",
+    });
+
+  return notification;
+});

server/api/v1/notifications/[id]/read.post.ts

@@ -0,0 +1,31 @@
+import prisma from "~/server/internal/db/database";
+
+export default defineEventHandler(async (h3) => {
+  const userId = await h3.context.session.getUserId(h3);
+  if (!userId) throw createError({ statusCode: 403 });
+
+  const notificationId = getRouterParam(h3, "id");
+  if (!notificationId)
+    throw createError({
+      statusCode: 400,
+      statusMessage: "Missing notification ID",
+    });
+
+  const notification = await prisma.notification.update({
+    where: {
+      id: notificationId,
+      userId,
+    },
+    data: {
+      read: true,
+    },
+  });
+
+  if (!notification)
+    throw createError({
+      statusCode: 400,
+      statusMessage: "Invalid notification ID",
+    });
+
+  return notification;
+});

server/api/v1/notifications/index.get.ts

@@ -0,0 +1,17 @@
+import prisma from "~/server/internal/db/database";
+
+export default defineEventHandler(async (h3) => {
+  const userId = await h3.context.session.getUserId(h3);
+  if (!userId) throw createError({ statusCode: 403 });
+
+  const notifications = await prisma.notification.findMany({
+    where: {
+      userId,
+    },
+    orderBy: {
+      created: "desc", // Newest first
+    },
+  });
+
+  return notifications;
+});

server/api/v1/notifications/readall.post.ts

@@ -0,0 +1,17 @@
+import prisma from "~/server/internal/db/database";
+
+export default defineEventHandler(async (h3) => {
+  const userId = await h3.context.session.getUserId(h3);
+  if (!userId) throw createError({ statusCode: 403 });
+
+  await prisma.notification.updateMany({
+    where: {
+      userId,
+    },
+    data: {
+      read: true,
+    },
+  });
+
+  return;
+});

server/api/v1/notifications/ws.get.ts

@@ -0,0 +1,42 @@
+import notificationSystem from "~/server/internal/notifications";
+import session from "~/server/internal/session";
+import { parse as parseCookies } from "cookie-es";
+
+// TODO add web socket sessions for horizontal scaling
+// Peer ID to user ID
+const socketSessions: { [key: string]: string } = {};
+
+export default defineWebSocketHandler({
+  async open(peer) {
+    const cookies = peer.request?.headers?.get("Cookie");
+    if (!cookies) {
+      peer.send("unauthenticated");
+      return;
+    }
+
+    const parsedCookies = parseCookies(cookies);
+    const token = parsedCookies[session.getDropTokenCookie()];
+
+    const userId = await session.getUserIdRaw(token);
+    if (!userId) {
+      peer.send("unauthenticated");
+      return;
+    }
+
+    socketSessions[peer.id] = userId;
+
+    notificationSystem.listen(userId, peer.id, (notification) => {
+      peer.send(JSON.stringify(notification));
+    });
+  },
+  async close(peer, details) {
+    const userId = socketSessions[peer.id];
+    if (!userId) {
+      console.log(`skipping websocket close for ${peer.id}`);
+      return;
+    }
+
+    notificationSystem.unlisten(userId, peer.id);
+    delete socketSessions[peer.id];
+  },
+});