ENCOA-283 or ENCOA-282, I don't know someone deleted the issue

src/pages/api/users/controller.ts

@@ -29,7 +29,14 @@ async function handler(req: NextApiRequest, res: NextApiResponse) {
         res.status(200).json(await entityCheck(req.body));
         break;
       case 'crossRefClassrooms':
-        res.status(200).json(await crossRefClassrooms(req.body.sets));
+        res.status(200).json(await crossRefClassrooms(req.body));
+        break;
+      case 'getIds':
+        res.status(200).json(await getIds(req.body.emails));
+        break;
+      case 'assignToEntity':
+        await assignToEntity(req.body);
+        res.status(200).json({"ok": true});
         break;
       default:
         res.status(400).json({ error: 'Invalid operation!' })
@@ -68,9 +75,12 @@ async function dontExist(emails: string[]): Promise<string[]> {
 }
 
 
-async function entityCheck(body: Record<string, any>): Promise<string[]> {
+async function entityCheck(body: Record<string, any>): Promise<Array<{
+  email: string;
+  names?: string[];
+}>> {
   const { entities, emails } = body;
-  
+
   const pipeline = [
     // Match users with the provided emails
     {
@@ -99,20 +109,65 @@ async function entityCheck(body: Record<string, any>): Promise<string[]> {
         ]
       }
     },
-    // Project only the email field
+    // Unwind the entities array (if it exists)
+    {
+      $unwind: {
+        path: "$entities",
+        preserveNullAndEmptyArrays: true
+      }
+    },
+    // Lookup entity details from entities collection
+    {
+      $lookup: {
+        from: 'entities',
+        localField: 'entities.id',
+        foreignField: 'id',
+        as: 'entityDetails'
+      }
+    },
+    // Unwind the entityDetails array
+    {
+      $unwind: {
+        path: "$entityDetails",
+        preserveNullAndEmptyArrays: true
+      }
+    },
+    // Group by email to collect all entity names in an array
+    {
+      $group: {
+        _id: "$email",
+        email: { $first: "$email" },
+        name: {
+          $push: {
+            $cond: [
+              { $ifNull: ["$entityDetails.label", false] },
+              "$entityDetails.label",
+              "$$REMOVE"
+            ]
+          }
+        }
+      }
+    },
+    // Final projection to clean up
     {
       $project: {
         _id: 0,
-        email: 1
+        email: 1,
+        name: 1
       }
     }
   ];
 
   const results = await db.collection('users').aggregate(pipeline).toArray();
-  return results.map((result: any) => result.email);
+  return results.map(result => ({
+    email: result.email as string,
+    names: result.name as string[] | undefined
+  }));
 }
 
-async function crossRefClassrooms(sets: { email: string, classroom: string }[]) {
+async function crossRefClassrooms(body: any) {
+  const { sets, entity } = body as { sets: { email: string, classroom: string }[], entity: string };
+  
   const pipeline = [
     // Match users with the provided emails
     {
@@ -132,7 +187,6 @@ async function crossRefClassrooms(sets: { email: string, classroom: string }[])
                 $and: [
                   { $in: ['$$userId', '$participants'] },
                   {
-                    // Match the classroom that corresponds to this user's email
                     $let: {
                       vars: {
                         matchingSet: {
@@ -153,6 +207,38 @@ async function crossRefClassrooms(sets: { email: string, classroom: string }[])
                 ]
               }
             }
+          },
+          // Lookup admin's entities
+          {
+            $lookup: {
+              from: 'users',
+              let: { adminId: '$admin' },
+              pipeline: [
+                {
+                  $match: {
+                    $expr: { $eq: ['$id', '$$adminId'] }
+                  }
+                }
+              ],
+              as: 'adminInfo'
+            }
+          },
+          // Filter where admin has the target entity
+          {
+            $match: {
+              $expr: {
+                $in: [
+                  entity,
+                  {
+                    $map: {
+                      input: { $arrayElemAt: ['$adminInfo.entities', 0] },
+                      as: 'entityObj',
+                      in: '$$entityObj.id'
+                    }
+                  }
+                ]
+              }
+            }
           }
         ],
         as: 'matchingGroups'
@@ -172,7 +258,54 @@ async function crossRefClassrooms(sets: { email: string, classroom: string }[])
       }
     }
   ];
-  
+
   const results = await db.collection('users').aggregate(pipeline).toArray();
   return results.map((result: any) => result.email);
 }
+
+
+async function getIds(emails: string[]): Promise<Array<{ email: string; id: string }>> {
+  const users = await db.collection('users')
+    .find({ email: { $in: emails } })
+    .project({ email: 1, id: 1, _id: 0 })
+    .toArray();
+
+  return users.map(user => ({
+    email: user.email,
+    id: user.id
+  }));
+}
+
+
+async function assignToEntity(body: any) {
+  const { ids, entity } = body;
+
+  if (!Array.isArray(ids) || ids.length === 0 || !entity) {
+    return;
+  }
+
+  const users = await db.collection('users')
+    .find({ id: { $in: ids } })
+    .project({ id: 1, entities: 1 })
+    .toArray();
+
+  const toUpdateUsers = users.filter((u) => u.entities[0].id !== entity);
+
+  if (toUpdateUsers.length > 0) {
+    const writes = users.map(user => ({
+      updateOne: {
+        filter: { id: user.id },
+        update: {
+          $set: {
+            entities: [{
+              id: entity,
+              role: user.entities?.[0]?.role
+            }]
+          }
+        }
+      }
+    }));
+  
+    db.collection('users').bulkWrite(writes);
+  }
+}