
    :"rid                    j   d dl mZ d dlZd dlmZ d dlmZmZ ddlm	Z	 ddl
mZmZmZmZmZmZ ddlmZmZmZ d	d
lmZmZmZ g dZdZ edd      D  cg c]
  }  e|        c} Z G d de      Z	 	 	 	 	 	 	 	 	 	 ddZ	 	 	 	 	 	 ddZ G d de      Z 	 	 	 	 ddZ! G d de      Z"	 	 	 	 ddZ#yc c} w )    )annotationsN)Sequence)AnyLiteral   )frames)DuplicateParameterInvalidParameterNameInvalidParameterValueNegotiationErrorPayloadTooBigProtocolError)	BytesLikeExtensionNameExtensionParameter   )ClientExtensionFactory	ExtensionServerExtensionFactory)PerMessageDeflateClientPerMessageDeflateFactory enable_client_permessage_deflateServerPerMessageDeflateFactory enable_server_permessage_deflates           c                  j    e Zd ZdZ ed      Z	 d		 	 	 	 	 	 	 	 	 	 	 d
dZddZdd	 	 	 	 	 ddZddZ	y)r   z)
    Per-Message Deflate extension.

    permessage-deflateNc                   |i }|dv sJ |dv sJ d|cxk  rdk  sJ  J d|cxk  rdk  sJ  J d|vsJ || _         || _        || _        || _        || _        | j                   s&t        j                  | j                         | _        | j                  s1t        j                  dd| j                   i| j                  | _	        d| _
        y)	z?
        Configure the Per-Message Deflate extension.

        N)FTr      wbitsr!   F )remote_no_context_takeoverlocal_no_context_takeoverremote_max_window_bitslocal_max_window_bitscompress_settingszlibdecompressobjdecodercompressobjencoderdecode_cont_data)selfr$   r%   r&   r'   r(   s         p/var/www/fortnox.pascalinesoft.com/venv/lib/python3.12/site-packages/websockets/extensions/permessage_deflate.py__init__zPerMessageDeflate.__init__)   s     $ ")]:::(M999*0b00000)/R/////////*D')B&&<#%:"!2..--T5P5P4PQDL--++ 111((DL !&    c           	     p    d| j                    d| j                   d| j                   d| j                   d	S )Nz-PerMessageDeflate(remote_no_context_takeover=z, local_no_context_takeover=z, remote_max_window_bits=z, local_max_window_bits=))r$   r%   r&   r'   r/   s    r0   __repr__zPerMessageDeflate.__repr__S   sV    **.*I*I)J K))-)G)G(H I&&*&A&A%B C%%)%?%?$@	C	
r2   )max_sizec                  |j                   t        j                  v r|S |j                   t        j                  u r"| j                  s|S |j
                  r[d| _        nS|j                  s|S |j
                  sd| _        | j                  r&t        j                  | j                         | _        |j
                  r5t        |j                        dk  rt        |j                        t        z   }n|j                  }|dn|}	 | j                  j!                  ||      }| j                  j"                  r|J t%        d|      |j
                  r7t        |j                        dk\  r| j                  j!                  t               |j
                  r| j                  r| `t        j*                  |j                   ||j
                  d|j,                  |j.                        S # t        j&                  $ r}t)        d      |d}~ww xY w)z,
        Decode an incoming frame.

        FTr"   i  Nr   zdecompression failed)opcoder   CTRL_OPCODESOP_CONTr.   finrsv1r$   r)   r*   r&   r+   lendatabytes_EMPTY_UNCOMPRESSED_BLOCK
decompressunconsumed_tailr   errorr   Framersv2rsv3)r/   framer7   r?   
max_lengthexcs         r0   decodezPerMessageDeflate.decode\   s    <<6...L
 <<6>>)((yy(-% ::99(,% ..#119T9T8TU 99UZZ4/ $'@@D::D"*Q
		A<<**4<D||+++++#D(33yyS_4''(AB
 9988||LLIIJJJJ
 	
 zz 	A 67S@	As   BG% %H	8HH	c                   |j                   t        j                  v r|S |j                   t        j                  ur=| j                  r1t        j                  dd| j                   i| j                  | _	        | j                  j                  |j                        | j                  j                  t
        j                        z   }|j                  r0|dd t        k(  sJ t!        |      dk  r|dd }nt#        |      dd }|j                  r| j                  r| `	t        j$                  |j                   ||j                  |j                   t        j                  u|j&                  |j(                        S )z,
        Encode an outgoing frame.

        r!   Ni   r#   )r9   r   r:   r;   r%   r)   r,   r'   r(   r-   compressr?   flushZ_SYNC_FLUSHr<   rA   r>   
memoryviewrE   rF   rG   )r/   rH   r?   s      r0   encodezPerMessageDeflate.encode   s7    <<6...L
 <<v~~---#//  555 ,,  ||$$UZZ04<<3E3EdFWFW3XX99 9 99994y4CRy!$', 9977||LLIILL.JJJJ
 	
r2   N)r$   boolr%   rT   r&   intr'   rU   r(   zdict[Any, Any] | NonereturnNone)rV   str)rH   frames.Framer7   
int | NonerV   rY   )rH   rY   rV   rY   )
__name__
__module____qualname____doc__r   namer1   r6   rK   rR   r#   r2   r0   r   r   !   s    
 -.D 48&&$(&& $(&& !$	&&
  #&& 1&& 
&&T
  $	G
G
 	G

 
G
R-
r2   r   c                    g }| r|j                  d       |r|j                  d       |r|j                  dt        |      f       |du r|j                  d       |S |r|j                  dt        |      f       |S )zS
    Build a list of ``(name, value)`` pairs for some compression parameters.

    )server_no_context_takeoverN)client_no_context_takeoverNserver_max_window_bitsT)client_max_window_bitsNrd   )appendrX   )ra   rb   rc   rd   paramss        r0   _build_parametersrg      s     (*F!:;!:;/5K1LMN%67 M 
 /5K1LMNMr2   c                  d}d}d}d}| D ]  \  }}|dk(  r|rt        |      |d}t        ||      |dk(  r|rt        |      |d}@t        ||      |dk(  r-|t        |      |t        v rt        |      }rt        ||      |dk(  r4|t        |      |r|d}|t        v rt        |      }t        ||      t	        |       ||||fS )z
    Extract compression parameters from a list of ``(name, value)`` pairs.

    If ``is_server`` is :obj:`True`, ``client_max_window_bits`` may be
    provided without a value. This is only allowed in handshake requests.

    FNra   Trb   rc   rd   )r	   r   _MAX_WINDOW_BITS_VALUESrU   r
   )rf   	is_serverra   rb   rc   rd   r_   values           r0   _extract_parametersrl      s5    (-',)-9= $-e//)(..}-1*+D%8811)(..}-1*+D%88--%1(..//),U&+D%88--%1(..U])-&11),U&+D%88 't,,I$-N 	#"	 r2   c                  f    e Zd ZdZ ed      Z	 	 	 	 	 d	 	 	 	 	 	 	 	 	 	 	 ddZd	dZ	 	 	 	 	 	 d
dZy)r   a  
    Client-side extension factory for the Per-Message Deflate extension.

    Parameters behave as described in `section 7.1 of RFC 7692`_.

    .. _section 7.1 of RFC 7692: https://datatracker.ietf.org/doc/html/rfc7692#section-7.1

    Set them to :obj:`True` to include them in the negotiation offer without a
    value or to an integer value to include them with this value.

    Args:
        server_no_context_takeover: Prevent server from using context takeover.
        client_no_context_takeover: Prevent client from using context takeover.
        server_max_window_bits: Maximum size of the server's LZ77 sliding window
            in bits, between 8 and 15.
        client_max_window_bits: Maximum size of the client's LZ77 sliding window
            in bits, between 8 and 15, or :obj:`True` to indicate support without
            setting a limit.
        compress_settings: Additional keyword arguments for :func:`zlib.compressobj`,
            excluding ``wbits``.

    r   Nc                   |"d|cxk  rdk  st        d       t        d      |&|du s"d|cxk  rdk  st        d       t        d      |d|v rt        d      || _        || _        || _        || _        || _        y)	G
        Configure the Per-Message Deflate extension factory.

        Nr   r    /server_max_window_bits must be between 8 and 15T/client_max_window_bits must be between 8 and 15r!   zLcompress_settings must not include wbits, set client_max_window_bits instead)
ValueErrorra   rb   rc   rd   r(   )r/   ra   rb   rc   rd   r(   s         r0   r1   z'ClientPerMessageDeflateFactory.__init__D  s     '.!7M2SQS2SNOO 3TNOO"*%-*0b0NOO 1NOO(W8I-I5 
 +E'*D'&<#&<#!2r2   c                n    t        | j                  | j                  | j                  | j                        S )z,
        Build request parameters.

        )rg   ra   rb   rc   rd   r5   s    r0   get_request_paramsz1ClientPerMessageDeflateFactory.get_request_paramsd  s5    
 !++++''''	
 	
r2   c                    t         fd|D              rt        d j                         t        |d      \  }}}} j                  r|st        d       j
                  r|sd} j                  n'|t        d      | j                  kD  rt        d       j                  |Ct        d	       j                  du rn)| j                  }n| j                  kD  rt        d
      t        |||xs d|xs d j                        S )zV
        Process response parameters.

        Return an extension instance.

        c              3  P   K   | ]  }|j                   j                   k(    y wrS   r_   .0otherr/   s     r0   	<genexpr>zIClientPerMessageDeflateFactory.process_response_params.<locals>.<genexpr>{       H5uzzTYY&H   #&zreceived duplicate Frj   z#expected server_no_context_takeoverTzexpected server_max_window_bitsz"unsupported server_max_window_bitsz!unexpected client_max_window_bitsz"unsupported client_max_window_bitsr    )
anyr   r_   rl   ra   rb   rc   rd   r   r(   r/   rf   accepted_extensionsra   rb   rc   rd   s   `      r0   process_response_paramsz6ClientPerMessageDeflateFactory.process_response_paramsp  s@    H4GHH"%8#DEE  %8	
&&"" **-&'LMM **--1* &&. &-&'HII'$*E*EE&'KLL &&.%1&'JKK((D0 &-)-)D)D&'$*E*EE&'KLL &&"(b"(b""
 	
r2   )FFNTN)ra   rT   rb   rT   rc   rZ   rd   int | Literal[True] | Noner(   dict[str, Any] | NonerV   rW   )rV   Sequence[ExtensionParameter])rf   r   r   Sequence[Extension]rV   r   )	r[   r\   r]   r^   r   r_   r1   rt   r   r#   r2   r0   r   r   *  s    . -.D ,1+0-1=A373$(3 %)3 !+	3
 !;3 13 
3@

g
,g
 1g
 
	g
r2   r   c                h    | g } t        d | D              st        |       t        ddi      gz   } | S )z
    Enable Per-Message Deflate with default settings in client extensions.

    If the extension is already present, perhaps with non-default settings,
    the configuration isn't changed.

    c              3  V   K   | ]!  }|j                   t        j                   k(   # y wrS   )r_   r   )ry   extension_factorys     r0   r{   z3enable_client_permessage_deflate.<locals>.<genexpr>  s*       	"@"E"EE   ')memLevel   )r(   )r   listr   
extensionss    r0   r   r     sQ     
 !+  *%*#-q/)
 


 r2   c                  d    e Zd ZdZ ed      Z	 	 	 	 	 	 d	 	 	 	 	 	 	 	 	 	 	 	 	 ddZ	 	 	 	 	 	 ddZy)	r   a  
    Server-side extension factory for the Per-Message Deflate extension.

    Parameters behave as described in `section 7.1 of RFC 7692`_.

    .. _section 7.1 of RFC 7692: https://datatracker.ietf.org/doc/html/rfc7692#section-7.1

    Set them to :obj:`True` to include them in the negotiation offer without a
    value or to an integer value to include them with this value.

    Args:
        server_no_context_takeover: Prevent server from using context takeover.
        client_no_context_takeover: Prevent client from using context takeover.
        server_max_window_bits: Maximum size of the server's LZ77 sliding window
            in bits, between 8 and 15.
        client_max_window_bits: Maximum size of the client's LZ77 sliding window
            in bits, between 8 and 15.
        compress_settings: Additional keyword arguments for :func:`zlib.compressobj`,
            excluding ``wbits``.
        require_client_max_window_bits: Do not enable compression at all if
            client doesn't advertise support for ``client_max_window_bits``;
            the default behavior is to enable compression without enforcing
            ``client_max_window_bits``.

    r   Nc                (   |"d|cxk  rdk  st        d       t        d      |"d|cxk  rdk  st        d       t        d      |d|v rt        d      ||rt        d      || _        || _        || _        || _        || _        || _        y)	ro   Nr   r    rp   rq   r!   zLcompress_settings must not include wbits, set server_max_window_bits insteadzVrequire_client_max_window_bits is enabled, but client_max_window_bits isn't configured)rr   ra   rb   rc   rd   r(   require_client_max_window_bits)r/   ra   rb   rc   rd   r(   r   s          r0   r1   z'ServerPerMessageDeflateFactory.__init__  s     '.!7M2SQS2SNOO 3TNOO&.!7M2SQS2SNOO 3TNOO(W8I-I5  ").L> 
 +E'*D'&<#&<#!2.L+r2   c                P    t         fd|D              rt        d j                         t        |d      \  }}}} j                  r|sd} j
                  r|sd} j                  n*| j                  }n| j                  kD  r j                  } j                  |du rR j                  }nE| j                  r7t        d      |du r j                  }n j                  |k  r j                  }t        ||||      t        |||xs d|xs d j                        fS )zi
        Process request parameters.

        Return response params and an extension instance.

        c              3  P   K   | ]  }|j                   j                   k(    y wrS   rw   rx   s     r0   r{   zHServerPerMessageDeflateFactory.process_request_params.<locals>.<genexpr>=  r|   r}   zskipped duplicate Tr~   zrequired client_max_window_bitsr    )r   r   r_   rl   ra   rb   rc   rd   r   rg   r   r(   r   s   `      r0   process_request_paramsz5ServerPerMessageDeflateFactory.process_request_params2  sf    H4GHH"%7		{#CDD  $7	
&&""" **--1* **--1* &&. &-)-)D)D&'$*E*EE)-)D)D& &&.%-)-)D)D& &-66*+LMM'4/)-)D)D&,,/EE)-)D)D& **&&	 **&,"&,"&&
 	
r2   )FFNNNF)ra   rT   rb   rT   rc   rZ   rd   rZ   r(   r   r   rT   rV   rW   )rf   r   r   r   rV   z2tuple[list[ExtensionParameter], PerMessageDeflate])r[   r\   r]   r^   r   r_   r1   r   r#   r2   r0   r   r     s    4 -.D ,1+0-1-137/4!M$(!M %)!M !+	!M
 !+!M 1!M )-!M 
!MFo
,o
 1o
 
<	o
r2   r   c                l    | g } t        d | D              st        |       t        ddddi      gz   } | S )z
    Enable Per-Message Deflate with default settings in server extensions.

    If the extension is already present, perhaps with non-default settings,
    the configuration isn't changed.

    c              3  V   K   | ]!  }|j                   t        j                   k(   # y wrS   )r_   r   )ry   ext_factorys     r0   r{   z3enable_server_permessage_deflate.<locals>.<genexpr>  s*       	:???r      r   r   )rc   rd   r(   )r   r   r   r   s    r0   r   r     sW     
 %  *%*')')#-q/)
 

 r2   )
ra   rT   rb   rT   rc   rZ   rd   r   rV   zlist[ExtensionParameter])rf   r   rj   rT   rV   z9tuple[bool, bool, int | None, int | Literal[True] | None])r   z'Sequence[ClientExtensionFactory] | NonerV   z Sequence[ClientExtensionFactory])r   z'Sequence[ServerExtensionFactory] | NonerV   z Sequence[ServerExtensionFactory])$
__future__r   r)   collections.abcr   typingr   r    r   
exceptionsr	   r
   r   r   r   r   r   r   r   baser   r   r   __all__rA   rangerX   ri   r   rg   rl   r   r   r   r   )bitss   0r0   <module>r      s   "  $    B A K K 0 16q">3t9> q
	 q
h $ $ ' 7	
 0:(:8<:>:zm
%; m
`7%0o
%; o
d7%M ?s   B0