From c2c71ab69cfc6f9b0422be3c1c3d055558e1dcf9 Mon Sep 17 00:00:00 2001
From: David Bomba <turbo124@gmail.com>
Date: Mon, 9 Aug 2021 20:33:59 +1000
Subject: [PATCH] Hosted email quotas

---
 app/Jobs/Mail/NinjaMailerJob.php |  4 ++++
 app/Models/Account.php           | 27 ++++++++++++++++-----------
 2 files changed, 20 insertions(+), 11 deletions(-)

diff --git a/app/Jobs/Mail/NinjaMailerJob.php b/app/Jobs/Mail/NinjaMailerJob.php
index 987e79f65716..0c3fd1928772 100644
--- a/app/Jobs/Mail/NinjaMailerJob.php
+++ b/app/Jobs/Mail/NinjaMailerJob.php
@@ -227,6 +227,10 @@ class NinjaMailerJob implements ShouldQueue
         if(Ninja::isHosted() && strpos($this->nmo->to_user->email, '@example.com') !== false)
             return true;
 
+        /* GMail users are uncapped */
+        if(Ninja::isHosted() && $this->nmo->settings->email_sending_method == 'gmail')
+            return false;
+
         /* On the hosted platform, if the user is over the email quotas, we do not send the email. */
         if(Ninja::isHosted() && $this->company->account->emailQuotaExceeded())
             return true;
diff --git a/app/Models/Account.php b/app/Models/Account.php
index 1e6119078e70..ae563d050eed 100644
--- a/app/Models/Account.php
+++ b/app/Models/Account.php
@@ -369,21 +369,26 @@ class Account extends BaseModel
         if(is_null(Cache::get($this->key)))
             return false;
 
-        if(Cache::get($this->key) > $this->getDailyEmailLimit()) {
+        try {
+            if(Cache::get($this->key) > $this->getDailyEmailLimit()) {
 
-            if(is_null(Cache::get("throttle_notified:{$this->key}"))) {
+                if(is_null(Cache::get("throttle_notified:{$this->key}"))) {
 
-                $nmo = new NinjaMailerObject;
-                $nmo->mailable = new EmailQuotaExceeded($this->companies()->first());
-                $nmo->company = $this->companies()->first();
-                $nmo->settings = $this->companies()->first()->settings;
-                $nmo->to_user = $this->companies()->first()->owner();
-                NinjaMailerJob::dispatch($nmo);
+                    $nmo = new NinjaMailerObject;
+                    $nmo->mailable = new EmailQuotaExceeded($this->companies()->first());
+                    $nmo->company = $this->companies()->first();
+                    $nmo->settings = $this->companies()->first()->settings;
+                    $nmo->to_user = $this->companies()->first()->owner();
+                    NinjaMailerJob::dispatch($nmo);
 
-                Cache::put("throttle_notified:{$this->key}", true, 60 * 24);
+                    Cache::put("throttle_notified:{$this->key}", true, 60 * 24);
+                }
+
+                return true;
             }
-
-            return true;
+        }
+        catch(\Exception $e){
+            \Sentry\captureMessage("I encountered an error with email quotas - defaulting to SEND");
         }
 
         return false;