Revamped the whole Solutions stuff with Zustand

src/pages/exam/index.tsx

@@ -40,16 +40,16 @@ export const getServerSideProps = withIronSessionSsr(({req, res}) => {
 }, sessionOptions);
 
 export default function Page() {
-	const [userSolutions, setUserSolutions] = useState<UserSolution[]>([]);
-	const [selectedModules, setSelectedModules] = useState<Module[]>([]);
 	const [hasBeenUploaded, setHasBeenUploaded] = useState(false);
 	const [moduleIndex, setModuleIndex] = useState(0);
 	const [sessionId, setSessionId] = useState("");
 	const [exam, setExam] = useState<Exam>();
 	const [timer, setTimer] = useState(-1);
 
-	const [showSolutions, setShowSolutions] = useExamStore((state) => [state.showSolutions, state.setShowSolutions]);
 	const [exams, setExams] = useExamStore((state) => [state.exams, state.setExams]);
+	const [userSolutions, setUserSolutions] = useExamStore((state) => [state.userSolutions, state.setUserSolutions]);
+	const [showSolutions, setShowSolutions] = useExamStore((state) => [state.showSolutions, state.setShowSolutions]);
+	const [selectedModules, setSelectedModules] = useExamStore((state) => [state.selectedModules, state.setSelectedModules]);
 
 	const {user} = useUser({redirectTo: "/login"});
 
@@ -66,15 +66,16 @@ export default function Page() {
 	}, [selectedModules, moduleIndex, exams]);
 
 	useEffect(() => {
-		async () => {
+		(async () => {
 			if (selectedModules.length > 0) {
 				const examPromises = selectedModules.map(getExam);
 				Promise.all(examPromises).then((values) => {
 					if (values.every((x) => !!x)) {
+						setExams(values.map((x) => x!));
 					}
 				});
 			}
-		};
+		})();
 	}, [selectedModules, setExams]);
 
 	useEffect(() => {
@@ -141,7 +142,7 @@ export default function Page() {
 	const onFinish = (solutions: UserSolution[]) => {
 		const solutionIds = solutions.map((x) => x.exercise);
 
-		setUserSolutions((prev) => [...prev.filter((x) => !solutionIds.includes(x.exercise)), ...solutions]);
+		setUserSolutions([...userSolutions.filter((x) => !solutionIds.includes(x.exercise)), ...solutions]);
 		setModuleIndex((prev) => prev + 1);
 	};
 
@@ -158,6 +159,7 @@ export default function Page() {
 					onViewResults={() => {
 						setShowSolutions(true);
 						setModuleIndex(0);
+						setExam(exams[0]);
 					}}
 					scores={userSolutions.map((x) => ({...x.score, module: x.module}))}
 				/>