vjU`bddlmZddlZddlZddlmZmZddlmZddl m Z m Z m Z m Z mZmZddlmZmZdd lmZmZmZgd Zd Zed d Dcgc] }e| c}ZGddeZ ddZ ddZGddeZ ddZGddeZ ddZ!ycc}w)) annotationsN)AnySequence)frames)DuplicateParameterInvalidParameterNameInvalidParameterValueNegotiationError PayloadTooBig ProtocolError) ExtensionNameExtensionParameter)ClientExtensionFactory ExtensionServerExtensionFactory)PerMessageDeflateClientPerMessageDeflateFactory enable_client_permessage_deflateServerPerMessageDeflateFactory enable_server_permessage_deflatescjeZdZdZedZ d d dZd dZdd d dZd dZ y)rz) Per-Message Deflate extension. permessage-deflateNc|i}|dvsJ|dvsJd|cxkrdksJJd|cxkrdksJJd|vsJ||_||_||_||_||_|js&t j |j |_|js1t jdd|j i|j|_ d|_ y) z? Configure the Per-Message Deflate extension. N)FTrwbitsrF) remote_no_context_takeoverlocal_no_context_takeoverremote_max_window_bitslocal_max_window_bitscompress_settingszlib decompressobjdecoder compressobjencoderdecode_cont_data)selfr"r#r$r%r&s y/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/websockets/extensions/permessage_deflate.py__init__zPerMessageDeflate.__init__)s  $ " )]:::(M999*0b00000)/R/////////*D')B&&<#%:"!2..--T5P5P4PQDL--++111((DL!&c pd|jd|jd|jd|jd 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 r.__repr__zPerMessageDeflate.__repr__SsV**.*I*I)JK))-)G)G(HI&&*&A&A%BC%%)%?%?$@  C r0)max_sizec|jtjvr|S|jtjur"|js|S|j rrd|_nj|j s|Stj|d}|j sd|_|jr&tj|j |_ |j}|j r |tz }|dn|} |jj!||}|jj&rt)d|d |j r|jr|` tj|| S#tj"$r}t%d|d}~wwxYw) z, Decode an incoming frame. Frsv1Tr Nrzdecompression failedzover size limit (? > z bytes)data)opcoder CTRL_OPCODESOP_CONTr,finr8 dataclassesreplacer"r'r(r$r)r:_EMPTY_UNCOMPRESSED_BLOCK decompresserrorr unconsumed_tailr )r-framer5r: max_lengthexcs r.decodezPerMessageDeflate.decode\sW <<6.. .L <<6>> )(( yy(-%:: ''E:E99(,%..#119T9T8TU zz 99 - -D"*Q  A<<**4endswithrA)r-rEr:s r.encodezPerMessageDeflate.encodes <<6.. .L <zIClientPerMessageDeflateFactory.process_response_params..^!H4G5uzzTYY&4G#&zreceived duplicate Frgz#expected server_no_context_takeoverTzexpected server_max_window_bitsz"unsupported server_max_window_bitsz!unexpected client_max_window_bitsz"unsupported client_max_window_bitsr) anyr r\rir^r_r`rarr&r-rcaccepted_extensionsr^r_r`ras` r.process_response_paramsz6ClientPerMessageDeflateFactory.process_response_paramsSs@ H4GH H"%8 #DE E % 8  & & " "  * *-&'LMM  * *--1*  & & . &-&'HII'$*E*EE&'KLL  & & .%1&'JKK  ( (D 0 &-)-)D)D&'$*E*EE&'KLL & & " (b " (b  " "   r0)FFNTN) r^rQr_rQr`rWraint | bool | Noner&dict[str, Any] | NonerSrT)rSlist[ExtensionParameter])rcSequence[ExtensionParameter]r~Sequence[Extension]rSr) rXrYrZr[rr\r/rqrr!r0r.rr s. - .D,1+0-14837 3$(3%)3!+ 3 !2 3 1 3 3@  g ,g 1g   g r0rch|g}td|Dst|tddigz}|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. c3VK|]!}|jtjk(#ywrP)r\r)rvextension_factorys r.rxz3enable_client_permessage_deflate..s+!+  "@"E"EE!+')memLevel)r&)r|listr extensionss r.rrsQ !+ *% *#-q/ )  r0cdeZdZdZedZ d ddZ ddZy) ra 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``. rNc(|"d|cxkrdkstdtd|"d|cxkrdkstdtd|d|vr td| |r td||_||_||_||_||_||_y) rlNrrrmrnrzLcompress_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)ror^r_r`rar&require_client_max_window_bits)r-r^r_r`rar&rs r.r/z'ServerPerMessageDeflateFactory.__init__s'.!7M2SQS2SNO O3TNO O&.!7M2SQS2SNO O3TNO O  (W8I-I5  " ).L>  +E'*D'&<#&<#!2.L+r0cPtfd|Drtdjt|d\}}}}jr|sd}j r|sd}j n*| j }n|j kDr j }j|durRj}nE|jr7td|dur j}nj|kr j}t||||t|||xsd|xsdjfS)zi Process request parameters. Return response params and an extension instance. c3PK|]}|jjk(ywrPrtrus r.rxzHServerPerMessageDeflateFactory.process_request_params.. ryrzzskipped duplicate Tr{zrequired client_max_window_bitsr) r|r r\rir^r_r`rarrdrr&r}s` r.process_request_paramsz5ServerPerMessageDeflateFactory.process_request_paramssf H4GH H"%7 {#CD D $ 7  & & " ""  * *--1*  * *--1*  & & . &-)-)D)D&'$*E*EE)-)D)D&  & & .%-)-)D)D&&-66*+LMM'4/)-)D)D&,,/EE)-)D)D& **&&   **&,"&,"&&    r0)FFNNNF)r^rQr_rQr`rWrarWr&rrrQrSrT)rcrr~rrSz2tuple[list[ExtensionParameter], PerMessageDeflate])rXrYrZr[rr\r/rr!r0r.rrs4 - .D,1+0-1-137/4!M$(!M%)!M!+ !M !+ !M 1 !M)-!M !MFo ,o 1o  < o r0rcl|g}td|Dst|tddddigz}|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. c3VK|]!}|jtjk(#ywrP)r\r)rv ext_factorys r.rxz3enable_server_permessage_deflate..s*%K :???%r rr)r`rar&)r|rrrs r.rrsW % *% *')')#-q/ )   r0) r^rQr_rQr`rWrarrSr)rcrrgrQrSz0tuple[bool, bool, int | None, int | bool | None])rz'Sequence[ClientExtensionFactory] | NonerSz Sequence[ClientExtensionFactory])rz'Sequence[ServerExtensionFactory] | NonerSz Sequence[ServerExtensionFactory])" __future__rr?r'typingrrr exceptionsrr r r r r rrbaserrr__all__rArangerUrfrrdrirrrr)bitss0r.rs" 7KK 016q">3t9>T5 T5n $ $'.   0: (:8<:5:zm %;m `7%0o %;o d7%S?sB,