Merge pull request #133 from TogetherCrew/add-populate-to-paginate-pl…

…ugin Add populate to paginate plugin
TogetherCrew · Dec 1, 2023 · d9a617c · d9a617c
2 parents 248f62a + 603dc87
commit d9a617c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@togethercrew.dev/db",
-  "version": "3.0.17",
+  "version": "3.0.18",
   "description": "All interactions with DB",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",

diff --git a/src/models/schemas/plugins/pagintae.plugin.ts b/src/models/schemas/plugins/pagintae.plugin.ts
@@ -21,7 +21,7 @@ function paginate(schema: any) {
    * @returns {Promise<QueryResult>}
    */
   schema.statics.paginate = async function (filter: any, options: any) {
-    let { limit, page, skip } = options;
+    let { limit, page, skip, populate } = options;
     const { sortBy } = options;
     limit = limit && parseInt(limit, 10) > 0 ? parseInt(limit, 10) : 10;
     page = page && parseInt(page, 10) > 0 ? parseInt(page, 10) : 1;
@@ -42,6 +42,16 @@ function paginate(schema: any) {
     const countPromise = this.countDocuments(filter).exec();
     let docsPromise = this.find(filter).sort(sort).skip(skip).limit(limit);
 
+    if (populate) {
+      if (Array.isArray(populate)) {
+        populate.forEach((p) => {
+          docsPromise = docsPromise.populate(p);
+        });
+      } else {
+        docsPromise = docsPromise.populate(populate);
+      }
+    }
+
     docsPromise = docsPromise.exec();
 
     // eslint-disable-next-line @typescript-eslint/return-await