From a35c85545e7332031d39efab2d2ad37ddd0033f5 Mon Sep 17 00:00:00 2001
From: Tiago Ribeiro <tiago.ribeiro@ecrop.dev>
Date: Mon, 29 Jan 2024 21:02:52 +0000
Subject: [PATCH] Updated the code to set the participant's expiration date to
 use the corporate's one if it is better

---
 src/pages/(admin)/Lists/GroupList.tsx |  17 +++-
 src/pages/api/groups/[id].ts          | 126 ++++++++++++++++----------
 src/pages/api/groups/index.ts         |  82 +++++++++++------
 src/utils/groups.be.ts                |  53 +++++++++++
 src/utils/groups.ts                   |  24 +++--
 5 files changed, 216 insertions(+), 86 deletions(-)
 create mode 100644 src/utils/groups.be.ts

diff --git a/src/pages/(admin)/Lists/GroupList.tsx b/src/pages/(admin)/Lists/GroupList.tsx
index f3a02401..35e9f106 100644
--- a/src/pages/(admin)/Lists/GroupList.tsx
+++ b/src/pages/(admin)/Lists/GroupList.tsx
@@ -39,6 +39,8 @@ const CreatePanel = ({ user, users, group, onClose }: CreateDialogProps) => {
   const [participants, setParticipants] = useState<string[]>(
     group?.participants || [],
   );
+  const [isLoading, setIsLoading] = useState(false);
+
   const { openFilePicker, filesContent, clear } = useFilePicker({
     accept: ".xlsx",
     multiple: false,
@@ -47,6 +49,8 @@ const CreatePanel = ({ user, users, group, onClose }: CreateDialogProps) => {
 
   useEffect(() => {
     if (filesContent.length > 0) {
+      setIsLoading(true);
+
       const file = filesContent[0];
       readXlsxFile(file.content).then((rows) => {
         const emails = uniq(
@@ -64,6 +68,7 @@ const CreatePanel = ({ user, users, group, onClose }: CreateDialogProps) => {
         if (emails.length === 0) {
           toast.error("Please upload an Excel file containing e-mails!");
           clear();
+          setIsLoading(false);
           return;
         }
 
@@ -86,16 +91,20 @@ const CreatePanel = ({ user, users, group, onClose }: CreateDialogProps) => {
             : "Added all students found in the file you've provided!",
           { toastId: "upload-success" },
         );
+        setIsLoading(false);
       });
     }
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }, [filesContent, user.type, users]);
 
   const submit = () => {
+    setIsLoading(true);
+
     if (name !== group?.name && (name === "Students" || name === "Teachers")) {
       toast.error(
         "That group name is reserved and cannot be used, please enter another one.",
       );
+      setIsLoading(false);
       return;
     }
 
@@ -113,7 +122,10 @@ const CreatePanel = ({ user, users, group, onClose }: CreateDialogProps) => {
         toast.error("Something went wrong, please try again later!");
         return false;
       })
-      .finally(onClose);
+      .finally(() => {
+        setIsLoading(false);
+        onClose();
+      });
   };
 
   return (
@@ -178,6 +190,7 @@ const CreatePanel = ({ user, users, group, onClose }: CreateDialogProps) => {
               <Button
                 className="w-full max-w-[300px]"
                 onClick={openFilePicker}
+                isLoading={isLoading}
                 variant="outline"
               >
                 {filesContent.length === 0
@@ -193,6 +206,7 @@ const CreatePanel = ({ user, users, group, onClose }: CreateDialogProps) => {
           variant="outline"
           color="red"
           className="w-full max-w-[200px]"
+          isLoading={isLoading}
           onClick={onClose}
         >
           Cancel
@@ -200,6 +214,7 @@ const CreatePanel = ({ user, users, group, onClose }: CreateDialogProps) => {
         <Button
           className="w-full max-w-[200px]"
           onClick={submit}
+          isLoading={isLoading}
           disabled={!name}
         >
           Submit
diff --git a/src/pages/api/groups/[id].ts b/src/pages/api/groups/[id].ts
index 7cbaef92..a9fda9f4 100644
--- a/src/pages/api/groups/[id].ts
+++ b/src/pages/api/groups/[id].ts
@@ -1,80 +1,108 @@
 // Next.js API route support: https://nextjs.org/docs/api-routes/introduction
-import type {NextApiRequest, NextApiResponse} from "next";
-import {app} from "@/firebase";
-import {getFirestore, collection, getDocs, getDoc, doc, deleteDoc, setDoc} from "firebase/firestore";
-import {withIronSessionApiRoute} from "iron-session/next";
-import {sessionOptions} from "@/lib/session";
-import {Group} from "@/interfaces/user";
+import type { NextApiRequest, NextApiResponse } from "next";
+import { app } from "@/firebase";
+import {
+  getFirestore,
+  collection,
+  getDocs,
+  getDoc,
+  doc,
+  deleteDoc,
+  setDoc,
+} from "firebase/firestore";
+import { withIronSessionApiRoute } from "iron-session/next";
+import { sessionOptions } from "@/lib/session";
+import { Group } from "@/interfaces/user";
+import { updateExpiryDateOnGroup } from "@/utils/groups.be";
 
 const db = getFirestore(app);
 
 export default withIronSessionApiRoute(handler, sessionOptions);
 
 async function handler(req: NextApiRequest, res: NextApiResponse) {
-	if (req.method === "GET") return await get(req, res);
-	if (req.method === "DELETE") return await del(req, res);
-	if (req.method === "PATCH") return await patch(req, res);
+  if (req.method === "GET") return await get(req, res);
+  if (req.method === "DELETE") return await del(req, res);
+  if (req.method === "PATCH") return await patch(req, res);
 
-	res.status(404).json(undefined);
+  res.status(404).json(undefined);
 }
 
 async function get(req: NextApiRequest, res: NextApiResponse) {
-	if (!req.session.user) {
-		res.status(401).json({ok: false});
-		return;
-	}
+  if (!req.session.user) {
+    res.status(401).json({ ok: false });
+    return;
+  }
 
-	const {id} = req.query as {id: string};
+  const { id } = req.query as { id: string };
 
-	const snapshot = await getDoc(doc(db, "groups", id));
+  const snapshot = await getDoc(doc(db, "groups", id));
 
-	if (snapshot.exists()) {
-		res.status(200).json({...snapshot.data(), id: snapshot.id});
-	} else {
-		res.status(404).json(undefined);
-	}
+  if (snapshot.exists()) {
+    res.status(200).json({ ...snapshot.data(), id: snapshot.id });
+  } else {
+    res.status(404).json(undefined);
+  }
 }
 
 async function del(req: NextApiRequest, res: NextApiResponse) {
-	if (!req.session.user) {
-		res.status(401).json({ok: false});
-		return;
-	}
+  if (!req.session.user) {
+    res.status(401).json({ ok: false });
+    return;
+  }
 
-	const {id} = req.query as {id: string};
+  const { id } = req.query as { id: string };
 
-	const snapshot = await getDoc(doc(db, "groups", id));
-	const group = {...snapshot.data(), id: snapshot.id} as Group;
+  const snapshot = await getDoc(doc(db, "groups", id));
+  const group = { ...snapshot.data(), id: snapshot.id } as Group;
 
-	const user = req.session.user;
-	if (user.type === "admin" || user.type === "developer" || user.id === group.admin) {
-		await deleteDoc(snapshot.ref);
+  const user = req.session.user;
+  if (
+    user.type === "admin" ||
+    user.type === "developer" ||
+    user.id === group.admin
+  ) {
+    await deleteDoc(snapshot.ref);
 
-		res.status(200).json({ok: true});
-		return;
-	}
+    res.status(200).json({ ok: true });
+    return;
+  }
 
-	res.status(403).json({ok: false});
+  res.status(403).json({ ok: false });
 }
 
 async function patch(req: NextApiRequest, res: NextApiResponse) {
-	if (!req.session.user) {
-		res.status(401).json({ok: false});
-		return;
-	}
+  if (!req.session.user) {
+    res.status(401).json({ ok: false });
+    return;
+  }
 
-	const {id} = req.query as {id: string};
+  const { id } = req.query as { id: string };
 
-	const snapshot = await getDoc(doc(db, "groups", id));
-	const group = {...snapshot.data(), id: snapshot.id} as Group;
+  const snapshot = await getDoc(doc(db, "groups", id));
+  const group = { ...snapshot.data(), id: snapshot.id } as Group;
 
-	const user = req.session.user;
-	if (user.type === "admin" || user.type === "developer" || user.id === group.admin) {
-		await setDoc(snapshot.ref, req.body, {merge: true});
+  const user = req.session.user;
+  if (
+    user.type === "admin" ||
+    user.type === "developer" ||
+    user.id === group.admin
+  ) {
+    if ("participants" in req.body) {
+      const newParticipants = (req.body.participants as string[]).filter(
+        (x) => !group.participants.includes(x),
+      );
+      await Promise.all(
+        newParticipants.map(
+          async (p) => await updateExpiryDateOnGroup(p, group.admin),
+        ),
+      );
+    }
 
-		res.status(200).json({ok: true});
-		return;
-	}
+    await setDoc(snapshot.ref, req.body, { merge: true });
 
-	res.status(403).json({ok: false});
+    res.status(200).json({ ok: true });
+    return;
+  }
+
+  res.status(403).json({ ok: false });
 }
diff --git a/src/pages/api/groups/index.ts b/src/pages/api/groups/index.ts
index f39ec3bd..db32967b 100644
--- a/src/pages/api/groups/index.ts
+++ b/src/pages/api/groups/index.ts
@@ -1,45 +1,73 @@
 // Next.js API route support: https://nextjs.org/docs/api-routes/introduction
-import type {NextApiRequest, NextApiResponse} from "next";
-import {app} from "@/firebase";
-import {getFirestore, collection, getDocs, setDoc, doc, query, where} from "firebase/firestore";
-import {withIronSessionApiRoute} from "iron-session/next";
-import {sessionOptions} from "@/lib/session";
-import {Group} from "@/interfaces/user";
-import {v4} from "uuid";
+import type { NextApiRequest, NextApiResponse } from "next";
+import { app } from "@/firebase";
+import {
+  getFirestore,
+  collection,
+  getDocs,
+  setDoc,
+  doc,
+  query,
+  where,
+} from "firebase/firestore";
+import { withIronSessionApiRoute } from "iron-session/next";
+import { sessionOptions } from "@/lib/session";
+import { Group } from "@/interfaces/user";
+import { v4 } from "uuid";
+import { updateExpiryDateOnGroup } from "@/utils/groups.be";
 
 const db = getFirestore(app);
 
 export default withIronSessionApiRoute(handler, sessionOptions);
 
 async function handler(req: NextApiRequest, res: NextApiResponse) {
-	if (!req.session.user) {
-		res.status(401).json({ok: false});
-		return;
-	}
+  if (!req.session.user) {
+    res.status(401).json({ ok: false });
+    return;
+  }
 
-	if (req.method === "GET") await get(req, res);
-	if (req.method === "POST") await post(req, res);
+  if (req.method === "GET") await get(req, res);
+  if (req.method === "POST") await post(req, res);
 }
 
 async function get(req: NextApiRequest, res: NextApiResponse) {
-	const {admin, participant} = req.query as {admin: string; participant: string};
+  const { admin, participant } = req.query as {
+    admin: string;
+    participant: string;
+  };
 
-	const queryConstraints = [
-		...(admin ? [where("admin", "==", admin)] : []),
-		...(participant ? [where("participants", "array-contains", participant)] : []),
-	];
-	const snapshot = await getDocs(queryConstraints.length > 0 ? query(collection(db, "groups"), ...queryConstraints) : collection(db, "groups"));
-	const groups = snapshot.docs.map((doc) => ({
-		id: doc.id,
-		...doc.data(),
-	})) as Group[];
+  const queryConstraints = [
+    ...(admin ? [where("admin", "==", admin)] : []),
+    ...(participant
+      ? [where("participants", "array-contains", participant)]
+      : []),
+  ];
+  const snapshot = await getDocs(
+    queryConstraints.length > 0
+      ? query(collection(db, "groups"), ...queryConstraints)
+      : collection(db, "groups"),
+  );
+  const groups = snapshot.docs.map((doc) => ({
+    id: doc.id,
+    ...doc.data(),
+  })) as Group[];
 
-	res.status(200).json(groups);
+  res.status(200).json(groups);
 }
 
 async function post(req: NextApiRequest, res: NextApiResponse) {
-	const body = req.body as Group;
+  const body = req.body as Group;
 
-	await setDoc(doc(db, "groups", v4()), {name: body.name, admin: body.admin, participants: body.participants});
-	res.status(200).json({ok: true});
+  await Promise.all(
+    body.participants.map(
+      async (p) => await updateExpiryDateOnGroup(p, body.admin),
+    ),
+  );
+
+  await setDoc(doc(db, "groups", v4()), {
+    name: body.name,
+    admin: body.admin,
+    participants: body.participants,
+  });
+  res.status(200).json({ ok: true });
 }
diff --git a/src/utils/groups.be.ts b/src/utils/groups.be.ts
new file mode 100644
index 00000000..8dae61c8
--- /dev/null
+++ b/src/utils/groups.be.ts
@@ -0,0 +1,53 @@
+import { app } from "@/firebase";
+import { CorporateUser, StudentUser, TeacherUser } from "@/interfaces/user";
+import { doc, getDoc, getFirestore, setDoc } from "firebase/firestore";
+import moment from "moment";
+
+const db = getFirestore(app);
+
+export const updateExpiryDateOnGroup = async (
+  participantID: string,
+  corporateID: string,
+) => {
+  const corporateRef = await getDoc(doc(db, "users", corporateID));
+  const participantRef = await getDoc(doc(db, "users", participantID));
+
+  if (!corporateRef.exists() || !participantRef.exists()) return;
+
+  const corporate = {
+    ...corporateRef.data(),
+    id: corporateRef.id,
+  } as CorporateUser;
+  const participant = { ...participantRef.data(), id: participantRef.id } as
+    | StudentUser
+    | TeacherUser;
+
+  if (
+    corporate.type !== "corporate" ||
+    (participant.type !== "student" && participant.type !== "teacher")
+  )
+    return;
+
+  if (
+    !corporate.subscriptionExpirationDate ||
+    !participant.subscriptionExpirationDate
+  ) {
+    return await setDoc(
+      doc(db, "users", participant.id),
+      { subscriptionExpirationDate: null },
+      { merge: true },
+    );
+  }
+
+  const corporateDate = moment(corporate.subscriptionExpirationDate);
+  const participantDate = moment(participant.subscriptionExpirationDate);
+
+  if (corporateDate.isAfter(participantDate))
+    return await setDoc(
+      doc(db, "users", participant.id),
+      { subscriptionExpirationDate: corporateDate.toISOString() },
+      { merge: true },
+    );
+
+  return;
+};
diff --git a/src/utils/groups.ts b/src/utils/groups.ts
index 5b7c894d..3742f2d7 100644
--- a/src/utils/groups.ts
+++ b/src/utils/groups.ts
@@ -1,18 +1,24 @@
-import {CorporateUser, Group, User} from "@/interfaces/user";
+import { CorporateUser, Group, User } from "@/interfaces/user";
 import axios from "axios";
 
 export const isUserFromCorporate = async (userID: string) => {
-	const groups = (await axios.get<Group[]>(`/api/groups?participant=${userID}`)).data;
-	const users = (await axios.get<User[]>("/api/users/list")).data;
+  const groups = (await axios.get<Group[]>(`/api/groups?participant=${userID}`))
+    .data;
+  const users = (await axios.get<User[]>("/api/users/list")).data;
 
-	const adminTypes = groups.map((g) => users.find((u) => u.id === g.admin)?.type);
-	return adminTypes.includes("corporate");
+  const adminTypes = groups.map(
+    (g) => users.find((u) => u.id === g.admin)?.type,
+  );
+  return adminTypes.includes("corporate");
 };
 
 export const getUserCorporate = async (userID: string) => {
-	const groups = (await axios.get<Group[]>(`/api/groups?participant=${userID}`)).data;
-	const users = (await axios.get<User[]>("/api/users/list")).data;
+  const groups = (await axios.get<Group[]>(`/api/groups?participant=${userID}`))
+    .data;
+  const users = (await axios.get<User[]>("/api/users/list")).data;
 
-	const admins = groups.map((g) => users.find((u) => u.id === g.admin)).filter((x) => x?.type === "corporate");
-	return admins.length > 0 ? (admins[0] as CorporateUser) : undefined;
+  const admins = groups
+    .map((g) => users.find((u) => u.id === g.admin))
+    .filter((x) => x?.type === "corporate");
+  return admins.length > 0 ? (admins[0] as CorporateUser) : undefined;
 };