Updated a bit more of the way the codes work

src/pages/api/code.ts

@@ -29,7 +29,7 @@ async function handler(req: NextApiRequest, res: NextApiResponse) {
 
 	const codePromises = codes.map(async (code, index) => {
 		const codeRef = doc(db, "codes", code);
-		await setDoc(codeRef, {type, code});
+		await setDoc(codeRef, {type, code, creator: req.session.user!.id});
 
 		if (emails && emails.length > index) {
 			const transport = prepareMailer();

src/pages/api/register.ts

@@ -1,9 +1,9 @@
 import {NextApiRequest, NextApiResponse} from "next";
-import {createUserWithEmailAndPassword, getAuth, sendPasswordResetEmail} from "firebase/auth";
+import {createUserWithEmailAndPassword, getAuth} from "firebase/auth";
 import {app} from "@/firebase";
 import {sessionOptions} from "@/lib/session";
 import {withIronSessionApiRoute} from "iron-session/next";
-import {getFirestore, getDoc, doc, setDoc, deleteDoc} from "firebase/firestore";
+import {getFirestore, doc, setDoc, query, collection, where, getDocs} from "firebase/firestore";
 import {DemographicInformation, Type} from "@/interfaces/user";
 
 const auth = getAuth(app);
@@ -28,13 +28,15 @@ const DEFAULT_LEVELS = {
 async function login(req: NextApiRequest, res: NextApiResponse) {
 	const {email, password, code} = req.body as {email: string; password: string; code: string; demographicInformation: DemographicInformation};
 
-	const codeRef = await getDoc(doc(db, "codes", code));
-	if (!codeRef.exists()) {
+	const codeQuery = query(collection(db, "codes"), where("code", "==", code));
+	const codeDocs = (await getDocs(codeQuery)).docs.filter((x) => !Object.keys(x.data()).includes("userId"));
+
+	if (codeDocs.length === 0) {
 		res.status(400).json({error: "Invalid Code!"});
 		return;
 	}
 
-	const codeData = codeRef.data() as {code: string; type: Type};
+	const codeData = codeDocs[0].data() as {code: string; type: Type};
 
 	createUserWithEmailAndPassword(auth, email, password)
 		.then(async (userCredentials) => {
@@ -52,7 +54,7 @@ async function login(req: NextApiRequest, res: NextApiResponse) {
 			};
 
 			await setDoc(doc(db, "users", userId), user);
-			await deleteDoc(codeRef.ref);
+			await setDoc(codeDocs[0].ref, {userId: userId}, {merge: true});
 
 			req.session.user = {...user, id: userId};
 			await req.session.save();

src/pages/exam.tsx

@@ -33,6 +33,7 @@ import {speakingReverseMarking, writingReverseMarking} from "@/utils/score";
 import AbandonPopup from "@/components/AbandonPopup";
 import {evaluateSpeakingAnswer, evaluateWritingAnswer} from "@/utils/evaluation";
 import {useRouter} from "next/router";
+import {getExam} from "@/utils/exams";
 
 export const getServerSideProps = withIronSessionSsr(({req, res}) => {
 	const user = req.session.user;
@@ -85,10 +86,13 @@ export default function Page() {
 	useEffect(() => {
 		(async () => {
 			if (selectedModules.length > 0 && exams.length === 0) {
-				const examPromises = selectedModules.map(getExam);
+				const examPromises = selectedModules.map((module) => getExam(module, avoidRepeated));
 				Promise.all(examPromises).then((values) => {
 					if (values.every((x) => !!x)) {
 						setExams(values.map((x) => x!));
+					} else {
+						toast.error("Something went wrong, please try again");
+						setTimeout(router.reload, 500);
 					}
 				});
 			}
@@ -115,27 +119,6 @@ export default function Page() {
 		// eslint-disable-next-line react-hooks/exhaustive-deps
 	}, [selectedModules, moduleIndex, hasBeenUploaded]);
 
-	const getExam = async (module: Module): Promise<Exam | undefined> => {
-		const examRequest = await axios<Exam[]>(`/api/exam/${module}?avoidRepeated=${avoidRepeated}`);
-		if (examRequest.status !== 200) {
-			toast.error("Something went wrong!");
-			return undefined;
-		}
-
-		const newExam = examRequest.data;
-
-		switch (module) {
-			case "reading":
-				return newExam.shift() as ReadingExam;
-			case "listening":
-				return newExam.shift() as ListeningExam;
-			case "writing":
-				return newExam.shift() as WritingExam;
-			case "speaking":
-				return newExam.shift() as SpeakingExam;
-		}
-	};
-
 	const updateExamWithUserSolutions = (exam: Exam): Exam => {
 		if (exam.module === "reading" || exam.module === "listening") {
 			const parts = exam.parts.map((p) =>

src/pages/exercises.tsx

@@ -36,6 +36,7 @@ import {speakingReverseMarking, writingReverseMarking} from "@/utils/score";
 import AbandonPopup from "@/components/AbandonPopup";
 import {evaluateSpeakingAnswer, evaluateWritingAnswer} from "@/utils/evaluation";
 import {useRouter} from "next/router";
+import {getExam} from "@/utils/exams";
 
 export const getServerSideProps = withIronSessionSsr(({req, res}) => {
 	const user = req.session.user;
@@ -89,10 +90,13 @@ export default function Page() {
 	useEffect(() => {
 		(async () => {
 			if (selectedModules.length > 0 && exams.length === 0) {
-				const examPromises = selectedModules.map(getExam);
+				const examPromises = selectedModules.map((module) => getExam(module, avoidRepeated));
 				Promise.all(examPromises).then((values) => {
 					if (values.every((x) => !!x)) {
 						setExams(values.map((x) => x!));
+					} else {
+						toast.error("Something went wrong, please try again");
+						setTimeout(router.reload, 500);
 					}
 				});
 			}
@@ -119,27 +123,6 @@ export default function Page() {
 		// eslint-disable-next-line react-hooks/exhaustive-deps
 	}, [selectedModules, moduleIndex, hasBeenUploaded]);
 
-	const getExam = async (module: Module): Promise<Exam | undefined> => {
-		const examRequest = await axios<Exam[]>(`/api/exam/${module}?avoidRepeated=${avoidRepeated}`);
-		if (examRequest.status !== 200) {
-			toast.error("Something went wrong!");
-			return undefined;
-		}
-
-		const newExam = examRequest.data;
-
-		switch (module) {
-			case "reading":
-				return newExam.shift() as ReadingExam;
-			case "listening":
-				return newExam.shift() as ListeningExam;
-			case "writing":
-				return newExam.shift() as WritingExam;
-			case "speaking":
-				return newExam.shift() as SpeakingExam;
-		}
-	};
-
 	const updateExamWithUserSolutions = (exam: Exam): Exam => {
 		if (exam.module === "reading" || exam.module === "listening") {
 			const parts = exam.parts.map((p) =>

src/pages/index.tsx

@@ -18,6 +18,7 @@ import Layout from "@/components/High/Layout";
 import {calculateAverageLevel} from "@/utils/score";
 import axios from "axios";
 import DemographicInformationInput from "@/components/DemographicInformationInput";
+import moment from "moment";
 
 export const getServerSideProps = withIronSessionSsr(({req, res}) => {
 	const user = req.session.user;
@@ -51,6 +52,34 @@ export default function Home() {
 		}
 	}, [user]);
 
+	const checkIfUserExpired = () => {
+		const expirationDate = user!.subscriptionExpirationDate;
+
+		if (expirationDate === null || expirationDate === undefined) return false;
+		if (moment(expirationDate).isAfter(moment(new Date()))) return false;
+
+		return true;
+	};
+
+	if (user && (user.isDisabled || checkIfUserExpired())) {
+		return (
+			<>
+				<Head>
+					<title>EnCoach | Muscat Training Institute</title>
+					<meta
+						name="description"
+						content="A training platform for the IELTS exam provided by the Muscat Training Institute and developed by eCrop."
+					/>
+					<meta name="viewport" content="width=device-width, initial-scale=1" />
+					<link rel="icon" href="/favicon.ico" />
+				</Head>
+				<Layout user={user} navDisabled>
+					<div className="flex flex-col items-center justify-center text-center"></div>
+				</Layout>
+			</>
+		);
+	}
+
 	if (user && showDemographicInput) {
 		return (
 			<>

src/pages/register.tsx

@@ -18,7 +18,7 @@ export const getServerSideProps = (context: any) => {
 	const {code} = context.query;
 
 	return {
-		props: {code},
+		props: {code: code || null},
 	};
 };