From 11a05a8e0980c876c7a69b29cf4af3ecd146c596 Mon Sep 17 00:00:00 2001
From: Manuel Gomez <34234948+lologf@users.noreply.github.com>
Date: Tue, 9 Apr 2019 12:09:55 +0200
Subject: [PATCH] Encrypt, decrypt and padding in model methods

---
 django_mailbox/models.py | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/django_mailbox/models.py b/django_mailbox/models.py
index a343989..221c410 100644
--- a/django_mailbox/models.py
+++ b/django_mailbox/models.py
@@ -17,6 +17,7 @@ import os.path
 import sys
 import uuid
 from tempfile import NamedTemporaryFile
+from Crypto.Cipher import AES
 
 import six
 from six.moves.urllib.parse import parse_qs, unquote, urlparse
@@ -113,6 +114,33 @@ class Mailbox(models.Model):
     objects = models.Manager()
     active_mailboxes = ActiveMailboxManager()
 
+    def pad(self, key):
+        length = 32 - (len(key) % 32)
+        return key + chr(length).encode('utf-8') * length
+
+    def unpad(self, key):
+        return key[0:-ord(key[-1])]
+
+    def encrypt_uri(self):
+        secret_key = self.pad(django_settings.SECRET_KEY)
+        self.uri = unicode(self.pad(self.uri)).encode('utf-8')
+
+        cipher = AES.new(secret_key)
+        self.uri = cipher.encrypt(self.uri)
+        self.uri = base64.b64encode(self.uri)
+
+        return self.uri
+
+    def decrypt_uri(self):
+        secret_key = self.pad(django_settings.SECRET_KEY)
+        self.uri = self.pad(self.uri)
+
+        cipher = AES.new(secret_key)
+        self.uri = base64.b64decode(self.uri)
+        self.uri = cipher.decrypt(self.uri)
+
+        return self.unpad(self.uri)
+    
     @property
     def _protocol_info(self):
         return urlparse(self.uri)