From b06b794de750d972a9dff9165e203815b79c5b62 Mon Sep 17 00:00:00 2001
From: Alexandre Pretto <anunes@thoughtworks.com>
Date: Thu, 8 Jan 2015 18:48:48 -0200
Subject: #157: mail api to return htmlBody and textPlainBody so the UI doesnt
 have to parse the multipart mails anymore

---
 service/pixelated/adapter/model/mail.py      | 54 ++++++++++++++++++----------
 service/pixelated/adapter/search/__init__.py |  2 +-
 2 files changed, 36 insertions(+), 20 deletions(-)

(limited to 'service/pixelated/adapter')

diff --git a/service/pixelated/adapter/model/mail.py b/service/pixelated/adapter/model/mail.py
index 4b128d8c..98533a74 100644
--- a/service/pixelated/adapter/model/mail.py
+++ b/service/pixelated/adapter/model/mail.py
@@ -25,7 +25,6 @@ import pixelated.support.date
 from email.MIMEMultipart import MIMEMultipart
 from pycryptopp.hash import sha256
 import re
-import base64
 from pixelated.support.functional import compact
 
 
@@ -65,7 +64,13 @@ class Mail(object):
         mime = MIMEMultipart()
         for key, value in self.headers.items():
             mime[str(key)] = str(value)
-        mime.attach(MIMEText(self.body, 'plain', self._charset()))
+
+        try:
+            body_to_use = self.body
+        except AttributeError:
+            body_to_use = self.text_plain_body
+
+        mime.attach(MIMEText(body_to_use, 'plain', self._charset()))
         self._mime = mime
         return mime
 
@@ -193,6 +198,7 @@ class InputMail(Mail):
 
 
 class PixelatedMail(Mail):
+
     @staticmethod
     def from_soledad(fdoc, hdoc, bdoc, parts=None, soledad_querier=None):
         mail = PixelatedMail()
@@ -205,24 +211,33 @@ class PixelatedMail(Mail):
         mail._mime = None
         return mail
 
+    def _decode_part(self, part):
+        encoding = part['headers'].get('Content-Transfer-Encoding', '')
+
+        decoding_map = {
+            'quoted-printable': lambda content: unicode(content.decode('quopri')),
+            'base64': lambda content: content.decode('base64').decode('utf-8')
+        }
+        if encoding:
+            return decoding_map[encoding](part['content'])
+        return part['content']
+
     @property
-    def body(self):
-        if self.parts and len(self.parts['alternatives']) > 1:
-            body = ''
-            for alternative in self.parts['alternatives']:
-                body += '--' + self.boundary + '\n'
-                for header, value in alternative['headers'].items():
-                    body += '%s: %s\n' % (header, value)
-                body += '\n'
-                body += alternative['content']
-                body += '\n'
-            body += '--' + self.boundary + '--'
-            return body
+    def alternatives(self):
+        return self.parts.get('alternatives')
+
+    @property
+    def text_plain_body(self):
+        if self.parts and len(self.alternatives) == 1:
+            return self._decode_part(self.alternatives[0])
         else:
-            if self.parts and self.parts['alternatives'][0]['headers'].get('Content-Transfer-Encoding', '') == 'base64':
-                return unicode(base64.b64decode(self.parts['alternatives'][0]['content']), 'utf-8')
-            else:
-                return self.bdoc.content['raw']
+            return self.bdoc.content['raw']  # plain
+
+    @property
+    def html_body(self):
+        if self.parts and len(self.alternatives) > 1:
+            html_part = [e for e in self.alternatives if re.match('text/html', e['headers']['Content-Type'])][0]
+            return self._decode_part(html_part)
 
     @property
     def headers(self):
@@ -352,7 +367,8 @@ class PixelatedMail(Mail):
                      'tags': list(self.tags),
                      'status': list(self.status),
                      'security_casing': self.security_casing,
-                     'body': self.body,
+                     'textPlainBody': self.text_plain_body,
+                     'htmlBody': self.html_body,
                      'mailbox': self.mailbox_name.lower(),
                      'attachments': self.parts['attachments'] if self.parts else []}
         dict_mail['replying'] = {'single': None, 'all': {'to-field': [], 'cc-field': []}}
diff --git a/service/pixelated/adapter/search/__init__.py b/service/pixelated/adapter/search/__init__.py
index bbefc487..a5eb4ef2 100644
--- a/service/pixelated/adapter/search/__init__.py
+++ b/service/pixelated/adapter/search/__init__.py
@@ -128,7 +128,7 @@ class SearchEngine(object):
             'cc': u','.join(header.get('cc', [''])),
             'bcc': u','.join(header.get('bcc', [''])),
             'tag': u','.join(unique(tags)),
-            'body': unicode(mdict['body']),
+            'body': unicode(mdict['textPlainBody']),
             'ident': unicode(mdict['ident']),
             'flags': unicode(','.join(unique(mail.flags))),
             'raw': unicode(mail.raw)
-- 
cgit v1.2.3