HEX

File: //lib/python3/dist-packages/hgext/remotefilelog/__pycache__/shallowbundle.cpython-310.pyc
o

�]Lb�+�@s�ddlmZddlmZddlmZmZddlmZm	Z	m
Z
mZmZm
Z
ddlmZmZmZdZdZdZdd	d
�ZGdd�de	j�Zd
d�Zdd�ZdS)�)�absolute_import)�_)�bin�hex)�
bundlerepo�changegroup�error�match�mdiff�pycompat�)�	constants�
remotefilelog�shallowutil�Nc
cs��t|tj�st||�j||||d�D]}|VqdSt|�dkr(|��VdSt�||j�}|�|d�d}|�	d|�t
�t|�d�D]}	||	||	d}
}||�}|�|||
|�D]}|Vq`qG|��VdS)N��unitsrr)
�
isinstancer�super�group�len�closer�	sortnodes�parents�insertr�xrange�	nodechunk)
�cls�self�nodelist�rlog�lookupr�reorder�c�p�i�prev�curr�linknode�r)�C/usr/lib/python3/dist-packages/hgext/remotefilelog/shallowbundle.py�shallowgroups$�
�r+csNeZdZ�fdd�Zddd�Z�fdd�Zdd	�Z�fd
d�Zdd
�Z�Z	S)�shallowcg1packercs.t�|j�rd}tt|�j||||fi|��S)NF)r�	isenabled�_reporr,�generate)r�
commonrevs�clnodes�fastpathlinkrev�source�kwargs��	__class__r)r*r/9s
��zshallowcg1packer.generateNcCstt|||||d�S)Nr)r+r,)rrr r!rr"r)r)r*rAs�zshallowcg1packer.groupc	s�z|\}}}Wnty|\}}}}}	}
Ynwt�|j�rO|j�t�tj�r;tj��d��jr9t	j
ddd��gS|�|�}|tkrOt
�fdd�|D��}tt|�j|g|�Ri|��S)Nsfooscannot pull from full bundlessuse `hg unbundle` instead)�hintcsg|]	}��|�s|�qSr))�shallowmatch)�.0�f��repor)r*�
<listcomp>\sz2shallowcg1packer.generatefiles.<locals>.<listcomp>)�
ValueErrorrr-r.rr�bundlerepository�file�_cgfilesposr�Abort�shouldaddfilegroups�NoFiles�listrr,�
generatefiles)r�changedfiles�argsr4�	linknodesr0r3�mfdictsr2�fnodes�clrevs�filestosendr5r;r*rFFs6��
�
���zshallowcg1packer.generatefilescCs�|j}t�|�s
tS|dks|dkrtS|dkr+|j�dd�}|dkr%tS|dkr+tS|jp/g}|dks8|d	krItj	|vr?tS|j�
td
��tStS)Nspushsbundlesstrips
remotefilelogsstrip.includefilesslocalsnone�servespullspulling from a shallow repo
)
r.rr-�AllFiles�ui�config�
LocalFiles�_bundlecapsr
�BUNDLE2_CAPABLITY�warnrrD)rr3r<�cfg�capsr)r)r*rCcs$


z$shallowcg1packer.shouldaddfilegroupscs^t|tj�stt|��|||�S|j}g}|D]}|j|j|d�}|��|vr,|�	|�q|S)N)�fileid)
rrrr,�pruner.�filectx�filename�linkrev�append)rr �missingr0r<�results�fnode�fctxr5r)r*rY�s
�
�zshallowcg1packer.pruneccs��d}||jkr|�|�}t�t|��}n|�||�}|�|�\}}|�|�}	|�||||||	�}
|
|7}
t|
�t|�}t	�
|�V|
V|VdS)N�)�nullid�rawdatar
�trivialdiffheaderr�revdiffr�flags�builddeltaheaderr�chunkheader)r�revlog�node�prevnoder(�prefix�delta�p1�p2rg�meta�lr)r)r*r�s�



zshallowcg1packer.nodechunk�NN)
�__name__�
__module__�__qualname__r/rrFrCrYr�
__classcell__r)r)r5r*r,8s
r,c
Os�t�|�s|||||g|�Ri|��S|j}zd|dkri|�d�}d}	d}
|p'gD]+}|�d�r?|td�d�}|r>|�d�}	q(|�d�rS|td�d�}|rS|�d�}
q(|	sX|
rdt�|jdd|	|
�|_nt�	�|_|||||g|�Ri|��W||_S||_w)NrN�
bundlecapssincludepattern=�sexcludepattern=rb)
rr-r8�get�
startswithr�splitr	�root�always)
�origr<�outgoing�versionr3rHr4�originalrx�includepattern�excludepattern�cap�rawr)r)r*�makechangegroup�s4



�

��
r�c%	st���s|�||||g|�Ri|��Sd}t�}	i�g��jjtd�|d�}
	|��}|s/n\|d}�j�d|�|
����	|�sU��
|�}
|��}|
�|||�q'd}	|�
|�}|s`n|d}|�||f<��||f�||	vr~|d7}|	�|�qX|dur�t�td���q(t������fd	d
�}d}g}�D]/\}}�||f}|d|d|dg}|D]}|�jks�||f�vr�q�|�|t|�f�q�q��j�|���rw��d�\}}||f�vr�q�|d7}|t��dkr�t�td
�����
|�}
�||f}|\	}}}}}}}}}|||||��sq�|
�|�}t�||�} t| t��s+t| �} t�| �\}!} d|!v�rJ|!d}"t|!d�}#||||"|#��sJq�||fD]}$|$�jk�r`|||||$��s`�qN�qN|
�| |!||||���||f�d}�s�|
��t��|fS)Nrsfiles)�totalTsfilenamesadding %s revisions
rs#received file revlog group is emptycsN|�jkr%||f�vr%||f�vrdS��d||f���d||f�dSdS)NTrF)rcr)r:rk�depf�depnode��	processed�queuer<�
revisiondatasr)r*�	availablesz&addchangegroupfiles.<locals>.availabler�scircular node dependencyscopyscopyrev) rr-�setrP�makeprogressr�
filelogheader�debug�	incrementr8r@�	deltaiter�addgroup�
deltachunkr]�addrrBrcr�fileservice�prefetch�poprrdr
�patchr�bytes�	parsemetar�complete)%rr<r3�revmap�trp�
expectedfilesrHr4�newfiles�visited�progress�	chunkdatar:�fl�deltas�chain�revisiondatar��	skipcount�
prefetchfilesrk�
dependents�	dependentrorpr(�	deltabasernrg�sidedata�proto_flags�base�textrq�copyfrom�copynoder$r)r�r*�addchangegroupfiles�s�
	



��+
�
�

��1r�rs)�
__future__r�mercurial.i18nr�mercurial.noderr�	mercurialrrrr	r
r�r
rrrDrRrOr+�cgpackerr,r�r�r)r)r)r*�<module>s 
j