backend-manager 2.0.16 → 2.0.19

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "backend-manager",
-  "version": "2.0.16",
+  "version": "2.0.19",
   "description": "Quick tools for developing Firebase functions",
   "main": "src/manager/index.js",
   "bin": {
@@ -31,7 +31,7 @@
     "@google-cloud/storage": "^5.19.4",
     "@sendgrid/mail": "^7.6.2",
     "@sentry/node": "^6.19.7",
-    "backend-assistant": "^0.0.62",
+    "backend-assistant": "^0.0.63",
     "busboy": "^1.6.0",
     "chalk": "^4.1.2",
     "cors": "^2.8.5",
@@ -64,4 +64,4 @@
     "src/",
     "templates/"
   ]
-}
+}

package/src/manager/helpers/api-manager.js

@@ -1,4 +1,5 @@
 const moment = require('moment');
+const fetch = require('node-fetch');
 const uuidv5 = require('uuid').v5;
 const { get, set, merge } = require('lodash');
 
@@ -18,10 +19,12 @@ let sampleUser = {
   }
 }
 
-function ApiManager() {
+function ApiManager(m) {
   const self = this;
+  self.Manager = m;
   self.options = {
-    planData: {},
+    appId: '',
+    plans: {},
     maxUsersStored: 10000,
     refetchInterval: 60,
     resetInterval: 60 * 24,
@@ -33,23 +36,58 @@ function ApiManager() {
 
 ApiManager.prototype.init = function (options) {
   const self = this;
-  options = options || {};
-  options.planData = options.planData || {};
-  options.planData.basic = options.planData.basic || {requests: 93};
+  return new Promise(async function(resolve, reject) {
+    options = options || {};
+    options.app = options.app || '';
+    options.plans = options.plans || {};
 
-  options.maxUsersStored = options.maxUsersStored || 10000;
-  options.refetchInterval = options.refetchInterval || 60;
-  options.resetInterval = options.resetInterval || (60 * 24);
-  options.officialAPIKeys = options.officialAPIKeys || [];
+    // await self.Manager.libraries.admin.firestore
+    // options.plans.basic = options.plans.basic || {requests: 93};
 
-  self.options = options;
+    options.maxUsersStored = options.maxUsersStored || 10000;
+    options.refetchInterval = options.refetchInterval || 60;
+    options.resetInterval = options.resetInterval || (60 * 24);
+    options.officialAPIKeys = options.officialAPIKeys || [];
+    options.whitelistedAPIKeys = options.whitelistedAPIKeys || [];
 
-  self.initialized = true;
+    await fetch('https://us-central1-itw-creative-works.cloudfunctions.net/getApp', {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({
+        id: options.app,
+      }),
+    })
+    .then(res => {
+      res.text()
+      .then(text => {
+        if (res.ok) {
+          const data = JSON.parse(text);
 
-  return self;
+          options.plans = {};
+
+          Object.keys(data.products)
+          .forEach((id, i) => {
+            const product = data.products[id]
+            options.plans[product.planId] = {}
+            options.plans[product.planId].limits = product.limits || {};
+          });
+
+          self.options = options;
+          self.initialized = true;
+
+          return resolve(self);
+        } else {
+          throw new Error(text || res.statusText || 'Unknown error.')
+        }
+      })
+    })
+    .catch(e => {
+      return reject(e)
+    })
+  });
 };
 
-ApiManager.prototype._createNewUser = function (authenticatedUser, planId, persistentData, isRefetch) {
+ApiManager.prototype._createNewUser = function (authenticatedUser, planId, persistentData, isRefetch, apiKey) {
   const self = this;
   persistentData = persistentData || {};
   persistentData._meta = persistentData._meta || {};
@@ -61,7 +99,7 @@ ApiManager.prototype._createNewUser = function (authenticatedUser, planId, persi
     plan: {
       id: planId,
       limits: {
-        requests: get(authenticatedUser, 'plan.limits.requests', get(self.options, `planData.${planId}.limits.requests`, 93)),
+        requests: get(authenticatedUser, 'plan.limits.requests', get(self.options, `plans.${planId}.limits.requests`, 93)),
       }
     },
     authenticated: authenticatedUser.authenticated,
@@ -73,7 +111,8 @@ ApiManager.prototype._createNewUser = function (authenticatedUser, planId, persi
     _meta: {
       lastStatsReset: new Date(),
       lastUserFetch: new Date(),
-    }
+    },
+    _providedAPIKey: apiKey,
   }
   // console.log('-----MIN', moment().diff(moment(persistentData._meta.lastStatsReset), 'minutes', true), self.options.resetInterval);
   if (moment().diff(moment(persistentData._meta.lastStatsReset), 'minutes', true) < self.options.resetInterval) {
@@ -145,14 +184,14 @@ ApiManager.prototype.getUser = async function (assistant) {
       if (moment().diff(moment(existingUser._meta.lastUserFetch), 'minutes', true) > self.options.refetchInterval) {
         // console.log('----REFETCHING');
         self.userList = self.userList.filter(user => user.auth.uid !== workingUID)
-        existingUser = self._createNewUser(authenticatedUser, planId, persistentData, true);
+        existingUser = self._createNewUser(authenticatedUser, planId, persistentData, true, apiKey);
         existingUser.auth.uid = workingUID;
         self.userList = self.userList.concat(existingUser);
       }
       newUser = existingUser
     } else {
       // console.log('---actually doesnt exist making new user');
-      newUser = self._createNewUser(authenticatedUser, planId, persistentData)
+      newUser = self._createNewUser(authenticatedUser, planId, persistentData, false, apiKey)
       newUser.auth.uid = workingUID;
       self.userList = self.userList.concat(newUser);
     }
@@ -164,15 +203,24 @@ ApiManager.prototype.getUser = async function (assistant) {
 
 };
 
+function _getUserStat(self, user, stat, def) {
+  const isWhitelistedAPIKey = self.options.whitelistedAPIKeys.includes(
+    get(user, `api.privateKey`, get(user, `_providedAPIKey`))
+  );
+  // console.log('----user', user);
+  // console.log('----isWhitelistedAPIKey', isWhitelistedAPIKey);
+  return {
+    current: !isWhitelistedAPIKey ? get(user, `_stats.${stat}`, typeof def !== 'undefined' ? def : 0) : 0,
+    limit: !isWhitelistedAPIKey ? get(user, `plan.limits.${stat}`, typeof def !== 'undefined' ? def : 0) : Infinity,
+  }
+}
+
 ApiManager.prototype.getUserStat = function (user, stat, def) {
   const self = this;
   if (!user || !stat) {
     throw new Error('<user> and <stat> required')
   }
-  return {
-    current: get(user, `_stats.${stat}`, def || 0),
-    limit: get(user, `plan.limits.${stat}`, def || 0),
-  }
+  return _getUserStat(self, user, stat, def);
 }
 
 ApiManager.prototype.incrementUserStat = function (user, stat, amount) {
@@ -181,10 +229,7 @@ ApiManager.prototype.incrementUserStat = function (user, stat, amount) {
     throw new Error('<user> and <stat> required')
   }
   set(user, `_stats.${stat}`, get(user, `_stats.${stat}`, 0) + amount)
-  return {
-    current: get(user, `_stats.${stat}`, 0),
-    limit: get(user, `plan.limits.${stat}`, 0),
-  }
+  return _getUserStat(self, user, stat, 0);
 }