o
    +if                     @   sf   d dl Z d dlmZmZ d dlZddlmZ ddlmZ ee	Z
dd Zdd	 Zd
d Zdd ZdS )    N)datetime	timedelta   )Mex)parse_responsec           	      K   s   |st d|d u rd|v rtj}nd|v rtj}|tjtjfvr't d| t| ||||}|j|f|d|dd|}|jdkrJtd	|j	 t
|j	S )
Nz)WsTrust endpoint address can not be emptyz/trust/2005/usernamemixedz/trust/13/usernamemixedzZUnsupported soap action: %s. Contact your administrator to check your ADFS's MEX settings.z#application/soap+xml; charset=utf-8)zContent-type
SOAPAction)dataheadersi  z)Unsuccessful WsTrust request receives: %s)
ValueErrorr   ACTION_2005	ACTION_13
_build_rstpoststatus_codeloggerdebugtextr   )	usernamepasswordcloud_audience_urnendpoint_addresssoap_actionhttp_clientkwargsr   resp r   M/var/www/html/corbot_env/lib/python3.10/site-packages/msal/wstrust_request.pysend_request%   s.   



r   c                 C   s,   |  dd dd dd dd d	d
S )N&z&amp;"z&quot;'z&apos;<z&lt;>z&gt;)replace)r   r   r   r   escape_password?   s   r$   c                 C   s
   |  dS )Nz%Y-%m-%dT%H:%M:%SZ)strftime)datetime_objr   r   r   wsu_time_formatE   s   
r'   c                 C   s   t  }djtjd tjd tjd |tt |t|t|t	dd | t
||tjkr2tjd ntjd ||tjkr>d	nd
|tjkrHddS ddS )Na  <s:Envelope xmlns:s='{s}' xmlns:wsa='{wsa}' xmlns:wsu='{wsu}'>
        <s:Header>
            <wsa:Action s:mustUnderstand='1'>{soap_action}</wsa:Action>
            <wsa:MessageID>urn:uuid:{message_id}</wsa:MessageID>
            <wsa:ReplyTo>
            <wsa:Address>http://www.w3.org/2005/08/addressing/anonymous</wsa:Address>
            </wsa:ReplyTo>
            <wsa:To s:mustUnderstand='1'>{endpoint_address}</wsa:To>

            <wsse:Security s:mustUnderstand='1'
            xmlns:wsse='http://docs.oasis-open.org/wss/2004/01/oasis-200401-wss-wssecurity-secext-1.0.xsd'>
                <wsu:Timestamp wsu:Id='_0'>
                    <wsu:Created>{time_now}</wsu:Created>
                    <wsu:Expires>{time_expire}</wsu:Expires>
                </wsu:Timestamp>
                <wsse:UsernameToken wsu:Id='ADALUsernameToken'>
                    <wsse:Username>{username}</wsse:Username>
                    <wsse:Password>{password}</wsse:Password>
                </wsse:UsernameToken>
            </wsse:Security>

        </s:Header>
        <s:Body>
        <wst:RequestSecurityToken xmlns:wst='{wst}'>
        <wsp:AppliesTo xmlns:wsp='http://schemas.xmlsoap.org/ws/2004/09/policy'>
            <wsa:EndpointReference>
                <wsa:Address>{applies_to}</wsa:Address>
            </wsa:EndpointReference>
        </wsp:AppliesTo>
        <wst:KeyType>{key_type}</wst:KeyType>
            <wst:RequestType>{request_type}</wst:RequestType>
        </wst:RequestSecurityToken>
        </s:Body>
        </s:Envelope>swsuwsa10
   )minuteswstwst2005z7http://docs.oasis-open.org/ws-sx/ws-trust/200512/Bearerz9http://schemas.xmlsoap.org/ws/2005/05/identity/NoProofKeyz6http://docs.oasis-open.org/ws-sx/ws-trust/200512/Issuez1http://schemas.xmlsoap.org/ws/2005/02/trust/Issue)r(   r)   wsar   
message_idr   time_nowtime_expirer   r   r-   
applies_tokey_typerequest_type)r   utcnowformatr   NSstruuiduuid4r'   r   r$   r   )r   r   r   r   r   nowr   r   r   r   N   s&   !

/r   )r:   r   r   loggingmexr   wstrust_responser   	getLogger__name__r   r   r$   r'   r   r   r   r   r   <module>   s   
	