o
    {g1                     @  s  U d dl mZ d dlZd dlZd dlZ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 zddlmZ W n eyN   ddlmZ Y nw g dZG d	d
 d
ejZejZejZejZejZej Z!ej"Z#eeefZ$ee!e#fZ%G dd dejZ&e&j'de&j(de&j)de&j*de&j+de&j,de&j-de&j.de&j/de&j0de&j1de&j2de&j3de&j4de&j5diZ6de7d< e&j'e&j(e&j)e&j*e&j-e&j.e&j/e&j0e&j1e&j2e&j3e&j4hZ8e&j'e&j(e&j+hZ9e:e;e<fZ=ej>G dd dZ?ej>G d d! d!Z@dd"lAmBZB dS )#    )annotationsN)	GeneratorSequence)CallableUnion   )PayloadTooBigProtocolError)
apply_mask)OpcodeOP_CONTOP_TEXT	OP_BINARYOP_CLOSEOP_PINGOP_PONGDATA_OPCODESCTRL_OPCODESFrameClosec                   @  s$   e Zd ZdZd\ZZZd\ZZZ	dS )r   z#Opcode values for WebSocket frames.)r   r      )   	   
   N)
__name__
__module____qualname____doc__CONTTEXTBINARYCLOSEPINGPONG r$   r$   n/var/www/bot.gig.net.ua/public_html/telegram/P1/HellBot/venv/lib/python3.10/site-packages/websockets/frames.pyr   $   s    
r   c                   @  sL   e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdS )	CloseCodez-Close code values for WebSocket close frames.i  i  i  i  i  i  i  i  i  i  i  i  i  i  i  N)r   r   r   r   NORMAL_CLOSURE
GOING_AWAYPROTOCOL_ERRORUNSUPPORTED_DATANO_STATUS_RCVDABNORMAL_CLOSUREINVALID_DATAPOLICY_VIOLATIONMESSAGE_TOO_BIGMANDATORY_EXTENSIONINTERNAL_ERRORSERVICE_RESTARTTRY_AGAIN_LATERBAD_GATEWAYTLS_HANDSHAKEr$   r$   r$   r%   r&   6   s"    r&   OKz
going awayzprotocol errorzunsupported datazno status received [internal]zabnormal closure [internal]zinvalid frame payload datazpolicy violationzmessage too bigzmandatory extensionzinternal errorzservice restartztry again laterzbad gatewayz TLS handshake failure [internal]zdict[int, str]CLOSE_CODE_EXPLANATIONSc                   @  s   e Zd ZU dZded< ded< dZded< d	Zded
< d	Zded< d	Zded< e	e
jddZd&ddZedddd'ddZddd(d!d"Zd)d$d%ZdS )*r   aN  
    WebSocket frame.

    Attributes:
        opcode: Opcode.
        data: Payload data.
        fin: FIN bit.
        rsv1: RSV1 bit.
        rsv2: RSV2 bit.
        rsv3: RSV3 bit.

    Only these fields are needed. The MASK bit, payload length and masking-key
    are handled on the fly when parsing and serializing frames.

    r   opcodez#Union[bytes, bytearray, memoryview]dataTboolfinFrsv1rsv2rsv3WEBSOCKETS_MAX_LOG_SIZE75returnstrc              
   C  s  d}t | j dt | jdkrdnd }| jrdnd}| jtu r+tt| j }n| jtu rc| j}t || j	d krX| j	d d d }d
|dd	|  d
|| d g}d
dd |D }n`| jtu rqtt| j}nR| jrztt| j }d}W nA ttfy   | j}t || j	d kr| j	d d d }d
|dd	|  d
|| d g}d
dd |D }d}Y nw d}t || j	kr| j	d d }|dd	|  d || d  }d
td|||g}| jj d| d| dS )zE
        Return a human-readable representation of a frame.

        Nz byter    s	continued       r   s      c                 s      | ]}|d V  qdS 02xNr$   .0byter$   r$   r%   	<genexpr>       z Frame.__str__.<locals>.<genexpr>textc                 s  rI   rJ   r$   rL   r$   r$   r%   rO      rP   binaryz''z...z, z [])lenr9   r;   r8   r   reprbytesdecoder   MAX_LOG_SIZEjoinr   rB   r   parseUnicodeDecodeErrorAttributeErrorfiltername)selfcodinglength	non_finalr9   rR   cutmetadatar$   r$   r%   __str__   s>   &

&
&"zFrame.__str__N)max_size
extensions
read_exact-Callable[[int], Generator[None, None, bytes]]maskrf   
int | Nonerg   %Sequence[extensions.Extension] | NoneGenerator[None, None, Frame]c             
   c  s   |dE dH }t d|\}}|d@ rdnd}|d@ rdnd}	|d@ r&dnd}
|d	@ r.dnd}zt|d
@ }W n tyI } ztd|d}~ww |d@ rPdnd|krXtd|d@ }|dkro|dE dH }t d|\}n|dkr|dE dH }t d|\}|dur||krt|||r|dE dH }||E dH }|rt||}| ||||	|
|}|du rg }t|D ]	}|j||d}q|	  |S )a  
        Parse a WebSocket frame.

        This is a generator-based coroutine.

        Args:
            read_exact: Generator-based coroutine that reads the requested
                bytes or raises an exception if there isn't enough data.
            mask: Whether the frame should be masked i.e. whether the read
                happens on the server side.
            max_size: Maximum payload size in bytes.
            extensions: List of extensions, applied in reverse order.

        Raises:
            EOFError: If the connection is closed without a full WebSocket frame.
            PayloadTooBig: If the frame's payload size exceeds ``max_size``.
            ProtocolError: If the frame contains incorrect values.

        r   N!BB   TF@             zinvalid opcodezincorrect masking   ~   !Hr   z!Q   )rf   )
structunpackr   
ValueErrorr	   r   r
   reversedrW   check)clsrh   rj   rf   rg   r9   head1head2r;   r<   r=   r>   r8   excra   
mask_bytesframe	extensionr$   r$   r%   rZ      sH   


zFrame.parserg   rV   c          
      C  s&  |    |du r
g }|D ]}|| } qt }| jrdnd| jr#dndB | jr*dndB | jr1dndB | jB }|r;dnd}t	| j
}|dk rS|td|||B  n|d	k re|td
||dB | n|td||dB | |r~td}|| |rt| j
|}	n| j
}	||	 | S )aH  
        Serialize a WebSocket frame.

        Args:
            mask: Whether the frame should be masked i.e. whether the write
                happens on the client side.
            extensions: List of extensions, applied in order.

        Raises:
            ProtocolError: If the frame contains incorrect values.

        Nro   r   rp   rq   rr   ru   rn   i   z!BBHz!BBQrt   rw   )r|   encodeioBytesIOr;   r<   r=   r>   r8   rT   r9   writerx   packsecretstoken_bytesr
   getvalue)
r_   rj   rg   r   outputr~   r   ra   r   r9   r$   r$   r%   	serialize  s>   



zFrame.serializeNonec                 C  sP   | j s	| js	| jrtd| jtv r$t| jdkrtd| js&tddS dS )z
        Check that reserved bits and opcode have acceptable values.

        Raises:
            ProtocolError: If a reserved bit or the opcode is invalid.

        zreserved bits must be 0}   zcontrol frame too longzfragmented control frameN)	r<   r=   r>   r	   r8   r   rT   r9   r;   r_   r$   r$   r%   r|   N  s   
zFrame.checkrA   rB   )
rh   ri   rj   r:   rf   rk   rg   rl   rA   rm   )rj   r:   rg   rl   rA   rV   rA   r   )r   r   r   r   __annotations__r;   r<   r=   r>   intosenvirongetrX   re   classmethodrZ   r   r|   r$   r$   r$   r%   r   {   s"   
 
2M;r   c                   @  sN   e Zd ZU dZded< ded< dddZedddZdddZdddZ	dS )r   z
    Code and reason for WebSocket close frames.

    Attributes:
        code: Close code.
        reason: Close reason.

    r   coderB   reasonrA   c                 C  sz   d| j   krdk rn nd}nd| j   krdk r n nd}nt| j d}| j  d| d}| jr;| d	| j }|S )
zU
        Return a human-readable representation of a close code and reason.

          i  
registered  zprivate useunknownz ()rH   )r   r7   r   r   )r_   explanationresultr$   r$   r%   re   n  s   zClose.__str__r9   rV   c                 C  sh   t |dkr$td|dd \}|dd  }| ||}|  |S t |dkr0| tjdS td)z
        Parse the payload of a close frame.

        Args:
            data: Payload of the close frame.

        Raises:
            ProtocolError: If data is ill-formed.
            UnicodeDecodeError: If the reason isn't valid UTF-8.

        r   rv   Nr   rC   zclose frame too short)rT   rx   ry   rW   r|   r&   r+   r	   )r}   r9   r   r   closer$   r$   r%   rZ     s   
zClose.parsec                 C  s    |    td| j| j  S )z:
        Serialize the payload of a close frame.

        rv   )r|   rx   r   r   r   r   r   r$   r$   r%   r     s   zClose.serializer   c                 C  s8   | j tv sd| j   krdk std tddS dS )z
        Check that the close code has a valid value for a close frame.

        Raises:
            ProtocolError: If the close code is invalid.

        r   r   zinvalid status codeN)r   EXTERNAL_CLOSE_CODESr	   r   r$   r$   r%   r|     s
   zClose.checkNr   )r9   rV   rA   r   )rA   rV   r   )
r   r   r   r   r   re   r   rZ   r   r|   r$   r$   r$   r%   r   `  s   
 	

r   r   )C
__future__r   dataclassesenumr   r   r   rx   collections.abcr   r   typingr   r   
exceptionsr   r	   speedupsr
   ImportErrorutils__all__IntEnumr   r   r   r   r   r    r   r!   r   r"   r   r#   r   r   r   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r7   r   r   OK_CLOSE_CODESrV   	bytearray
memoryview	BytesLike	dataclassr   r   rC   rg   r$   r$   r$   r%   <module>   s    


 eL