|
7 | 7 |
|
8 | 8 | import cherrypy |
9 | 9 |
|
10 | | -from oidcmsg.key_jar import build_keyjar |
| 10 | +from oidcmsg.key_jar import build_keyjar, init_key_jar |
11 | 11 | from oidcmsg.key_jar import KeyJar |
12 | 12 |
|
13 | 13 | from oidcrp import RPHandler |
|
25 | 25 | SIGKEY_NAME = 'sigkey.jwks' |
26 | 26 |
|
27 | 27 |
|
28 | | -def get_jwks(private_path, keydefs, public_path): |
29 | | - if os.path.isfile(private_path): |
30 | | - _jwks = open(private_path, 'r').read() |
31 | | - _kj = KeyJar() |
32 | | - _kj.import_jwks(json.loads(_jwks), '') |
33 | | - else: |
34 | | - _kj = build_keyjar(keydefs)[1] |
35 | | - jwks = _kj.export_jwks(private=True) |
36 | | - head, tail = os.path.split(private_path) |
37 | | - if not os.path.isdir(head): |
38 | | - os.makedirs(head) |
39 | | - fp = open(private_path, 'w') |
40 | | - fp.write(json.dumps(jwks)) |
41 | | - fp.close() |
42 | | - |
43 | | - jwks = _kj.export_jwks() # public part |
44 | | - fp = open(public_path, 'w') |
45 | | - fp.write(json.dumps(jwks)) |
46 | | - fp.close() |
47 | | - |
48 | | - return _kj |
49 | | - |
50 | | - |
51 | 28 | if __name__ == '__main__': |
52 | 29 | import argparse |
53 | 30 |
|
@@ -103,8 +80,9 @@ def get_jwks(private_path, keydefs, public_path): |
103 | 80 |
|
104 | 81 | _base_url = config.BASEURL |
105 | 82 |
|
106 | | - _kj = get_jwks(config.PRIVATE_JWKS_PATH, config.KEYDEFS, |
107 | | - config.PUBLIC_JWKS_PATH) |
| 83 | + _kj = init_key_jar(public_path=config.PUBLIC_JWKS_PATH, |
| 84 | + private_path=config.PRIVATE_JWKS_PATH, |
| 85 | + key_defs=config.KEYDEFS) |
108 | 86 |
|
109 | 87 | if args.insecure: |
110 | 88 | _kj.verify_ssl = False |
|
0 commit comments