Added setup.py and changed app to pixelated because it will be a package

1 files changed, 35 insertions, 0 deletions

diff --git a/service/pixelated/bitmask_libraries/nicknym.py b/service/pixelated/bitmask_libraries/nicknym.py
new file mode 100644
index 00000000..c4939e9a
--- /dev/null
+++ b/service/pixelated/bitmask_libraries/nicknym.py
@@ -0,0 +1,35 @@
+from leap.keymanager import KeyManager, openpgp, KeyNotFound
+from .certs import which_bundle
+
+
+class NickNym(object):
+    def __init__(self, provider, config, soledad_session, srp_session):
+        nicknym_url = _discover_nicknym_server(provider)
+        self._email = '%s@%s' % (srp_session.user_name, provider.domain)
+        self.keymanager = KeyManager('%s@%s' % (srp_session.user_name, provider.domain), nicknym_url,
+                                     soledad_session.soledad,
+                                     srp_session.token, which_bundle(provider), provider.api_uri,
+                                     provider.api_version,
+                                     srp_session.uuid, config.gpg_binary)
+
+    def generate_openpgp_key(self):
+        if not self._key_exists(self._email):
+            self._gen_key()
+            self._send_key_to_leap()
+
+    def _key_exists(self, email):
+        try:
+            self.keymanager.get_key(email, openpgp.OpenPGPKey, private=True, fetch_remote=False)
+            return True
+        except KeyNotFound:
+            return False
+
+    def _gen_key(self):
+        self.keymanager.gen_key(openpgp.OpenPGPKey)
+
+    def _send_key_to_leap(self):
+        self.keymanager.send_key(openpgp.OpenPGPKey)
+
+
+def _discover_nicknym_server(provider):
+    return 'https://nicknym.%s:6425/' % provider.domain