backend-manager 3.2.111 → 3.2.112

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "backend-manager",
-  "version": "3.2.111",
+  "version": "3.2.112",
   "description": "Quick tools for developing Firebase functions",
   "main": "src/manager/index.js",
   "bin": {
@@ -76,4 +76,4 @@
     "wonderful-log": "^1.0.5",
     "yargs": "^17.7.2"
   }
-}
+}

package/src/manager/helpers/usage.js

@@ -43,6 +43,7 @@ Usage.prototype.init = function (assistant, options) {
     options.today = typeof options.today === 'undefined' ? undefined : options.today;
     options.key = typeof options.key === 'undefined' ? undefined : options.key;
     options.unauthenticatedMode = typeof options.unauthenticatedMode === 'undefined' ? 'firestore' : options.unauthenticatedMode;
+    options.whitelistKeys = options.whitelistKeys || [];
     options.log = typeof options.log === 'undefined' ? false : options.log;
 
     // Check for required options
@@ -50,6 +51,12 @@ Usage.prototype.init = function (assistant, options) {
       return reject(new Error('Missing required {assistant} parameter'));
     }
 
+    // Add BEM to whitelist keys
+    options.whitelistKeys.push(
+      Manager?.config?.backend_manager?.key
+      || process.env.BACKEND_MANAGER_KEY
+    );
+
     // Set options
     self.options = options;
 
@@ -171,6 +178,14 @@ Usage.prototype.validate = function (name, options) {
       return _reject();
     }
 
+    // Check if they have a white list key
+    const hasWhitelistKey = self.options.whitelistKeys.some((key) => key && key === self?.user?.api?.privateKey);
+    if (hasWhitelistKey) {
+      self.log(`Usage.validate(): Whitelist key found for ${name}`);
+
+      return resolve(true);
+    }
+
     // If they are under the limit, resolve
     if (period < allowed) {
       self.log(`Usage.validate(): Valid for ${name}`);
@@ -339,6 +354,21 @@ Usage.prototype.update = function () {
   });
 };
 
+Usage.prototype.addWhitelistKeys = function (keys) {
+  const self = this;
+
+  const options = self.options;
+
+  // Make keys and array if not already
+  keys = Array.isArray(keys) ? keys : [keys];
+
+  // Add keys to whitelist
+  options.whitelistKeys = options.whitelistKeys.concat(keys);
+
+  // Log
+  return self;
+};
+
 Usage.prototype.log = function () {
   const self = this;