ENCOA-96: License Distribuition system from Master Corporate to Corporate

src/utils/users.be.ts

@@ -1,43 +1,55 @@
-import { app } from "@/firebase";
+import {app} from "@/firebase";
 
-import {
-  collection,
-  doc,
-  getDoc,
-  getDocs,
-  getFirestore,
-  query,
-  where,
-} from "firebase/firestore";
-import { User } from "@/interfaces/user";
+import {collection, doc, getDoc, getDocs, getFirestore, query, where} from "firebase/firestore";
+import {CorporateUser, Group, User} from "@/interfaces/user";
+import {getGroupsForUser} from "./groups.be";
+import {uniq, uniqBy} from "lodash";
+import {getUserCodes} from "./codes.be";
 const db = getFirestore(app);
 
 export async function getUsers() {
-  const snapshot = await getDocs(collection(db, "users"));
+	const snapshot = await getDocs(collection(db, "users"));
 
-  return snapshot.docs.map((doc) => ({
-    id: doc.id,
-    ...doc.data(),
-  })) as User[];
+	return snapshot.docs.map((doc) => ({
+		id: doc.id,
+		...doc.data(),
+	})) as User[];
 }
 
 export async function getUser(id: string) {
-  const userDoc = await getDoc(doc(db, "users", id));
+	const userDoc = await getDoc(doc(db, "users", id));
 
-  return { ...userDoc.data(), id } as User;
+	return {...userDoc.data(), id} as User;
 }
 
 export async function getSpecificUsers(ids: string[]) {
-  if (ids.length === 0) return [];
+	if (ids.length === 0) return [];
 
-  const snapshot = await getDocs(
-    query(collection(db, "users"), where("id", "in", ids))
-  );
+	const snapshot = await getDocs(query(collection(db, "users"), where("id", "in", ids)));
 
-  const groups = snapshot.docs.map((doc) => ({
-    id: doc.id,
-    ...doc.data(),
-  })) as User[];
+	const groups = snapshot.docs.map((doc) => ({
+		id: doc.id,
+		...doc.data(),
+	})) as User[];
 
-  return groups;
+	return groups;
+}
+
+export async function getUserBalance(user: User) {
+	const codes = await getUserCodes(user.id);
+	if (user.type !== "corporate" && user.type !== "mastercorporate") return codes.length;
+
+	const groups = await getGroupsForUser(user.id);
+	const participants = uniq(groups.flatMap((x) => x.participants));
+
+	if (user.type === "corporate") return participants.length + codes.length;
+
+	const participantUsers = await Promise.all(participants.map(getUser));
+	const corporateUsers = participantUsers.filter((x) => x.type === "corporate") as CorporateUser[];
+
+	return (
+		corporateUsers.reduce((acc, curr) => acc + curr.corporateInformation?.companyInformation?.userAmount || 0, 0) +
+		corporateUsers.length +
+		codes.length
+	);
 }