Merge pull request #12087 from Automattic/ts-aggregation-let

fix(types): add $let as a possible expression to $addFields
Automattic · Jul 15, 2022 · c48461a · c48461a
2 parents 3e5fa8c + 0897fea
commit c48461a
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 1 deletion.
diff --git a/test/types/PipelineStage.test.ts b/test/types/PipelineStage.test.ts
@@ -392,3 +392,26 @@ const stages3: PipelineStage[] = [
     }
   }
 ];
+
+const stages4: PipelineStage[] = [
+  {
+    $addFields: {
+      usersCount: {
+        $let: {
+          vars: {
+            users: { $push: '$user' }
+          },
+          in: {
+            $reduce: {
+              input: '$users',
+              initialValue: 0,
+              in: {
+                $cond: { if: { $isArray: '$$this' }, then: { $size: '$$this' }, else: '$$this' }
+              }
+            }
+          }
+        }
+      }
+    }
+  }
+];
diff --git a/types/expressions.d.ts b/types/expressions.d.ts
@@ -2448,7 +2448,8 @@ declare module 'mongoose' {
     BinaryExpression |
     FunctionExpression |
     ObjectIdExpression |
-    ConditionalExpressionOperator;
+    ConditionalExpressionOperator |
+    Expression.Let;
 
   export type ObjectIdExpression =
     TypeExpressionOperatorReturningObjectId;