/api/groups and /api/invites refactored, fixed some inserts/updates in which I didn't include the id

src/pages/api/invites/accept/[id].ts

@@ -1,17 +1,15 @@
 // Next.js API route support: https://nextjs.org/docs/api-routes/introduction
-import type {NextApiRequest, NextApiResponse} from "next";
-import {app} from "@/firebase";
-import {getFirestore, getDoc, doc, deleteDoc, setDoc, getDocs, collection, where, query} from "firebase/firestore";
-import {withIronSessionApiRoute} from "iron-session/next";
-import {sessionOptions} from "@/lib/session";
-import {Ticket} from "@/interfaces/ticket";
-import {Invite} from "@/interfaces/invite";
-import {CorporateUser, Group, User} from "@/interfaces/user";
-import {v4} from "uuid";
-import {sendEmail} from "@/email";
-import {updateExpiryDateOnGroup} from "@/utils/groups.be";
+import type { NextApiRequest, NextApiResponse } from "next";
+import client from "@/lib/mongodb";
+import { withIronSessionApiRoute } from "iron-session/next";
+import { sessionOptions } from "@/lib/session";
+import { Invite } from "@/interfaces/invite";
+import { CorporateUser, Group, User } from "@/interfaces/user";
+import { v4 } from "uuid";
+import { sendEmail } from "@/email";
+import { updateExpiryDateOnGroup } from "@/utils/groups.be";
 
-const db = getFirestore(app);
+const db = client.db(process.env.MONGODB_DB);
 
 export default withIronSessionApiRoute(handler, sessionOptions);
 
@@ -22,12 +20,7 @@ async function handler(req: NextApiRequest, res: NextApiResponse) {
 }
 
 async function addToInviterGroup(user: User, invitedBy: User) {
-	const invitedByGroupsRef = await getDocs(query(collection(db, "groups"), where("admin", "==", invitedBy.id)));
-	const invitedByGroups = invitedByGroupsRef.docs.map((g) => ({
-		...g.data(),
-		id: g.id,
-	})) as Group[];
-
+	const invitedByGroups = await db.collection("groups").find<Group>({ admin: invitedBy.id }).toArray();
 	const typeGroupName = user.type === "student" ? "Students" : user.type === "teacher" ? "Teachers" : undefined;
 
 	if (typeGroupName) {
@@ -38,14 +31,18 @@ async function addToInviterGroup(user: User, invitedBy: User) {
 			participants: [],
 			disableEditing: true,
 		};
-		await setDoc(
-			doc(db, "groups", typeGroup.id),
+
+		await db.collection("groups").updateOne(
+			{ id: typeGroup.id },
 			{
-				...typeGroup,
-				participants: [...typeGroup.participants.filter((x) => x !== user.id), user.id],
+				$set: {
+					...typeGroup,
+					participants: [...typeGroup.participants.filter((x) => x !== user.id), user.id],
+				},
 			},
-			{merge: true},
+			{ upsert: true }
 		);
+
 	}
 
 	const invitationsGroup: Group = invitedByGroups.find((g) => g.name === "Invited") || {
@@ -55,54 +52,58 @@ async function addToInviterGroup(user: User, invitedBy: User) {
 		participants: [],
 		disableEditing: true,
 	};
-	await setDoc(
-		doc(db, "groups", invitationsGroup.id),
+
+	await db.collection("groups").updateOne(
+		{ id: invitationsGroup.id },
 		{
-			...invitationsGroup,
-			participants: [...invitationsGroup.participants.filter((x) => x !== user.id), user.id],
-		},
-		{
-			merge: true,
+			$set: {
+				...invitationsGroup,
+				participants: [...invitationsGroup.participants.filter((x) => x !== user.id), user.id],
+			}
 		},
+		{ upsert: true }
 	);
 }
 
 async function deleteFromPreviousCorporateGroups(user: User, invitedBy: User) {
-	const corporatesRef = await getDocs(query(collection(db, "users"), where("type", "==", "corporate")));
-	const corporates = (corporatesRef.docs.map((x) => ({...x.data(), id: x.id})) as CorporateUser[]).filter((x) => x.id !== invitedBy.id);
+	const corporatesRef = await db.collection("users").find<CorporateUser>({ type: "corporate" }).toArray();
+	const corporates = corporatesRef.filter((x) => x.id !== invitedBy.id);
 
-	const userGroupsRef = await getDocs(query(collection(db, "groups"), where("participants", "array-contains", user.id)));
-	const userGroups = userGroupsRef.docs.map((x) => ({...x.data(), id: x.id})) as Group[];
+	const userGroups = await db.collection("groups").find<Group>({
+		participants: user.id
+	}).toArray();
 
 	const corporateGroups = userGroups.filter((x) => corporates.map((c) => c.id).includes(x.admin));
 	await Promise.all(
 		corporateGroups.map(async (group) => {
-			await setDoc(doc(db, "groups", group.id), {participants: group.participants.filter((x) => x !== user.id)}, {merge: true});
+			await db.collection("groups").updateOne(
+				{ id: group.id },
+				{ $set: { participants: group.participants.filter((x) => x !== user.id) } },
+				{ upsert: true }
+			);
 		}),
 	);
 }
 
 async function get(req: NextApiRequest, res: NextApiResponse) {
 	if (!req.session.user) {
-		res.status(401).json({ok: false});
+		res.status(401).json({ ok: false });
 		return;
 	}
 
-	const {id} = req.query as {id: string};
-	const snapshot = await getDoc(doc(db, "invites", id));
+	const { id } = req.query as { id: string };
+	const invite = await db.collection("invites").findOne<Invite>({ id: id});
 
-	if (snapshot.exists()) {
-		const invite = {...snapshot.data(), id: snapshot.id} as Invite;
-		if (invite.to !== req.session.user.id) return res.status(403).json({ok: false});
+	if (invite) {
+		if (invite.to !== req.session.user.id) return res.status(403).json({ ok: false });
 
-		await deleteDoc(snapshot.ref);
-		const invitedByRef = await getDoc(doc(db, "users", invite.from));
-		if (!invitedByRef.exists()) return res.status(404).json({ok: false});
+		await db.collection("invites").deleteOne({ id: id });
+
+		const invitedBy = await db.collection("users").findOne<User>({ id: invite.from});
+		if (!invitedBy) return res.status(404).json({ ok: false });
 
 		await updateExpiryDateOnGroup(invite.to, invite.from);
 
-		const invitedBy = {...invitedByRef.data(), id: invitedByRef.id} as User;
-
 		if (invitedBy.type === "corporate") await deleteFromPreviousCorporateGroups(req.session.user, invitedBy);
 		await addToInviterGroup(req.session.user, invitedBy);
 
@@ -122,7 +123,7 @@ async function get(req: NextApiRequest, res: NextApiResponse) {
 			console.log(e);
 		}
 
-		res.status(200).json({ok: true});
+		res.status(200).json({ ok: true });
 	} else {
 		res.status(404).json(undefined);
 	}