U
    bz                     @   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                   @   sn   e Zd ZdeeedddZedddZe	dd	d
Z
ee dddZeedddZeedddZdS )PyJWKClientT   )uri
cache_keysmax_cached_keysc                 C   s    || _ |rt|d| j| _d S )N)maxsize)r   r   get_signing_key)selfr   r   r    r   3/tmp/pip-unpacked-wheel-ivybibab/jwt/jwks_client.py__init__   s    zPyJWKClient.__init__)returnc              
   C   s0   t j| j}t|W  5 Q R  S Q R X 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 kr|jr|qS ))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)kidr   c                 C   s@   |   }d }|D ]}|j|kr|} q(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)tokenr   c                 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   r0   r   r   r   r   r
      s   r
   )r   urllib.requestr   	functoolsr   typingr   r   Zapi_jwkr   r   Zapi_jwtr   r.   
exceptionsr	   r
   r   r   r   r   <module>   s   