1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
"""Utilities dealing with JSON Web Keys (JWK)"""
import os
from pathlib import Path
from typing import Any, Union
from datetime import datetime
from authlib.jose import JsonWebKey
from pymonad.either import Left, Right, Either
def jwks_directory(appsecretsdir: Path) -> Path:
"""Compute the directory where the JWKs are stored."""
if appsecretsdir.exists() and appsecretsdir.is_dir():
jwksdir = Path(appsecretsdir, "jwks/")
if not jwksdir.exists():
jwksdir.mkdir()
return jwksdir
raise ValueError(
"The `appsecretsdir` value should be a directory that actually exists.")
def generate_and_save_private_key(
storagedir: Path,
kty: str = "RSA",
crv_or_size: Union[str, int] = 2048,
options: tuple[tuple[str, Any]] = (("iat", datetime.now().timestamp()),)
) -> JsonWebKey:
"""Generate a private key and save to `storagedir`."""
privatejwk = JsonWebKey.generate_key(
kty, crv_or_size, dict(options), is_private=True)
keyname = f"{privatejwk.thumbprint()}.private.pem"
with open(Path(storagedir, keyname), "wb") as pemfile:
pemfile.write(privatejwk.as_pem(is_private=True))
return privatejwk
def pem_to_jwk(filepath: Path) -> JsonWebKey:
"""Parse a PEM file into a JWK object."""
with open(filepath, "rb") as pemfile:
return JsonWebKey.import_key(pemfile.read())
def __sorted_jwks_paths__(storagedir: Path) -> tuple[tuple[float, Path], ...]:
"""A sorted list of the JWK file paths with their creation timestamps."""
return tuple(sorted(((os.stat(keypath).st_ctime, keypath)
for keypath in (Path(storagedir, keyfile)
for keyfile in os.listdir(storagedir)
if keyfile.endswith(".pem"))),
key=lambda tpl: tpl[0]))
def list_jwks(storagedir: Path) -> tuple[JsonWebKey, ...]:
"""
List all the JWKs in a particular directory in the order they were created.
"""
return tuple(pem_to_jwk(keypath) for ctime,keypath in
__sorted_jwks_paths__(storagedir))
def newest_jwk(storagedir: Path) -> Either:
"""
Return an Either monad with the newest JWK or a message if none exists.
"""
existingkeys = __sorted_jwks_paths__(storagedir)
if len(existingkeys) > 0:
return Right(pem_to_jwk(existingkeys[-1][1]))
return Left("No JWKs exist")
|
