��c@s>dd
lZdd
l
Z
dd
lZdd
lZddlmZ
ddlmZmZ
dd
l	Z
dd
lZdd
lZdd
l
Z
dd
lZdd
lZdef
d��YZdef
d��YZdef
d	��YZd
ef
d��YZd�Zd
d�Zd
d�Zd�Zd�Zd�Zdejf
d��YZdef
d��YZdef
d��YZdef
d��YZdef
d��YZ d�Z!d�Z"d�Z#d �Z$d!�Z%d"�Z&d#�Z'd$�Z	d%�Z(d&�Z)d'�Z*d(�Z+d)�Z,d*�Z-d+�Z.d,�Z/d-�Z0d.�Z1d/�Z2d0�Z3ie#d1fd26e$d3fd46e%d5fd66e&d7fd86e'd5fd96e	d:fd;6e(d<fd=6e)d>fd?6e*d@fdA6e+d5fdB6e,d5fdC6e/dDfdE6e-dFfdG6e.dHfdI6e0dJfdK6e2d5fdL6e3dBfdM6Z4d
S(Ni�N(
t
_(tbinthextfuturec
BseZ
dZd
�ZRS(s'placeholder for a value to be set latercCs1tj
|d
�r$tjd��n|
|_dS(Ntvaluesfuture is already set(tutiltsafehasattrterrort	RepoErrorR(tselfR((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pytsets


(t__name__t
__module__t__doc__R
(((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRs

tbatcherc
Bs)eZ
dZd
�Zd�Zd�ZRS(s'
base class for batches of commands submittable in a single request

    All methods invoked on instances of this class are simply queued and
    return a a future for the result. Once you call submit(), all the queued
    calls are performed and the results set in their respective futures.
    c

Cs
g|_dS(
N(
tcalls(
R	((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyt__init__s
cs��fd
�}|S(Ncs)t�}�j
j�||
|f�
|S(
N(RRtappend(targstoptstresref(tnameR	(s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pytcall!s
	

((R	RR((RR	s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyt__getattr__ s
c


CsdS(
N((
R	((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pytsubmit&s
(RRR
RRR(((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRs
		t
localbatchc
Bs eZ
dZd
�Zd�ZRS(s"performs the queued calls directlycCstj
|�
|
|_dS(
N(RRtlocal(R	R((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR+s


c
CsFx?|jD]4\}
}}}|j
t|j|
�||��
q
WdS(
N(RR
tgetattrR(R	RRRR((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR.s

(RRR
RR(((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR)s

	tremotebatchc
Bs)eZ
dZd
�Zd�Zd�ZRS(s<batches the queued calls; uses as few roundtrips as possiblecCstj
|�
|
|_d
S(sNremote must support _submitbatch(encbatch) and
        _submitone(op, encargs)N(RRtremote(R	R((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR4s

c
Cs
gg}
}x�jD]�}}}}t
|j|�}t
|d
d�}|dk	r�||j||�}	|	j�\}
}|r�|
j||
f�
|j|	||f�
q�j|
�
q|
r�j|
|�
gg}
}n|j|||��
qW|
r
|j|
|�
ndS(Nt	batchable(	RRRtNonetim_selftnextRR
t
_submitreq(R	treqtrspRRRRtmtdtbatchablefnRtencargsorrest	encresref((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR9s"















c	Cse|jj
|
�}xLt||�D];\}}|\}}}|j|�
|j|j��
q"WdS(
N(Rt_submitbatchtzipR
R!(	R	R#R$t
encresultstencrest
rRR(R((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR"Ms






(RRR
RRR"(((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR2s

		c
s#�f
d
�}
t|
d��
|
S(s�annotation for batchable methods

    Such methods must implement a coroutine as follows:

    @batchable
    def sample(self, one, two=None):
        # Handle locally computable results first:
        if not one:
            yield "a local result", 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.
    cs[�||
�}|j�\}}|s+|S|d
}|j
|j�j|��
|j�S(Ni(R!R
t
_submitonet	func_name(RRRR'R(R	(
t
f(s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pytplainms







R(
tsetattr(R0R1((
R0s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRTs
t
 cCs |rtt
|j|
��SgS(
N(tmapR
tsplit(t
ltsep((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyt
decodelistzs


cCs|
jt
t|��S(
N(tjoinR4R(R6R7((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyt
encodelists
c

Cs4|jd
d�jdd�jdd�jdd�S(	Nt
:s::t
,s:,t
;s:;t
=s:=(
treplace(
R1((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyt	escapearg�s





c

Cs4|jd
d�jdd�jdd�jdd�S(	Ns:=R>s:;R=s:,R<s::R;(
R?(
tescaped((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pytunescapearg�s





c

Ks|S(
N((
R((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyttodict�s
twirepeercBs�eZ
d�Zd
�Zd�Zed��Zed��Zed��Zed��Z	d�Z
d�Zed	��Zed
��Z
d�Zd�Zd
�Zddd�Zd�Zdddd�ZRS(c

Cs
t|�S(
N(
R(
R	((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pytbatch�s
cCs~g}xJ|
D]B\}}d
jd�|j
�D��}|jd||f�
q
W|jdddj|��
}|jd�S(NR<c
ss|]}
d|
V
qd
S(s%s=%sN((t.0t
p((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pys	<genexpr>�ss%s %sREtcmdsR=(R9t	iteritemsRt_callR5(R	R#RHtoptargsdictRR$((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR)�s





cCs|j|
|�S(
N(
RJ(R	RKR((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR.�s
ccs�|jd
t
d��
t�}tdtj|
��
|fV
|j}|d jdd�\}}t|�r{t	|�V
n|j
tj|��
dS(Ntlookupslook up remote revisiontkeyi�R3i
(
t
requirecapRRRCtencodingt	fromlocalRR5tintR
t_abortRR(R	RNR0t
dtsuccesstdata((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRM�s
	

	



c
csgt�}
i|
fV
|
j
}yt|d
 �V
Wn0tk
rb


|jtjtd�|��
n
XdS(Ni�sunexpected response:(RRR8t
ValueErrorRSRt
ResponseErrorR(R	R0RT((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pytheads�s	

	




ccs�t�}t
d
t|
��
|fV
|j}y(g|D]}tt|��^q6V
Wn0tk
r�


|jtj	t
d�|��
n
XdS(Ntnodessunexpected response:(RRCR:RtboolRRRWRSRRXR(R	RZR0RT((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pytknown�s	

	

(


c
cs�t�}
i|
fV
|
j
}yli}xZ|j�D]L}|jd
d�\}}tjtj|��}t|�}|||<q3W|V
Wn0t	k
r�


|j
tjt
d�|��
n
XdS(NR3i
sunexpected response:(RRt
splitlinesR5RPttolocalturllibtunquoteR8t	TypeErrorRSRRXR(R	R0RTt	branchmapt
branchpartt
branchnametbranchheads((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRb�s	

	







	


cCs�t|
�}|j
d
d|�
}y3g|j�D]}tt|��^q1}|SWn0tk
r�


|jtjt	d�|��
n
XdS(NtbranchesRZsunexpected response:(
R:RJR]ttupleR8RWRSRRXR(R	RZt
nRTt
btbr((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRf�s



+



cCs�d
}g}x�tdt
|
�|�D]�}djg|
|||!D]}t|d�^qC�}|jdd|�
}y!|jd�|j�D��
Wq%tk
r�


|jt	j
td�|��
q%Xq%W|S(	NiiR3t
-tbetweentpairsc
ss'|]}
|
rt|
�pgV
qdS(
N(
R8(RFR6((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pys	<genexpr>�ssunexpected response:(txrangetlenR9R:RJtextendR]RWRSRRXR(R	RmRER-t
iRGRhRT((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRl�s



6


!


'
c	
cs)
|jd
�st
dfV
nt�}|jjd|
|f�
tdtj|
�dtj|�dtj|�dtj|��|fV
|j	}|j
dd�\}}ytt|��}Wn)t
k
r�

tjtd	�|��n
Xx0|jt�D]}|jjtd
�|�
q�V
dS(Ntpushkeyspreparing pushkey for "%s:%s"
t	namespaceRNtoldtnews

i
s"push failed (unexpected response):sremote: (tcapabletFalseRRtuitdebugRCRPRQRR5R[RRRWRRXRR]tTruetstatus(	R	RsRNRtRuR0RTtoutputR6((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRr�s$

	





	









ccs�|jd
�sidfV
nt�}|jjd|
�
tdtj|
��
|fV
|j}i}xE|j	�D]7}|j
d�\}}tj|�|tj|�<qsW|V
dS(NRrspreparing listkeys for "%s"
Rss
	(RvRRRxRyRCRPRQRR]R5R^(R	RsR0RTR-R6t
kt
v((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pytlistkeys�s

	


	



 
c

Cs
|jd
�S(Nt
stream_out(
t_callstream(
R	((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR�
s
cCs:t|
�}|j
d
d|�
}tj|j|�d�S(NtchangegrouptrootstUN(R:R�tchangegroupmodt
unbundle10t_decompress(R	RZtkindRhR0((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR�
s


cCsb|jd
t
d��
t|
�}
t|�}|jd
d|
d|�}tj|j|�d�S(Ntchangegroupsubsetslook up remote changestbasesRYR�(RORR:R�R�R�R�(R	R�RYR�R0((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR�
s





cCs�|jd
t
d��
i}|dk	r;t|�|d<n|dk	rZt|�|d<n|jd
|�}tj|j|�d�S(Nt	getbundleslook up remote changesRYtcommonR�(RORRR:R�R�R�R�(R	tsourceRYR�RR0((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR�
s







cCs
|d
g
krQ|jd�rQt
dtjdjt|���j�g�}nt
|�}|jd|
d|�
\}}|dkr�tj	t
d�|��nyt|�}Wn)tk
r�


tj	t
d�|��n
Xx0|j
t�D]}|jjt
d	�|�
q�|S(
s
Send cg (a readable file-like object representing the
        changegroup to push, typically a chunkbuffer object) to the
        remote server as a bundle. Return an integer indicating the
        result of the push (see localrepository.addchangegroup()).tforcetunbundlehashthashedttunbundleRYspush failed:s"push failed (unexpected response):sremote: (RvR:Rtsha1R9tsortedtdigestt	_callpushRRXRRRRWR]RzRxR{(R	tcgRYR�tretR|R6((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR�&
s 

-










cCsTi}|dk	r||d
<n|dk	r8||d<n|j
dd|
d||�S(Ntthreetfourt
debugwireargstonettwo(RRJ(R	R�R�R�R�tfiveR((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR�@
s






N(RRRER)R.RRMRYR\RbRfRlRrRR�R�R�RR�R�R�(((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRD�s"			

						
	t	streamresc
BseZ
d�ZRS(
cCs
|
|_dS(
N(
tgen(R	R�((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRL
s
(RRR(((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR�K
s
tpushresc
BseZ
d�ZRS(
cCs
|
|_dS(
N(
tres(R	R�((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRP
s
(RRR(((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR�O
s
tpusherrc
BseZ
d�ZRS(
cCs
|
|_dS(
N(
R�(R	R�((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRT
s
(RRR(((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR�S
s
tooberrorc
BseZ
d�ZRS(
cCs
|
|_dS(
N(
tmessage(R	R�((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRX
s
(RRR(((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR�W
s
cCs>|jd
�}t
|\}}|
j|�}|||
|�S(Ntserved(tfilteredtcommandstgetargs(trepotprototcommandtfunctspecR((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pytdispatch[
s



cCski}x2|
D]*}||kr
||||<||=q
q
W|rgtj
jd
|dj|�f�
n|S(Ns&abort: %s got unexpected arguments %s
R<(tsyststderrtwriteR9(tcmdtkeystothersRR}((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pytoptionsa
s









cCs�|jd
�}g}xx
|j
d�D]g
}|j
dd�\}}i}xE|j
d�D]4}	|	rY|	j
d�\}
}t|�||
<qYqYWt|\}}
|
rW
|
j
�}i}xo|D]g}|dkr
i}x1|j�D]#}||kr�|||<q��||d<q�||||<q�W|||
g|D]}||^q;
�}n|||
�}t|t�ry
|S|jt|��
q%Wdj	|�S(NR�R=R3i
R<R>t
*(
R�R5RBR�R�t
isinstanceR�RR@R9(R�R�RHR�R�tpairRKRtvalst
aRhR~R�R�R�RVR}tstarRNtresult((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyREl
s6






















,



cCsog|jd
�D]}t
|d�^q}g}x.|j|�D]}|jt|�d�
qAWdj|�S(NR3Rks

R�(R5R8RlRR:R9(R�R�RmRGR-Ri((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRl�s

+



cCsw|j�}g}xU|j
�D]G\}}tjtj|��}t|�}|jd
||f�
qWdj|�S(Ns%s %ss

(	RbRIR_tquoteRPRQR:RR9(R�R�RbRYtbranchRZRdtbranchnodes((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRb�s






cCsPt|�}g}x.|j
|�D]}|jt|�d
�
q"Wdj|�S(Ns

R�(R8RfRR:R9(R�R�RZR-Ri((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRf�s





cCs�d
j�}t
|j�r�|jjddt�rC|jd�
n|j|j@}|td�ss|jd�
q�|jddj	|��
n|jd	dj	t
j��
|jd
�
dj	|�S(
NsMlookup changegroupsubset branchmap pushkey known getbundle unbundlehash batchtservertpreferuncompressedsstream-preferredtrevlogv1tstreams
streamreqs=%sR<sunbundle=%sshttpheader=1024R3(
R�(R5t_allowstreamRxt
configboolRwRtrequirementstsupportedformatsR
R9R�tbundlepriority(R�R�tcapstrequiredformats((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pytcapabilities�s








cCs1t|�}|j
|d
�}t|
j|��S(Ntserve(R8R�R�tgroupchunks(R�R�R�RZR�((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR��s


cCs@t|�}t|�}|j
||d
�}t|
j|��S(NR�(R8R�R�R�(R�R�R�RYR�((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR��s



cCs+td
ddg|�}|j
|||�S(NR�R�R�(R�R�(R�R�R�R�R�R((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR��s
cCsjtd
ddg|�}x*|j
�D]\}}t|�||<q%W|jd|�}t|
j|��S(NR�RYR�R�(R�RIR8R�R�R�(R�R�R�RR}R~R�((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR��s





cCs|j�}t
|�d
S(Ns

(RYR:(R�R�t
h((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRY�s

cCsd
t||
�S(s
the hello command returns a set of lines describing various
    interesting things about the server, in an RFC822-like format.
    Currently the only one defined is "capabilities", which
    consists of a line in the form:

    capabilities: space separated list of tokens
    scapabilities: %s
(
R�(R�R�((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pythello�scCsf|jt
j|��j�}d
jg|D].\}}dt
j|�t
j|�f^q+�}|S(Ns

s%s	%s(RRPR^titemsR9RQ(R�R�RsRTR}R~t
t((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR�s

	
;
cCsey/tj
|�}||}|j�}d
}Wn%tk
rV
}
t|�}d}n
Xd||fS(Ni
is%s %s
(RPR^Rt	Exceptiontstr(R�R�RNR}t
cR-RUtinst((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRM�s











cCs&d
jd�|j
t|��D��S(NR�c
ss!|]}
|
rdpd
V
qdS(t
1t
0N((RFRi((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pys	<genexpr>�s(R9R\R8(R�R�RZR�((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR\�s
cCsD
t|�d
kr^|j
d�|kr^y |jd�
tj|�}Wqmtk
rZ


qmXntj|�}tj|
d�r
|
j�
y=|j	tj|�tj|�tj|�|�p�t
}Wntjk
r�

t
}n
X|
j�}dt
|�|fS|j	tj|�tj|�tj|�|�}dt
|�S(Nis
string-escapesutf-8trestores%s
%ss%s
(RotencodetdecodeRPR^tUnicodeDecodeErrorRRtredirectRrRwtAbortR�RR(R�R�RsRNRtRuR-R|((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyRr�s&'













c

Cs|jd
dt
dt
�
S(NR�tuncompressedt	untrusted(R�Rz(
Rx((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR�s
cs�t|j
�sd
Sg}d�y{|j�}z]|j
jd�
xF|jj�D]5\}}}|rQ|j||f�
�|7�qQqQWWd|j�
XWntj	k
r�


dSX�f
d�}t
|||���S(s
If the server supports streaming clone, it advertises the "stream"
    capability with a value representing the version and flags of the repo
    it is serving. Client checks to see if it understands the format.

    The format is simple: the server writes out a line with the amount
    of files, then the total amount of bytes to be transferred (separated
    by a space). Then, for each file, the server first writes the filename
    and filesize (separated by the null character), then the file contents.
    s1
is	scanning
Ns2
c3sP
d
V
|jj
dt|
��f�
dt|
��fV
|j}|j}|jj}t|_y�x�|
D]�\}}|r�|jj
d||f�
ndtj|�|fV
|dk
r�|�}z|j	|�}	Wd|j
�
X|	V
qkx(tj||�d|�
D]}
|
V
q
WqkWWnt
k
rB



||_�n
X||_dS(	s,stream out all metadata files in repository.s0
s%d files, %d bytes to transfer
s%d %d
ssending %s (%d bytes)
s%s%d
i
Ntlimit(RxRyRotsopenert	mustauditt	debugflagRwtstoret	encodedirtreadtcloseRt
filechunkiterR�(R�tentriesttotalR�toldauditR�RtsizetfpRVtchunk(
ttotal_bytes(s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pytstreamer,s2


	
	

	







"


	

(R�RxtlockRyR�twalkRtreleaseRt	LockErrorR�(R�R�R�R�RtenameR�R�((
R�s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR�s"










"csA
t|����fd
�}|
j
�
|�s;td�Stjdd�
\}}tj|d�}d}z�|
j|�
�j�}z�|�s�td�S|j	d�
t
j|d�}	y�j
|	d|
j��}Wn*tjk
r

}

tjjd	|
�
n
XWd|j�
Xt|�SWd|j�
tj|�
XdS(
Ncs[�j�}t
jd
jt|���j�}
�dg
kpZ�|kpZ�d|
gkS(NR�R�R�(RYRR�R9R�R�(RYt
heads_hash(R�ttheir_heads(s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pytcheck_headsSs

$

s=repository changed while preparing changes - please try againtprefixshg-unbundle-swb+is=repository changed while uploading changes - please try againR�s
abort: %s
(R8R�R�ttempfiletmkstemptostfdopentgetfileR�tseekR�t
readbundleRtaddchangegroupt_clientRR�R�R�R�R�R�R�tunlink(R�R�RYR�tfdttempnameR�R-R�R�R�((R�R�s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyR�Ps0

	









	








scmds *RERmRlR�RbRZRfR�R�R�sbases headsR�s	one two *R�R�R�RYR�snodes *R\RsRRNRMsnamespace key old newRrR�R�(5R_R�R�R�ti18nRtnodeR
RR�R�tpeerRRPRR�tobjectRRRRRR8R:R@RBRCtpeerrepositoryRDR�R�R�R�R�R�RERlRbRfR�R�R�R�RYR�RRMR\RrR�R�R�R�(((s7/sys/lib/python2.7/site-packages/mercurial/wireproto.pyt<module>sr0



<	"	&				�														
						B	.