*Gj=UddlZddlZddlZddlmZmZddlmZddlm Z m Z m Z m Z m Z ejdkrddlmZnddlmZeeedd fZ ddlZn#e$rddlZYnwxYwd Zn #e$rd ZYnwxYw ejd krdd lmZndd lmZd Zn #e$rd ZYnwxYwdZdZdZGdde ZGdde Z Gdde Z!Gdde d Z"GddZ#e#eZ$e#e%d<de!ddfdZ& d,de'dzd e(de)fd!Z*Gd"d#eZ+Gd$d%Z,Gd&d'e+Z-Gd(d)e+Z.Gd*d+e+Z/dS)-N)ABCabstractmethod)Executor)AnyFinalProtocol TypedDictcast) )Buffer)Unionzmemoryview[int]zmemoryview[bytes]TF)r )ZstdDecompressoric0eZdZdedefdZddedefdZdS) ZLibCompressObjProtocoldatareturncdSNselfrs a/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/aiohttp/compression_utils.pycompressz ZLibCompressObjProtocol.compress,.modecdSrrrrs rflushzZLibCompressObjProtocol.flush-rrN.)__name__ __module__ __qualname__r bytesrintr"rrrrr+s?2V2222255#55555555rrceZdZd dededefdZd dedefdZede fdZ edefd Z edefd Z d S) ZLibDecompressObjProtocol.r max_lengthrcdSrrrrr+s r decompressz$ZLibDecompressObjProtocol.decompress1rrlengthcdSrrrr/s rr"zZLibDecompressObjProtocol.flush2rrcdSrrrs reofzZLibDecompressObjProtocol.eof4s3rcdSrrr3s runconsumed_tailz)ZLibDecompressObjProtocol.unconsumed_tail7s(+rcdSrrr3s r unused_dataz%ZLibDecompressObjProtocol.unused_data:s$'CrNr#) r$r%r&r r(r'r.r"propertyboolr4r6r8rrrr*r*0sKKvK3KKKKK77C7U7777 TX ++++X+ 'U'''X'''rr*ceZdZUeed<eed<eed<eed<eed< ddeded ed ed ed ed zdefdZ dd ed edefdZ dded edede fdZ dd ededede fdZ d S)ZLibBackendProtocol MAX_WBITS Z_FULL_FLUSH Z_SYNC_FLUSH Z_BEST_SPEEDZ_FINISH.levelmethodwbitsmemLevelstrategyzdictNrcdSrr)rrBrCrDrErFrGs r compressobjzZLibBackendProtocol.compressobjEs #&#rcdSrr)rrDrGs r decompressobjz!ZLibBackendProtocol.decompressobjNs$'CrrcdSrr)rrrBrDs rrzZLibBackendProtocol.compressRrbufsizecdSrr)rrrDrNs rr.zZLibBackendProtocol.decompressUrMr)......)..) r$r%r&r(__annotations__r rrIr*rKr'rr.rrrr<r<>svNNNMMM"&&&& &  &  &}& !&&&&14((('-( "(((( ?B&)8; AD&):= rr<c.eZdZUeed<eed<eed<dS)CompressObjArgsrDrFrBN)r$r%r&r(rPrrrrRrRZs+ JJJMMM JJJJJrrR)totalc*eZdZdefdZedefdZedefdZ edefdZ edefdZ edefdZ edefd Z d ed edefd Zd ed edefd Zded ed edefdZded ed edefdZdedefdZdS)ZLibBackendWrapper _zlib_backendc||_dSr)rV)rrVs r__init__zZLibBackendWrapper.__init__as2?rrc.t|jddS)Nr$ undefinedgetattrrVr3s rnamezZLibBackendWrapper.namedst):{CCCrc|jjSr)rVr=r3s rr=zZLibBackendWrapper.MAX_WBITShs!++rc|jjSr)rVr>r3s rr>zZLibBackendWrapper.Z_FULL_FLUSHl!..rc|jjSr)rVr?r3s rr?zZLibBackendWrapper.Z_SYNC_FLUSHpr`rc|jjSr)rVr@r3s rr@zZLibBackendWrapper.Z_BEST_SPEEDtr`rc|jjSr)rVrAr3s rrAzZLibBackendWrapper.Z_FINISHxs!**rargskwargsc&|jj|i|Sr)rVrIrrdres rrIzZLibBackendWrapper.compressobj|s-t!-t>v>>>rc&|jj|i|Sr)rVrKrgs rrKz ZLibBackendWrapper.decompressobjs/t!/@@@@rrc.|jj|g|Ri|Sr)rVrrrrdres rrzZLibBackendWrapper.compresss(*t!*4A$AAA&AAArc.|jj|g|Ri|Sr)rVr.rjs rr.zZLibBackendWrapper.decompresss(,t!,TCDCCCFCCCrattrnamec,t|j|Srr[)rrls r __getattr__zZLibBackendWrapper.__getattr__st)8444rN)r$r%r&r<rXr9strr]r(r=r>r?r@rArrrIr*rKr r'rr.rnrrrrUrU`s@&9@@@@DcDDDXD,3,,,X,/c///X//c///X//c///X/+#+++X+???8O????A3A#A:SAAAABVBCB3B5BBBBDvDcDSDUDDDD5C5C555555rrU ZLibBackendnew_zlib_backendrc|t_dSr)rprV)rqs rset_zlib_backendrss 0Krencodingsuppress_deflate_headercb|dkrdtjzS|r tj n tjS)Ngzip)rpr=)rtrus rencoding_to_moderys66K)))%< WK ! ! !+BWWrceZdZdefdedzdedzfdZeefde dede fdZ efde dede fd Z e edefd ZdS) DecompressionBaseHandlerNexecutormax_sync_chunk_sizec"||_||_dS)z&Base class for decompression handlers.N) _executor_max_sync_chunk_size)rr|r}s rrXz!DecompressionBaseHandler.__init__s "$7!!!rrr+rcdSDecompress the given data.Nrr-s rdecompress_syncz(DecompressionBaseHandler.decompress_syncrrcK|jQt||jkr9tj|j|j||d{VS|||Sr)rlenasyncioget_event_looprun_in_executorrrr-s rr.z#DecompressionBaseHandler.decompresss  % 1D D555 /11AA 4dJ ##D*555rcdS)z7Return True if more output is available by passing b"".Nrr3s rdata_availablez'DecompressionBaseHandler.data_availablerr)r$r%r&MAX_SYNC_CHUNK_SIZErr(rXrZLIB_MAX_LENGTH_UNLIMITEDr r'rr.r9r:rrrrr{r{s%)*=88T/8!4Z8888.G)))(+) )))^) /H 6 6 6(+ 6  6 6 6 6FFFF^XFFFrr{ceZdZddddddefdedzdededzdedzdedzdedzd edzfd Zd e d e fd Z d e d e fdZ ddedzd e fdZ dS)ZLibCompressorNFrtrurBrDrFr|r}c||_||_|t||n||_t t j|_i}|j|d<|||d<|||d<|jjdi||_dS)NrDrFrBr) rrry_moderUrprVrI _compressor) rrtrurBrDrFr|r}res rrXzZLibCompressor.__init__s"$7!} X'> ? ? ? %7{7P$Q$Q"$*w  !)F:   #F7O94-9CCFCCrrrc6|j|Sr)rrrs r compress_synczZLibCompressor.compress_syncs((...rcK|jduot||jk}|r=tj|j|jj|d{VS||S)aCompress the data and returned the compressed bytes. Note that flush() must be called after the last call to compress() If the data size is large than the max_sync_chunk_size, the compression will be done in the executor. Otherwise, the compression will be done in the event loop. **WARNING: This method is NOT cancellation-safe when used with flush().** If this operation is cancelled, the compressor state may be corrupted. The connection MUST be closed after cancellation to avoid data corruption in subsequent compress operations. For cancellation-safe compression (e.g., WebSocket), the caller MUST wrap compress() + flush() + send operations in a shield and lock to ensure atomicity. N) rrrget_running_looprrrrr)rrshould_use_executors rrzZLibCompressor.compresss&  %T 1 6D D55    133CC 0 94 !!$'''rrcR|j||n |jjS)a Flush the compressor synchronously. **WARNING: This method is NOT cancellation-safe when called after compress().** The flush() operation accesses shared compressor state. If compress() was cancelled, calling flush() may result in corrupted data. The connection MUST be closed after compress() cancellation. For cancellation-safe compression (e.g., WebSocket), the caller MUST wrap compress() + flush() + send operations in a shield and lock to ensure atomicity. )rr"rVrAr!s rr"zZLibCompressor.flushs1%%$DD$*<*E   rr)r$r%r&rror:r(rrXr r'rrr"rrrrrs $(-  #$(*=DD*D"&DTz D Tz D * DT/D!4ZDDDD6/&/U////(6(e((((8   #*         rrc eZdZdddefdedzdededzdedzffd Ze fde d ed e fd Z dd ed e fdZ ed efdZed efdZxZS)ZLibDecompressorNFrtrur|r}ct||t|||_t t j|_|j|j|_d|_ d|_ dS)Nr|r}rDF) superrXryrrUrprVrK _decompressor _last_empty_pending_unused_data)rrtrur|r} __class__s rrXzZLibDecompressor.__init__ s| (@STTT%h0GHH $6{7P$Q$Q!/==DJ=OO 26!!!rrr+rc|j|jt|z}d|_|j|jj|z|}|dk|_|jjr|jjr|jj}|j |j |_|tkr |t|z}|dkr||_nC|j||}|dk|_||z }|jjr |jj|jjr:|j |jj kr%|j |j |_|S)Nrrr)rr'rr.r6rr4r8rVrKrrrr=)rrr+resultunusedchunks rrz ZLibDecompressor.decompress_syncsd  $ 0,uT{{:D(,D %#..   . 5z  "S=  $ );)G '3F!%!3!A!A !A!S!SD 666c&kk) ??06D-&11&*EEE$|D  eOF $ );)G    ! Tdj43E3O&O&O!%!3!A!A !A!S!SD  rrr/ct|dkr|j|n|jSNr)rr"r1s rr"zZLibDecompressor.flush?s=zz   $ $V , , ,#))++ rcVt|jjp|j p|jduSr)r:rr6rrr3s rrzZLibDecompressor.data_availableFs8 #3 4 4 5## 5(4 rc|jjSr)rr4r3s rr4zZLibDecompressor.eofNs!%%r)r)r$r%r&rror:rr(rXrr r'rr"r9rr4 __classcell__rs@rrr s/ $(-$(*= 7 7* 7"& 7T/ 7 !4Z 7 7 7 7 7 7/H"""(+" """"H  C         X &T&&&X&&&&&rrc~eZdZdefdedzdedzddffd Zefdedede fdZ de fd Z e de fd ZxZS) BrotliDecompressorNr|r}rctstdtj|_d|_t ||dS)z)Decompress data using the Brotli library.zIThe brotli decompression is not available. Please install `Brotli` moduleFrN) HAS_BROTLI RuntimeErrorbrotli Decompressor_objrrrXrr|r}rs rrXzBrotliDecompressor.__init__Wsb  1 '))   (@STTTTTrrr+ct|jdrh|tkr.tt|j|}ntt|j||}ng|tkr.tt|j|}n.tt|j||}|dk|_|S)rr.r)hasattrrrr r'r.processr)rrr+rs rrz"BrotliDecompressor.decompress_syncfs 49l + + J666eTY%9%9$%?%?@@eTY%9%9$ %K%KLL666eTY%6%6t%<%<==eTY%6%6tZ%H%HII!S= rct|jdr,tt|jSdS)zFlush the decompressor.r"r)rrr r'r"r3s rr"zBrotliDecompressor.flushxs7 49g & & 2ty0011 1srcF|j o|j Sr)r is_finishedrr3s rrz!BrotliDecompressor.data_available~s$9((***C43C/CCrr$r%r&rrr(rXrr r'rr"r9r:rrrs@rrrSs %)*= U UT/ U!4Z U  U U U U U U /H(+ $u DDDDXDDDDDrrc~eZdZdefdedzdedzddffd Zefdedede fdZ de fd Z e de fd ZxZS) ZSTDDecompressorNr|r}rctstdt|_d|_t ||dS)NzOThe zstd decompression is not available. Please install `backports.zstd` moduler)HAS_ZSTDrrrrrrXrs rrXzZSTDDecompressor.__init__s_  9 %&& 26! (@STTTTTrrr+c|tkrtn|}|j|j|z}d|_|j||}|jjr|jjr|jj}t|_|tkr |t|z}|dkr||_n6||j||z }|jjr |jj|jjrt|_|Sr) rZSTD_MAX_LENGTH_UNLIMITEDrrr.r4r8rr)rrr+zstd_max_lengthrr8s rrz ZSTDDecompressor.decompress_syncs666 & %   $ 0,t3D(,D %%%dO<<im I 5 I)/K(**DI";;;3v;;."a''0;D- di**;HH HFim I 5 I 9= +(**DI rcdS)Nrrr3s rr"zZSTDDecompressor.flushssrcH|jj o |jj p|jduSr)r needs_inputr4rr3s rrzZSTDDecompressor.data_availables0 % % ;dim*;3  &d 2 3rrrs@rrrs%)*= U UT/ U!4Z U  U U U U U U/H$$$(+$ $$$$Lu3333X33333rr)NF)0rsyszlibabcrrconcurrent.futuresrtypingrrrr r version_infocollections.abcr rr' bytearray brotlicffir ImportErrorrcompression.zstdrbackports.zstdrrrrrr*r<rRrUrprPrsror:r(ryr{rrrrrrrrs  ########''''''88888888888888w&&&&&&& 5)%68KK LF#####  JJJJJ 7""5555555333333HHHHH66666h666 ( ( ( ( ( ( ( ((8iu *5*5*5*5*5*5*5*5Z#5"4T":": :::1':1t1111  $)XXDjX!X XXXX F F F F Fs F F FFH H H H H H H H VD&D&D&D&D&/D&D&D&N-D-D-D-D-D1-D-D-D`<3<3<3<3<3/<3<3<3<3<3sBAA' A!A' A!!A''A10A15BBB