From d54e3622e6ec27880085cfc3d8f2ab63a4ad771c Mon Sep 17 00:00:00 2001
From: titanism <101466223+titanism@users.noreply.github.com>
Date: Mon, 11 Nov 2024 13:00:34 -0600
Subject: [PATCH] fix: optimized validUserId lookup on API endpoint

---
 app/controllers/api/v1/lookup.js | 37 +++++++++++++++++++-------------
 1 file changed, 22 insertions(+), 15 deletions(-)

diff --git a/app/controllers/api/v1/lookup.js b/app/controllers/api/v1/lookup.js
index 38c45a8c5f..85a7dcc840 100644
--- a/app/controllers/api/v1/lookup.js
+++ b/app/controllers/api/v1/lookup.js
@@ -138,23 +138,30 @@ async function lookup(ctx) {
   if (domain.is_global) {
     // TODO: implement caching here
     // TODO: maxTimeMS: 30000
-    const validUserIds = await Users.distinct('id', {
-      $or: [
-        // get all paid users with expiration is >= 30 days ago
-        {
-          plan: { $in: ['enhanced_protection', 'team'] },
-          [config.userFields.planExpiresAt]: {
-            $gte: dayjs().subtract(30, 'days').toDate()
-          }
-        },
-        // get all admin users
-        {
-          group: 'admin'
+    const arr = await Users.aggregate([
+      {
+        $match: {
+          $or: [
+            // get all paid users with expiration is >= 30 days ago
+            {
+              plan: { $in: ['enhanced_protection', 'team'] },
+              [config.userFields.planExpiresAt]: {
+                $gte: dayjs().subtract(30, 'days').toDate()
+              }
+            },
+            // get all admin users
+            {
+              group: 'admin'
+            }
+          ]
         }
-      ]
-    });
+      },
+      { $group: { _id: '$id' } }
+    ])
+      .allowDiskUse(true)
+      .exec();
 
-    const validUserIdSet = new Set(validUserIds);
+    const validUserIdSet = new Set(arr.map((v) => v._id));
 
     aliases = aliases.filter((a) => {
       return (