fix(index.d.ts): support implicit $in in FilterQuery

Fix #10826
Automattic · Oct 18, 2021 · 8865bd5 · 8865bd5
1 parent 8337f1d
commit 8865bd5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/index.d.ts b/index.d.ts
@@ -2555,10 +2555,10 @@ declare module 'mongoose' {
 
   type MatchKeysAndValues<TSchema> = ReadonlyPartial<TSchema> & AnyObject;
 
-  type ApplyBasicQueryCasting<T> = T extends mongodb.ObjectId ? T | string : // Allow strings for ObjectIds
-    T extends string ? T | RegExp : // Allow RegExps for strings
+  type ApplyBasicQueryCasting<T> = T extends mongodb.ObjectId ? T | string | AnyArray<T | string | Document> : // Allow strings for ObjectIds
+    T extends string ? T | RegExp | AnyArray<T> : // Allow RegExps for strings
     T extends (infer U)[] ? T | U : // Allow single array elements for arrays
-    T;
+    T | AnyArray<T>;
   type Condition<T> = ApplyBasicQueryCasting<T> | QuerySelector<ApplyBasicQueryCasting<T>>;
 
   type _FilterQuery<T> = {

diff --git a/test/typescript/queries.ts b/test/typescript/queries.ts
@@ -64,6 +64,9 @@ Test.find({ name: { $in: ['Test'] } }).exec().then((res: Array<ITest>) => consol
 Test.find({ tags: 'test' }).exec();
 Test.find({ tags: { $in: ['test'] } }).exec();
 
+// Implicit `$in`
+Test.find({ name: ['Test1', 'Test2'] }).exec();
+
 Test.find({ name: 'test' }, (err: Error, docs: ITest[]) => {
   console.log(!!err, docs[0].age);
 });