MddlZddlZddlmZddlmZddlmZddlm Z m Z ddl m Z m ZmZmZmZmZmZmZddlmZmZdd lmZej4Zd Zd ZGd d ej:Zej>ej@GddZ!ej>ejDejFGddejHZ%y)N)futures)_)bin)getattrsetattr)bundle2 changegroupencodingerrorpushkeypycompatutilwireprototypes) repositoryr)hashutilcZfd}t|dt|dj|S)aannotation for batchable methods Such methods must implement a coroutine as follows: @batchable def sample(self, one, two=None): # Build list of encoded arguments suitable for your wire protocol: encoded_args = [('one', encode(one),), ('two', encode(two),)] # Return it, along with a function that will receive the result # from the batched request. return encoded_args, decode The decorator returns a function which wraps this coroutine as a plain method, but adds the original method as an attribute called "batchable", which is used by remotebatch to split the call into separate encoding and decoding phases. c|i|\}}|s|S|d}tjj}|j||}||S)Nr)rbytesurl__name__ _submitone)argsoptsencoded_args_or_resdecodeselfcmd encoded_resfs =usr/lib/python3.12/site-packages/mercurial/wireprotov1peer.pyplainzbatchable..plain9sZ&'&6&6#V& &Aw +ooc+>? k"" batchabler)rr)rr!s` r r#r#&s+&# E;" E:qzz* Lr"ctjg}|D]Y\}}tfd|DsJdjfd|j D}|j d||fz[dj|S)z;Return a ``cmds`` argument value for the ``batch`` command.c34K|]}||k(ywN).0k escapeargs r z"encodebatchcmds..Ps7h9Q<1$hs,c3JK|]\}}d||fzyw)s%s=%sNr')r(r)vr*s r r+z"encodebatchcmds..Rs, @P1H ! il3 3@Ps #s%s %s;)rescapebatchargalljoinitemsappend)reqcmdsopargsdictrr*s @r encodebatchcmdsr9Gs--I D H7h7777yy @H@P   HDz)* 99T?r"ceZdZdZddZy) unsentfuturea0A Future variation to represent an unsent command. Because we buffer commands and don't submit them immediately, calling ``result()`` on an unsent future could deadlock. Futures for buffered commands are represented by this type, which wraps ``result()`` to call ``sendcommands()``. Nc|jr tjj||S|jj |j|Sr&)donerFutureresult _peerexecutor sendcommands)rtimeouts r r?zunsentfuture.resultcsE 99;>>((w7 7 '') {{7##r"r&)r __module__ __qualname____doc__r?r'r"r r;r;Zs  $r"r;c6eZdZdZdZdZdZdZdZdZ y) peerexecutorc||_d|_d|_g|_t j |_d|_d|_y)NF) _peer_sent_closed_callsweakrefWeakSet_futures_responseexecutor _responsef)rpeers r __init__zpeerexecutor.__init__qs;    ) !%r"c|Sr&r'rs r __enter__zpeerexecutor.__enter__zs r"c$|jyr&)close)rexctypeexcvaleeexctbs r __exit__zpeerexecutor.__exit__}s  r"cjrtjdjrtjdt j t jdstjdzfd}t ddr|}t|_ |_ |Sjrtjdz|}j|S)Ns4callcommand() cannot be used after commands are sents*callcommand() cannot be used after close()sAcannot call command %s: method of same name not available on peerctj}jj|jj |f|Sr&)rr>rOaddrLr4)rrcommandfnrs r addcallz)peerexecutor.callcommand..addcallsA A MM  a KK  r15 6Hr"r#FsX%s is not batchable and cannot be called on a command executor along with other commands) rJr ProgrammingErrorrKrrIrsysstrr; __class__r@rLrA)rr`rrbrras``` @r callcommandzpeerexecutor.callcommands ::((G  <<((=  TZZ!94 @(($%   2{E * A'AK"AO{{,,:rer@lenset_running_or_notify_cancelr strkwargs set_result Exceptionfuture_set_exception_infosysexc_infor#__self__r4rI _submitbatchThreadPoolExecutorrPsubmit_readbatchresponserQ) rrcallsr`rrar?requestsstatesrr wireresultss r rAzpeerexecutor.sendcommandss :: {{  A!\*%nn "&    u:?#(8 GT2q113 %7h0067 V$ $) GT2q113 .:bllKK/#+#5#5d#;/+#V 01*= >? w9f=>#%*& jj--h7 ")!;!;A!>0077  # #V[ a J221cllnQR6HI   J4 221cllnQR6HI s$G*3G>4G;:G;>4H65H6c |j|jryd|_|jsy |jj|jj dd|_d|_|j D]A}|jr|jtjtddCd|_y#|jj dd|_d|_|j D]A}|jr|jtjtddCd|_wxYw)NT)wait"unfulfilled batch command response) rArKrQr?rPshutdownrOr= set_exceptionr ResponseErrorr)rrs r rXzpeerexecutor.close s  <<    ! OO " " $  " " + + + 6"DO%)D "]]vvxOO++CDd#!DM  " " + + + 6"DO%)D "]]vvxOO++CDd#!DMsCA E8Ec V|D]-\}}}} t|} ||}|j|/y#t$r.tj|t j ddYgwxYw#t$r1|jtjtddYwxYw)Nrr|) nextrlrmrrnrorp StopIterationr~r rr) rrxryr`rr#r remoteresultr?s r ruzpeerexecutor._readbatchresponse)s.4 )GQ 6 )#K0 )#L1FLL().4"!N66q#,,.:LMN! ''?@$  s A.44A+*A+.7B('B(N) rrCrDrSrVr\rfrArXrur'r"r rGrGos)5nP d!@)r"rGceZdZdZdZdZdZdZedZ edZ edZ ed Z ed Z ed Zd Zd ZdZdZdZdZdZdZdZddZdZdZdZdZdZdZy)wirepeerzClient-side interface for communicating with a peer repository. Methods commonly call wire protocol commands of the same name. See also httppeer.py and sshpeer.py for protocol-specific implementations of this interface. ct|Sr&)rGrUs r commandexecutorzwirepeer.commandexecutorOs D!!r"c|jdr|jdS|jdtd|jdS)Nsclonebundles_manifests clonebundless clone bundles)capable_call requirecaprrUs r clonebundleszwirepeer.clonebundlesTsB <<0 1::67 7 OOOQ/?-@ A::o. .r"cyr&r')rstreams r _finish_inline_clone_bundlez$wirepeer._finish_inline_clone_bundle[s r"c#K|jd|}tj|}tj||D]}||j |yw)Nsget_cached_bundle_inline)path)limit) _callstreamruvarintdecodestream filechunkiterr)rrrlengthchunks r get_cached_bundle_inlinez!wirepeer.get_cached_bundle_inline^sZ!!"=D!I))&1''f=EK> ((0sAAcvjdtdfd}dtj|i|fS)Nslookupslook up remote revisionc|ddjdd\}}t|r t|Sjt j |y)N r)splitintr_abortr RepoError)dsuccessdatars r rzwirepeer.lookup..decodelsCcrFLLq1MGT7|4y  EOOD12r"key)rrr fromlocal)rkeyrs` r lookupzwirepeer.lookuphs9  1%?#@A 3**3/0&88r"cfd}i|fS)Nc tj|ddS#t$r1jt j t d|YywxYw)Nrunexpected response:)r decodelist ValueErrorrr rr)rrs r rzwirepeer.heads..decodewsP P%003B88 P E//2I0JANO Ps7AAr'rrs` r headszwirepeer.headsus P 6zr"c@fd}dtj|i|fS)Nc tj|Dcgc]}tt|c}Scc}w#t$r1j t jtd|YywxYw)Nr) r iterbytestrboolrrrr rr)rbrs r rzwirepeer.known..decodesf P.6.B.B1.EF.ESV .EFFF P E//2I0JANO Ps=8==7A76A7nodes)r encodelist)rnodesrs` r knownzwirepeer.knowns& P .33E:;VCCr"cfd}i|fS)Nc` i}|jD]Y}|jdd\}}tjtj |}t j|}|||<[|S#t$r1jtjtd|YywxYw)Nrrr) splitlinesrr tolocalurlrequnquoterr TypeErrorrr rr)r branchmap branchpart branchname branchheadsrs r rz"wirepeer.branchmap..decodes P "#,,.J.8.>.>tQ.G+J !)!1!1&..2L!MJ"0";";K"HK,7Ij) #1 !  P E//2I0JANO PsA/A337B-,B-r'rs` r rzwirepeer.branchmaps P6zr"cjdsidfSjjdzfd}dtji|fS)Npushkeyspreparing listkeys for "%s" cjjdt|fztj|S)Ns$received listkey for "%s": %i bytes )uidebugri pushkeymod decodekeys)r namespacers r rz!wirepeer.listkeys..decodes7 GGMM8Is1v;NN ((+ +r" namespacerrrr r)rrrs`` r listkeyszwirepeer.listkeyssP||J't8O  6BC , h00;.decodesq)IAv QL &&t,q~q1-H  ));K|]}|jdyw)sHG2N) startswith)r(caps r r+z%wirepeer.getbundle..s=*3v&*sUN)r)r byteskwargsrrgetsetr3rGETBUNDLE_ARGUMENTSr rcrr2sortedrKeyError_callcompressablerkanyr getunbundlerrchangegroupmod cg1unpacker) rsourcekwargsr bundlecapsrvaluekeytypers r getbundlezwirepeer.getbundlesn%%f-  a(A&BCZZ .7#%  ,,.JC}$88<''3 3!--a7 7r"c |dgk7rb|jdrQtjdtjdj t |jg}ntj|}tj|dr|jd||\}}|dk(rtjtd| t|}|j!d D]'}|j"j%td |)|S|j'd||}t)j*|j"|}|S#t$r tjtd |wxYw) aSend cg (a readable file-like object representing the changegroup to push, typically a chunkbuffer object) to the remote server as a bundle. When pushing a bundle10 stream, return an integer indicating the result of the push (see changegroup.apply()). When pushing a bundle20 stream, return a bundle20 stream. `url` is the url the client thinks it's pushing to, which is visible to hooks. sforces unbundlehashshashedr" deltaheadersunbundle)rs push failed:rTr)rrrrsha1r2rdigestr safehasattr _callpushr rrrrrrr_calltwowaystreamr r)rbundlerurlretrrrs r unbundlezwirepeer.unbundlesN XJ 4<<#@"--HMM#((6%=*ABIIKLE#--e4E   FM 2..fE.JKCcz))!O*.#s0+6.33A6<"<+s')r) rangerir2rrrextendrrrr rr)rrbatchriprrs r betweenzwirepeer.betweens q#e*e,A #1q5y11#--a61A  :Q /A P\\^- P E//2I0JANO PsB +&B7CCc~tj|}|jd|}tj|dS)Ns changegroup)rootsr)rrrrr)rrrrrs r r zwirepeer.changegroup+s:  % %e ,  " "> " ;))!U33r"c|jdtdtj|}tj|}|j d||}t j |dS)Nschangegroupsubsetr)basesrr)rrrrrrr)rrrrrs r changegroupsubsetzwirepeer.changegroupsubset0sh ,a0I.JK))%0))%0  " " U # ))!U33r"c#K|j}|jrN|jddr<|jd|D]&\}}d}|j||t |fz(t j }|jdt|}|jd}|g} |rd|vr+|r)|jd}| j|d|vr|r)d j| } d| vr$| jdd \} } || d| vr$|jd}| |g} |r|d j| y w) zqrun batch request on the server Returns an iterator of the raw responses from the server. sdevelsdebug.peer-requests$devel-peer-request: batched-content s+devel-peer-request: - %s (%d arguments) sbatch)r6ir/r"rN) r debugflag configboolrrirunescapebatchargrr9readr4r2r) rr5rr7rmsg unescapeargrsprworkmergedones r rrzwirepeer._submitbatch;s< WW <DECI./ %55 xoc.BCwe# E"e#XXd^F&.$ll43 V!#&&&.HHTNEE?D#((4.))sCE8EE+EcN|j|fitj|Sr&)rrrk)rr7rs r rzwirepeer._submitoneXs#tzz"9 2 24 899r"NcLi}|||d<|||d<|jd||d|S)Nthreefour)rtwo)s debugwireargs)r)rrr"r r!fivers r debugwireargszwirepeer.debugwireargs[s@  !DM  DLtzzEEEEr"c t)zexecute on the server The command is expected to return a simple string. returns the server reply as a string.NotImplementedErrorrrrs r rzwirepeer._callds "##r"c t)a execute on the server The command is expected to return a stream. Note that if the command doesn't return a stream, _callstream behaves differently for ssh and http peers. returns the server reply as a file like object. r&r(s r rzwirepeer._callstreaml "##r"c t)aEexecute on the server The command is expected to return a stream. The stream may have been compressed in some implementations. This function takes care of the decompression. This is the only difference with _callstream. returns the server reply as a file like object. r&r(s r rzwirepeer._callcompressablews "##r"c t)zexecute a on server The command is expected to be related to a push. Push has a special return method. returns the server reply as a (ret, output) tuple. ret is either empty (error) or a stringified int. r&rrfprs r rzwirepeer._callpushr*r"c t)zqexecute on server The command will send a stream to the server and get a stream in reply. r&r-s r rzwirepeer._calltwowaystreams "##r"ct)zBclearly abort the wire protocol connection and raise the exceptionr&)r exceptions r rzwirepeer._aborts !##r")NNN)rrCrDrErrrrr#rrrrrr rrrrr r rrrrr$rrrrrrr'r"r rrCs" / 1 9 9DD   E E0/8<'ZL(4 4*::F$ $ $ $$$r"r)&rorM concurrentri18nrnoderrrrr r rr r r rrr interfacesr interfaceutilutilsrrr#r9r>r; implementeripeercommandexecutorrG ipeercommandsipeerlegacycommandsrRrr'r"r r=s     B&$7>>$*:::;P)P)<P)fj<<R$zR$R$r"