o
    'hSY                     @  s  U d Z ddlmZ ddlZddlZddlmZ ddlmZ ddl	m
Z
mZ ddlZddlmZmZmZmZ e
r=ddlmZ d	d
lmZ d	dlmZ 	 dZdZdZdZdZeedjddZ G dd de!Z"G dd de"Z#G dd de"Z$dSddZ%G d d! d!ej&Z'G d"d# d#Z(G d$d% d%ej&Z)dTd'd(Z*dTd)d*Z+dTd+d,Z,dTd-d.Z-dZ.d/e/d0< e- re+ re* se, se( Z.ndZ.d1Z0dUd5d6Z1d7a2dVd9d:Z3d7a4dVd;d<Z5dWd>d?Z6dXdAdBZ7dYdJdKZ8dTdLdMZ9dZdOdPZ:dTdQdRZ;dS )[a  
Support for streaming http requests in emscripten.

A few caveats -

If your browser (or Node.js) has WebAssembly JavaScript Promise Integration enabled
https://github.com/WebAssembly/js-promise-integration/blob/main/proposals/js-promise-integration/Overview.md
*and* you launch pyodide using `pyodide.runPythonAsync`, this will fetch data using the
JavaScript asynchronous fetch api (wrapped via `pyodide.ffi.call_sync`). In this case
timeouts and streaming should just work.

Otherwise, it uses a combination of XMLHttpRequest and a web-worker for streaming.

This approach has several caveats:

Firstly, you can't do streaming http in the main UI thread, because atomics.wait isn't allowed.
Streaming only works if you're running pyodide in a web worker.

Secondly, this uses an extra web worker and SharedArrayBuffer to do the asynchronous fetch
operation, so it requires that you have crossOriginIsolation enabled, by serving over https
(or from localhost) with the two headers below set:

    Cross-Origin-Opener-Policy: same-origin
    Cross-Origin-Embedder-Policy: require-corp

You can tell if cross origin isolation is successfully enabled by looking at the global crossOriginIsolated variable in
JavaScript console. If it isn't, streaming requests will fallback to XMLHttpRequest, i.e. getting the whole
request into a buffer and then returning it. it shows a warning in the JavaScript console in this case.

Finally, the webworker which does the streaming fetch is created on initial import, but will only be started once
control is returned to javascript. Call `await wait_for_streaming_ready()` to wait for streaming fetch.

NB: in this code, there are a lot of JavaScript objects. They are named js_*
to make it clear what type of object they are.
    )annotationsN)Parser)files)TYPE_CHECKINGAny)JsArrayJsExceptionJsProxyto_js)Buffer   )EmscriptenRequest)EmscriptenResponse)z
user-agentzemscripten_fetch_worker.jszutf-8)encodingc                      s*   e Zd Z	ddddd fd	d
Z  ZS )_RequestErrorNrequestresponsemessage
str | Noner   EmscriptenRequest | Noner   EmscriptenResponse | Nonec                  s$   || _ || _|| _t | j d S N)r   r   r   super__init__)selfr   r   r   	__class__ \/var/www/html/olx_land/venv/lib/python3.10/site-packages/urllib3/contrib/emscripten/fetch.pyr   N   s   z_RequestError.__init__r   )r   r   r   r   r   r   )__name__
__module____qualname__r   __classcell__r"   r"   r    r#   r   M   s    r   c                   @     e Zd ZdS )_StreamingErrorNr$   r%   r&   r"   r"   r"   r#   r)   [       r)   c                   @  r(   )_TimeoutErrorNr*   r"   r"   r"   r#   r,   _   r+   r,   dict_valdict[str, Any]returnr	   c                 C  s   t | tjjdS )N)dict_converter)r
   jsObjectfromEntries)r-   r"   r"   r#   _obj_from_dictc      r4   c                      sr   e Zd Zd#ddZd$ddZd%ddZed%ddZd$ fddZd%ddZ	d%ddZ
d%ddZd&d!d"Z  ZS )'_ReadStream
int_bufferr   byte_buffertimeoutfloatworkerr	   connection_idintr   r   c                 C  sT   || _ || _d| _d| _|| _|| _|dkrtd| nd | _d| _d| _	|| _
d S )Nr     TF)r7   r8   read_posread_lenr<   r;   r=   r9   is_live
_is_closedr   )r   r7   r8   r9   r;   r<   r   r"   r"   r#   r   h   s   	
z_ReadStream.__init__r/   Nonec                 C     |    d S r   closer   r"   r"   r#   __del__|      z_ReadStream.__del__boolc                 C     | j S r   rB   rG   r"   r"   r#   	is_closed      z_ReadStream.is_closedc                 C     |   S r   rM   rG   r"   r"   r#   closed      z_ReadStream.closedc                   s`   |   rd S d| _d| _d | _d | _d| _d | _| jr)| j	t
d| ji d| _t   d S )Nr   TrF   F)rM   r@   r?   r7   r8   rB   r   rA   r;   postMessager4   r<   r   rF   rG   r    r"   r#   rF      s   z_ReadStream.closec                 C     dS NTr"   rG   r"   r"   r#   readable      z_ReadStream.readablec                 C  rT   NFr"   rG   r"   r"   r#   writable   rW   z_ReadStream.writablec                 C  rT   rX   r"   rG   r"   r"   r#   seekable   rW   z_ReadStream.seekablebyte_objr   c                 C  s<  | j std| jd d| jdkrotj| j dt | j	t
d| ji tj| j dt| jdkr3t| j d }|dkrC|| _d| _n,|tkrf| j d }tj }|| jd|}td| | jd dd| _|   dS t| jtt|}| j| j| j|  }|t|d|< |  j|8  _|  j|7  _|S )	Nz,No buffer for stream in _ReadStream.readintor   r   getMorez	timed-outr   Exception thrown in fetch: F)r7   r)   r   r@   r1   AtomicsstoreERROR_TIMEOUTr;   rS   r4   r<   waitr9   r,   r?   ERROR_EXCEPTIONTextDecodernewdecoder8   slicerA   rF   minlen
memoryviewsubarrayto_py)r   r[   data_len
string_len
js_decoderjson_str
ret_lengthrj   r"   r"   r#   readinto   sN   



z_ReadStream.readinto)r7   r   r8   r   r9   r:   r;   r	   r<   r=   r   r   r/   rC   r/   rJ   r[   r   r/   r=   )r$   r%   r&   r   rH   rM   propertyrQ   rF   rV   rY   rZ   rq   r'   r"   r"   r    r#   r6   g   s    





r6   c                   @  s    e Zd ZdddZddd	Zd
S )_StreamingFetcherr/   rC   c                   sd   d _ tjttgddtddi}d fd
d}tj|}tj	j
| _tj	j| _d S )NF)create_pyproxiestypezapplication/javascriptjs_resolve_fnr	   js_reject_fnr/   rC   c                   s2   d	fdd}d	 fdd}|j _|j _d S )
Ner	   r/   rC   c                   s   d_  |  d S rU   )streaming_readyr{   )ry   r   r"   r#   onMsg   s   zC_StreamingFetcher.__init__.<locals>.promise_resolver.<locals>.onMsgc                   s    |  d S r   r"   r}   )rz   r"   r#   onErr   rI   zC_StreamingFetcher.__init__.<locals>.promise_resolver.<locals>.onErr)r{   r	   r/   rC   )	js_worker	onmessageonerror)ry   rz   r~   r   rG   )rz   ry   r#   promise_resolver   s   z4_StreamingFetcher.__init__.<locals>.promise_resolver)ry   r	   rz   r	   r/   rC   )r|   r1   Blobrd   r
   _STREAMING_WORKER_CODEr4   URLcreateObjectURL
globalThisWorkerr   Promisejs_worker_ready_promise)r   js_data_blobr   js_data_urlr"   rG   r#   r      s   
z_StreamingFetcher.__init__r   r   r   c                 C  s  dd |j  D }|j}|t||jd}|jdkr"td|j nd }tj	d}tj
	|}tj	|d}tj|dt tj|d tj	|jtjj}	| jt||	|d tj|dt| |d tkrrtd	|d d
|d tkr|d }
tj	 }||d|
}t|}t||d |d t |||j| j|d |dS |d t!kr|d }
tj	 }||d|
}t"d| |d d
t"d|d  |d d
)Nc                 S     i | ]\}}|t vr||qS r"   HEADERS_TO_IGNORE.0kvr"   r"   r#   
<dictcomp>   s    z*_StreamingFetcher.send.<locals>.<dictcomp>)headersbodymethodr   r>   i      )bufferurlfetchParamsz'Timeout connecting to streaming requestr   r   statusr   connectionID)r   status_coder   r   r]   z%Unknown status from worker in fetch: )#r   itemsr   r
   r   r9   r=   r1   SharedArrayBufferrd   
Int32Array
Uint8Arrayr^   r_   r`   notifyr   r   locationhrefr   rS   r4   ra   r,   SUCCESS_HEADERrc   re   rf   jsonloadsr   r6   rb   r)   )r   r   r   r   
fetch_datar9   js_shared_bufferjs_int_bufferjs_byte_bufferjs_absolute_urlrm   rn   ro   response_objr"   r"   r#   send   sr   



z_StreamingFetcher.sendNrr   r   r   r/   r   )r$   r%   r&   r   r   r"   r"   r"   r#   rv      s    
rv   c                      s   e Zd ZdZd%ddZd&ddZd'ddZed'ddZd& fddZ	d'ddZ
d'ddZd'ddZd'ddZd(d#d$Z  ZS ))_JSPIReadStreamaF  
    A read stream that uses pyodide.ffi.run_sync to read from a JavaScript fetch
    response. This requires support for WebAssembly JavaScript Promise Integration
    in the containing browser, and for pyodide to be launched via runPythonAsync.

    :param js_read_stream:
        The JavaScript stream reader

    :param timeout:
        Timeout in seconds

    :param request:
        The request we're handling

    :param response:
        The response this stream relates to

    :param js_abort_controller:
        A JavaScript AbortController object, used for timeouts
    js_read_streamr   r9   r:   r   r   r   r   js_abort_controllerc                 C  s:   || _ || _d| _d| _|| _|| _d | _d| _|| _d S )NFr   )	r   r9   rB   _is_doner   r   current_buffercurrent_buffer_posr   )r   r   r9   r   r   r   r"   r"   r#   r   F  s   
z_JSPIReadStream.__init__r/   rC   c                 C  rD   r   rE   rG   r"   r"   r#   rH   X  rI   z_JSPIReadStream.__del__rJ   c                 C  rK   r   rL   rG   r"   r"   r#   rM   \  rN   z_JSPIReadStream.is_closedc                 C  rO   r   rP   rG   r"   r"   r#   rQ   `  rR   z_JSPIReadStream.closedc                   sN   |   rd S d| _d| _| j  d | _d| _d| _d | _d | _t	 
  d S )Nr   T)rM   r@   r?   r   cancelrB   r   r   r   r   rF   rG   r    r"   r#   rF   d  s   
z_JSPIReadStream.closec                 C  rT   rU   r"   rG   r"   r"   r#   rV   q  rW   z_JSPIReadStream.readablec                 C  rT   rX   r"   rG   r"   r"   r#   rY   t  rW   z_JSPIReadStream.writablec                 C  rT   rX   r"   rG   r"   r"   r#   rZ   w  rW   z_JSPIReadStream.seekablec                 C  sF   t | j | j| j| j| jd}|jrd| _dS |j	
 | _d| _dS )Nr   TFr   )_run_sync_with_timeoutr   readr9   r   r   r   doner   valuerk   r   r   )r   	result_jsr"   r"   r#   _get_next_bufferz  s   z _JSPIReadStream._get_next_bufferr[   r   r=   c                 C  s   | j d u r|  r| j d u r|   dS tt|t| j | j }| j | j| j|  |d|< |  j|7  _| jt| j krBd | _ |S )Nr   )r   r   rF   rg   rh   r   )r   r[   rp   r"   r"   r#   rq     s   
z_JSPIReadStream.readinto)
r   r   r9   r:   r   r   r   r   r   r   rr   rs   rt   )r$   r%   r&   __doc__r   rH   rM   ru   rQ   rF   rV   rY   rZ   r   rq   r'   r"   r"   r    r#   r   0  s    






r   rJ   c                   C  s    t tdot tdotjtjkS )Nwindowr   )hasattrr1   r   r   r"   r"   r"   r#   is_in_browser_main_thread  s    r   c                   C  s   t tdotjS )NcrossOriginIsolated)r   r1   r   r"   r"   r"   r#   is_cross_origin_isolated  r5   r   c                   C  s2   t tdot tjdot tjjdotjjjdkS )Nprocessreleasenamenode)r   r1   r   r   r   r"   r"   r"   r#   
is_in_node  s   

r   c                   C  s   t tdo	t tdS )Nr   r   )r   r1   r"   r"   r"   r#   is_worker_available  s   r   z_StreamingFetcher | None_fetcherzurllib3 only works in Node.js with pyodide.runPythonAsync and requires the flag --experimental-wasm-stack-switching in  versions of node <24.r   r   r   c                 C  sB   t  rt| dS t rtt| d dtrt rt| S t  d S )NTr   r   r   )	has_jspisend_jspi_requestr   r   NODE_JSPI_ERRORr   r|   r   _show_streaming_warningr   r"   r"   r#   send_streaming_request  s   


r   FrC   c                  C  s    t sda d} tj|  d S d S )NTz8Warning: Timeout is not available on main browser thread)_SHOWN_TIMEOUT_WARNINGr1   consolewarn)r   r"   r"   r#   _show_timeout_warning  s
   r   c                  C  sf   t s1da d} t s| d7 } t r| d7 } t s| d7 } t du r$| d7 } dd	lm} ||  d S d S )
NTz%Can't stream HTTP requests because: 
z$  Page is not cross-origin isolated
z+  Python is running in main browser thread
z> Worker or Blob classes are not available in this environment.Fz Streaming fetch worker isn't ready. If you want to be sure that streaming fetch
is working, you need to call: 'await urllib3.contrib.emscripten.fetch.wait_for_streaming_ready()`r   )r   )_SHOWN_STREAMING_WARNINGr   r   r   r|   r1   r   r   )r   r   r"   r"   r#   r     s   
r   r   c              
   C  sX  t  rt| dS t rtt| d dzntj }t s*d|_	| j
r)t| j
d |_
n|d | j
r5t  || j| jd | j D ]\}}| tvrS||| qC|t| j tt | }t sq|j  }n|j d}t!|j"||| dW S  t#y } z|j$dkrt%|j&| d	|j$d
krt|j&| d	t|j&| d	d }~ww )NFr   arraybufferr>   ztext/plain; charset=ISO-8859-15zISO-8859-15r   r   r   r   TimeoutErrorr   NetworkError)'r   r   r   r   r   r1   XMLHttpRequestrd   r   responseTyper9   r=   overrideMimeTyper   openr   r   r   r   lowerr   setRequestHeaderr   r
   r   dictr   parsestrgetAllResponseHeadersr   rk   tobytesencoder   r   r   r   r,   r   )r   js_xhrr   r   r   r   errr"   r"   r#   send_request  sN   





r   	streamingc                 C  s  | j }tj }dd | j D }| j}|t|| j|j	d}t
| jt|}t|||| dd}i }|j }		 |	 }
t|
ddrEnt|
jd	 |t|
jd
 < q:|j}d}t||d| d}|rv|jduru|j }t||| ||}nt| ||| |d }||_|S )a7  
    Send a request using WebAssembly JavaScript Promise Integration
    to wrap the asynchronous JavaScript fetch api (experimental).

    :param request:
        Request to send

    :param streaming:
        Whether to stream the response

    :return: The response object
    :rtype: EmscriptenResponse
    c                 S  r   r"   r   r   r"   r"   r#   r   8  s    z%send_jspi_request.<locals>.<dictcomp>)r   r   r   signalNr   Tr   Fr   r       r   )r9   r1   AbortControllerrd   r   r   r   r
   r   r   fetchr   r4   r   entriesnextgetattrstrr   r   r   	getReaderr   arrayBufferrk   )r   r   r9   r   r   req_bodyr   fetcher_promise_jsresponse_jsheader_iteriter_value_jsr   r   r   body_stream_jsr"   r"   r#   r   &  s`   




r   promiser   r9   r:   r   r   r   c              
   C  s   d}|dkrt |j|t|d }z7zddlm} || W W |dur+t | S S  tyK } z|j	dkr?t
d||dt|j||dd}~ww |durVt | w w )ak  
    Await a JavaScript promise synchronously with a timeout which is implemented
    via the AbortController

    :param promise:
        Javascript promise to await

    :param timeout:
        Timeout in seconds

    :param js_abort_controller:
        A JavaScript AbortController object, used on timeout

    :param request:
        The request being handled

    :param response:
        The response being handled (if it exists yet)

    :raises _TimeoutError: If the request times out
    :raises _RequestError: If the request raises a JavaScript exception

    :return: The result of awaiting the promise.
    Nr   r>   )run_sync
AbortErrorzRequest timed outr   )r1   
setTimeoutabortbindr=   pyodide.ffir   clearTimeoutr   r   r,   r   r   )r   r9   r   r   r   timer_idr   r   r"   r"   r#   r   o  s,   
	
r   c                  C  s2   zddl m} m} t|  W S  ty   Y dS w )a  
    Return true if jspi can be used.

    This requires both browser support and also WebAssembly
    to be in the correct state - i.e. that the javascript
    call into python was async not sync.

    :return: True if jspi can be used.
    :rtype: bool
    r   can_run_syncr   F)r  r  r   rJ   ImportErrorr  r"   r"   r#   r     s   r   bool | Nonec                   C  s   t rt jS d S r   )r   r|   r"   r"   r"   r#   r|     s   r|   c                     s   t rt jI d H  dS dS )NTF)r   r   r"   r"   r"   r#   wait_for_streaming_ready  s
   r  )r-   r.   r/   r	   rs   )r   r   r/   r   rr   r   )r   r   r   rJ   r/   r   )r   r   r9   r:   r   r   r   r   r   r   r/   r   )r/   r  )<r   
__future__r   ior   email.parserr   importlib.resourcesr   typingr   r   r1   r  r   r   r	   r
   typing_extensionsr   r   r   r   r   r   r   SUCCESS_EOFr`   rb   __package__joinpath	read_textr   	Exceptionr   r)   r,   r4   	RawIOBaser6   rv   r   r   r   r   r   r   __annotations__r   r   r   r   r   r   r   r   r   r   r|   r  r"   r"   r"   r#   <module>   st    $
gb
l


	




1
I
6
