a &0É_ÉVã@sddlmZddlZddlZddlmZddlmZddlm Z m Z ddl m Z m ZmZmZmZmZmZmZddlmZmZdd lmZejZd d „ZGd d „d eƒZdd„ZGdd„dejj ƒZ!e "ej#¡Gdd„deƒƒZ$e "ej%ej&¡Gdd„dej'ƒƒZ(dS)é)Úabsolute_importNé)Ú_)Úbin)ÚgetattrÚsetattr)Úbundle2Ú changegroupÚencodingÚerrorÚpushkeyÚpycompatÚutilÚwireprototypes)Ú repositoryr)Úhashutilcs*‡fdd„}t|dˆƒt|dˆjƒ|S)aUannotation 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: encargs = [('one', encode(one),), ('two', encode(two),)] # Create future for injection of encoded result: encresref = future() # Return encoded arguments and future: yield encargs, encresref # Assuming the future to be filled with the result from the batched # request now. Decode it: yield decode(encresref.value) 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. csPˆ|i|¤Ž}t|ƒ\}}|s"|S|d}t ˆj¡}| | ||¡¡t|ƒS)Nr)Únextr ZbytesurlÚ__name__ÚsetÚ _submitone)ÚargsÚoptsÚ batchableÚ encargsorresZ encresrefÚselfÚcmd©Úf©ú;/usr/lib/python3/dist-packages/mercurial/wireprotov1peer.pyÚplain=s  zbatchable..plainrr)rr)rr rrrr&s rc@seZdZdZdd„ZdS)Úfuturez'placeholder for a value to be set latercCs t |d¡rt d¡‚||_dS)Nsvaluesfuture is already set)rÚ safehasattrr Ú RepoErrorÚvalue)rr$rrrrOs  z future.setN)rÚ __module__Ú __qualname__Ú__doc__rrrrrr!Lsr!csltj‰g}|D]R\}}t‡fdd„|Dƒƒs0J‚d ‡fdd„t |¡Dƒ¡}| d||f¡qd |¡S)z;Return a ``cmds`` argument value for the ``batch`` command.c3s|]}ˆ|ƒ|kVqdS©Nr)Ú.0Úk©Z escapeargrrÚ ^óz"encodebatchcmds..ó,c3s&|]\}}dˆ|ƒˆ|ƒfVqdS)s%s=%sNr)r)r*Úvr+rrr,`sÿs%s %só;)rZescapebatchargÚallÚjoinr Ú iteritemsÚappend)ÚreqÚcmdsÚopZargsdictrrr+rÚencodebatchcmdsUs þr8c@seZdZdZddd„ZdS)Ú 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()``. NcCs,| ¡rtjj ||¡S|j ¡| |¡Sr()Údoner ÚfuturesÚFutureÚresultÚ _peerexecutorÚ sendcommands)rZtimeoutrrrr=rs zunsentfuture.result)N)rr%r&r'r=rrrrr9isr9c@sDeZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dS)Ú peerexecutorcCs2||_d|_d|_g|_t ¡|_d|_d|_dS)NF) Ú_peerÚ_sentÚ_closedÚ_callsÚweakrefÚWeakSetÚ_futuresÚ_responseexecutorÚ _responsef)rÚpeerrrrÚ__init__€s zpeerexecutor.__init__cCs|Sr(r©rrrrÚ __enter__‰szpeerexecutor.__enter__cCs | ¡dSr()Úclose)rÚexctypeZexcvaleeÚexctbrrrÚ__exit__Œszpeerexecutor.__exit__csžˆjrt d¡‚ˆjr t d¡‚tˆjt ˆ¡dƒ‰ˆsFt dˆ¡‚‡‡‡‡fdd„}tˆddƒrx|ƒ}t|_ ˆ|_ n"ˆj rŒt dˆ¡‚|ƒ}ˆ  ¡|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 peercs.tj ¡}ˆj |¡ˆj ˆˆˆ|f¡|Sr()r r;r<rGÚaddrDr4r©rÚcommandÚfnrrrÚaddcallªs  z)peerexecutor.callcommand..addcallrFsX%s is not batchable and cannot be called on a command executor along with other commands) rBr ÚProgrammingErrorrCrrAr Zsysstrr9Ú __class__r>rDr?)rrTrrVrrrSrÚ callcommands:ÿÿÿÿ  ÿÿzpeerexecutor.callcommandc Csê|jr dS|jsdSd|_|jD]}t|tƒr tjj|_d|_ q |j}d|_t |ƒdkrÆ|d\}}}}|  ¡stdSz|fit  |¡¤Ž}Wn*t y¶t |t ¡dd…¡Yn 0| |¡dSg}g}|D]Ú\}}}}|  ¡sèqÒz|j|jfit  |¡¤Ž} Wn0t y6t |t ¡dd…¡YdS0zt| ƒ\} } Wn0t yxt |t ¡dd…¡YdS0| sŒ| | ¡qÒ| || f¡| ||| | f¡qÒ|s¸dS|j |¡} tj d¡|_|j |j|| ¡|_dS)NTrr)rBrDrGÚ isinstancer9r r;r<rXr>ÚlenZset_running_or_notify_cancelÚ strkwargsÚ ExceptionÚfuture_set_exception_infoÚsysÚexc_infoÚ set_resultrÚ__self__rr4rAÚ _submitbatchZThreadPoolExecutorrHZsubmitÚ_readbatchresponserI) rrZcallsrTrrUr=ZrequestsÚstatesrrÚfremoteÚ wireresultsrrrr?Æsf       ÿÿ     ÿzpeerexecutor.sendcommandsc CsÈ| ¡|jrdSd|_|js"dSzV|j ¡W|jjddd|_d|_|jD] }| ¡sP| t   t dƒ¡¡qPd|_nJ|jjddd|_d|_|jD] }| ¡sš| t   t dƒ¡¡qšd|_0dS)NT)Úwaits"unfulfilled batch command response) r?rCrIr=rHZshutdownrGr:Z set_exceptionr Ú ResponseErrorr)rrrrrrNs<  ÿÿò ÿÿzpeerexecutor.closec Csj|D]`\}}}}t|ƒ}| |¡z t|ƒ}Wn*tyXt |t ¡dd…¡Yq0| |¡qdS)Nr)rrr]r r^r_r`ra) rrergrTrrrfZ remoteresultr=rrrrd?s   zpeerexecutor._readbatchresponseN) rr%r&rKrMrQrYr?rNrdrrrrr@~s 7Y r@c@sêeZdZdZdd„Zdd„Zedd„ƒZedd „ƒZed d „ƒZ ed d „ƒZ edd„ƒZ edd„ƒZ dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd3d%d&„Zd'd(„Zd)d*„Zd+d,„Zd-d.„Zd/d0„Zd1d2„Zd$S)4ÚwirepeerzîClient-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. cCst|ƒSr()r@rLrrrÚcommandexecutor]szwirepeer.commandexecutorcCs| dtdƒ¡| d¡S)Ns clonebundless clone bundles)Ú requirecaprÚ_callrLrrrÚ clonebundlesbszwirepeer.clonebundlesccsp| dtdƒ¡tƒ}dt |¡i|fV|j}|dd… dd¡\}}t|ƒr\t|ƒVn|  t   |¡¡dS)Nslookupslook up remote revisionókeyéÿÿÿÿó r) rlrr!r Ú fromlocalr$ÚsplitÚintrÚ_abortr r#)rÚkeyrÚdZsuccessÚdatarrrÚlookupfs zwirepeer.lookupc cs\tƒ}i|fV|j}zt |dd…¡VWn(tyV| t tdƒ|¡¡Yn0dS)Nrpóunexpected response:) r!r$rÚ decodelistÚ ValueErrorrur rir)rrrwrrrÚheadsrs  zwirepeer.headsc cshtƒ}dt |¡i|fV|j}zdd„t |¡DƒVWn(tyb| t  t dƒ|¡¡Yn0dS)NónodescSsg|]}tt|ƒƒ‘qSr)Úboolrt©r)ÚbrrrÚ ‚r-z"wirepeer.known..rz) r!rÚ encodelistr$r Z iterbytestrr|rur rir)rÚnodesrrwrrrÚknown|s zwirepeer.knownc cs’tƒ}i|fV|j}zNi}| ¡D]6}| dd¡\}}t t |¡¡}t  |¡}|||<q$|VWn(t yŒ|  t   tdƒ|¡¡Yn0dS)Nrqrrz)r!r$Ú splitlinesrsr ZtolocalÚurlreqZunquoterr{Ú TypeErrorrur rir)rrrwÚ branchmapZ branchpartZ branchnameZ branchheadsrrrr‰†s      zwirepeer.branchmapccsl| d¡sidfVtƒ}|j d|¡dt |¡i|fV|j}|j d|t|ƒf¡t  |¡VdS)Nópushkeyspreparing listkeys for "%s" ó namespaces$received listkey for "%s": %i bytes ) Úcapabler!ÚuiÚdebugr rrr$r[Ú pushkeymodZ decodekeys)rÚ namespacerrwrrrÚlistkeys–s  ÿzwirepeer.listkeysc csÌ| d¡sdVtƒ}|j d||f¡t |¡t |¡t |¡t |¡dœ|fV|j}| dd¡\}}ztt |ƒƒ}Wn"t yžt   t dƒ|¡‚Yn0| d¡D]}|j t d ƒ|¡qª|VdS) NrŠ)FNspreparing pushkey for "%s:%s" )r‹rosoldsnewó ró"push failed (unexpected response):Tóremote: )rŒr!rrŽr rrr$rsrrtr|r rirr†Ústatus) rrrvÚoldÚnewrrwÚoutputÚlrrrr £s, üû ÿ zwirepeer.pushkeycCs | d¡S)Ns stream_out)Ú _callstreamrLrrrÚ stream_out»szwirepeer.stream_outc Ks*t |¡}| dtdƒ¡i}| d¡p,tƒ}t |¡D]¨\}}|durJq8tj |¡}|durnt   d|¡‚nj|dkr‚t  |¡}nV|dkr–d  |¡}nB|dkr®d  t |ƒ¡}n*|d krÄd t|ƒ}n|d krØtd |ƒ‚|||<q8|jdit |¡¤Ž}td d„|Dƒƒrt |j|¡St |d¡SdS)Nó getbundleólook up remote changess bundlecapss$Unexpectedly None keytype for key %sr~scsvr.sscsvsbooleans%isplains unknown getbundle option type %scss|]}| d¡VqdS)sHG2N)Ú startswith)r)Zcaprrrr,×r-z%wirepeer.getbundle..óUN)rœ)r Z byteskwargsrlrÚgetrr3rZGETBUNDLE_ARGUMENTSr rWrƒr2ÚsortedrÚKeyErrorÚ_callcompressabler\ÚanyrÚ getunbundlerrÚchangegroupmodÚ cg1unpacker) rÚsourceÚkwargsrZ bundlecapsrvr$ZkeytyperrrrÚ getbundle¾s6  ÿ    zwirepeer.getbundlec Csð|dgkr8| d¡r8t dt d t|ƒ¡¡ ¡g¡}n t |¡}t  |d¡rÎ|j d||d\}}|dkrzt   t dƒ|¡‚z t|ƒ}Wn"ty¨t   t d ƒ|¡‚Yn0| d ¡D]}|j t d ƒ|¡q´n|jd||d}t |j|¡}|S) aÈSend 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-s deltaheadersunbundle)r}s push failed:r“Tr”)rŒrrƒrZsha1r2r¡Zdigestrr"Ú _callpushr rirrtr|r†rr•Ú_calltwowaystreamrr¥)rZbundler}ZurlZretr˜r™ÚstreamrrrÚunbundleÜs(ÿ    ÿ zwirepeer.unbundlec Cs^t |¡}|jd|d}zdd„| ¡Dƒ}|WStyX| t tdƒ|¡¡Yn0dS)Nsbranches)r„cSsg|]}tt |¡ƒ‘qSr)Útuplerr{r€rrrr‚ r-z%wirepeer.branches..rz) rrƒrmr†r|rur rir)rr„ÚnrwÚbrrrrÚbranches s  zwirepeer.branchesc Cs˜d}g}t dt|ƒ|¡D]x}d dd„||||…Dƒ¡}|jd|d}z| dd „| ¡Dƒ¡Wqty| t   t d ƒ|¡¡Yq0q|S) NérrqcSsg|]}t |d¡‘qS)ó-)rrƒ)r)Úprrrr‚sÿz$wirepeer.between..sbetween)Úpairscss |]}|rt |¡pgVqdSr()rr{)r)r™rrrr,sÿz#wirepeer.between..rz) r Zxranger[r2rmÚextendr†r|rur rir)rr¶ZbatchÚrÚir°rwrrrÚbetweens þÿ þ  zwirepeer.betweencCs$t |¡}|jd|d}t |d¡S)Ns changegroup)ÚrootsrŸ)rrƒr£r¦r§)rr„r¨r°rrrrr &s zwirepeer.changegroupcCs@| dtdƒ¡t |¡}t |¡}|jd||d}t |d¡S)Nschangegroupsubsetr)Úbasesr}rŸ)rlrrrƒr£r¦r§)rr¼r}r¨rrrrÚchangegroupsubset+s  ÿzwirepeer.changegroupsubsetc csî|j}|jrJ| dd¡rJ| d¡|D]"\}}d}| ||t|ƒf¡q&tj}|jdt|ƒd}|  d¡}|g} |rÚd|vr˜|r˜|  d¡}|   |¡qvd   | ¡} d| vrÆ|   dd ¡\} } || ƒVq¢|  d¡}| |g} qr|d   | ¡ƒVd S) 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)r6ir0r-rN) rZ debugflagZ configboolrŽr[rZunescapebatchargršr8Úreadr4r2rs) rr5rr7rÚmsgZ unescapeargZrspÚchunkZworkZmergedÚonerrrrc6s*          zwirepeer._submitbatchcCs|j|fit |¡¤ŽSr()rmr r\)rr7rrrrrSszwirepeer._submitoneNcCs:i}|dur||d<|dur$||d<|jd||dœ|¤ŽS)NÚthreeÚfouró debugwireargs)rÁÚtwo)rÄ)rm)rrÁrÅrÂrÃZfiverrrrÚ debugwireargsVs zwirepeer.debugwireargscKs tƒ‚dS)z†execute on the server The command is expected to return a simple string. returns the server reply as a string.N©ÚNotImplementedError©rrrrrrrm_szwirepeer._callcKs tƒ‚dS)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. NrÇrÉrrrršgs zwirepeer._callstreamcKs tƒ‚dS)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. NrÇrÉrrrr£rs zwirepeer._callcompressablecKs tƒ‚dS)züexecute 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. NrÇ©rrÚfprrrrr«s zwirepeer._callpushcKs tƒ‚dS)zqexecute on server The command will send a stream to the server and get a stream in reply. NrÇrÊrrrr¬Šszwirepeer._calltwowaystreamcCs tƒ‚dS)zKclearly abort the wire protocol connection and raise the exception NrÇ)rZ exceptionrrrru‘szwirepeer._abort)NNN)rr%r&r'rkrnrryr}r…r‰r‘r r›rªr®r²rºr r½rcrrÆrmršr£r«r¬rurrrrrjQs>      -      rj))Z __future__rr_rEZi18nrZnoderr rrÚrr r¦r r r rrrZ interfacesrZ interfaceutilZutilsrr‡rÚobjectr!r8r;r<r9Z implementerZipeercommandexecutorr@Z ipeercommandsZipeerlegacycommandsrJrjrrrrÚs(   (  &  Sÿ