@strapi/database 4.4.3 → 4.5.0-beta.0

package/lib/entity-manager/entity-repository.js

@@ -1,5 +1,8 @@
 'use strict';
 
+const { isString } = require('lodash/fp');
+const { isAnyToMany } = require('../metadata/relations');
+
 const withDefaultPagination = (params) => {
   const { page = 1, pageSize = 10, ...rest } = params;
 
@@ -10,6 +13,21 @@ const withDefaultPagination = (params) => {
   };
 };
 
+const withOffsetLimit = (params) => {
+  const { page, pageSize, ...rest } = withDefaultPagination(params);
+
+  const offset = Math.max(page - 1, 0) * pageSize;
+  const limit = pageSize;
+
+  const query = {
+    ...rest,
+    limit,
+    offset,
+  };
+
+  return [query, { page, pageSize }];
+};
+
 const createRepository = (uid, db) => {
   return {
     findOne(params) {
@@ -28,16 +46,7 @@ const createRepository = (uid, db) => {
     },
 
     async findPage(params) {
-      const { page, pageSize, ...rest } = withDefaultPagination(params);
-
-      const offset = Math.max(page - 1, 0) * pageSize;
-      const limit = pageSize;
-
-      const query = {
-        ...rest,
-        limit,
-        offset,
-      };
+      const [query, { page, pageSize }] = withOffsetLimit(params);
 
       const [results, total] = await Promise.all([
         db.entityManager.findMany(uid, query),
@@ -99,8 +108,38 @@ const createRepository = (uid, db) => {
       return db.entityManager.populate(uid, entity, populate);
     },
 
-    load(entity, field, params) {
-      return db.entityManager.load(uid, entity, field, params);
+    load(entity, fields, params) {
+      return db.entityManager.load(uid, entity, fields, params);
+    },
+
+    async loadPages(entity, field, params) {
+      if (!isString(field)) {
+        throw new Error(`Invalid load. Expected ${field} to be a string`);
+      }
+
+      const { attributes } = db.metadata.get(uid);
+      const attribute = attributes[field];
+
+      if (!attribute || attribute.type !== 'relation' || !isAnyToMany(attribute)) {
+        throw new Error(`Invalid load. Expected ${field} to be an anyToMany relational attribute`);
+      }
+
+      const [query, { page, pageSize }] = withOffsetLimit(params);
+
+      const [results, { count: total }] = await Promise.all([
+        db.entityManager.load(uid, entity, field, query),
+        db.entityManager.load(uid, entity, field, { ...query, count: true }),
+      ]);
+
+      return {
+        results,
+        pagination: {
+          page,
+          pageSize,
+          pageCount: Math.ceil(total / pageSize),
+          total,
+        },
+      };
     },
   };
 };