�ڧOc@sd
dd
lm
Z

ddlmZ
ddlZddlZddlZddlZddlZddl	Z	ddl
Z
dZd�Zd�Z
d�Zd�Zd	ef
d
��YZidefd6defd6d
d�fd6dd�fd6ZdddgZd�Zd�Zdef
d��YZdef
d��YZd�Zdef
d��YZdS(i�(
t
_(
tnullrevNt20s20s20s20scCsM|j|
�}t
|�|
krItjtd
�t
|�|
f��n|S(s=read n bytes from stream.read and abort if less was availables5stream ended unexpectedly (got %d bytes, expected %d)(treadtlentutiltAbortR(tstreamt
nt
s((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pytreadexactlys



c
Csht|d
�}
t
jd|
�d}|d
k
rW|rStjtd�|��ndSt||d
�S(s-return the next chunk from stream as a stringis>lisinvalid chunk length %dt(R
tstructtunpackRRR(Rt
dt
l((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pytgetchunks





c

Cstj
d
|d�S(s*return a changegroup chunk header (string)s>li(Rtpack(
tlength((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pytchunkheader"scCstj
d
d�S(sBreturn a changegroup chunk header (string) for a zero-length chunks>li(RR(((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyt
closechunk&st
nocompressc
BseZ
d�Zd
�ZRS(c
Cs|
S(
N((tselft
x((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pytcompress+s
c


Csd
S(NR((
R((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pytflush-s
(t__name__t
__module__RR(((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyR*s
	RtHG10UNtHG10c
Cs
tj
�S(
N(tbz2t
BZ2Compressor(((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyt<lambda>5stHG10BZtHG10GZc
Cs
tj
�S(
N(tzlibtcompressobj(((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyR 6scCs�d
}d
}z{
|
r't
|
d
�}n0tjdddd�\}}
tj|d
�}|
}t|\}}|j|�
|�}t}	d}
x�|	s�|
dk
rh
t	}	|
d7}
x�t	rK
t
|�}|s�Pnt}	|j|jtt
|����
d}xC|t
|�krG
|d}
|j|j|||
!��
|
}q
Wq�W|j|jt���
q�W|j|j��
d
}|
SWd
|d
k	r�|j�
n|d
k	r�tj|�
nXd
S(s�Write a bundle file and return its filename.

    Existing files will not be overwritten.
    If no filename is specified, a temporary file is created.
    bz2 compression can be turned off.
    The bundle file will be deleted in case of errors.
    twbtprefixs
hg-bundle-tsuffixs.hgiii
iNi(tNonetopenttempfiletmkstemptostfdopentbundletypestwritetFalsetTrueRRRRRRtclosetunlink(tcgtfilenamet
bundletypetfhtcleanuptfdtheadert
compressort
ztemptytcounttchunktpostnext((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pytwritebundle<sF	








	





	




"













cCsf|
d
kr|S|
dkr(d�}n+|
dkr@d�}ntj
d|
��tj||��S(NtUNtGZc
ss8tj
�}
x%tj|�D]}|
j|�V
qWdS(
N(R#t
decompressobjRt
filechunkitert
decompress(t
ftzdR?((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyt	generatorws


tBZc
ssHtj
�}
|
jd
�
x(tj|d�D]}|
j|�V
q,WdS(NRKi(RtBZ2DecompressorRGRRF(RHRIR?((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyRJ|s




sunknown bundle compression '%s'(RRtchunkbuffer(R7talgRJ((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pytdecompressorss





t
unbundle10cBs�eZ
eZeje�Zd�Zd
�Zd�Z	d�Z
d�Zd�Zd�Z
d�Zd�Zd	�Zd
�Zd�ZRS(cCs(t|
|�|_
||_d|_dS(
N(ROt_streamt_typeR(tcallback(RR7RN((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyt__init__�s

	
c

Cs
|jd
kS(NRC(
RR(
R((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyt
compressed�s
cCs|jj
|
�S(
N(RQR(RR((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyR�s
cCs|jj
|
�S(
N(RQtseek(RR@((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyRV�s
c


Cs
|jj
�S(
N(RQttell(
R((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyRW�s
c


Cs
|jj
�S(
N(RQR2(
R((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyR2�s
c
Csxt|j
d
�}
tjd|
�d}|d
k
rZ|rVtjtd�|��ndS|jrp|j�
n|d
S(Nis>lisinvalid chunk length %d(R
RQRR
RRRRS(RRR((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pytchunklength�s






	


c


CsiS(
s*v10 does not have a changelog header chunk((
R((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pytchangelogheader�sc


CsiS(
s)v10 does not have a manifest header chunk((
R((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pytmanifestheader�sc
Cs5|j�}
|
siSt
|j|
�}td
|�
S(sBreturn the header of the filelogs chunk, v10 only has the filenameR5(RXR
RQtdict(RRtfname((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyt
filelogheader�s




cCs@|
\}}}}|dkr'|}n|}|||||fS(
N(
R((Rtheadertupletprevnodetnodetp1tp2tcst	deltabase((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyt_deltaheader�s



	
c
Cs�|j�}|siSt
|j|j�}tj|j|�}t
|j||j�}|j||
�\}}}}	}
td
|d|d|d|
d|	d|�S(NR`RaRbRcRdtdelta(	RXR
RQtdeltaheadersizeRR
tdeltaheaderReR[(RR_Rt
headerdataR:RfR`RaRbRdRc((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyt
deltachunk�s






!

(RRt_BUNDLE10_DELTA_HEADERRhRtcalcsizeRgRTRURRVRWR2RXRYRZR]ReRj(((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyRP�s


											theaderlessfixupc
BseZ
d�Zd
�ZRS(cCs||_|
|_
dS(
N(t_ht_fh(RR7t
h((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyRT�s
	
cCsp|jr`|j|
 |j|
}|_t
|�|
kr\|t|j|
t
|��7}n|St|j|
�S(
N(RnRR
Ro(RRR((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyR�s
	


#

(RRRTR(((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyRm�s
	cCs�t|d
�}|
sUd}
|j
d�rU|j
d�rUt||�}d}qUn|dd!|dd!|dd
!}}}|dkr�tjtd	�|
��n|d
kr�tjtd�|
|f��nt||�S(NiRtHGt
Riiis%s: not a Mercurial bundlet10s%s: unknown bundle version %s(R
t
startswithRmRRRRP(R7R\R:tmagictversionRN((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyt
readbundle�s




)


"
tbundle10c
Bs;eZ
eZd�Zd
�Zd�Zd�Zd�ZRS(cCs
|
|_dS(
N(
t_lookup(Rtlookup((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyRT�s
c


Cst�S(
N(
R(
R((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyR2�s
cCstt
|
��|
S(
N(RR(RR\((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyt
fileheader�s
ccs�|
j|�}|
j
|�\}}|}d
}|tkrc|
j|�}	tjt|	��}n|
j||�}	|j|
|�}
|
j	|�\}}|
j|�}
|j
||||
|
�}||7}t|�t|	�}t|�V
|V
|	V
dS(NR(R`t
parentrevsR
trevisiontmdiffttrivialdiffheaderRtrevdiffRytparentstbuilddeltaheaderR(RtrevlogtrevtprevR`RaRbtbaseR&Rftlinknodetp1ntp2ntbasenodetmetaR((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pytrevchunk�s"















cCstj
|j|
|||�S(
N(RRRh(RR`R�R�R�R�((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyR��s(	RRRkRhRTR2R{R�R�(((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyRx�s

				(ti18nRR`R
R~RRR,RR#R*RkR
RRRtobjectRR.tbundlepriorityRBRORPRmRwRx(((s9/sys/lib/python2.7/site-packages/mercurial/changegroup.pyt<module>s*


<			
		




	7	@