/api/evaluate refactor

src/pages/api/evaluate/speaking.ts

@@ -6,12 +6,12 @@ import axios, {AxiosResponse} from "axios";
 import formidable from "formidable-serverless";
 import {getDownloadURL, ref, uploadBytes} from "firebase/storage";
 import fs from "fs";
-import {app, storage} from "@/firebase";
-import {doc, getDoc, getFirestore, setDoc} from "firebase/firestore";
+import {storage} from "@/firebase";
+import client from "@/lib/mongodb";
 import {Stat} from "@/interfaces/user";
 import {speakingReverseMarking} from "@/utils/score";
 
-const db = getFirestore(app);
+const db = client.db(process.env.MONGODB_DB);
 export default withIronSessionApiRoute(handler, sessionOptions);
 
 function delay(ms: number) {
@@ -51,8 +51,8 @@ async function handler(req: NextApiRequest, res: NextApiResponse) {
 			solution: url,
 		}));
 
-		await setDoc(
-			doc(db, "stats", fields.id),
+		await db.collection("stats").updateOne(
+			{ id: fields.id },
 			{
 				solutions,
 				score: {
@@ -62,7 +62,7 @@ async function handler(req: NextApiRequest, res: NextApiResponse) {
 				},
 				isDisabled: false,
 			},
-			{merge: true},
+			{upsert: true},
 		);
 		console.log("🌱 - Updated the DB");
 	});
@@ -70,9 +70,9 @@ async function handler(req: NextApiRequest, res: NextApiResponse) {
 
 async function getCorrespondingStat(id: string, index: number): Promise<Stat> {
 	console.log(`🌱 - Try number ${index} - ${id}`);
-	const correspondingStat = await getDoc(doc(db, "stats", id));
+	const correspondingStat = await db.collection("stats").findOne<Stat>({ id: id });
 
-	if (correspondingStat.exists()) return {...correspondingStat.data(), id} as Stat;
+	if (correspondingStat) return correspondingStat;
 	await delay(3 * 10000);
 	return getCorrespondingStat(id, index + 1);
 }