It is now possible to generate and save both Reading and Writing exams

src/pages/(generation)/ReadingGeneration.tsx

@@ -1,12 +1,16 @@
 import Input from "@/components/Low/Input";
-import {ReadingPart} from "@/interfaces/exam";
+import {ReadingExam, ReadingPart} from "@/interfaces/exam";
+import useExamStore from "@/stores/examStore";
+import {getExamById} from "@/utils/exams";
 import {convertCamelCaseToReadable} from "@/utils/string";
 import {Tab} from "@headlessui/react";
 import axios from "axios";
 import clsx from "clsx";
+import {useRouter} from "next/router";
 import {useState} from "react";
 import {BsArrowRepeat} from "react-icons/bs";
 import {toast} from "react-toastify";
+import {v4} from "uuid";
 
 const PartTab = ({part, types, index, setPart}: {part?: ReadingPart; types: string[]; index: number; setPart: (part?: ReadingPart) => void}) => {
 	const [topic, setTopic] = useState("");
@@ -80,6 +84,13 @@ const ReadingGeneration = () => {
 	const [part2, setPart2] = useState<ReadingPart>();
 	const [part3, setPart3] = useState<ReadingPart>();
 	const [types, setTypes] = useState<string[]>([]);
+	const [isLoading, setIsLoading] = useState(false);
+	const [resultingExam, setResultingExam] = useState<ReadingExam>();
+
+	const router = useRouter();
+
+	const setExams = useExamStore((state) => state.setExams);
+	const setSelectedModules = useExamStore((state) => state.setSelectedModules);
 
 	const availableTypes = [
 		{type: "fillBlanks", label: "Fill the Blanks"},
@@ -90,6 +101,57 @@ const ReadingGeneration = () => {
 
 	const toggleType = (type: string) => setTypes((prev) => (prev.includes(type) ? [...prev.filter((x) => x !== type)] : [...prev, type]));
 
+	const loadExam = async (examId: string) => {
+		const exam = await getExamById("reading", examId.trim());
+		if (!exam) {
+			toast.error("Unknown Exam ID! Please make sure you selected the right module and entered the right exam ID", {
+				toastId: "invalid-exam-id",
+			});
+
+			return;
+		}
+
+		setExams([exam]);
+		setSelectedModules(["reading"]);
+
+		router.push("/exercises");
+	};
+
+	const submitExam = () => {
+		if (!part1 || !part2 || !part3) {
+			toast.error("Please generate all three passages before submitting");
+			return;
+		}
+
+		setIsLoading(true);
+		const exam: ReadingExam = {
+			parts: [part1, part2, part3],
+			isDiagnostic: false,
+			minTimer: 60,
+			module: "reading",
+			id: v4(),
+			type: "academic",
+		};
+
+		axios
+			.post(`/api/exam/reading`, exam)
+			.then((result) => {
+				console.log(`Generated Exam ID: ${result.data.id}`);
+				toast.success("This new exam has been generated successfully! Check the ID in our browser's console.");
+				setResultingExam(result.data);
+
+				setPart1(undefined);
+				setPart2(undefined);
+				setPart3(undefined);
+				setTypes([]);
+			})
+			.catch((error) => {
+				console.log(error);
+				toast.error("Something went wrong while generating, please try again later.");
+			})
+			.finally(() => setIsLoading(false));
+	};
+
 	return (
 		<>
 			<div className="flex flex-col gap-3">
@@ -156,17 +218,38 @@ const ReadingGeneration = () => {
 					))}
 				</Tab.Panels>
 			</Tab.Group>
-			<button
-				disabled={!part1 || !part2 || !part3}
-				data-tip="Please generate all three passages"
-				className={clsx(
-					"bg-ielts-reading/70 border border-ielts-reading text-white w-full max-w-[200px] px-6 py-6 rounded-xl h-[70px] self-end",
-					"hover:bg-ielts-reading disabled:bg-ielts-reading/40 disabled:cursor-not-allowed",
-					"transition ease-in-out duration-300",
-					(!part1 || !part2 || !part3) && "tooltip",
-				)}>
-				Submit
-			</button>
+			<div className="w-full flex justify-end gap-4">
+				{resultingExam && (
+					<button
+						disabled={isLoading}
+						onClick={() => loadExam(resultingExam.id)}
+						className={clsx(
+							"bg-white border border-ielts-reading text-ielts-reading w-full max-w-[200px] rounded-xl h-[70px] self-end",
+							"hover:bg-ielts-reading hover:text-white disabled:bg-ielts-reading/40 disabled:cursor-not-allowed",
+							"transition ease-in-out duration-300",
+						)}>
+						Perform Exam
+					</button>
+				)}
+				<button
+					disabled={!part1 || !part2 || !part3 || isLoading}
+					data-tip="Please generate all three passages"
+					onClick={submitExam}
+					className={clsx(
+						"bg-ielts-reading/70 border border-ielts-reading text-white w-full max-w-[200px] rounded-xl h-[70px] self-end",
+						"hover:bg-ielts-reading disabled:bg-ielts-reading/40 disabled:cursor-not-allowed",
+						"transition ease-in-out duration-300",
+						(!part1 || !part2 || !part3) && "tooltip",
+					)}>
+					{isLoading ? (
+						<div className="flex items-center justify-center">
+							<BsArrowRepeat className="text-white animate-spin" size={25} />
+						</div>
+					) : (
+						"Submit"
+					)}
+				</button>
+			</div>
 		</>
 	);
 };

src/pages/(generation)/WritingGeneration.tsx

@@ -1,63 +1,148 @@
 import Input from "@/components/Low/Input";
+import {WritingExam} from "@/interfaces/exam";
+import useExamStore from "@/stores/examStore";
+import {getExamById} from "@/utils/exams";
 import {Tab} from "@headlessui/react";
 import axios from "axios";
 import clsx from "clsx";
+import {useRouter} from "next/router";
 import {useState} from "react";
 import {BsArrowRepeat} from "react-icons/bs";
 import {toast} from "react-toastify";
+import {v4} from "uuid";
+
+const TaskTab = ({task, index, setTask}: {task?: string; index: number; setTask: (task: string) => void}) => {
+	const [isLoading, setIsLoading] = useState(false);
+
+	const generate = () => {
+		setIsLoading(true);
+		axios
+			.get(`/api/exam/writing/generate/writing_task${index}_general`)
+			.then((result) => setTask(result.data.question))
+			.catch((error) => {
+				console.log(error);
+				toast.error("Something went wrong!");
+			})
+			.finally(() => setIsLoading(false));
+	};
+
+	return (
+		<Tab.Panel className="w-full bg-ielts-writing/20 min-h-[600px] h-full rounded-xl p-6 flex flex-col gap-4">
+			<div className="flex gap-4 items-end">
+				<button
+					onClick={generate}
+					disabled={isLoading}
+					className={clsx(
+						"bg-ielts-writing/70 border border-ielts-writing text-white w-full px-6 py-6 rounded-xl h-[70px]",
+						"hover:bg-ielts-writing disabled:bg-ielts-writing/40 disabled:cursor-not-allowed",
+						"transition ease-in-out duration-300",
+					)}>
+					{isLoading ? (
+						<div className="flex items-center justify-center">
+							<BsArrowRepeat className="text-white animate-spin" size={25} />
+						</div>
+					) : (
+						"Generate"
+					)}
+				</button>
+			</div>
+			{isLoading && (
+				<div className="w-fit h-fit mt-12 self-center animate-pulse flex flex-col gap-8 items-center">
+					<span className={clsx("loading loading-infinity w-32 text-ielts-writing")} />
+					<span className={clsx("font-bold text-2xl text-ielts-writing")}>Generating...</span>
+				</div>
+			)}
+			{task && (
+				<div className="flex flex-col gap-2 w-full overflow-y-scroll scrollbar-hide">
+					<span className="w-full h-96">{task}</span>
+				</div>
+			)}
+		</Tab.Panel>
+	);
+};
 
 const WritingGeneration = () => {
 	const [task1, setTask1] = useState<string>();
 	const [task2, setTask2] = useState<string>();
+	const [isLoading, setIsLoading] = useState(false);
+	const [resultingExam, setResultingExam] = useState<WritingExam>();
 
-	const TaskTab = ({task, index, setTask}: {task?: string; index: number; setTask: (task: string) => void}) => {
-		const [isLoading, setIsLoading] = useState(false);
+	const router = useRouter();
 
-		const generate = () => {
-			setIsLoading(true);
-			axios
-				.get(`/api/exam/writing/generate/writing_task${index}_general`)
-				.then((result) => setTask(result.data.question))
-				.catch((error) => {
-					console.log(error);
-					toast.error("Something went wrong!");
-				})
-				.finally(() => setIsLoading(false));
+	const setExams = useExamStore((state) => state.setExams);
+	const setSelectedModules = useExamStore((state) => state.setSelectedModules);
+
+	const loadExam = async (examId: string) => {
+		const exam = await getExamById("writing", examId.trim());
+		if (!exam) {
+			toast.error("Unknown Exam ID! Please make sure you selected the right module and entered the right exam ID", {
+				toastId: "invalid-exam-id",
+			});
+
+			return;
+		}
+
+		setExams([exam]);
+		setSelectedModules(["writing"]);
+
+		router.push("/exercises");
+	};
+
+	const submitExam = () => {
+		if (!task1 || !task2) {
+			toast.error("Please generate all tasks before submitting");
+			return;
+		}
+
+		setIsLoading(true);
+		const exam: WritingExam = {
+			isDiagnostic: false,
+			minTimer: 60,
+			module: "writing",
+			exercises: [
+				{
+					id: v4(),
+					type: "writing",
+					prefix: `You should spend about 20 minutes on this task.`,
+					prompt: task1,
+					userSolutions: [],
+					suffix: "You should write at least 150 words.",
+					wordCounter: {
+						limit: 150,
+						type: "min",
+					},
+				},
+				{
+					id: v4(),
+					type: "writing",
+					prefix: `You should spend about 40 minutes on this task.`,
+					prompt: task2,
+					userSolutions: [],
+					suffix: "You should write at least 250 words.",
+					wordCounter: {
+						limit: 250,
+						type: "min",
+					},
+				},
+			],
+			id: v4(),
 		};
 
-		return (
-			<Tab.Panel className="w-full bg-ielts-writing/20 min-h-[600px] h-full rounded-xl p-6 flex flex-col gap-4">
-				<div className="flex gap-4 items-end">
-					<button
-						onClick={generate}
-						disabled={isLoading}
-						className={clsx(
-							"bg-ielts-writing/70 border border-ielts-writing text-white w-full px-6 py-6 rounded-xl h-[70px]",
-							"hover:bg-ielts-writing disabled:bg-ielts-writing/40 disabled:cursor-not-allowed",
-							"transition ease-in-out duration-300",
-						)}>
-						{isLoading ? (
-							<div className="flex items-center justify-center">
-								<BsArrowRepeat className="text-white animate-spin" size={25} />
-							</div>
-						) : (
-							"Generate"
-						)}
-					</button>
-				</div>
-				{isLoading && (
-					<div className="w-fit h-fit mt-12 self-center animate-pulse flex flex-col gap-8 items-center">
-						<span className={clsx("loading loading-infinity w-32 text-ielts-writing")} />
-						<span className={clsx("font-bold text-2xl text-ielts-writing")}>Generating...</span>
-					</div>
-				)}
-				{task && (
-					<div className="flex flex-col gap-2 w-full overflow-y-scroll scrollbar-hide">
-						<span className="w-full h-96">{task}</span>
-					</div>
-				)}
-			</Tab.Panel>
-		);
+		axios
+			.post(`/api/exam/writing`, exam)
+			.then((result) => {
+				console.log(`Generated Exam ID: ${result.data.id}`);
+				toast.success("This new exam has been generated successfully! Check the ID in our browser's console.");
+				setResultingExam(result.data);
+
+				setTask1(undefined);
+				setTask2(undefined);
+			})
+			.catch((error) => {
+				console.log(error);
+				toast.error("Something went wrong while generating, please try again later.");
+			})
+			.finally(() => setIsLoading(false));
 	};
 
 	return (
@@ -96,17 +181,38 @@ const WritingGeneration = () => {
 					))}
 				</Tab.Panels>
 			</Tab.Group>
-			<button
-				disabled={!task1 || !task2}
-				data-tip="Please generate all three passages"
-				className={clsx(
-					"bg-ielts-writing/70 border border-ielts-writing text-white w-full max-w-[200px] px-6 py-6 rounded-xl h-[70px] self-end",
-					"hover:bg-ielts-writing disabled:bg-ielts-writing/40 disabled:cursor-not-allowed",
-					"transition ease-in-out duration-300",
-					(!task1 || !task2) && "tooltip",
-				)}>
-				Submit
-			</button>
+			<div className="w-full flex justify-end gap-4">
+				{resultingExam && (
+					<button
+						disabled={isLoading}
+						onClick={() => loadExam(resultingExam.id)}
+						className={clsx(
+							"bg-white border border-ielts-writing text-ielts-writing w-full max-w-[200px] rounded-xl h-[70px] self-end",
+							"hover:bg-ielts-writing hover:text-white disabled:bg-ielts-writing/40 disabled:cursor-not-allowed",
+							"transition ease-in-out duration-300",
+						)}>
+						Perform Exam
+					</button>
+				)}
+				<button
+					disabled={!task1 || !task2 || isLoading}
+					data-tip="Please generate all three passages"
+					onClick={submitExam}
+					className={clsx(
+						"bg-ielts-writing/70 border border-ielts-writing text-white w-full max-w-[200px] rounded-xl h-[70px] self-end",
+						"hover:bg-ielts-writing disabled:bg-ielts-writing/40 disabled:cursor-not-allowed",
+						"transition ease-in-out duration-300",
+						(!task1 || !task2) && "tooltip",
+					)}>
+					{isLoading ? (
+						<div className="flex items-center justify-center">
+							<BsArrowRepeat className="text-white animate-spin" size={25} />
+						</div>
+					) : (
+						"Submit"
+					)}
+				</button>
+			</div>
 		</>
 	);
 };

src/pages/api/exam/[module]/index.ts

@@ -1,18 +1,26 @@
 // 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, query, where} from "firebase/firestore";
+import {getFirestore, collection, getDocs, query, where, setDoc, doc} from "firebase/firestore";
 import {withIronSessionApiRoute} from "iron-session/next";
 import {sessionOptions} from "@/lib/session";
 import {shuffle} from "lodash";
 import {Exam} from "@/interfaces/exam";
 import {Stat} from "@/interfaces/user";
+import {v4} from "uuid";
 
 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 === "POST") return await POST(req, res);
+
+	res.status(404).json({ok: false});
+}
+
+async function GET(req: NextApiRequest, res: NextApiResponse) {
 	if (!req.session.user) {
 		res.status(401).json({ok: false});
 		return;
@@ -45,3 +53,21 @@ async function handler(req: NextApiRequest, res: NextApiResponse) {
 
 	res.status(200).json(exams);
 }
+
+async function POST(req: NextApiRequest, res: NextApiResponse) {
+	if (!req.session.user) {
+		res.status(401).json({ok: false});
+		return;
+	}
+
+	if (req.session.user.type !== "developer") {
+		res.status(403).json({ok: false});
+		return;
+	}
+	const {module} = req.query as {module: string};
+
+	const exam = {...req.body, module: module};
+	await setDoc(doc(db, module, req.body.id), exam);
+
+	res.status(200).json(exam);
+}

src/pages/api/exam/index.ts

@@ -13,6 +13,12 @@ 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);
+
+	res.status(404).json({ok: false});
+}
+
+async function GET(req: NextApiRequest, res: NextApiResponse) {
 	if (!req.session.user) {
 		res.status(401).json({ok: false});
 		return;