From b44af0658a26b1148da6b37b2fc38d7315da2c39 Mon Sep 17 00:00:00 2001
From: Wolfgang Hottgenroth <wolfgang.hottgenroth@icloud.com>
Date: Fri, 7 May 2021 13:28:12 +0200
Subject: [PATCH] jwe

---
 auth.py    | 20 ++++++++++++++------
 testjwe.py |  9 +++++++++
 2 files changed, 23 insertions(+), 6 deletions(-)
 create mode 100644 testjwe.py

diff --git a/auth.py b/auth.py
index 357399c..bbdda61 100755
--- a/auth.py
+++ b/auth.py
@@ -30,13 +30,20 @@ class PasswordMismatchException(Exception):
 
 UserEntry = namedtuple('UserEntry', ['id', 'login', 'expiry', 'claims'])
 
+
 JWT_PRIV_KEY = ""
-with open('/opt/app/config/authservice.key', 'r') as f:
-    JWT_PRIV_KEY = f.read()
+try:
+    JWT_PRIV_KEY = os.environ["JWT_PRIV_KEY"]
+except KeyError:
+    with open('/opt/app/config/authservice.key', 'r') as f:
+        JWT_PRIV_KEY = f.read()
 
 JWT_PUB_KEY = ""
-with open('/opt/app/config/authservice.pub', 'r') as f:
-    JWT_PUB_KEY = f.read()
+try:
+    JWT_PUB_KEY = os.environ["JWT_PUB_KEY"]
+except KeyError:
+    with open('/opt/app/config/authservice.pub', 'r') as f:
+        JWT_PUB_KEY = f.read()
 
 
 def getUserEntryFromDB(application: str, login: str):
@@ -129,8 +136,9 @@ def generateToken(**args):
         print("ERROR: generateToken: unspecific exception: {}".format(str(e)))
         raise werkzeug.exceptions.Unauthorized()
 
-def generateTokenFromEnc(content):
-    return content
+def generateTokenFromEnc(**args):
+    cryptContent = args["body"]
+    return str(cryptContent)
 
 def getPubKey():
     return JWT_PUB_KEY
diff --git a/testjwe.py b/testjwe.py
new file mode 100644
index 0000000..6c6b53e
--- /dev/null
+++ b/testjwe.py
@@ -0,0 +1,9 @@
+from jose import jwe
+
+
+JWT_PUB_KEY = os.environ["JWT_PUB_KEY"]
+
+plainText = "BlaBlaBla123"
+cryptText = jwe.encrypt(plainText, JWT_PUB_KEY, "A256GCM", "RSA-OAEP")
+
+print(cryptText)
\ No newline at end of file