o
    i'ed                     @  s   d dl mZ d dlZd dlZd dlmZ ddlmZmZm	Z	 ddl
mZmZmZmZ ddlmZ G dd	 d	ZG d
d dZG dd dZdS )    )annotationsN)Any   )get_default_algorithms
has_cryptorequires_cryptography)InvalidKeyError
PyJWKErrorPyJWKSetError
PyJWTError)JWKDictc                   @  sb   e Zd Zdddd	ZeddddZeddddZedddZedddZ	edddZ
dS )PyJWKNjwk_datar   	algorithm
str | NonereturnNonec                 C  sf  t  | _|| _| jdd }|std| j |s't| jtr'| jdd }|s| jdd }|dkrY|dks:|s=d}nM|dkrDd	}nF|d
krKd}n?|dkrRd}n8td| |dkr`d}n*|dkrgd}n#|dkr|sutd| j |dkr|d}ntd| td| ts|tv rt	| d| j|| _
| j
st	d| j | j
| j| _d S )Nktyzkty is not found: ZalgcrvZECzP-256ZES256zP-384ZES384zP-521ZES512Z	secp256k1ZES256KzUnsupported crv: ZRSAZRS256octZHS256ZOKPzcrv is not found: ZEd25519ZEdDSAzUnsupported kty: z) requires 'cryptography' to be installed.z%Unable to find an algorithm for key: )r   Z_algorithms	_jwk_datagetr   
isinstancedictr   r   r	   Z	AlgorithmZfrom_jwkkey)selfr   r   r   r    r   +lib/python3.10/site-packages/jwt/api_jwk.py__init__   sH   zPyJWK.__init__obj'PyJWK'c                 C  s
   t | |S N)r   )r   r   r   r   r   	from_dict>   s   
zPyJWK.from_dictdatastrc                 C  s   t | }t||S r!   )jsonloadsr   r"   )r#   r   r   r   r   r   	from_jsonB   s   
zPyJWK.from_jsonc                 C     | j dd S )Nr   r   r   r   r   r   r   key_typeG      zPyJWK.key_typec                 C  r(   )Nkidr)   r*   r   r   r   key_idK   r,   zPyJWK.key_idc                 C  r(   )NZuser)   r*   r   r   r   public_key_useO   r,   zPyJWK.public_key_user!   )r   r   r   r   r   r   )r   r   r   r   r   r    )r#   r$   r   r   r   r    )r   r   )__name__
__module____qualname__r   staticmethodr"   r'   propertyr+   r.   r/   r   r   r   r   r      s    1r   c                   @  s<   e Zd ZdddZedd
dZedddZdddZdS )PyJWKSetkeyslist[JWKDict]r   r   c              	   C  sn   g | _ |s	tdt|tstd|D ]}z
| j t| W q ty)   Y qw t| j dkr5tdd S )Nz$The JWK Set did not contain any keyszInvalid JWK Set valuer   zUThe JWK Set did not contain any usable keys. Perhaps 'cryptography' is not installed?)r6   r
   r   listappendr   r   len)r   r6   r   r   r   r   r   U   s    
zPyJWKSet.__init__r   dict[str, Any]
'PyJWKSet'c                 C  s   |  dg }t|S )Nr6   )r   r5   )r   r6   r   r   r   r"   j   s   zPyJWKSet.from_dictr#   r$   c                 C  s   t | }t|S r!   )r%   r&   r5   r"   )r#   r   r   r   r   r'   o   s   

zPyJWKSet.from_jsonr-   r    c                 C  s,   | j D ]}|j|kr|  S qtd| )Nzkeyset has no key for kid: )r6   r.   KeyError)r   r-   r   r   r   r   __getitem__t   s
   

zPyJWKSet.__getitem__N)r6   r7   r   r   )r   r;   r   r<   )r#   r$   r   r<   )r-   r$   r   r    )r0   r1   r2   r   r3   r"   r'   r>   r   r   r   r   r5   T   s    
r5   c                   @  s*   e Zd ZdddZdddZdd	d
ZdS )PyJWTSetWithTimestampjwk_setr5   c                 C  s   || _ t | _d S r!   )r@   time	monotonic	timestamp)r   r@   r   r   r   r   |   s   zPyJWTSetWithTimestamp.__init__r   c                 C     | j S r!   )r@   r*   r   r   r   get_jwk_set      z!PyJWTSetWithTimestamp.get_jwk_setfloatc                 C  rD   r!   )rC   r*   r   r   r   get_timestamp   rF   z#PyJWTSetWithTimestamp.get_timestampN)r@   r5   )r   r5   )r   rG   )r0   r1   r2   r   rE   rH   r   r   r   r   r?   {   s    

r?   )Z
__future__r   r%   rA   typingr   Z
algorithmsr   r   r   
exceptionsr   r	   r
   r   typesr   r   r5   r?   r   r   r   r   <module>   s    H'