From 1e3fca15c5a18b837b9222a72e35c09cb8dad16f Mon Sep 17 00:00:00 2001
From: Martin McKeaveney <martin@shogunsystems.co.uk>
Date: Mon, 27 Sep 2021 17:35:22 +0100
Subject: [PATCH] abstract usagedoc creation into auth, create the doc if it
 doesn't exist

---
 packages/auth/src/db/constants.js             |  1 +
 packages/auth/src/db/utils.js                 | 25 +++++++++++++++++++
 packages/server/src/api/routes/user.js        |  2 --
 packages/server/src/api/routes/view.js        |  1 -
 packages/server/src/utilities/usageQuota.js   | 18 ++++++++++++-
 .../src/api/controllers/global/users.js       | 23 ++---------------
 6 files changed, 45 insertions(+), 25 deletions(-)

diff --git a/packages/auth/src/db/constants.js b/packages/auth/src/db/constants.js
index 77643ce4c5..ad4f6c9f66 100644
--- a/packages/auth/src/db/constants.js
+++ b/packages/auth/src/db/constants.js
@@ -12,6 +12,7 @@ exports.StaticDatabases = {
     name: "global-info",
     docs: {
       tenants: "tenants",
+      usageQuota: "usage_quota",
     },
   },
 }
diff --git a/packages/auth/src/db/utils.js b/packages/auth/src/db/utils.js
index a1a831523e..09e2ff6314 100644
--- a/packages/auth/src/db/utils.js
+++ b/packages/auth/src/db/utils.js
@@ -368,8 +368,33 @@ async function getScopedConfig(db, params) {
   return configDoc && configDoc.config ? configDoc.config : configDoc
 }
 
+function generateNewUsageQuotaDoc() {
+  return {
+    _id: StaticDatabases.PLATFORM_INFO.docs.usageQuota,
+    quotaReset: Date.now() + 2592000000,
+    usageQuota: {
+      automationRuns: 0,
+      rows: 0,
+      storage: 0,
+      apps: 0,
+      users: 0,
+      views: 0,
+      emails: 0,
+    },
+    usageLimits: {
+      automationRuns: 1000,
+      rows: 4000,
+      apps: 4,
+      storage: 1000,
+      users: 10,
+      emails: 50,
+    },
+  }
+}
+
 exports.Replication = Replication
 exports.getScopedConfig = getScopedConfig
 exports.generateConfigID = generateConfigID
 exports.getConfigParams = getConfigParams
 exports.getScopedFullConfig = getScopedFullConfig
+exports.generateNewUsageQuotaDoc = generateNewUsageQuotaDoc
diff --git a/packages/server/src/api/routes/user.js b/packages/server/src/api/routes/user.js
index 465fef82c8..d171870215 100644
--- a/packages/server/src/api/routes/user.js
+++ b/packages/server/src/api/routes/user.js
@@ -27,13 +27,11 @@ router
   .post(
     "/api/users/metadata/self",
     authorized(PermissionTypes.USER, PermissionLevels.WRITE),
-    // usage,
     controller.updateSelfMetadata
   )
   .delete(
     "/api/users/metadata/:id",
     authorized(PermissionTypes.USER, PermissionLevels.WRITE),
-    // usage,
     controller.destroyMetadata
   )
 
diff --git a/packages/server/src/api/routes/view.js b/packages/server/src/api/routes/view.js
index f096dd4185..b72fe1ac26 100644
--- a/packages/server/src/api/routes/view.js
+++ b/packages/server/src/api/routes/view.js
@@ -8,7 +8,6 @@ const {
   PermissionTypes,
   PermissionLevels,
 } = require("@budibase/auth/permissions")
-const usage = require("../../middleware/usageQuota")
 
 const router = Router()
 
diff --git a/packages/server/src/utilities/usageQuota.js b/packages/server/src/utilities/usageQuota.js
index 4e645ec67d..80fddb8303 100644
--- a/packages/server/src/utilities/usageQuota.js
+++ b/packages/server/src/utilities/usageQuota.js
@@ -1,5 +1,9 @@
 const env = require("../environment")
 const { getGlobalDB } = require("@budibase/auth/tenancy")
+const {
+  StaticDatabases,
+  generateNewUsageQuotaDoc,
+} = require("@budibase/auth/db")
 
 function getNewQuotaReset() {
   return Date.now() + 2592000000
@@ -15,6 +19,18 @@ exports.Properties = {
   EMAILS: "emails",
 }
 
+async function getUsageQuotaDoc(db) {
+  let quota
+  try {
+    quota = await db.get(StaticDatabases.PLATFORM_INFO.docs.usageQuota)
+  } catch (err) {
+    // doc doesn't exist. Create it
+    quota = await db.post(generateNewUsageQuotaDoc())
+  }
+
+  return quota
+}
+
 /**
  * Given a specified tenantId this will add to the usage object for the specified property.
  * @param {string} property The property which is to be added to (within the nested usageQuota object).
@@ -29,7 +45,7 @@ exports.update = async (property, usage) => {
 
   try {
     const db = getGlobalDB()
-    const quota = await db.get("usage_quota")
+    const quota = await getUsageQuotaDoc(db)
 
     // Check if the quota needs reset
     if (Date.now() >= quota.quotaReset) {
diff --git a/packages/worker/src/api/controllers/global/users.js b/packages/worker/src/api/controllers/global/users.js
index e00fd3ecc0..1d3f38698b 100644
--- a/packages/worker/src/api/controllers/global/users.js
+++ b/packages/worker/src/api/controllers/global/users.js
@@ -2,6 +2,7 @@ const {
   generateGlobalUserID,
   getGlobalUserParams,
   StaticDatabases,
+  generateNewUsageQuotaDoc,
 } = require("@budibase/auth/db")
 const { hash, getGlobalUserByEmail } = require("@budibase/auth").utils
 const { UserStatus, EmailTemplatePurpose } = require("../../../constants")
@@ -141,27 +142,7 @@ exports.adminUser = async ctx => {
 
   // write usage quotas for cloud
   if (!env.SELF_HOSTED) {
-    await db.post({
-      _id: "usage_quota",
-      quotaReset: Date.now() + 2592000000,
-      usageQuota: {
-        automationRuns: 0,
-        rows: 0,
-        storage: 0,
-        apps: 0,
-        users: 0,
-        views: 0,
-        emails: 0,
-      },
-      usageLimits: {
-        automationRuns: 1000,
-        rows: 4000,
-        apps: 4,
-        storage: 1000,
-        users: 10,
-        emails: 50,
-      },
-    })
+    await db.post(generateNewUsageQuotaDoc())
   }
 
   if (response.rows.some(row => row.doc.admin)) {