�
u��g7 � � � d dl mZ n# e$ r d dlmZ Y nw xY w d dlmZ n# e$ r d dlmZ Y nw xY wd dlZ ej e � � Z
d� Z d� Z G d� de
� � ZdS ) � )�urlparse)� cElementTree)� ElementTreeNc �Z � d� | dgt | � � dz
z z � � S )N�/z..� )�join�len)�
route_to_leafs �X/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/msal/mex.py�_xpath_of_rootr
) s- � � �8�8�M�T�F�c�-�.@�.@��.B�$C�C� D� D�D� c � � |j | fi |��}|� � � t |j � � � � � S # t
j $ r( t � d|j |j � � � w xY w)Nz Malformed MEX document: %s, %s)
�get�raise_for_status�Mex�text�&get_wstrust_username_password_endpoint�ET�
ParseError�logger� exception� status_code)� mex_endpoint� http_client�kwargs�mex_resps r � send_requestr . s� � � �{��|�6�6�v�6�6�H� � � � � � ���8�=�!�!�H�H�J�J�J��
�=� � � ���� ,�h�.B�H�M� S� S� S�
����s �%A
�
7Bc �b � e Zd Zdddddddddd d
d
d d d
�ZdZdZd� Zd� Zd� Zd� Z d� Z
d� Z d� Z dS )r z http://schemas.xmlsoap.org/wsdl/z9http://docs.oasis-open.org/ws-sx/ws-securitypolicy/200702z4http://schemas.xmlsoap.org/ws/2005/07/securitypolicyzRhttp://docs.oasis-open.org/wss/2004/01/oasis-200401-wss-wssecurity-utility-1.0.xsdz$http://www.w3.org/2005/08/addressingz3http://schemas.microsoft.com/ws/06/2004/policy/httpz'http://schemas.xmlsoap.org/wsdl/soap12/z,http://schemas.xmlsoap.org/ws/2004/09/policyz'http://www.w3.org/2003/05/soap-envelopez0http://docs.oasis-open.org/ws-sx/ws-trust/200512z%urn:oasis:names:tc:SAML:1.0:assertionz+http://schemas.xmlsoap.org/ws/2005/02/trust)�wsdl�sp�sp2005�wsu�wsa�wsa10�http�soap12�wsp�s�wst�trust�saml�wst2005z:http://docs.oasis-open.org/ws-sx/ws-trust/200512/RST/Issuez5http://schemas.xmlsoap.org/ws/2005/02/trust/RST/Issuec �8 � t j |� � | _ d S )N)r �
fromstring�dom)�self� mex_documents r �__init__z Mex.__init__N s � ��=� �.�.����r c � � ��� d� j d z �t ��� fd�� j � t |� � � j � � D � � � � S )Nz{%s}Idr# c � �� g | ]E}|� ��j � � � d� |� �� � � � ��FS )Nz#{})�find�NS�formatr )�.0�policy�
binding_xpath�id_attrr1 s ���r �
<listcomp>z'Mex._get_policy_ids.<locals>.<listcomp>S sV �� � @� @� @���{�{�=�$�'�2�2�>� �L�L����G�!4�!4�5�5� ?�>�>r )r7 �setr0 �findallr
)r1 �components_to_leafr; r< s ` `@r �_get_policy_idszMex._get_policy_idsQ s� ���� ��T�W�U�^�+��� @� @� @� @� @� @� �(�*�*�>�:L�+M�+M�t�w�W�W�@� @� @� A� A� Ar c � � g d�}| � |d� � }g d�}|� | � |d� � � � |S )N)�
wsp:Policy�wsp:ExactlyOne�wsp:Allz"sp:SignedEncryptedSupportingTokensrC zsp:UsernameTokenrC zsp:WssUsernameToken10z.//sp:TransportBinding)rC rD rE z sp2005:SignedSupportingTokensrC zsp2005:UsernameTokenrC zsp2005:WssUsernameToken10�.//sp2005:TransportBinding)rA �update)r1 �path�policies�path2005s r �!_get_username_password_policy_idsz%Mex._get_username_password_policy_idsX sl � �G� G� G�� �'�'��.F�G�G��O� O� O�� ����,�,�X�7S�T�T�U�U�U��r c �2 � | � g d�d� � S )N)rC rD rE z http:NegotiateAuthenticationrF )rA )r1 s r �_get_iwa_policy_idszMex._get_iwa_policy_idsc s& � ��#�#� W� W� W� (�*� *� *r c � � i }| j � d| j � � D ]�}|� d| j � � � d� � dk r�5|� d| j � � � d� � }|� d| j � � D ]0}|� d� � |d �||� d
� � < �1��|S ) Nz wsdl:bindingzsoap12:binding� transportz$http://schemas.xmlsoap.org/soap/httpz wsdl:operation/soap12:operation�
soapActionzwsp:PolicyReference�URI)�
policy_uri�action�name)r0 r? r7 r6 r )r1 �bindings�bindingrS �prs r �
_get_bindingszMex._get_bindingsh s� � ����x�'�'����@�@� C� C�G�� � � -�t�w�7�7�;�;�K�H�H�:�;� ;�� �\�\�1�4�7�<� <�<?�C� �<M�<M�
� �o�o�&;�T�W�E�E�
C�
C��"$�&�&��-�-�6�1C� 1C�����V�,�,�-�-�
C� �r c �� � g }| j � d| j � � D ]�}|� d� � � d� � d }|� |� � }|rv|d |v rl|� d| j � � }|�O|j � � � � d� � r#|� |j |d d �� � ��|S )
Nzwsdl:service/wsdl:portrV �:�����rR z%wsa10:EndpointReference/wsa10:Addresszhttps://rS )�addressrS )
r0 r? r7 r �splitr6 r �lower�
startswith�append)r1 rU �
policy_ids� endpoints�port� binding_namerV r\ s r �_get_endpointszMex._get_endpointsu s� � �� ��H�$�$�%=�t�w�G�G� P� P�D� �8�8�I�.�.�4�4�S�9�9�"�=�L� �l�l�<�0�0�G��
P�7�<�0�J�>�>� �)�)�$K�T�W�U�U���&�7�<�+=�+=�+?�+?�+J�+J�:�+V�+V�&� �$�$�$+�L�G�H�<M�N�N�P� P� P���r c �� � | � | � � � | � � � � � }|D ]}|d | j k r|c S �|r|d ndS )zDReturns {"address": "https://...", "action": "the soapAction value"}rS r N)re rX rK � ACTION_13)r1 rb �es r r z*Mex.get_wstrust_username_password_endpoint� sw � ��'�'��"�"�$�$�d�&L�&L�&N�&N�P� P� �� � �A���{�d�n�,�,����� -� (�2�y��|�|�d�2r N)
�__name__�
__module__� __qualname__r7 rg � ACTION_2005r3 rA rK rM rX re r � r r r r 9 s� � � � � � � 3�I�H�c�5�7�E�;�=�
6�A�C�7�@�
�
�B� M�I�I�K�/� /� /�A� A� A� � � �*� *� *�
� � �
�
�
�3� 3� 3� 3� 3r r )� urllib.parser � ImportError� xml.etreer r r �logging� getLoggerri r r
r �objectr rm r r �<module>rt s ��8"�%�%�%�%�%�%�%��� "� "� "�!�!�!�!�!�!�!�!�"����,�,�,�,�,�,�,�,��� ,� ,� ,�+�+�+�+�+�+�+�+�,���� ����
�� �8� $� $��E� E� E�
� � �O3� O3� O3� O3� O3�&� O3� O3� O3� O3� O3s � � ��" � 0�0