From 2657cb409c8230ab0ef7be35d8ecd3ed0e6e59a2 Mon Sep 17 00:00:00 2001 From: Tiago Ribeiro Date: Thu, 28 Mar 2024 08:21:45 +0000 Subject: [PATCH] Solved a bug where it would not send the correct link to the e-mail --- src/pages/api/code/index.ts | 212 ++++++++++++++++-------------------- src/pages/api/stripe.ts | 1 + 2 files changed, 97 insertions(+), 116 deletions(-) diff --git a/src/pages/api/code/index.ts b/src/pages/api/code/index.ts index ff53ec05..bad31fc9 100644 --- a/src/pages/api/code/index.ts +++ b/src/pages/api/code/index.ts @@ -1,143 +1,123 @@ // Next.js API route support: https://nextjs.org/docs/api-routes/introduction -import type { NextApiRequest, NextApiResponse } from "next"; -import { app } from "@/firebase"; -import { - getFirestore, - setDoc, - doc, - query, - collection, - where, - getDocs, -} from "firebase/firestore"; -import { withIronSessionApiRoute } from "iron-session/next"; -import { sessionOptions } from "@/lib/session"; -import { Type } from "@/interfaces/user"; -import { PERMISSIONS } from "@/constants/userPermissions"; -import { uuidv4 } from "@firebase/util"; -import { prepareMailer, prepareMailOptions } from "@/email"; +import type {NextApiRequest, NextApiResponse} from "next"; +import {app} from "@/firebase"; +import {getFirestore, setDoc, doc, query, collection, where, getDocs} from "firebase/firestore"; +import {withIronSessionApiRoute} from "iron-session/next"; +import {sessionOptions} from "@/lib/session"; +import {Type} from "@/interfaces/user"; +import {PERMISSIONS} from "@/constants/userPermissions"; +import {uuidv4} from "@firebase/util"; +import {prepareMailer, prepareMailOptions} from "@/email"; const db = getFirestore(app); export default withIronSessionApiRoute(handler, sessionOptions); async function handler(req: NextApiRequest, res: NextApiResponse) { - if (req.method === "GET") return get(req, res); - if (req.method === "POST") return post(req, res); + if (req.method === "GET") return get(req, res); + if (req.method === "POST") return post(req, res); - return res.status(404).json({ ok: false }); + return res.status(404).json({ok: false}); } async function get(req: NextApiRequest, res: NextApiResponse) { - if (!req.session.user) { - res - .status(401) - .json({ ok: false, reason: "You must be logged in to generate a code!" }); - return; - } + if (!req.session.user) { + res.status(401).json({ok: false, reason: "You must be logged in to generate a code!"}); + return; + } - const { creator } = req.query as { creator?: string }; - const q = query(collection(db, "codes"), where("creator", "==", creator)); - const snapshot = await getDocs(creator ? q : collection(db, "codes")); + const {creator} = req.query as {creator?: string}; + const q = query(collection(db, "codes"), where("creator", "==", creator)); + const snapshot = await getDocs(creator ? q : collection(db, "codes")); - res.status(200).json(snapshot.docs.map((doc) => doc.data())); + res.status(200).json(snapshot.docs.map((doc) => doc.data())); } async function post(req: NextApiRequest, res: NextApiResponse) { - if (!req.session.user) { - res - .status(401) - .json({ ok: false, reason: "You must be logged in to generate a code!" }); - return; - } + if (!req.session.user) { + res.status(401).json({ok: false, reason: "You must be logged in to generate a code!"}); + return; + } - const { type, codes, infos, expiryDate } = req.body as { - type: Type; - codes: string[]; - infos?: { email: string; name: string; passport_id?: string }[]; - expiryDate: null | Date; - }; - const permission = PERMISSIONS.generateCode[type]; + const {type, codes, infos, expiryDate} = req.body as { + type: Type; + codes: string[]; + infos?: {email: string; name: string; passport_id?: string}[]; + expiryDate: null | Date; + }; + const permission = PERMISSIONS.generateCode[type]; - if (!permission.includes(req.session.user.type)) { - res - .status(403) - .json({ - ok: false, - reason: - "Your account type does not have permissions to generate a code for that type of user!", - }); - return; - } + if (!permission.includes(req.session.user.type)) { + res.status(403).json({ + ok: false, + reason: "Your account type does not have permissions to generate a code for that type of user!", + }); + return; + } - if (req.session.user.type === "corporate") { - const codesGeneratedByUserSnapshot = await getDocs( - query( - collection(db, "codes"), - where("creator", "==", req.session.user.id), - ), - ); - const totalCodes = codesGeneratedByUserSnapshot.docs.length + codes.length; - const allowedCodes = - req.session.user.corporateInformation?.companyInformation.userAmount || 0; + if (req.session.user.type === "corporate") { + const codesGeneratedByUserSnapshot = await getDocs(query(collection(db, "codes"), where("creator", "==", req.session.user.id))); + const totalCodes = codesGeneratedByUserSnapshot.docs.length + codes.length; + const allowedCodes = req.session.user.corporateInformation?.companyInformation.userAmount || 0; - if (totalCodes > allowedCodes) { - res.status(403).json({ - ok: false, - reason: `You have or would have exceeded your amount of allowed codes, you currently are allowed to generate ${ - allowedCodes - codesGeneratedByUserSnapshot.docs.length - } codes.`, - }); - return; - } - } + if (totalCodes > allowedCodes) { + res.status(403).json({ + ok: false, + reason: `You have or would have exceeded your amount of allowed codes, you currently are allowed to generate ${ + allowedCodes - codesGeneratedByUserSnapshot.docs.length + } codes.`, + }); + return; + } + } - const codePromises = codes.map(async (code, index) => { - const codeRef = doc(db, "codes", code); - const codeInformation = { - type, - code, - creator: req.session.user!.id, - expiryDate, - }; + const codePromises = codes.map(async (code, index) => { + const codeRef = doc(db, "codes", code); + const codeInformation = { + type, + code, + creator: req.session.user!.id, + expiryDate, + }; - if (infos && infos.length > index) { - const { email, name, passport_id } = infos[index]; + if (infos && infos.length > index) { + const {email, name, passport_id} = infos[index]; - const transport = prepareMailer(); - const mailOptions = prepareMailOptions( - { - type, - code, - }, - [email.toLowerCase().trim()], - "EnCoach Registration", - "main", - ); + const transport = prepareMailer(); + const mailOptions = prepareMailOptions( + { + type, + code, + environment: process.env.ENVIRONMENT, + }, + [email.toLowerCase().trim()], + "EnCoach Registration", + "main", + ); - try { - await transport.sendMail(mailOptions); - await setDoc( - codeRef, - { - ...codeInformation, - email: email.trim().toLowerCase(), - name: name.trim(), - ...(passport_id ? { passport_id: passport_id.trim() } : {}), - }, - { merge: true }, - ); + try { + await transport.sendMail(mailOptions); + await setDoc( + codeRef, + { + ...codeInformation, + email: email.trim().toLowerCase(), + name: name.trim(), + ...(passport_id ? {passport_id: passport_id.trim()} : {}), + }, + {merge: true}, + ); - return true; - } catch (e) { - return false; - } - } else { - await setDoc(codeRef, codeInformation); - } - }); + return true; + } catch (e) { + return false; + } + } else { + await setDoc(codeRef, codeInformation); + } + }); - Promise.all(codePromises).then((results) => { - res.status(200).json({ ok: true, valid: results.filter((x) => x).length }); - }); + Promise.all(codePromises).then((results) => { + res.status(200).json({ok: true, valid: results.filter((x) => x).length}); + }); } diff --git a/src/pages/api/stripe.ts b/src/pages/api/stripe.ts index e3ed1797..5e5dbbae 100644 --- a/src/pages/api/stripe.ts +++ b/src/pages/api/stripe.ts @@ -65,6 +65,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse) { type: "student", code, + environment: process.env.ENVIRONMENT, }, [email], "EnCoach Registration",