o
    h                     @   sr   d dl Z d dlZd dlZd dlmZmZ d dlmZmZ d dl	m
Z
mZmZmZmZ dZh dZG dd dZdS )	    N)HeadersMutableHeaders)PlainTextResponseResponse)ASGIAppMessageReceiveScopeSend)DELETEGETHEADOPTIONSPATCHPOSTPUT>   Content-TypeAccept-LanguageContent-LanguageAcceptc                   @   s   e Zd Z							d$dedeje deje d	eje d
edeje deje de	ddfddZ
dedededdfddZdedefddZdedefddZdededededdf
ddZdedededdfdd Zed!ededdfd"d#ZdS )%CORSMiddleware r   FNX  appallow_originsallow_methodsallow_headersallow_credentialsallow_origin_regexexpose_headersmax_agereturnc	                 C   s  d|v rt }d }	|d urt|}	d|v }
d|v }|
 p|}i }|
r&d|d< |r,d|d< |r5d||d< i }|r>d|d< nd|d< |d|t|d	 ttt|B }|rb|sbd||d
< |rhd|d< || _	|| _
|| _dd |D | _|
| _|| _|| _|	| _|| _|| _d S )N*Access-Control-Allow-Origintruez Access-Control-Allow-Credentials, zAccess-Control-Expose-HeadersOriginVary)zAccess-Control-Allow-MethodszAccess-Control-Max-AgeAccess-Control-Allow-Headersc                 S      g | ]}|  qS r   lower.0hr   r   O/var/www/html/ai/venv/lib/python3.10/site-packages/starlette/middleware/cors.py
<listcomp>A       z+CORSMiddleware.__init__.<locals>.<listcomp>)ALL_METHODSrecompilejoinupdatestrsortedSAFELISTED_HEADERSsetr   r   r   r   allow_all_originsallow_all_headerspreflight_explicit_allow_originr   simple_headerspreflight_headers)selfr   r   r   r   r   r   r    r!   compiled_allow_origin_regexr<   r=   r>   r?   r@   r   r   r0   __init__   sN   



zCORSMiddleware.__init__scopereceivesendc                    s   |d dkr|  |||I d H  d S |d }t|d}|d}|d u r1|  |||I d H  d S |dkrJd|v rJ| j|d}||||I d H  d S | j||||dI d H  d S )	NtypehttpmethodrD   originr   access-control-request-method)request_headers)r   r   getpreflight_responsesimple_response)rA   rD   rE   rF   rI   headersrK   responser   r   r0   __call__I   s   

zCORSMiddleware.__call__rK   c                 C   s.   | j rdS | jd ur| j|rdS || jv S )NT)r<   r   	fullmatchr   )rA   rK   r   r   r0   is_allowed_origin]   s   
z CORSMiddleware.is_allowed_originrM   c           	      C   s   |d }|d }| d}t| j}g }| j|dr"| jr!||d< n|d || jvr1|d | jr=|d ur=||d< n|d ur\dd	 |d
D D ]}|	 | j
vr[|d  nqK|rldd| }t|d|dS tdd|dS )NrK   rL   zaccess-control-request-headersrK   r$   rI   r)   c                 S   r*   r   r+   r-   r   r   r0   r1      r2   z5CORSMiddleware.preflight_response.<locals>.<listcomp>,rQ   zDisallowed CORS r&   i  )status_coderQ   OK   )rN   dictr@   rU   r>   appendr   r=   splitstripr   r6   r   )	rA   rM   requested_originrequested_methodrequested_headersrQ   failuresheaderfailure_textr   r   r0   rO   h   s0   






z!CORSMiddleware.preflight_responsec                    s,   t j| j||d}| |||I d H  d S )N)rF   rM   )	functoolspartialrF   r   )rA   rD   rE   rF   rM   r   r   r0   rP      s   zCORSMiddleware.simple_responsemessagec                    s   |d dkr||I d H  d S | dg  t|d}|| j |d }d|v }| jr5|r5| || n| jsD| j|drD| || ||I d H  d S )NrG   zhttp.response.startrQ   rJ   r'   cookierV   )
setdefaultr   r7   r?   r<   allow_explicit_originrU   )rA   rg   rF   rM   rQ   rK   
has_cookier   r   r0   rF      s   

zCORSMiddleware.sendrQ   c                 C   s   || d< |  d d S )Nr$   r'   )add_vary_header)rQ   rK   r   r   r0   rj      s   z$CORSMiddleware.allow_explicit_origin)r   r   r   FNr   r   )__name__
__module____qualname__r   typingSequencer8   boolOptionalintrC   r	   r   r
   rS   rU   r   r   rO   rP   r   rF   staticmethodr   rj   r   r   r   r0   r      sh    	

;&

r   )re   r4   rp   starlette.datastructuresr   r   starlette.responsesr   r   starlette.typesr   r   r   r	   r
   r3   r:   r   r   r   r   r0   <module>   s    