backend-manager 3.0.43 → 3.0.45

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "backend-manager",
-  "version": "3.0.43",
+  "version": "3.0.45",
   "description": "Quick tools for developing Firebase functions",
   "main": "src/manager/index.js",
   "bin": {

package/src/manager/helpers/utilities.js

@@ -23,9 +23,12 @@ Utilities.prototype.iterateCollection = function (callback, options) {
     options.orderBy = options.orderBy || null;
     options.startAt = options.startAt || null;
     options.startAfter = options.startAfter || null;
+    options.prefetchCursor = typeof options.prefetchCursor === 'undefined'
+      ? (!!options.startAt || !!options.startAfter) && !options.orderBy
+      : options.prefetchCursor;
     options.log = options.log;
 
-    function listAllDocuments(nextPageToken) {
+    async function listAllDocuments(nextPageToken) {
       let query = admin.firestore().collection(options.collection)
 
       options.where
@@ -37,12 +40,24 @@ Utilities.prototype.iterateCollection = function (callback, options) {
         query = query.orderBy(options.orderBy);
       }
 
-      if (batch === -1 && options.startAt) {
-        query = query.startAt(options.startAt);
-      }
+      if (batch === -1) {
+        let prefetchedCursor = null;
+
+        if (options.prefetchCursor) {
+          prefetchedCursor = await admin.firestore().doc(`${options.collection}/${options.startAt || options.startAfter}`)
+          .get()
+          .catch(e => e);
+
+          if (prefetchedCursor instanceof Error) {
+            return reject(prefetchedCursor);
+          }
+        }
 
-      if (batch === -1 && options.startAfter) {
-        query = query.startAfter(options.startAfter);
+        if (options.startAt) {
+          query = query.startAt(prefetchedCursor || options.startAt);
+        } else if (options.startAfter) {
+          query = query.startAfter(prefetchedCursor || options.startAfter);
+        }
       }
 
       // Start at next page
@@ -174,19 +189,41 @@ Utilities.prototype.get = function (docPath, options) {
   const self = this;
 
   return new Promise(function(resolve, reject) {
+    const Manager = self.Manager;
+    const { admin } = Manager.libraries;
 
     options = options || {};
     options.maxAge = options.maxAge || (1000 * 60 * 5); // 5 minutes
+    options.readTime = typeof options.readTime === 'undefined' ? null : options.readTime;
+    options.log = typeof options.log === 'undefined' ? false : options.log;
 
-    self.cache = self.cache || self.Manager.storage({name: 'cache', temporary: true, clear: false});
+    self.cache = self.cache || Manager.storage({name: 'cache', temporary: true, clear: false});
 
     const item = self.cache.get(docPath).value();
     const age = item ? Date.now() - item.time : null;
 
-    if (item && age && age < options.maxAge) {
+    if (options.readTime) {
+      const { Timestamp } = require('firebase-admin/firestore')
+      const time = Math.round(new Date(options.readTime).getTime() / 1000 / 60);
+      const logg = new Date(time * 1000 * 60);
+
+      if (options.log) {
+        console.log('Read time:', logg);
+      }
+
+      // loop docs
+      admin.firestore().runTransaction(
+        updateFunction => updateFunction.get(admin.firestore().doc(docPath)),
+        {readOnly: true, readTime: new Timestamp(time * 60, 0)}
+      )
+      .then(snap => {
+        return resolve(snap);
+      })
+      .catch(e => reject(e));
+    } else if (item && age && age < options.maxAge) {
       return resolve(item.doc);
     } else {
-      self.Manager.libraries.admin.firestore().doc(docPath)
+      admin.firestore().doc(docPath)
         .get()
         .then(async (doc) => {
           const data = doc.data();