postgres-ai
diff --git a/‎ui/packages/platform/src/actions/actions.js
Lines changed: 37 additions & 0 deletions b/‎ui/packages/platform/src/actions/actions.js
Lines changed: 37 additions & 0 deletions
diff --git a/‎ui/packages/platform/src/api/api.js
Lines changed: 29 additions & 0 deletions b/‎ui/packages/platform/src/api/api.js
Lines changed: 29 additions & 0 deletions
@@ -56,6 +56,7 @@ const Actions = Reflux.createActions([{
   createOrg: ASYNC_ACTION,
   updateAiBotSettings: ASYNC_ACTION,
   updateAuditSettings: ASYNC_ACTION,
+  updateDBLabSettings: ASYNC_ACTION,
   inviteUser: ASYNC_ACTION,
   useDemoData: ASYNC_ACTION,
   setReportsProject: {},
@@ -730,6 +731,42 @@ Actions.updateAuditSettings.listen(function (token, orgId, orgData) {
     });
 });
 
+Actions.updateDBLabSettings.listen(function (token, orgId, orgData) {
+  let action = this;
+
+  if (!api) {
+    settings.init(function () {
+      api = new Api(settings);
+    });
+  }
+
+  action.progressed({ orgId } + orgData);
+  timeoutPromise(REQUEST_TIMEOUT, api.updateDBLabSettings(token, orgId, orgData))
+
+    .then(result => {
+      result.json()
+        .then(json => {
+          if (json) {
+            action.completed(json);
+          } else {
+            action.failed(new Error('wrong_reply'));
+          }
+        })
+        .catch(err => {
+          console.error(err);
+          action.failed(new Error('wrong_reply'));
+        });
+    })
+    .catch(err => {
+      console.error(err);
+      if (err && err.message && err.message === 'timeout') {
+        action.failed(new Error('failed_fetch'));
+      } else {
+        action.failed(new Error('wrong_reply'));
+      }
+    });
+});
+
 Actions.createOrg.listen(function (token, orgData) {
   let action = this;
 
 
@@ -132,6 +132,14 @@ class Api {
       body.last_name = data.last_name;
     }
 
+    if (data.dblab_low_disk_space_notifications_enabled !== 'undefined') {
+      body.dblab_low_disk_space_notifications_enabled = data.dblab_low_disk_space_notifications_enabled;
+    }
+
+    if (data.dblab_old_clones_notifications_enabled !== 'undefined') {
+      body.dblab_old_clones_notifications_enabled = data.dblab_old_clones_notifications_enabled;
+    }
+
     return this.post(`${this.apiServer}/rpc/update_user_profile`, body, {
       headers: headers
     });
@@ -528,6 +536,27 @@ class Api {
     });
   }
 
+  updateDBLabSettings(token, orgId, orgData) {
+    let params = {};
+    let headers = {
+      Authorization: 'Bearer ' + token,
+      prefer: 'return=representation'
+    };
+
+    if (typeof orgData.dblab_low_disk_space_notifications_threshold_percent !== 'undefined') {
+      params.dblab_low_disk_space_notifications_threshold_percent = orgData.dblab_low_disk_space_notifications_threshold_percent
+    }
+
+    if (typeof orgData.dblab_old_clones_notifications_threshold_hours !== 'undefined') {
+      params.dblab_old_clones_notifications_threshold_hours = orgData.dblab_old_clones_notifications_threshold_hours
+    }
+
+    return this.patch(`${this.apiServer}/orgs?id=eq.` + orgId, params, {
+      headers: headers
+    });
+  }
+
+
   testSiemServiceConnection(token, data) {
     let params = {};
     let headers = {