�
L�g� � � � d dl Z d dlZd dlmZ d dlmZ d dlmZm Z m
Z
m Z d dl m
Z
ddlmZmZ ddlmZ dd lmZmZ dd
lmZ G d � d � � ZdS )
� N)� lru_cache)�
SSLContext)�Any�Dict�List�Optional)�URLError� )�PyJWK�PyJWKSet)�decode_complete)�PyJWKClientConnectionError�PyJWKClientError)� JWKSetCachec � � e Zd Z ddeded ed
ed ed eeeef d
edee fd�Z
defd�Z d dede fd�Z
d dedee fd�Zdedefd�Zdedefd�Zedee dedee fd�� � ZdS ) � PyJWKClientF� T�, N� �uri�
cache_keys�max_cached_keys�
cache_jwk_set�lifespan�headers�timeout� ssl_contextc � � |�i }|| _ d | _ || _ || _ || _ |r.|dk rt d|� d�� � �t
|� � | _ nd | _ |r% t |�� � | j � � | _ d S d S )Nr z/Lifespan must be greater than 0, the input is "�")�maxsize) r �
jwk_set_cacher r r r r r �get_signing_key) �selfr r r r r r r r s �_/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/jwt/jwks_client.py�__init__zPyJWKClient.__init__ s� � � �?��G����48��� �� � �� �&��� � &� �1�}�}�&�Q�h�Q�Q�Q�� � � "-�X�!6�!6�D� � �!%�D� � � � $F�9�_�#E�#E�#E��$�$� $�D� � � � � � �returnc �� � d } t j � | j | j �� � }t j � || j | j �� � 5 }t j |� � }d d d � � n # 1 swxY w Y || j
�| j
� |� � S S # t t f$ r}t d|� d�� � |�d }~ww xY w# | j
�| j
� |� � w w xY w)N)�urlr )r �contextz'Fail to fetch data from the url, err: "r )�urllib�request�Requestr r �urlopenr r �json�loadr! �putr � TimeoutErrorr )r# �jwk_set�r�response�es r$ �
fetch_datazPyJWKClient.fetch_data4 sY � ��� 0���&�&�4�8�T�\�&�J�J�A�� �'�'��4�<��1A� (� � �
.�� �)�H�-�-��
.�
.�
.�
.�
.�
.�
.�
.�
.�
.�
.����
.�
.�
.�
.� ��!�-��"�&�&�w�/�/�/�/� .�� �,�'� � � �,�>�!�>�>�>�� ��
����� ����� �!�-��"�&�&�w�/�/�/�/� .���sN �AB+ �A<�0 B+ �<B � B+ �B �B+ �C �+C�<C�C�C �$C<�refreshc �� � d }| j �|s| j � � � }|�| � � � }t |t � � st d� � �t
j |� � S )Nz.The JWKS endpoint did not return a JSON object)r! �getr7 �
isinstance�dictr r � from_dict)r# r8 �datas r$ � get_jwk_setzPyJWKClient.get_jwk_setF sp � ��� � � )�'� )��%�)�)�+�+�D� �<��?�?�$�$�D��$��%�%� U�"�#S�T�T� T��!�$�'�'�'r& c �t � | � |� � }d� |j D � � }|st d� � �|S )Nc �2 � g | ]}|j d v � |j �|��S ))�sigN)�public_key_use�key_id)�.0� jwk_set_keys r$ �
<listcomp>z0PyJWKClient.get_signing_keys.<locals>.<listcomp>U s8 � �
�
�
���)�]�:�:�{�?Q�:�
�:�:�:r& z2The JWKS endpoint did not contain any signing keys)r? �keysr )r# r8 r3 � signing_keyss r$ �get_signing_keysz PyJWKClient.get_signing_keysS sU � ��"�"�7�+�+��
�
� &�|�
�
�
� � � Y�"�#W�X�X� X��r& �kidc �� � | � � � }| � ||� � }|sA| � d�� � }| � ||� � }|st d|� d�� � �|S )NT)r8 z,Unable to find a signing key that matches: "r )rJ � match_kidr )r# rK rI � signing_keys r$ r" zPyJWKClient.get_signing_key` s� � ��,�,�.�.� ��n�n�\�3�7�7� �� � �0�0��0�>�>�L� �.�.� �s�;�;�K� �
�&�I�3�I�I�I�� � � �r& �tokenc � � t |ddi�� � }|d }| � |� d� � � � S )N�verify_signatureF)�options�headerrK )� decode_tokenr" r: )r# rO �
unverifiedrS s r$ �get_signing_key_from_jwtz$PyJWKClient.get_signing_key_from_jwtp sD � �!�%�2D�e�1L�M�M�M�
��H�%���#�#�F�J�J�u�$5�$5�6�6�6r& rI c �2 � d }| D ]}|j |k r|} n�|S )N)rD )rI rK rN �keys r$ rM zPyJWKClient.match_kidu s: � �� � � � �C��z�S� � � !� ��� !� �r& )Fr Tr Nr N)F)�__name__�
__module__� __qualname__�str�bool�intr r r r r% r7 r r? r r rJ r" rV � staticmethodrM � r& r$ r r s� � � � � � � !� !� "��,0��,0�#� #�
�#� �#� � #�
� #� �
#� �$�s�C�x�.�)�#� �#� �j�)�#� #� #� #�J0�C� 0� 0� 0� 0�$ (� (�4� (�H� (� (� (� (� � �� ��e�� � � � ��3� �5� � � � � 7�c� 7�e� 7� 7� 7� 7�
���U� � �#� �(�5�/� � � � �\�� � r& r )r/ �urllib.requestr+ � functoolsr �sslr �typingr r r r � urllib.errorr �api_jwkr r �api_jwtr
rT �
exceptionsr r r! r r r` r& r$ �<module>ri s �� � � � � � � � � � � � � � � � � � � � � ,� ,� ,� ,� ,� ,� ,� ,� ,� ,� ,� ,� !� !� !� !� !� !� $� $� $� $� $� $� $� $� 4� 4� 4� 4� 4� 4� D� D� D� D� D� D� D� D� &� &� &� &� &� &�p� p� p� p� p� p� p� p� p� pr&