o
�J�h� � @ s� d dl mZ d dlZd dlZd dlmZmZ d dlmZm Z d dl
m Z d dl m
Z
d dlmZmZmZmZ G dd � d ej�ZG d
d � d e�ZG d d
� d
e �ZG dd� d�ZdS )� )� annotationsN)�
AsyncIterator�Iterable)�Any�cast)�HTTPConnection)�Response)�Message�Receive�Scope�Sendc @ s e Zd ZdZdZdZdZdS )�WebSocketStater � � � N)�__name__�
__module__� __qualname__�
CONNECTING� CONNECTED� DISCONNECTED�RESPONSE� r r �RC:\pinokio\api\whisper-webui.git\app\env\lib\site-packages\starlette\websockets.pyr
s
r
c @ s e Zd Zd d d d
�ZdS )
�WebSocketDisconnect�� N�code�int�reason�
str | None�return�Nonec C � || _ |pd| _d S �N� �r r ��selfr r r r r �__init__ � z WebSocketDisconnect.__init__�r N�r r r r r r! )r r r r( r r r r r s r c s� e Zd Zd?� fd d
� Zd@d d
�ZdAdd�Z dBdCdd�ZdAdd�ZdDdd �ZdEd d �Z dFdGd#d$�Z
dHd&d'�Z dId)d*�Z dJd,d-�Z
dKd/d0�ZdLd1d2�ZdFdMd3d4�ZdNdOd9d:�ZdPd=d>�Z� ZS )Q� WebSocket�scoper �receiver
�sendr r r! c s< t � �|� |d dksJ �|| _|| _tj| _tj| _d S )N�type� websocket)�superr( �_receive�_sendr
r � client_state�application_state�r' r- r. r/ �� __class__r r r( s zWebSocket.__init__r c � s� �| j tjkr#| �� I dH }|d }|dkr td|����tj| _ |S | j tjkrI| �� I dH }|d }|dvr?td|����|dkrGtj| _ |S td��) zT
Receive ASGI websocket messages, ensuring valid state transitions.
Nr0 zwebsocket.connectz3Expected ASGI message "websocket.connect", but got > zwebsocket.receive�websocket.disconnectzMExpected ASGI message "websocket.receive" or "websocket.disconnect", but got r: zBCannot call "receive" once a disconnect message has been received.)r5 r
r r3 � RuntimeErrorr r �r' �message� message_typer r r r. # s& � �zWebSocket.receiver= c � s8 �| j tjkr6|d }|dvrtd|����|dkr tj| _ n
|dkr(tj| _ ntj| _ | �|�I dH dS | j tjkro|d }|dvrKtd|����|dkrStj| _ z | �|�I dH W dS tyn tj| _ t d d
��w | j tjkr�|d }|d kr�td |����|�
d
d�s�tj| _ | �|�I dH dS td��)zQ
Send ASGI websocket messages, ensuring valid state transitions.
r0 > �websocket.close� websocket.http.response.start�websocket.acceptzhExpected ASGI message "websocket.accept", "websocket.close" or "websocket.http.response.start", but got r? r@ N> r? �websocket.sendzEExpected ASGI message "websocket.send" or "websocket.close", but got i� )r z websocket.http.response.bodyz>Expected ASGI message "websocket.http.response.body", but got � more_bodyFz6Cannot call "send" once a close message has been sent.) r6 r
r r; r r r r4 �OSErrorr �getr< r r r r/ ; sJ � ��
�
� zWebSocket.sendN� subprotocolr �headers�$Iterable[tuple[bytes, bytes]] | Nonec � s@ �|pg }| j tjkr| �� I d H | �d||d��I d H d S )NrA )r0 rF rG )r5 r
r r. r/ )r' rF rG r r r �acceptd s
� zWebSocket.acceptc C s$ |d dkrt |d |�d���d S )Nr0 r: r r )r rE �r' r= r r r �_raise_on_disconnectp s �z WebSocket._raise_on_disconnect�strc � �< �| j tjkr td��| �� I d H }| �|� tt|d �S )N�8WebSocket is not connected. Need to call "accept" first.�text)r6 r
r r; r. rK r rL rJ r r r � receive_textt � �
zWebSocket.receive_text�bytesc � rM )NrN rR )r6 r
r r; r. rK r rR rJ r r r �
receive_bytes{ rQ zWebSocket.receive_bytesrO �moder c � sh �|dvr t d��| jtjkrt d��| �� I d H }| �|� |dkr(|d }n|d �d�}t�|�S )N> �binaryrO �1The "mode" argument should be "text" or "binary".rN rO rR �utf-8) r; r6 r
r r. rK �decode�json�loads)r' rT r= rO r r r � receive_json� s �
zWebSocket.receive_json�AsyncIterator[str]c C �, �z
| � � I d H V q ty Y d S w �N)rP r �r' r r r � iter_text� � �� �zWebSocket.iter_text�AsyncIterator[bytes]c C r] r^ )rS r r_ r r r �
iter_bytes� ra zWebSocket.iter_bytes�AsyncIterator[Any]c C r] r^ )r[ r r_ r r r � iter_json� ra zWebSocket.iter_json�datac � � �| � d|d��I d H d S )NrB �r0 rO �r/ �r' rf r r r � send_text� � �zWebSocket.send_textc � rg )NrB �r0 rR ri rj r r r �
send_bytes� rl zWebSocket.send_bytesc � sd �|dvr t d��tj|ddd�}|dkr"| �d|d��I d H d S | �d|�d �d
��I d H d S ) N> rU rO rV )�,�:F)�
separators� ensure_asciirO rB rh rW rm )r; rY �dumpsr/ �encode)r' rf rT rO r r r � send_json� s � zWebSocket.send_jsonr r r r c � s" �| � d||pdd��I d H d S )Nr? r$ �r0 r r ri r& r r r �close� s � zWebSocket.close�responser c � s8 �d| j �di �v r|| j | j| j�I d H d S td��)Nzwebsocket.http.response�
extensionszCThe server doesn't support the Websocket Denial Response extension.)r- rE r. r/ r; )r' rx r r r �send_denial_response� s � z WebSocket.send_denial_response�r- r r. r
r/ r r r! )r r )r= r r r! )NN)rF r rG rH r r! )r rL )r rR )rO )rT rL r r )r r\ )r rb )r rd )rf rL r r! )rf rR r r! )rf r rT rL r r! r* r+ )rx r r r! )r r r r( r. r/ rI rK rP rS r[ r` rc re rk rn ru rw rz �
__classcell__r r r8 r r, s&
+ �
r, c @ s" e Zd Zddd d
�Zddd�ZdS )�WebSocketCloser Nr r r r r r! c C r" r# r% r& r r r r( � r) zWebSocketClose.__init__r- r r. r
r/ r c � s �|d| j | jd��I d H d S )Nr? rv r% r7 r r r �__call__� s � zWebSocketClose.__call__r* r+ r{ )r r r r( r~ r r r r r} � s r} )�
__future__r �enumrY �collections.abcr r �typingr r �starlette.requestsr �starlette.responsesr �starlette.typesr r
r r �Enumr
� Exceptionr r, r} r r r r �<module> s %