update pipeline stage types

Author: princecodes247

src/collection/types/expressions.ts

@@ -16,60 +16,68 @@ export type Expression =
 export type AnyExpression = Expression;
 
 export type AccumulatorOperator = {
-  $accumulator?: any;
+  $accumulator?: {
+    init: Expression;
+    initArgs?: Expression[];
+    accumulate: Expression;
+    accumulateArgs: Expression[];
+    merge: Expression;
+    finalize?: Expression;
+    lang: "js";
+  };
   $addToSet?: Expression;
   $avg?: Expression;
-  $bottom?: any;
-  $bottomN?: any;
-  $count?: any;
+  $bottom?: { sortBy: Record<string, 1 | -1>; output: Expression | Expression[] };
+  $bottomN?: { sortBy: Record<string, 1 | -1>; output: Expression | Expression[]; n: Expression };
+  $count?: Record<string, never>;
   $first?: Expression;
-  $firstN?: any;
+  $firstN?: { input: Expression; n: Expression };
   $last?: Expression;
-  $lastN?: any;
+  $lastN?: { input: Expression; n: Expression };
   $max?: Expression;
-  $maxN?: any;
+  $maxN?: { input: Expression; n: Expression };
   $mergeObjects?: Expression;
   $min?: Expression;
-  $minN?: any;
+  $minN?: { input: Expression; n: Expression };
   $push?: Expression;
   $stdDevPop?: Expression;
   $stdDevSamp?: Expression;
   $sum?: Expression;
-  $top?: any;
-  $topN?: any;
+  $top?: { sortBy: Record<string, 1 | -1>; output: Expression | Expression[] };
+  $topN?: { sortBy: Record<string, 1 | -1>; output: Expression | Expression[]; n: Expression };
 };
 
 export type WindowOperator = {
   $addToSet?: Expression;
   $avg?: Expression;
-  $bottom?: any;
-  $bottomN?: any;
-  $count?: any;
-  $covariancePop?: any;
-  $covarianceSamp?: any;
-  $denseRank?: any;
-  $derivative?: any;
-  $documentNumber?: any;
-  $expMovingAvg?: any;
+  $bottom?: { sortBy: Record<string, 1 | -1>; output: Expression | Expression[] };
+  $bottomN?: { sortBy: Record<string, 1 | -1>; output: Expression | Expression[]; n: Expression };
+  $count?: Record<string, never>;
+  $covariancePop?: Expression | [Expression, Expression];
+  $covarianceSamp?: Expression | [Expression, Expression];
+  $denseRank?: Record<string, never>;
+  $derivative?: { input: Expression; unit?: string };
+  $documentNumber?: Record<string, never>;
+  $expMovingAvg?: { input: Expression; N: number } | { input: Expression; alpha: number };
   $first?: Expression;
-  $firstN?: any;
-  $integral?: any;
+  $firstN?: { input: Expression; n: Expression };
+  $integral?: { input: Expression; unit?: string };
   $last?: Expression;
-  $lastN?: any;
-  $linearFill?: any;
-  $locf?: any;
+  $lastN?: { input: Expression; n: Expression };
+  $linearFill?: Expression;
+  $locf?: Expression;
   $max?: Expression;
-  $maxN?: any;
+  $maxN?: { input: Expression; n: Expression };
   $min?: Expression;
-  $minN?: any;
+  $minN?: { input: Expression; n: Expression };
   $push?: Expression;
-  $rank?: any;
-  $shift?: any;
+  $rank?: Record<string, never>;
+  $shift?: { output: Expression; by: number; default?: Expression };
   $stdDevPop?: Expression;
   $stdDevSamp?: Expression;
   $sum?: Expression;
-  $top?: any;
-  $topN?: any;
+  $top?: { sortBy: Record<string, 1 | -1>; output: Expression | Expression[] };
+  $topN?: { sortBy: Record<string, 1 | -1>; output: Expression | Expression[]; n: Expression };
 };
 
 export type ObjectExpressionOperator = {

src/collection/types/pipeline-stage.ts

@@ -62,8 +62,8 @@ export interface Bucket {
   /** [`$bucket` reference](https://www.mongodb.com/docs/manual/reference/operator/aggregation/bucket/) */
   $bucket: {
     groupBy: Expression;
-    boundaries: any[];
-    default?: any;
+    boundaries: Expression[];
+    default?: Expression;
     output?: Record<string, AccumulatorOperator>;
   };
 }
@@ -134,7 +134,7 @@ export interface Facet {
   $facet: Record<string, FacetPipelineStage[]>;
 }
 
-export type FacetPipelineStage<T = any> = Exclude<
+export type FacetPipelineStage<T = Document> = Exclude<
   PipelineStage<T>,
   CollStats | Facet | GeoNear | IndexStats | Out | Merge<T> | PlanCacheStats
 >;
@@ -175,7 +175,7 @@ export interface GraphLookup {
 
 export interface Group {
   /** [`$group` reference](https://www.mongodb.com/docs/manual/reference/operator/aggregation/group) */
-  $group: { _id: any } | { [key: string]: AccumulatorOperator };
+  $group: { _id: Expression } | { [key: string]: AccumulatorOperator };
 }
 
 export interface IndexStats {
@@ -273,7 +273,7 @@ export interface Search {
       maxCharsToExamine?: number;
       maxNumPassages?: number;
     };
-    [operator: string]: any;
+    [operator: string]: unknown;
   };
 }
 
@@ -287,19 +287,19 @@ export interface SearchMeta {
       maxCharsToExamine?: number;
       maxNumPassages?: number;
     };
-    [operator: string]: any;
+    [operator: string]: unknown;
   };
 }
 
 export interface Set {
   /** [`$set` reference](https://www.mongodb.com/docs/manual/reference/operator/aggregation/set/) */
-  $set: Record<string, AnyExpression | any>;
+  $set: Record<string, AnyExpression>;
 }
 
 export interface SetWindowFields {
   /** [`$setWindowFields` reference](https://www.mongodb.com/docs/manual/reference/operator/aggregation/setWindowFields/) */
   $setWindowFields: {
-    partitionBy?: any;
+    partitionBy?: Expression;
     sortBy?: Record<string, 1 | -1>;
     output: Record<
       string,