from jwcrypto import jwk, jwe, jwt
from io import open

if __name__ == "__main__":
    claims = {
        "exp": 1607828221,
        "jti": "de2fc838ca4ce51e798e5ae4810fcfe3ca531d951eb2da99b2f9e7584adac8ec245a20d5e6dedc107003d52af33b3ab5",
        "scp": "xxx",
        "hsh": "de2fc838ca4ce51e798e5ae4810fcfe3ca531d951eb2da99b2f9e7584adac8ec245a20d5e6dedc107003d52af33b3ab5",
        "sfp": "de2fc838ca4ce51e798e5ae4810fcfe3ca531d951eb2da99b2f9e7584adac8ec245a20d5e6dedc107003d52af33b3ab5",
        "nbu": 3
    }
    # vin = hash(vin + bid + pid)

    key1 = jwk.JWK(generate='oct', size=256)
    key2 = jwk.JWK(generate='oct', size=256)

    EToken = jwt.JWT(
        header={"alg": "A256KW", "enc": "A256CBC-HS512"},
        claims=claims
    )
    EToken.make_encrypted_token(key=key1)
    etoken = EToken.serialize()

    token = jwt.JWT(header={"alg": "HS256"}, claims={'t': etoken})
    token.make_signed_token(key=key2)
    stoken = token.serialize()

    print(etoken)
    print(len(etoken))
    print(stoken)
    print(len(stoken))