Added comission to user card and updated user update to insert a new payment entry

src/pages/api/users/update.ts

@@ -1,13 +1,16 @@
 // 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, getDoc, doc, setDoc} from "firebase/firestore";
+import {getFirestore, collection, getDocs, getDoc, doc, setDoc, query, where} from "firebase/firestore";
 import {withIronSessionApiRoute} from "iron-session/next";
 import {sessionOptions} from "@/lib/session";
 import {User} from "@/interfaces/user";
 import {getDownloadURL, getStorage, ref, uploadBytes} from "firebase/storage";
 import {getAuth, signInWithEmailAndPassword, updateEmail, updatePassword} from "firebase/auth";
 import {errorMessages} from "@/constants/errors";
+import moment from "moment";
+import ShortUniqueId from "short-unique-id";
+import {Payment} from "@/interfaces/paypal";
 
 const db = getFirestore(app);
 const storage = getStorage(app);
@@ -15,6 +18,52 @@ const auth = getAuth(app);
 
 export default withIronSessionApiRoute(handler, sessionOptions);
 
+const addPaymentRecord = async (data: Payment) => {
+	const shortUID = new ShortUniqueId();
+	await setDoc(doc(db, "payments", shortUID.randomUUID(8)), data);
+}
+const managePaymentRecords = async (user: User, userId: string | undefined): Promise<boolean> => {
+	try {
+		if(user.type === 'corporate' && userId) {
+			const data = {
+				corporate: userId,
+				agent: user.corporateInformation.referralAgent,
+				agentCommission: user.corporateInformation.payment!.commission,
+				agentValue: (user.corporateInformation.payment!.commission / 100) * user.corporateInformation.payment!.value,
+				currency: user.corporateInformation.payment!.currency,
+				value: user.corporateInformation.payment!.value,
+				isPaid: false,
+				date: new Date(),
+			} as Payment;
+			
+			const corporatePayments = await getDocs(query(collection(db, "payments"), where("corporate", "==", userId)));
+			if(corporatePayments.docs.length === 0) {
+				await addPaymentRecord(data);
+				return true;
+			}
+			
+			const hasPaymentPaidAndExpiring = corporatePayments.docs.filter((doc) => {
+				const data = doc.data();
+				return data.isPaid
+				&& moment().isAfter(moment(user.subscriptionExpirationDate).subtract(30, "days"))
+				&& moment().isBefore(moment(user.subscriptionExpirationDate));
+			});
+
+			if(hasPaymentPaidAndExpiring.length > 0) {
+				await addPaymentRecord(data);
+				return true;
+			}
+		}
+
+		return false;
+	} catch(e) {
+		// if this process fails it should not stop the rest of the process
+		console.log(e);
+		return false;
+	}
+
+}
+
 async function handler(req: NextApiRequest, res: NextApiResponse) {
 	if (!req.session.user) {
 		res.status(401).json({ok: false});
@@ -25,7 +74,8 @@ async function handler(req: NextApiRequest, res: NextApiResponse) {
 	const updatedUser = req.body as User & {password?: string; newPassword?: string};
 
 	if (!!req.query.id) {
-		await setDoc(userRef, updatedUser, {merge: true});
+		const user = await setDoc(userRef, updatedUser, {merge: true});
+		await managePaymentRecords(updatedUser, updatedUser.id);
 		res.status(200).json({ok: true});
 		return;
 	}
@@ -74,6 +124,8 @@ async function handler(req: NextApiRequest, res: NextApiResponse) {
 		await req.session.save();
 	}
 
+	await managePaymentRecords(user, req.query.id);
+
 	res.status(200).json({user});
 }