Simple tool which will produce JWT tokens for testing.

1 files changed, 45 insertions, 0 deletions

diff --git a/tools/jwt_producer.py b/tools/jwt_producer.py
new file mode 100644
index 0000000..3f8094d
--- /dev/null
+++ b/tools/jwt_producer.py
@@ -0,0 +1,45 @@
+import sys
+import jwt
+import getopt
+
+
+def usage():
+    progname = sys.argv[0]
+
+    print(f'{progname} -p <path to public key> -s <path to private key>' +
+          '-d <domain>, for example sunet.se>')
+    sys.exit(0)
+
+
+def create_token(private_key, domain):
+    payload = {
+        'type': 'access',
+        'domains': [domain]  # We'll just do one domain now
+    }
+
+    with open(private_key, "r") as fd:
+        key = fd.read()
+
+    return jwt.encode(payload=payload, algorithm='ES256', key=key)
+
+
+if __name__ == '__main__':
+    try:
+        opts, args = getopt.getopt(sys.argv[1:], 'p:d:')
+    except getopt.GetoptError:
+        usage()
+
+    if len(sys.argv) != 5:
+        usage()
+
+    for opt, arg in opts:
+        if opt == '-p':
+            private_key = arg
+        elif opt == '-d':
+            domain = arg
+        else:
+            usage()
+
+    token = create_token(private_key, domain).decode('utf-8')
+
+    print(f'{token}')