[feat] use a config class for the protect function

1 files changed, 4 insertions, 8 deletions

diff --git a/memoryhole/__init__.py b/memoryhole/__init__.py
index fd621ea..a7b0df5 100644
--- a/memoryhole/__init__.py
+++ b/memoryhole/__init__.py
@@ -1,12 +1,8 @@
-from memoryhole.protection import protect
-from memoryhole.openpgp import OpenPGP
+from memoryhole.protection import protect, ProtectConfig
+from memoryhole.openpgp import IOpenPGP
 from memoryhole.gpg import Gnupg
 
 
-PROTECTED_HEADERS = ('Subject', 'Message-ID', 'Date', 'To', 'From')
-OBSCURED_HEADERS = ('Subject', 'Message-ID', 'Date', 'To', 'From')
-
-
 def unwrap(msg, opengp=Gnupg()):
     """
     Unwrap an email replacing and verifying memory hole headers.
@@ -17,10 +13,10 @@ def unwrap(msg, opengp=Gnupg()):
                     verification
     :type openpgp: OpenPGP
 
-    :return: a dencrypted email
+    :return: a decrypted email
     :rtype: Message
     """
     raise NotImplementedError()
 
 
-__all__ = ["protect", "unwrap", "OpenPGP"]
+__all__ = ["protect", "ProtectConfig", "unwrap", "IOpenPGP"]