q&fWddlZddlZddlmZddlmZddlmZddlm Z m Z m Z m Z mZmZmZmZddlmZmZddlmZej0Zd Zd ZGd d ej6Zej:ej<Gd dZej:ej@ejBGddejDZ#y)N)futures)_)bin)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)r bytesurl__name__ _submitone)argsoptsencoded_args_or_resdecodeselfcmd encoded_resfs ;/usr/lib/python3/dist-packages/mercurial/wireprotov1peer.pyplainzbatchable..plain5sZ&'&6&6#V& &Aw +ooc+>? k"" batchabler)setattrr)rrs` rr!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..Ls79Q<1$7s,c3JK|]\}}d||fzyw)s%s=%sNr&)r'r(vr)s rr*z"encodebatchcmds..Ns, 8<1H ! il3 3 s #s%s %s;)rescapebatchargalljoinitemsappend)reqcmdsopargsdictrr)s @rencodebatchcmdsr8Cs--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 rr>zunsentfuture.result_sE 99;>>((w7 7 '') {{7##r r%)r __module__ __qualname____doc__r>r&r rr:r:Vs  $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__ms;    ) !%r c|Sr%r&rs r __enter__zpeerexecutor.__enter__vs r c$|jyr%)close)rexctypeexcvaleeexctbs r__exit__zpeerexecutor.__exit__ys  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=rNaddrKr3)rrcommandfnrs raddcallz)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) rIr ProgrammingErrorrJgetattrrHr sysstrr: __class__r?rKr@)rr_rrarr`s``` @r callcommandzpeerexecutor.callcommand|s ::((G  <<((=  TZZ!94 @(($%   2{E * A'AK"AO{{,,:requestsstatesrr wireresultss rr@zpeerexecutor.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) r@rJrPr>rOshutdownrNr< set_exceptionr ResponseErrorr)rrs rrWzpeerexecutor.closes&  <<    ! 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|) nextrlrmr rnrorp StopIterationr~r rr) rrxryr_rr!r remoteresultr>s rruzpeerexecutor._readbatchresponse%s.4 ) )GQ 6 )#K0 )#L1FLL() )"!N66q#,,.:LMN! ''?@$  s A.44A+*A+.7B('B(N) rrBrCrRrUr[rfr@rWrur&r rrFrFks)5nP d!@)r rFceZdZdZdZdZdZdZedZ 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%)rFrTs rcommandexecutorzwirepeer.commandexecutorKs D!!r c|jdr|jdS|jdtd|jdS)Nsclonebundles_manifests clonebundless clone bundles)capable_call requirecaprrTs r clonebundleszwirepeer.clonebundlesPsB <<0 1::67 7 OOOQ/?-@ A::o. .r cyr%r&)rstreams r_finish_inline_clone_bundlez$wirepeer._finish_inline_clone_bundleWs r c#K|jd|}tj|}tj||D]}||j |yw)Nsget_cached_bundle_inline)path)limit) _callstreamr uvarintdecodestream filechunkiterr)rrrlengthchunks rget_cached_bundle_inlinez!wirepeer.get_cached_bundle_inlineZs_!!"=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 rrzwirepeer.lookup..decodehsCcrFLLq1MGT7|4y  EOOD12r key)rrr fromlocal)rkeyrs` rlookupzwirepeer.lookupds9  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 rrzwirepeer.heads..decodessP P%003B88 P E//2I0JANO Ps7AAr&rrs` rheadszwirepeer.headsqs P 6zr c@fd}dtj|i|fS)Nc tj|Dcgc]}tt|c}Scc}w#t$r1j t jtd|YywxYw)Nr) r iterbytestrboolrrrr rr)rbrs rrzwirepeer.known..decode}s` P.6.B.B1.EFSV FFF P E//2I0JANO Ps=8==7A76A7nodes)r encodelist)rnodesrs` rknownzwirepeer.known{s& 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 rrz"wirepeer.branchmap..decodes P "#,,.8J.8.>.>tQ.G+J !)!1!1&..2L!MJ"0";";K"HK,7Ij) 8 !  P E//2I0JANO PsA/A337B-,B-r&rs` rrzwirepeer.branchmaps P6zr cjdsidfSjjdzfd}dtji|fS)Npushkeyspreparing listkeys for "%s" cjjdt|fztj|S)Ns$received listkey for "%s": %i bytes )uidebugri pushkeymod decodekeys)r namespacers rrz!wirepeer.listkeys..decodes7 GGMM8Is1v;NN ((+ +r namespacerrrr r)rrrs`` rlistkeyszwirepeer.listkeyssP||J't8O  6BC , h00;.decodesq)IAv QL &&t, 2q~q1 2H  ));K|]}|jdyw)sHG2N) startswith)r'caps rr*z%wirepeer.getbundle..s=3v&=sUN)r)r byteskwargsrrgetsetr2rGETBUNDLE_ARGUMENTSr rbrr1sortedrKeyError_callcompressablerkanyr getunbundlerrchangegroupmod cg1unpacker) rsourcekwargsr bundlecapsrvaluekeytypers r getbundlezwirepeer.getbundless%%f-  a(A&BCZZ .7#%  ,,. JC}$88<''3 3!--a7 7r c |dgk7rb|jdrQtjdtjdj t |jg}ntj|}t|dr|jd||\}}|dk(rtjtd| t|}|jd 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)rrrrsha1r1rdigesthasattr _callpushr rrrrrrr_calltwowaystreamrr)rbundlerurlretrrrs runbundlezwirepeer.unbundlesM XJ 4<<#@"--HMM#((6%=*ABIIKLE#--e4E 6= )..fE.JKCcz))!O*.s06.33A6<"<s')r) rangerir1rrrextendrrrr rr)rrbatchriprrs rbetweenzwirepeer.betweens q#e*e, PA #1q5y1#--a6A  :Q /A P\\^ P P E//2I0JANO PsB +&B7CCc~tj|}|jd|}tj|dS)Ns changegroup)rootsr)rrrrr)rrrrrs rrzwirepeer.changegroup's:  % %e ,  " "> " ;))!U33r c|jdtdtj|}tj|}|j d||}t j |dS)Nschangegroupsubsetr)basesrr)rrrrrrr)rrrrrs rchangegroupsubsetzwirepeer.changegroupsubset,sh ,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)r5ir.r rN) r debugflag configboolrrirunescapebatchargrr8readr3r1r) rr4rr6rmsg unescapeargrsprworkmergedones rrrzwirepeer._submitbatch7sA WW < 0DECI./ 0%55 xoc.BCwe# E"e#XXd^F&.$ll43 V!#&&&.HHTNEE?D#((4.))sCE8EE+EcN|j|fitj|Sr%)rr rk)rr6rs rrzwirepeer._submitoneTs#tzz"9 2 24 899r NcLi}|||d<|||d<|jd||d|S)Nthreefour)rtwo)s debugwireargs)r)rrr"r r!fivers r debugwireargszwirepeer.debugwireargsWs@  !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 rrzwirepeer._call`s "##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 rrzwirepeer._callstreamh "##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 rrzwirepeer._callcompressabless "##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 rrzwirepeer._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 rrzwirepeer._calltwowaystreams "##r ct)zBclearly abort the wire protocol connection and raise the exceptionr&)r exceptions rrzwirepeer._aborts !##r )NNN)rrBrCrDrrrrr!rrrrrr rrrrr rrrrrr$rrrrrrr&r rrr?s" / 1 9 9DD   E E0/8<'ZL(4 4*::F$ $ $ $$$r r)$rorL concurrentri18nrnoderrrrr r r rr r r interfacesr interfaceutilutilsrrr!r8r=r: implementeripeercommandexecutorrF ipeercommandsipeerlegacycommandsrQrr&r rr=s     B&$7>>$*:::;P)P)<P)fj<<R$zR$R$r