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)urllibrequesturlopenr   jsonload)r   responser   r   r   
fetch_data   s   $zPyJWKClient.fetch_datac                 C   s   |   }t|S r   )r   r   	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 ))sigN)public_key_usekey_id).0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   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)   signing_keykeyr   r   r   r   (   s   

zPyJWKClient.get_signing_keytokenc                 C   s(   t |ddid}|d }| |dS )Nverify_signatureF)optionsheaderr+   )decode_tokenr   get)r   r/   
unverifiedr2   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   r6   r   r   r   r   r
      s    r
   )r   urllib.requestr   	functoolsr   typingr   r   api_jwkr   r   api_jwtr   r3   
exceptionsr	   r
   r   r   r   r   <module>   s    