o
    tBh                     @   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   f/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-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   sJ   |   }g }|jD ]}|jdkr|jr|| q	t|dkr#td|S )Nsigr   z2The JWKS endpoint did not contain any signing keys)r    keyspublic_key_usekey_idappendlenr	   )r   jwk_setsigning_keysjwk_set_keyr   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    