#216 when trying to save a draft too fast (before the previous version is saved) simply ignore the attempt

4 files changed, 29 insertions, 14 deletions

diff --git a/service/pixelated/adapter/services/mail_service.py b/service/pixelated/adapter/services/mail_service.py
index f4f37d77..cb92228c 100644
--- a/service/pixelated/adapter/services/mail_service.py
+++ b/service/pixelated/adapter/services/mail_service.py
@@ -41,6 +41,9 @@ class MailService:
     def mail(self, mail_id):
         return self.mailboxes.mail(mail_id)
 
+    def mail_exists(self, mail_id):
+        return not(not(self.querier.get_header_by_chash(mail_id)))
+
     def send(self, last_draft_ident, mail):
         self.mail_sender.sendmail(mail)
         if last_draft_ident:
diff --git a/service/pixelated/adapter/soledad/soledad_facade_mixin.py b/service/pixelated/adapter/soledad/soledad_facade_mixin.py
index 33588d1e..1df038ea 100644
--- a/service/pixelated/adapter/soledad/soledad_facade_mixin.py
+++ b/service/pixelated/adapter/soledad/soledad_facade_mixin.py
@@ -23,17 +23,20 @@ class SoledadDbFacadeMixin(object):
     def get_all_flags_by_mbox(self, mbox):
         return self.soledad.get_from_index('by-type-and-mbox', 'flags', mbox)
 
-    def get_all_headers_by_chash(self, chash):
-        return self.soledad.get_from_index('by-type-and-contenthash', 'head', chash)
-
     def get_content_by_phash(self, phash):
-        return self.soledad.get_from_index('by-type-and-payloadhash', 'cnt', phash)
+        content = self.soledad.get_from_index('by-type-and-payloadhash', 'cnt', phash)
+        if len(content):
+            return content[0]
 
     def get_flags_by_chash(self, chash):
-        return self.soledad.get_from_index('by-type-and-contenthash', 'flags', chash)[0]
+        flags = self.soledad.get_from_index('by-type-and-contenthash', 'flags', chash)
+        if len(flags):
+            return flags[0]
 
     def get_header_by_chash(self, chash):
-        return self.soledad.get_from_index('by-type-and-contenthash', 'head', chash)[0]
+        header = self.soledad.get_from_index('by-type-and-contenthash', 'head', chash)
+        if len(header):
+            return header[0]
 
     def get_recent_by_mbox(self, mbox):
         return self.soledad.get_from_index('by-type-and-mbox', 'rct', mbox)
diff --git a/service/pixelated/adapter/soledad/soledad_reader_mixin.py b/service/pixelated/adapter/soledad/soledad_reader_mixin.py
index 9cb20a9a..24010e2d 100644
--- a/service/pixelated/adapter/soledad/soledad_reader_mixin.py
+++ b/service/pixelated/adapter/soledad/soledad_reader_mixin.py
@@ -34,26 +34,29 @@ class SoledadReaderMixin(SoledadDbFacadeMixin, object):
 
         fdocs_hdocs = []
         for fdoc, chash in fdocs_chash:
-            hdoc = self.get_all_headers_by_chash(chash)
-            if len(hdoc) == 0:
+            hdoc = self.get_header_by_chash(chash)
+            if not hdoc:
                 continue
-            fdocs_hdocs.append((fdoc, hdoc[0]))
+            fdocs_hdocs.append((fdoc, hdoc))
 
         fdocs_hdocs_bodyphash = [(f[0], f[1], f[1].content.get('body')) for f in fdocs_hdocs]
         fdocs_hdocs_bdocs_parts = []
         for fdoc, hdoc, body_phash in fdocs_hdocs_bodyphash:
             bdoc = self.get_content_by_phash(body_phash)
-            if len(bdoc) == 0:
+            if not bdoc:
                 continue
             parts = self._extract_parts(hdoc.content)
-            fdocs_hdocs_bdocs_parts.append((fdoc, hdoc, bdoc[0], parts))
+            fdocs_hdocs_bdocs_parts.append((fdoc, hdoc, bdoc, parts))
 
         return [PixelatedMail.from_soledad(*raw_mail, soledad_querier=self) for raw_mail in fdocs_hdocs_bdocs_parts]
 
+    def mail_exists(self, ident):
+        return self.get_flags_by_chash(ident)
+
     def mail(self, ident):
         fdoc = self.get_flags_by_chash(ident)
         hdoc = self.get_header_by_chash(ident)
-        bdoc = self.get_content_by_phash(hdoc.content['body'])[0]
+        bdoc = self.get_content_by_phash(hdoc.content['body'])
         parts = self._extract_parts(hdoc.content)
 
         return PixelatedMail.from_soledad(fdoc, hdoc, bdoc, parts=parts, soledad_querier=self)
@@ -65,7 +68,7 @@ class SoledadReaderMixin(SoledadDbFacadeMixin, object):
         return self._build_mails_from_fdocs(fdocs_chash)
 
     def attachment(self, attachment_ident, encoding):
-        bdoc = self.get_content_by_phash(attachment_ident)[0]
+        bdoc = self.get_content_by_phash(attachment_ident)
         return {'content': self._try_decode(bdoc.content['raw'], encoding),
                 'content-type': bdoc.content['content-type']}
 
@@ -94,7 +97,7 @@ class SoledadReaderMixin(SoledadDbFacadeMixin, object):
         return parts
 
     def _extract_alternative(self, hdoc, headers_dict):
-        bdoc = self.get_content_by_phash(hdoc['phash'])[0]
+        bdoc = self.get_content_by_phash(hdoc['phash'])
         raw_content = bdoc.content['raw']
         return {'headers': headers_dict, 'content': raw_content}
 
diff --git a/service/pixelated/controllers/mails_controller.py b/service/pixelated/controllers/mails_controller.py
index 9d649a1f..29e25d76 100644
--- a/service/pixelated/controllers/mails_controller.py
+++ b/service/pixelated/controllers/mails_controller.py
@@ -15,6 +15,8 @@
 # along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
 
 import json
+import random
+import time
 
 from pixelated.adapter.model.mail import InputMail
 from pixelated.controllers import respond_json
@@ -105,11 +107,15 @@ class MailsController:
         return respond_json(mail.as_dict(), request)
 
     def update_draft(self, request):
+        if bool(random.getrandbits(1)):
+            time.sleep(3)
         content_dict = json.loads(request.content.read())
         _mail = InputMail.from_dict(content_dict)
         draft_id = content_dict.get('ident')
 
         if draft_id:
+            if not self._mail_service.mail_exists(draft_id):
+                return respond_json("", request, status_code=422)
             pixelated_mail = self._draft_service.update_draft(draft_id, _mail)
             self._search_engine.remove_from_index(draft_id)
         else: