Continued creating the entity system

src/utils/entities.be.ts

@@ -0,0 +1,35 @@
+import {Entity, EntityWithRoles, Role} from "@/interfaces/entity";
+import client from "@/lib/mongodb";
+import {getRolesByEntities, getRolesByEntity} from "./roles.be";
+
+const db = client.db(process.env.MONGODB_DB);
+
+export const getEntityWithRoles = async (id: string): Promise<{entity: Entity; roles: Role[]} | undefined> => {
+	const entity = await getEntity(id);
+	if (!entity) return undefined;
+
+	const roles = await getRolesByEntity(id);
+	return {entity, roles};
+};
+
+export const getEntity = async (id: string) => {
+	return (await db.collection("entities").findOne<Entity>({id})) ?? undefined;
+};
+
+export const getEntitiesWithRoles = async (ids?: string[]): Promise<EntityWithRoles[]> => {
+	const entities = await db
+		.collection("entities")
+		.find<Entity>(ids ? {id: {$in: ids}} : {})
+		.toArray();
+
+	const roles = await getRolesByEntities(entities.map((x) => x.id));
+
+	return entities.map((x) => ({...x, roles: roles.filter((y) => y.entityID === x.id) || []}));
+};
+
+export const getEntities = async (ids?: string[]) => {
+	return await db
+		.collection("entities")
+		.find<Entity>(ids ? {id: {$in: ids}} : {})
+		.toArray();
+};