From 88b1a1aa213cc70443254a05f2a8968fb7c43a48 Mon Sep 17 00:00:00 2001
From: Kristofer Hallin <kristofer@sunet.se>
Date: Tue, 18 Jan 2022 10:42:14 +0100
Subject: Added -e flag to jwt_producer.

---
 tools/jwt_producer.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

(limited to 'tools/jwt_producer.py')

diff --git a/tools/jwt_producer.py b/tools/jwt_producer.py
index 9f30439..a024c13 100644
--- a/tools/jwt_producer.py
+++ b/tools/jwt_producer.py
@@ -10,7 +10,8 @@ def usage():
     print(f'Usage: {progname} [-p <path>] [-w <value>] [-r <value>]\n' +
           '  -p <path to private key>\n' +
           '  -w <write, comma separated list of domains>\n' +
-          '  -r <read, comma separated list of domains>')
+          '  -r <read, comma separated list of domains>\n' +
+          '  -e <print export statement')
     sys.exit(0)
 
 
@@ -42,9 +43,10 @@ if __name__ == '__main__':
     read_domains = None
     write_domains = None
     private_key = None
+    export = False
 
     try:
-        opts, args = getopt.getopt(sys.argv[1:], 'p:w:r:h')
+        opts, args = getopt.getopt(sys.argv[1:], 'p:w:r:eh')
     except getopt.GetoptError:
         usage()
 
@@ -55,6 +57,8 @@ if __name__ == '__main__':
             write_domains = arg
         elif opt == '-r':
             read_domains = arg
+        elif opt == '-e':
+            export = True
         elif opt == '-h':
             usage()
         else:
@@ -69,4 +73,7 @@ if __name__ == '__main__':
     token = create_token(private_key, write_domains,
                          read_domains)
 
-    print(f'{token}')
+    if export:
+        print(f'export JWT_TOKEN={token}')
+    else:
+        print(f'{token}')
-- 
cgit v1.1