Monkey patching fetch decrypt to add encryption info

1 files changed, 14 insertions, 0 deletions

diff --git a/service/pixelated/support/ext_fetch.py b/service/pixelated/support/ext_fetch.py
new file mode 100644
index 00000000..ab0def9f
--- /dev/null
+++ b/service/pixelated/support/ext_fetch.py
@@ -0,0 +1,14 @@
+import leap.mail.imap.fetch as fetch
+
+
+def mark_as_encrypted(f):
+
+    def w(*args, **kwargs):
+        msg, was_decrypted = f(*args)
+        msg.add_header('X-Pixelated-encryption-status', 'true' if was_decrypted else 'false')
+        return msg, was_decrypted
+    return w
+
+
+fetch.LeapIncomingMail._maybe_decrypt_inline_encrypted_msg = mark_as_encrypted(fetch.LeapIncomingMail._maybe_decrypt_inline_encrypted_msg)
+fetch.LeapIncomingMail._decrypt_multipart_encrypted_msg = mark_as_encrypted(fetch.LeapIncomingMail._decrypt_multipart_encrypted_msg)