From eaf0b4e56e6a9b4146f038e4e41d89ecb7595dae Mon Sep 17 00:00:00 2001
From: Girish Ramakrishnan <girish@cloudron.io>
Date: Wed, 13 Jun 2018 07:51:22 -0700
Subject: [PATCH] mail api: handle not found errors

---
 src/mail.js | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/mail.js b/src/mail.js
index 2dba37e73..050a7a013 100644
--- a/src/mail.js
+++ b/src/mail.js
@@ -811,6 +811,7 @@ function setMailRelay(domain, relay, callback) {
         if (error) return callback(error);
 
         maildb.update(domain, { relay: relay }, function (error) {
+            if (error && error.reason === DatabaseError.NOT_FOUND) return callback(new MailError(MailError.NOT_FOUND));
             if (error) return callback(new MailError(MailError.INTERNAL_ERROR, error));
 
             restartMail(NOOP_CALLBACK);
@@ -826,6 +827,7 @@ function setMailEnabled(domain, enabled, callback) {
     assert.strictEqual(typeof callback, 'function');
 
     maildb.update(domain, { enabled: enabled }, function (error) {
+        if (error && error.reason === DatabaseError.NOT_FOUND) return callback(new MailError(MailError.NOT_FOUND));
         if (error) return callback(new MailError(MailError.INTERNAL_ERROR, error));
 
         restartMail(NOOP_CALLBACK);