From f11a1327c29e9eb638dc007394f98c7d54129771 Mon Sep 17 00:00:00 2001
From: Brendan Howell <brendan@howell-ersatz.com>
Date: Sat, 21 Jan 2017 19:44:10 +0100
Subject: [PATCH] fix reconnect exception handling. layout tweaks. html
 entities in headers.

---
 screenless/bureau/mailroom/email.html  |  8 ++++++--
 screenless/bureau/mailroom/mailroom.py | 24 +++++++++++++++++++-----
 2 files changed, 25 insertions(+), 7 deletions(-)

diff --git a/screenless/bureau/mailroom/email.html b/screenless/bureau/mailroom/email.html
index 2edff32..cf5d828 100644
--- a/screenless/bureau/mailroom/email.html
+++ b/screenless/bureau/mailroom/email.html
@@ -5,8 +5,11 @@
   <meta charset="utf-8" />
   <style>
     body {
-      font-family: "FreeUniversal";
-      font-size: 11pt;
+      font-family: "CMU Serif";
+      font-size: 10pt;
+    }
+    #header {
+      font-family: "CMU Sans Serif";
     }
     #main {
       margin-top: 2em;
@@ -15,6 +18,7 @@
       float: left;
       width: 63%;
       padding-right: 3%;
+      line-height: 0.9em;
     }
     #menu {
       width: 30%;
diff --git a/screenless/bureau/mailroom/mailroom.py b/screenless/bureau/mailroom/mailroom.py
index a8e1a70..5842d19 100644
--- a/screenless/bureau/mailroom/mailroom.py
+++ b/screenless/bureau/mailroom/mailroom.py
@@ -7,6 +7,16 @@ import imapclient
 from bureau import Bureau, add_command, add_api
 
 
+def clean_header(header):
+    """
+    Converts header string to unicode with HTML entities
+    """
+    header = str(make_header(decode_header(header)))
+    header = header.replace("<", "&lt;")
+    header = header.replace(">", "&gt;")
+    return header
+
+
 class Message(object):
     """
     This is just a convenience class for holding email message data.
@@ -57,8 +67,9 @@ class MailRoom(Bureau):
     def _connect_imap(self):
         try:
             self.imapserv.select_folder("INBOX")
-        except imaplib.error:
+        except imaplib.abort:
             self.imapserv.login(self.login, self.password)
+            self.imapserv.select_folder("INBOX")
 
     def get_imap_id(self, msgid):
         """
@@ -94,11 +105,11 @@ class MailRoom(Bureau):
 
         # format and tidy header data
         msg = Message()
-        msg.fromstr = make_header(decode_header(msg_obj['From']))
-        msg.tostr = make_header(decode_header(msg_obj['To']))
-        msg.subject = make_header(decode_header(msg_obj['Subject']))
+        msg.fromstr = clean_header(msg_obj['From'])
+        msg.tostr = clean_header(msg_obj['To'])
+        msg.subject = clean_header(msg_obj['Subject'])
         if 'Cc' in msg_obj:
-            msg.cc = make_header(decode_header(msg_obj['Cc']))
+            msg.cc = clean_header(msg_obj['Cc'])
         else:
             msg.cc = None
         msg.date = internaldate
@@ -114,6 +125,9 @@ class MailRoom(Bureau):
             #       should clean these up on delete
             if part.get_content_type() == "text/plain":
                 msg.content = part.get_payload(decode=True).decode("utf-8")
+                msg.content = msg.content.replace("<", "&lt;")
+                msg.content = msg.content.replace(">", "&gt;")
+                msg.content = msg.content.replace("\n", "<br />")
 
         self.print_full("email.html", msg=msg, shortcode=shortcode)