@dnax/core 0.18.7 → 0.18.8

package/driver/mongo/rest.ts

@@ -633,6 +633,7 @@ class useRest {
             io: Cfg.io,
             driver: "mongodb",
             action: "find",
+            count: result?.docs?.length || 0,
             params: params,
             session: sessionStorage(),
             result: result.docs,
@@ -673,6 +674,109 @@ class useRest {
       }
     });
   }
+
+  async count(
+    collection: string,
+    params: findParam,
+    options?: optionCb
+  ): Promise<number> {
+    return new Promise(async (resolve, reject) => {
+      try {
+        if (options?.cleanDeep) {
+          params = cleanDeep(params);
+        }
+        let meta: { total: number; count: number } = {
+          total: 0,
+          count: 0,
+        };
+        let useHook = options?.useHook ?? this.#useHook;
+        let useCustomApi = options?.useCustomApi ?? this.#useCustomApi;
+        let sharedData = {};
+        let result = {
+          docs: [],
+        };
+
+        let col = getCollection(collection, this.#tenant_id);
+        if (col?.hooks?.beforeOperation && useHook) {
+          await col.hooks.beforeOperation({
+            sharedData: sharedData,
+            c: this.#c,
+            driver: "mongodb",
+            io: Cfg.io,
+            action: "find",
+            params: params,
+            session: sessionStorage(),
+            rest: new useRest({
+              useHook: false,
+              tenant_id: this.#tenant_id,
+            }),
+          });
+        }
+
+        if (col?.hooks?.beforeCount && useHook) {
+          await col.hooks.beforeCount({
+            sharedData: sharedData,
+            c: this.#c,
+            driver: "mongodb",
+            action: "find",
+            io: Cfg.io,
+            params: params,
+            session: sessionStorage(),
+            rest: new useRest({
+              useHook: false,
+              tenant_id: this.#tenant_id,
+            }),
+          });
+        }
+
+        if (col?.customApi?.count && useCustomApi) {
+          let result = (await col?.customApi?.count({
+            io: Cfg.io,
+            session: sessionStorage(),
+            params: params,
+            rest: new useRest({
+              useHook: false,
+              tenant_id: this.#tenant_id,
+              useCustomApi: false,
+            }),
+          })) as any;
+          return resolve(result);
+        }
+
+        result.docs = await this.#tenant.database.db
+          ?.collection(collection)
+          .aggregate(formatData(buildPipeline(params, col)), {
+            allowDiskUse: true,
+          })
+          .toArray();
+
+        result.docs = toJson(result.docs);
+
+        if (col?.hooks?.afterCount && useHook) {
+          await col.hooks.afterCount({
+            sharedData: sharedData,
+            c: this.#c,
+            io: Cfg.io,
+            driver: "mongodb",
+            action: "find",
+            params: params,
+            session: sessionStorage(),
+            result: result.docs,
+            count: result?.docs?.length || 0,
+            rest: new useRest({
+              useHook: false,
+              tenant_id: this.#tenant_id,
+            }),
+          });
+        }
+
+        let resultDocs = toJson([...(result?.docs || [])]);
+        return resolve(resultDocs?.length);
+      } catch (err) {
+        return reject(err);
+      }
+    });
+  }
   /**
    *
    * @param collection

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dnax/core",
-  "version": "0.18.7",
+  "version": "0.18.8",
   "module": "index.ts",
   "type": "module",
   "bin": {

package/types/index.ts

@@ -192,6 +192,7 @@ export type hooksCtx = (ctx: {
   data?: object;
   params?: findParam;
   options?: {};
+  count?: number;
   id?: string;
   ids?: string[];
   update?: updateParams;
@@ -250,6 +251,7 @@ export type Collection = {
     deleteMany?: (ctx: ctxApi) => object | null | undefined;
     find?: (ctx: ctxApi) => Array<object> | null | undefined;
     findOne?: (ctx: ctxApi) => object;
+    count: (ctx: ctxApi) => number;
   };
   cache?: {
     ttl?: string;
@@ -280,6 +282,8 @@ export type Collection = {
     afterInsert?: hooksCtx;
     beforeAggregate?: hooksCtx;
     afterAggregate?: hooksCtx;
+    beforeCount?: hooksCtx;
+    afterCount?: hooksCtx;
   };
   access?: {
     "*"?: accessCtx;
@@ -296,6 +300,7 @@ export type Collection = {
     deleteMany?: accessCtx;
     aggregate?: accessCtx;
     upload?: accessCtx;
+    count?: accessCtx;
   };
   tenant_id?: string;
   slug: string;