o
    \|]az                     @   sf   d dl Z d dlZd dlmZ d dlmZmZ ddlm	Z	m
Z
 ddlmZ ddlmZ G dd	 d	ZdS )
    N)	lru_cache)AnyList   )PyJWKPyJWKSet)decode_complete)PyJWKClientErrorc                   @   sv   e Zd ZddededefddZdefd	d
Zde	fddZ
dee fddZdedefddZdedefddZdS )PyJWKClientT   uri
cache_keysmax_cached_keysc                 C   s$   || _ |rt|d| j| _d S d S )N)maxsize)r   r   get_signing_key)selfr   r   r    r   1/usr/lib/python3/dist-packages/jwt/jwks_client.py__init__   s   zPyJWKClient.__init__returnc                 C   s<   t j| j}t|W  d    S 1 sw   Y  d S N)urllibZrequestZurlopenr   jsonload)r   Zresponser   r   r   
fetch_data   s   $zPyJWKClient.fetch_datac                 C   s   |   }t|S r   )r   r   Z	from_dict)r   datar   r   r   get_jwk_set   s   
zPyJWKClient.get_jwk_setc                 C   s(   |   }dd |jD }|std|S )Nc                 S   s    g | ]}|j d v r|jr|qS ))ZsigN)Zpublic_key_usekey_id).0Zjwk_set_keyr   r   r   
<listcomp>   s    z0PyJWKClient.get_signing_keys.<locals>.<listcomp>z2The JWKS endpoint did not contain any signing keys)r   keysr	   )r   Zjwk_setsigning_keysr   r   r   get_signing_keys   s   zPyJWKClient.get_signing_keyskidc                 C   s@   |   }d }|D ]}|j|kr|} nq|std| d|S )Nz,Unable to find a signing key that matches: "")r"   r   r	   )r   r#   r!   Zsigning_keykeyr   r   r   r   (   s   

zPyJWKClient.get_signing_keytokenc                 C   s(   t |ddid}|d }| |dS )NZverify_signatureF)optionsheaderr#   )decode_tokenr   get)r   r&   Z
unverifiedr(   r   r   r   get_signing_key_from_jwt8   s   z$PyJWKClient.get_signing_key_from_jwtN)Tr   )__name__
__module____qualname__strboolintr   r   r   r   r   r   r   r"   r   r+   r   r   r   r   r
      s    r
   )r   Zurllib.requestr   	functoolsr   typingr   r   Zapi_jwkr   r   Zapi_jwtr   r)   
exceptionsr	   r
   r   r   r   r   <module>   s    