@objectstack/driver-memory 9.0.0 → 9.1.0

package/dist/index.d.mts

@@ -180,7 +180,7 @@ declare class InMemoryDriver implements IDataDriver {
      *   { $group: { _id: null, avgPrice: { $avg: '$price' } } }
      * ]);
      */
-    aggregate(object: string, pipeline: Record<string, any>[], options?: DriverOptions): Promise<any[]>;
+    aggregate(object: string, pipeline: Record<string, any>[] | QueryAST, options?: DriverOptions): Promise<any[]>;
     /**
      * Convert ObjectQL filter format to MongoDB query format for Mingo.
      *

package/dist/index.d.ts

@@ -180,7 +180,7 @@ declare class InMemoryDriver implements IDataDriver {
      *   { $group: { _id: null, avgPrice: { $avg: '$price' } } }
      * ]);
      */
-    aggregate(object: string, pipeline: Record<string, any>[], options?: DriverOptions): Promise<any[]>;
+    aggregate(object: string, pipeline: Record<string, any>[] | QueryAST, options?: DriverOptions): Promise<any[]>;
     /**
      * Convert ObjectQL filter format to MongoDB query format for Mingo.
      *

package/dist/index.js

@@ -657,6 +657,22 @@ var _InMemoryDriver = class _InMemoryDriver {
    * ]);
    */
   async aggregate(object, pipeline, options) {
+    if (!Array.isArray(pipeline)) {
+      const query = pipeline;
+      this.logger.debug("Aggregate operation (QueryAST)", {
+        object,
+        groupBy: query.groupBy,
+        aggregations: query.aggregations?.length ?? 0
+      });
+      let results2 = this.getTable(object).map((r) => ({ ...r }));
+      if (query.where) {
+        const mongoQuery = this.convertToMongoQuery(query.where);
+        if (mongoQuery && Object.keys(mongoQuery).length > 0) {
+          results2 = new import_mingo.Query(mongoQuery).find(results2).all();
+        }
+      }
+      return this.performAggregation(results2, query);
+    }
     this.logger.debug("Aggregate operation", { object, stageCount: pipeline.length });
     const records = this.getTable(object).map((r) => ({ ...r }));
     const aggregator = new import_mingo.Aggregator(pipeline);