Updated the user balance to be based not only on the amount of codes

This commit is contained in:
Tiago Ribeiro
2024-08-15 19:24:08 +01:00
parent d94a9bb88a
commit 85f684dff5
2 changed files with 134 additions and 139 deletions

View File

@@ -57,6 +57,7 @@ export default function CorporateDashboard({user}: Props) {
const [corporateUserToShow, setCorporateUserToShow] = useState<CorporateUser>(); const [corporateUserToShow, setCorporateUserToShow] = useState<CorporateUser>();
const [selectedAssignment, setSelectedAssignment] = useState<Assignment>(); const [selectedAssignment, setSelectedAssignment] = useState<Assignment>();
const [isCreatingAssignment, setIsCreatingAssignment] = useState(false); const [isCreatingAssignment, setIsCreatingAssignment] = useState(false);
const [userBalance, setUserBalance] = useState(0);
const {stats} = useStats(); const {stats} = useStats();
const {users, reload} = useUsers(); const {users, reload} = useUsers();
@@ -71,6 +72,14 @@ export default function CorporateDashboard({user}: Props) {
setShowModal(!!selectedUser && page === ""); setShowModal(!!selectedUser && page === "");
}, [selectedUser, page]); }, [selectedUser, page]);
useEffect(() => {
const relatedGroups = groups.filter((x) => x.name === "Students" || x.name === "Teachers" || x.name === "Corporate");
const usersInGroups = relatedGroups.map((x) => x.participants).flat();
const filteredCodes = codes.filter((x) => !x.userId || !usersInGroups.includes(x.userId));
setUserBalance(usersInGroups.length + filteredCodes.length);
}, [codes, groups]);
useEffect(() => { useEffect(() => {
// in this case it fetches the master corporate account // in this case it fetches the master corporate account
getUserCorporate(user.id).then(setCorporateUserToShow); getUserCorporate(user.id).then(setCorporateUserToShow);
@@ -350,7 +359,7 @@ export default function CorporateDashboard({user}: Props) {
<IconCard <IconCard
Icon={BsPersonCheck} Icon={BsPersonCheck}
label="User Balance" label="User Balance"
value={`${codes.length}/${user.corporateInformation?.companyInformation?.userAmount || 0}`} value={`${userBalance}/${user.corporateInformation?.companyInformation?.userAmount || 0}`}
color="purple" color="purple"
/> />
<IconCard <IconCard

View File

@@ -1,174 +1,160 @@
// Next.js API route support: https://nextjs.org/docs/api-routes/introduction // Next.js API route support: https://nextjs.org/docs/api-routes/introduction
import type { NextApiRequest, NextApiResponse } from "next"; import type {NextApiRequest, NextApiResponse} from "next";
import { app } from "@/firebase"; import {app} from "@/firebase";
import { import {getFirestore, setDoc, doc, query, collection, where, getDocs, getDoc, deleteDoc} from "firebase/firestore";
getFirestore, import {withIronSessionApiRoute} from "iron-session/next";
setDoc, import {sessionOptions} from "@/lib/session";
doc, import {Code, Group, Type} from "@/interfaces/user";
query, import {PERMISSIONS} from "@/constants/userPermissions";
collection, import {uuidv4} from "@firebase/util";
where, import {prepareMailer, prepareMailOptions} from "@/email";
getDocs,
getDoc,
deleteDoc,
} from "firebase/firestore";
import { withIronSessionApiRoute } from "iron-session/next";
import { sessionOptions } from "@/lib/session";
import { Code, Type } from "@/interfaces/user";
import { PERMISSIONS } from "@/constants/userPermissions";
import { uuidv4 } from "@firebase/util";
import { prepareMailer, prepareMailOptions } from "@/email";
const db = getFirestore(app); const db = getFirestore(app);
export default withIronSessionApiRoute(handler, sessionOptions); export default withIronSessionApiRoute(handler, sessionOptions);
async function handler(req: NextApiRequest, res: NextApiResponse) { async function handler(req: NextApiRequest, res: NextApiResponse) {
if (req.method === "GET") return get(req, res); if (req.method === "GET") return get(req, res);
if (req.method === "POST") return post(req, res); if (req.method === "POST") return post(req, res);
if (req.method === "DELETE") return del(req, res); if (req.method === "DELETE") return del(req, res);
return res.status(404).json({ ok: false }); return res.status(404).json({ok: false});
} }
async function get(req: NextApiRequest, res: NextApiResponse) { async function get(req: NextApiRequest, res: NextApiResponse) {
if (!req.session.user) { if (!req.session.user) {
res res.status(401).json({ok: false, reason: "You must be logged in to generate a code!"});
.status(401) return;
.json({ ok: false, reason: "You must be logged in to generate a code!" }); }
return;
}
const { creator } = req.query as { creator?: string }; const {creator} = req.query as {creator?: string};
const q = query( const q = query(collection(db, "codes"), where("creator", "==", creator || ""));
collection(db, "codes"), const snapshot = await getDocs(creator ? q : 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) { async function post(req: NextApiRequest, res: NextApiResponse) {
if (!req.session.user) { if (!req.session.user) {
res res.status(401).json({ok: false, reason: "You must be logged in to generate a code!"});
.status(401) return;
.json({ ok: false, reason: "You must be logged in to generate a code!" }); }
return;
}
const { type, codes, infos, expiryDate } = req.body as { const {type, codes, infos, expiryDate} = req.body as {
type: Type; type: Type;
codes: string[]; codes: string[];
infos?: { email: string; name: string; passport_id?: string }[]; infos?: {email: string; name: string; passport_id?: string}[];
expiryDate: null | Date; expiryDate: null | Date;
}; };
const permission = PERMISSIONS.generateCode[type]; const permission = PERMISSIONS.generateCode[type];
if (!permission.includes(req.session.user.type)) { if (!permission.includes(req.session.user.type)) {
res.status(403).json({ res.status(403).json({
ok: false, ok: false,
reason: reason: "Your account type does not have permissions to generate a code for that type of user!",
"Your account type does not have permissions to generate a code for that type of user!", });
}); return;
return; }
}
const codesGeneratedByUserSnapshot = await getDocs( const codesGeneratedByUserSnapshot = await getDocs(query(collection(db, "codes"), where("creator", "==", req.session.user.id)));
query(collection(db, "codes"), where("creator", "==", req.session.user.id)), const creatorGroupsSnapshot = await getDocs(query(collection(db, "groups"), where("admin", "==", req.session.user.id)));
);
const userCodes = codesGeneratedByUserSnapshot.docs.map((x) => ({
...x.data(),
}));
if (req.session.user.type === "corporate") { const creatorGroups = (
const totalCodes = codesGeneratedByUserSnapshot.docs.length + codes.length; creatorGroupsSnapshot.docs.map((x) => ({
const allowedCodes = ...x.data(),
req.session.user.corporateInformation?.companyInformation.userAmount || 0; })) as Group[]
).filter((x) => x.name === "Students" || x.name === "Teachers" || x.name === "Corporate");
if (totalCodes > allowedCodes) { const usersInGroups = creatorGroups.flatMap((x) => x.participants);
res.status(403).json({ const userCodes = codesGeneratedByUserSnapshot.docs.map((x) => ({
ok: false, ...x.data(),
reason: `You have or would have exceeded your amount of allowed codes, you currently are allowed to generate ${ })) as Code[];
allowedCodes - codesGeneratedByUserSnapshot.docs.length
} codes.`,
});
return;
}
}
const codePromises = codes.map(async (code, index) => { if (req.session.user.type === "corporate") {
const codeRef = doc(db, "codes", code); const totalCodes = userCodes.filter((x) => !x.userId || !usersInGroups.includes(x.userId)).length + usersInGroups.length + codes.length;
let codeInformation = { const allowedCodes = req.session.user.corporateInformation?.companyInformation.userAmount || 0;
type,
code,
creator: req.session.user!.id,
creationDate: new Date().toISOString(),
expiryDate,
};
if (infos && infos.length > index) { if (totalCodes > allowedCodes) {
const { email, name, passport_id } = infos[index]; res.status(403).json({
const previousCode = userCodes.find((x) => x.email === email) as Code; 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 transport = prepareMailer(); const codePromises = codes.map(async (code, index) => {
const mailOptions = prepareMailOptions( const codeRef = doc(db, "codes", code);
{ let codeInformation = {
type, type,
code: previousCode ? previousCode.code : code, code,
environment: process.env.ENVIRONMENT, creator: req.session.user!.id,
}, creationDate: new Date().toISOString(),
[email.toLowerCase().trim()], expiryDate,
"EnCoach Registration", };
"main",
);
try { if (infos && infos.length > index) {
await transport.sendMail(mailOptions); const {email, name, passport_id} = infos[index];
const previousCode = userCodes.find((x) => x.email === email) as Code;
if (!previousCode) { const transport = prepareMailer();
await setDoc( const mailOptions = prepareMailOptions(
codeRef, {
{ type,
...codeInformation, code: previousCode ? previousCode.code : code,
email: email.trim().toLowerCase(), environment: process.env.ENVIRONMENT,
name: name.trim(), },
...(passport_id ? { passport_id: passport_id.trim() } : {}), [email.toLowerCase().trim()],
}, "EnCoach Registration",
{ merge: true }, "main",
); );
}
return true; try {
} catch (e) { await transport.sendMail(mailOptions);
return false;
}
} else {
await setDoc(codeRef, codeInformation);
}
});
Promise.all(codePromises).then((results) => { if (!previousCode) {
res.status(200).json({ ok: true, valid: results.filter((x) => x).length }); 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);
}
});
Promise.all(codePromises).then((results) => {
res.status(200).json({ok: true, valid: results.filter((x) => x).length});
});
} }
async function del(req: NextApiRequest, res: NextApiResponse) { async function del(req: NextApiRequest, res: NextApiResponse) {
if (!req.session.user) { if (!req.session.user) {
res res.status(401).json({ok: false, reason: "You must be logged in to generate a code!"});
.status(401) return;
.json({ ok: false, reason: "You must be logged in to generate a code!" }); }
return;
}
const codes = req.query.code as string[]; const codes = req.query.code as string[];
for (const code of codes) { for (const code of codes) {
const snapshot = await getDoc(doc(db, "codes", code as string)); const snapshot = await getDoc(doc(db, "codes", code as string));
if (!snapshot.exists()) continue; if (!snapshot.exists()) continue;
await deleteDoc(snapshot.ref); await deleteDoc(snapshot.ref);
} }
res.status(200).json({ codes }); res.status(200).json({codes});
} }