@prisma-next/mongo-query-ast 0.0.1

package/src/visitors.ts

@@ -0,0 +1,135 @@
+import type {
+  MongoAggAccumulator,
+  MongoAggArrayFilter,
+  MongoAggCond,
+  MongoAggExpr,
+  MongoAggFieldRef,
+  MongoAggLet,
+  MongoAggLiteral,
+  MongoAggMap,
+  MongoAggMergeObjects,
+  MongoAggOperator,
+  MongoAggReduce,
+  MongoAggSwitch,
+} from './aggregation-expressions';
+import type {
+  MongoAndExpr,
+  MongoExistsExpr,
+  MongoExprFilter,
+  MongoFieldFilter,
+  MongoFilterExpr,
+  MongoNotExpr,
+  MongoOrExpr,
+} from './filter-expressions';
+import type {
+  MongoAddFieldsStage,
+  MongoBucketAutoStage,
+  MongoBucketStage,
+  MongoCountStage,
+  MongoDensifyStage,
+  MongoFacetStage,
+  MongoFillStage,
+  MongoGeoNearStage,
+  MongoGraphLookupStage,
+  MongoGroupStage,
+  MongoLimitStage,
+  MongoLookupStage,
+  MongoMatchStage,
+  MongoMergeStage,
+  MongoOutStage,
+  MongoProjectStage,
+  MongoRedactStage,
+  MongoReplaceRootStage,
+  MongoSampleStage,
+  MongoSearchMetaStage,
+  MongoSearchStage,
+  MongoSetWindowFieldsStage,
+  MongoSkipStage,
+  MongoSortByCountStage,
+  MongoSortStage,
+  MongoUnionWithStage,
+  MongoUnwindStage,
+  MongoVectorSearchStage,
+} from './stages';
+
+export interface MongoAggExprVisitor<R> {
+  fieldRef(expr: MongoAggFieldRef): R;
+  literal(expr: MongoAggLiteral): R;
+  operator(expr: MongoAggOperator): R;
+  accumulator(expr: MongoAggAccumulator): R;
+  cond(expr: MongoAggCond): R;
+  switch_(expr: MongoAggSwitch): R;
+  filter(expr: MongoAggArrayFilter): R;
+  map(expr: MongoAggMap): R;
+  reduce(expr: MongoAggReduce): R;
+  let_(expr: MongoAggLet): R;
+  mergeObjects(expr: MongoAggMergeObjects): R;
+}
+
+export interface MongoAggExprRewriter {
+  fieldRef?(expr: MongoAggFieldRef): MongoAggExpr;
+  literal?(expr: MongoAggLiteral): MongoAggExpr;
+  operator?(expr: MongoAggOperator): MongoAggExpr;
+  accumulator?(expr: MongoAggAccumulator): MongoAggExpr;
+  cond?(expr: MongoAggCond): MongoAggExpr;
+  switch_?(expr: MongoAggSwitch): MongoAggExpr;
+  filter?(expr: MongoAggArrayFilter): MongoAggExpr;
+  map?(expr: MongoAggMap): MongoAggExpr;
+  reduce?(expr: MongoAggReduce): MongoAggExpr;
+  let_?(expr: MongoAggLet): MongoAggExpr;
+  mergeObjects?(expr: MongoAggMergeObjects): MongoAggExpr;
+}
+
+export interface MongoFilterVisitor<R> {
+  field(expr: MongoFieldFilter): R;
+  and(expr: MongoAndExpr): R;
+  or(expr: MongoOrExpr): R;
+  not(expr: MongoNotExpr): R;
+  exists(expr: MongoExistsExpr): R;
+  expr(expr: MongoExprFilter): R;
+}
+
+export interface MongoFilterRewriter {
+  field?(expr: MongoFieldFilter): MongoFilterExpr;
+  and?(expr: MongoAndExpr): MongoFilterExpr;
+  or?(expr: MongoOrExpr): MongoFilterExpr;
+  not?(expr: MongoNotExpr): MongoFilterExpr;
+  exists?(expr: MongoExistsExpr): MongoFilterExpr;
+  expr?(expr: MongoExprFilter): MongoFilterExpr;
+}
+
+export interface MongoStageRewriterContext {
+  filter?: MongoFilterRewriter;
+  aggExpr?: MongoAggExprRewriter;
+}
+
+export interface MongoStageVisitor<R> {
+  match(stage: MongoMatchStage): R;
+  project(stage: MongoProjectStage): R;
+  sort(stage: MongoSortStage): R;
+  limit(stage: MongoLimitStage): R;
+  skip(stage: MongoSkipStage): R;
+  lookup(stage: MongoLookupStage): R;
+  unwind(stage: MongoUnwindStage): R;
+  group(stage: MongoGroupStage): R;
+  addFields(stage: MongoAddFieldsStage): R;
+  replaceRoot(stage: MongoReplaceRootStage): R;
+  count(stage: MongoCountStage): R;
+  sortByCount(stage: MongoSortByCountStage): R;
+  sample(stage: MongoSampleStage): R;
+  redact(stage: MongoRedactStage): R;
+  out(stage: MongoOutStage): R;
+  unionWith(stage: MongoUnionWithStage): R;
+  bucket(stage: MongoBucketStage): R;
+  bucketAuto(stage: MongoBucketAutoStage): R;
+  geoNear(stage: MongoGeoNearStage): R;
+  facet(stage: MongoFacetStage): R;
+  graphLookup(stage: MongoGraphLookupStage): R;
+  merge(stage: MongoMergeStage): R;
+  setWindowFields(stage: MongoSetWindowFieldsStage): R;
+  densify(stage: MongoDensifyStage): R;
+  fill(stage: MongoFillStage): R;
+  search(stage: MongoSearchStage): R;
+  searchMeta(stage: MongoSearchMetaStage): R;
+  vectorSearch(stage: MongoVectorSearchStage): R;
+}