|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/mercurial/__pycache__/ |
Upload File : |
o
�]Lb� � @ s� d dl mZ d dlZd dlZd dlZddlmZ ddlmZm Z m
Z
ddl m Z ddl
mZmZmZmZm Z mZmZmZ ddlmZ dd lmZ dd
lmZ dd l m Z e� d �Z e� d
�Z!e� d�Z"e� d�Z#dZ$ej%Z%dd� Z&dd� Z'dd� Z(dd� Z)dOdd�Z*G dd � d e+�Z,G d d � d e,�Z-G d d � d e-�Z.G d!d"� d"e.�Z/G d#d$� d$e+�Z0d%d&� Z1d'd(� Z2d)d*� Z3 + dPd,d-�Z4G d.d/� d/e+�Z5 + + dQd0d1�Z6 + + dQd2d3�Z7 + + dQd4d5�Z8 + + dQd6d7�Z9e6e,fe7e-fe8e.fe9e/fd8�Z:d9d:� Z;d;d<� Z<d=d>� Z=d?d@� Z>dAdB� Z? + + dRdCdD�Z@dOdEdF�ZAdGdH� ZB dSdIdJ�ZC + dTdKdL�ZD dOdMdN�ZEdS )U� )�absolute_importN� )�_)�hex�nullrev�short)�open)�error�match�mdiff�phases�pycompat� requirements�scmutil�util)�
repository)�sidedata)� constants)� storageutils 20s20s20s20ss 20s20s20s20s20ss >20s20s20s20s20sHs >B20s20s20s20s20sHs lfsc C sJ t | d�}t�d|�d }|dkr |r t�td�| ��dS t | |d �S )z-return the next chunk from stream as a string� � >lr � invalid chunk length %d� )� readexactly�struct�unpackr �Abortr )�stream�d�l� r �7/usr/lib/python3/dist-packages/mercurial/changegroup.py�getchunk0 s
r" c C s t �d| d �S )z*return a changegroup chunk header (string)r r �r �pack)�lengthr r r! � chunkheader; � r& c C s t �dd�S )zBreturn a changegroup chunk header (string) for a zero-length chunkr r r# r r r r! �
closechunk@ s r( c C s t t| ��| S )z3Obtain a changegroup chunk header for a named path.)r& �len)�pathr r r! � _fileheaderE r' r+ c C s� d}d}zN|r|r|� |d�}nt |dd�}ntjddd�\}}t�|d�}|}|D ]}|�|� q*d}|W |dur>|�� |durR|rL|rL|�|� S t�|� S S |dur[|�� |duro|ri|ri|�|� w t�|� w w )z�Write chunks to a file and return its filename.
The stream is assumed to be a bundle file.
Existing files will not be overwritten.
If no filename is specified, a temporary file is created.
Ns wbi s
hg-bundle-s .hg)�prefix�suffix�wb)r r
�mkstemp�os�fdopen�write�close�unlink)�ui�chunks�filename�vfs�fh�cleanup�fd�cr r r! � writechunksJ s8 �� �r= c @ s� e Zd ZdZeZejZdZdZ d%dd�Z
dd� Z d d
� Z d d � Z
d
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd%d d �Zejddfd!d"�Zd#d$� ZdS )&� cg1unpackera� Unpacker for cg1 changegroup streams.
A changegroup unpacker handles the framing of the revision data in
the wire format. Most consumers will want to use the apply()
method to add the changes from the changegroup to a repository.
If you're forwarding a changegroup unmodified to another consumer,
use getchunks(), which returns an iterator of changegroup
chunks. This is mostly useful for cases where you need to know the
data stream has ended by observing the end of the changegroup.
deltachunk() is useful only if you're applying delta data. Most
consumers should prefer apply() instead.
A few other public methods exist. Those are used only for
bundlerepo and some debug commands - their use is discouraged.
� 01r Nc C sh |d u rd}|t jjvrt�td�| ��|dkrd}t j�|�}|�|�| _|| _ |p-i | _
d | _ d S )N� UNs# unknown stream compression type: %ss BZs _truncatedBZ) r � compengines�supportedbundletypesr r r �
forbundletype�decompressorreader�_stream�_type�extras�callback)�selfr9 �algrG �
compenginer r r! �__init__� s
zcg1unpacker.__init__c C s | j d uo | j dkS )Nr@ )rF �rI r r r! �
compressed� s zcg1unpacker.compressedc C � | j �|�S �N)rE �read)rI r r r r! rQ � � zcg1unpacker.readc C rO rP )rE �seek)rI �posr r r! rS � rR zcg1unpacker.seekc C �
| j �� S rP )rE �tellrM r r r! rV � �
zcg1unpacker.tellc C rU rP )rE r3 rM r r r! r3 � rW zcg1unpacker.closec C sT t | jd�}t�d|�d }|dkr |r t�td�| ��dS | jr&| �� |d S )Nr r r r )r rE r r r r r rH )rI r r r r r! � _chunklength� s zcg1unpacker._chunklengthc C � i S )z*v10 does not have a changelog header chunkr rM r r r! �changelogheader� � zcg1unpacker.changelogheaderc C rY )z)v10 does not have a manifest header chunkr rM r r r! �manifestheader� r[ zcg1unpacker.manifestheaderc C s$ | � � }|si S t| j|�}d|iS )zBreturn the header of the filelogs chunk, v10 only has the filename� filename)rX r rE )rI r �fnamer r r! �
filelogheader� s
zcg1unpacker.filelogheaderc
C s8 |\}}}}|d u r
|}n|}d}d} ||||||| fS �Nr r )
rI � headertuple�prevnode�node�p1�p2�cs� deltabase�flags�protocol_flagsr r r! � _deltaheader� s zcg1unpacker._deltaheaderc
C sp | � � }|si S t| j| j�}| j�|�}t| j|| j �}| �||�}|\}}}} }
} } ||||
| || i | f S rP )rX r rE �deltaheadersize� deltaheaderr rj )
rI rb r �
headerdata�header�deltarc rd re rg rf rh ri r r r! �
deltachunk� s zcg1unpacker.deltachunkc c s� � d}|d| j k rUd} t| �}|s#|dk r |d7 }n|r"|d7 }n%d}tt|��V d}|t|�k rG|d }|||� V |}|t|�k s4q
t� V |d| j k s
dS dS )a" returns all the chunks contains in the bundle
Used when you need to forward the binary stream to a file or another
network API. To do so, it parse the changegroup data, otherwise it will
block in case of sshrepo because it don't know the end of the stream.
r � Tr Fi N)�_grouplistcountr" r& r) r( )rI �parts� noentries�chunkrT �nextr r r! � getchunks� s. �
���zcg1unpacker.getchunksc C sH |j | _| �� | �� }|j�d�}|j||||d� |�� d | _d S )Nr ��
addrevisioncb)� incrementrH r\ � deltaiter� manifestlog�
getstorage�addgroup�complete)rI �repo�revmap�trp�progry �deltas�storager r r! �_unpackmanifests s
z cg1unpacker._unpackmanifestsc ( s� �� � �tj�jv ptj�jv o| jdko|dk}|r'tj�|p"t� dd�} nd} ��fdd�}
�fdd �} �z�|j� d
|�}|j� d |� �j
d4d
dit � |j��� �j
���|� t��� ��t�|�} �j�td�� t����jjtd�td�|d�}
|
j| _t� �g ���fdd�}��fdd�}| �� | �� }�j||
| d||d�s��jj ddd� t��}t��}|� }|
� � ~d�d| _i �
i � �j�td�� �jjtd�td�|d�}
d}| r�t j | d v r�
fd d �}| j �| | |
|d � i }�j�!d d!��r8�j
��j"}t �#�|�D ]$}��$|�j%}|| �&� }t �'|�D ]\}}|� |t� ��(|� �q&�qd} | �rKt j)| d v �rK� fd"d#�} �j�td$�� t*�| | | ||| d �\} } | �r�t j+| d v �ru��,| | �|d � �
�-� D ]\} \} }!| �,| | | |!� �qy� �-� D ]\}"\} }!|"�,| | | |!� �q�|j.� d%d&� |j.� d'd&� |j.� d(d&� |j.� d)d&� �jj/�s�|j.d% |7 < |j.d' | 7 < |j.d( | 7 < d&}#��r��� }$|#t|$�t�� 7 }#|$D ]}%|%�v�r �|% �0� �r |#d 8 }#�q�jj/�s|j.d) |#7 < ��1� |d&k�rbd*|jv�r;t2��3���|jd*< t2��3|d ��|jd+< t4|j��nt4|j��t2��3����d*< t2��3|d ���d+< �j
,d5d
dit � ���� t �#�|�� d}&|d-v �r��5� �rzt6j7 }}&nt6j8 }}&� �r�t6�9�||� � |&du�r���r���:� � n� �t6j;�||&g �d.� g �|d&k�rÇ �����fd/d0��|�<d1� ��fd2d3�� W �j�=� n�j�=� w |#d&k �r�|#d }'|'S |#d }'|'S )6a? Add the changegroup returned by source.read() to this repo.
srctype is a string like 'push', 'pull', or 'unbundle'. url is
the URL of the repo where this changegroup is coming from.
Return an integer summarizing the change to this repo:
- nothing changed or no source: 0
- more heads than before: 1+added heads (2..n)
- fewer heads than before: -1-removed heads (-2..-n)
- number of heads stays the same: 1
`sidedata_categories` is an optional set of the remote's sidedata wanted
categories.
� 04s pullT)�pullNc s �j �dt| � � t� �S )Ns add changeset %s
)r5 �debugr r) ��x��clr� r r! �csmap; s z cg1unpacker.apply.<locals>.csmapc s
� � | �S rP ��revr� �r� r r! r� ? rW z!cg1unpacker.apply.<locals>.revmaps sources url� prechangegroup�throws adding changesets
�
changesets� chunks��unit�totalc s |� k r �� |� d S d S rP )�append)r� r� )�clstart�duprevsr r! �ondupchangelog_ s �z)cg1unpacker.apply.<locals>.ondupchangelogc s2 | � |�}� d us J �� �|j� ��||� d S rP )�changelogrevision�update�files�register_changeset)r� r� �ctx)� efilessetr� r r! � onchangelogc s
z&cg1unpacker.apply.<locals>.onchangelog)� alwayscachery �duplicaterevisioncbs( applied empty changelog from changegroups warn-empty-changegroup)�configs adding manifests
� manifestsr c �F � � | �}|s||f� | < d S ||d d ksJ �|d |f� | < d S �Nr r ��get)�manifestr� �range)�touched_manifestsr r! �on_manifest_rev� �
z*cg1unpacker.apply.<locals>.on_manifest_revrx s servers validatec r� r� r� )�filelogr� r� )�touched_filelogsr r! �on_filelog_rev� r� z)cg1unpacker.apply.<locals>.on_filelog_revs adding file changes
s changegroup-count-changesetsr s changegroup-count-revisionss changegroup-count-filess changegroup-count-heads� node� node_last� pretxnchangegroup)s pushs serve)�revsc s� �t ��krd S �jd
i t����� � D ] }��� }t��|��|d< |d= �jd i t�|��� q�fdd���� D �}�j� ddt � �d�
d d� |D ��� d S ) N� changegroupr� r� � incomingc s g | ]}|� vr|�qS r r )�.0�h)�oldheadsr r! �
<listcomp>/ � z7cg1unpacker.apply.<locals>.runhooks.<locals>.<listcomp>s$ %d incoming changes - new heads: %s
s , c S s g | ]
}t |d d� ��qS )N� )r �r� r< r r r! r� 4 s )r� )r� ) r) �hookr
� strkwargs�copyr rc �headsr5 �log�join)�unused_successr� �args�newheads)�addedr� r� �hookargsr� r� r r! �runhooks s �z#cg1unpacker.apply.<locals>.runhookss changegroup-runhooks-%020ic s
� � ��S rP )�
_afterlock)�tr)r� r� r r! �<lambda>9 �
z#cg1unpacker.apply.<locals>.<lambda>)r� )r� )>�
unfilteredr �REVLOGV2_REQUIREMENT�CHANGELOGV2_REQUIREMENT�version� sidedatamod�get_sidedata_helpers�setr� �
setdefaultr� r
r� � changelog� delayupdater� �weakref�proxyr5 �statusr r) � makeprogressrz rH rZ r{ r~ � develwarnr �revlog_constants�KIND_MANIFESTLOGr� �
configboolr| �xranger� r� � readdelta� iteritems�add� KIND_FILELOG�_addchangegroupfiles�KIND_CHANGELOG�rewrite_sidedata�items�changes�quiet� closesbranch�invalidatevolatilesetsr rc �dict�
publishingr �public�draft� registernew�extend�advanceboundary� addpostclose�flush)(rI r� r� �srctype�url� targetphase�
expectedtotal�sidedata_categories�adding_sidedata�sidedata_helpersr� r� r� �progressr� r� r� �efiles�clend�
changesetsr� � needfiles�ml�cset�mfnode�mfest�f�nr� �newrevs�newfiles�mf�startrev�endrev�fl�
deltaheadsr� r� �phaseall�retr ) r� r� r� r� r� r� r� r� r� r� r� r! �apply sZ
���� ��
�
���� � �
�
�
�
���
��
�zcg1unpacker.applyc # s: � d� t � �fdd�i �D ]
}|dd� V |d � q
dS )z�
returns an iterator of the deltas in this changegroup
Useful for passing to the underlying storage system to be stored.
Nc s
�� � �S rP )rp r ��chainrI r r! r� K r� z'cg1unpacker.deltaiter.<locals>.<lambda>� r )�iter)rI � chunkdatar r r! r{ D s �
�zcg1unpacker.deltaiterrP ) �__name__�
__module__� __qualname__�__doc__�_CHANGEGROUPV1_DELTA_HEADERrl �sizerk r� rr rL rN rQ rS rV r3 rX rZ r\ r_ rj rp rw r� r r� r r{ r r r r! r> m s6
-
� 7r> c @ s& e Zd ZdZeZejZdZdd� Z dS )� cg2unpackerz�Unpacker for cg2 streams.
cg2 streams add support for generaldelta, so the delta header
format is slightly different. All other features about the data
remain the same.
� 02c
C s( |\}}}}}d}d} ||||||| fS r` r �
rI ra rb rc rd re rg rf rh ri r r r! rj ] s zcg2unpacker._deltaheaderN)
r r r r �_CHANGEGROUPV2_DELTA_HEADERrl r rk r� rj r r r r! r Q s r c s< e Zd ZdZeZejZdZdZ dd� Z
d � fdd� Z � Z S )
� cg3unpackerz�Unpacker for cg3 streams.
cg3 streams add support for exchanging treemanifests and revlog
flags. It adds the revlog flags to the delta header and an empty chunk
separating manifests and files.
� 03rq c
C s& |\}}}}}}d} ||||||| fS r` r r" r r r! rj q s zcg3unpacker._deltaheaderNc sx t t| �j|||||d� t| ji �D ]&}|d }|j�d| � | �� }|j� |�j
||||d�s9t � t
d���qd S )Nrx r] � adding %s revisions
s" received dir revlog group is empty)�superr$ r� r r_ r5 r� r{ r| r} r~ r r r ) rI r� r� r� r� ry r r r� �� __class__r r! r� v s
� ���z cg3unpacker._unpackmanifestsrP )
r r r r �_CHANGEGROUPV3_DELTA_HEADERrl r rk r� rr rj r� �
__classcell__r r r( r! r$ d s r$ c s6 e Zd ZdZeZejZdZdd� Z � fdd�Z
� Z S )� cg4unpackerzTUnpacker for cg4 streams.
cg4 streams add support for exchanging sidedata.
r� c
C s$ |\}}}}}}} |||||| |fS rP r )
rI ra rb ri rc rd re rg rf rh r r r! rj � s zcg4unpacker._deltaheaderc
sn t t| ��|�}|s |S |\ }}}}}}} }
} |
rJ �i }
| tj@ r,t| j�} t�| �}
||||||| |
| f S rP ) r' r, rp r �CG_FLAG_SIDEDATAr" rE r� �deserialize_sidedata)
rI rb �resrc rd re rf rg ro rh r ri � sidedata_rawr( r r! rp � s: �
�zcg4unpacker.deltachunk) r r r r �_CHANGEGROUPV4_DELTA_HEADERrl r rk r� rj rp r+ r r r( r! r, � s r, c @ s e Zd Zdd� Zdd� ZdS )�headerlessfixupc C s || _ || _d S rP )�_h�_fh)rI r9 r� r r r! rL � s
zheaderlessfixup.__init__c C sZ | j r'| j d |� | j |d � }| _ t|�|k r%|t| j|t|� �7 }|S t| j|�S rP )r3 r) r r4 )rI r
r r r r! rQ � s zheaderlessfixup.readN)r r r rL rQ r r r r! r2 � s r2 c c s� � |j dur
d|j }}n |j| jkr |j}t�t|��}n |j}t�|jt|��}||�}t t|�t|� t|� �V |V |rE|V |V |j
t j @ r]|j
}t t|��V |V dS dS )z0Serialize a revisiondelta to changegroup chunks.Nr )ro �basenode�nullid�revisionr �trivialdiffheaderr) �replacediffheader�baserevisionsizer& ri r r- r )r� ro �headerfnr, �data�metar r r r! �_revisiondeltatochunks� s& �
�r> c s � �fdd�}t ||d�S )z&Sort nodes for changegroup generation.c s � � �| ��S rP r� )r
�r� �lookupr r! r� � s z$_sortnodesellipsis.<locals>.<lambda>��key)�sorted)�store�nodesr� r@ rB r r? r! �_sortnodesellipsis� s rF c s� �| } � ������fdd��| r�� ��ttfkr!tt}
} n"t| �dkr6t�fdd�| D ��\}
t} n
t�fdd�| D ��\}
} ��|
���| �} }
| |
|fS )Nc s� | t krt S �r
| S | g}|rs|d }|dd� }|�v r!�| S |�v r3|�dd� � �|�D �� n>|�v rD|�dd� �| D �� n-t��dd�D ]
}��|�| krX| S qKt��d�rc�j}n�j j}t
� d |�| f ��|st S )
a� Turn a changelog revnum into a local revnum.
The ellipsis dag is stored as revnums on the changelog,
but when we're producing ellipsis entries for
non-changelog revlogs, we need to turn those numbers into
something local. This does that for us, and during the
changelog sending phase will also expand the stored
mappings as needed.
r r Nc S � g | ]}|t kr|�qS r �r �r� �ppr r r! r� + r� z=_resolvenarrowrevisioninfo.<locals>.local.<locals>.<listcomp>c S rG r rH rI r r r! r� . r� ����targets? unable to resolve parent while packing '%s' %r for changeset %r) r r� �
parentrevsr
r� �linkrevr � safehasattr�
display_id�_revlogr r )�clrev�walk�p�irL )r� �clrevtolocalrev� fullclnodes� ischangelog�precomputedellipsisr� rD r r! �local
s<
�� ���(z)_resolvenarrowrevisioninfo.<locals>.localr c 3 � � | ]}� |�V qd S rP r �r� rT �rZ r r! � <genexpr>R � � z-_resolvenarrowrevisioninfo.<locals>.<genexpr>c 3 r[ rP r r\ r] r r! r^ U r_ )rM r r) rC rc )r� rD rX r� rN �linknoderV rW rY � linkparentsrd re �p1node�p2noder )r� rV rW rX rZ rY r� rD r! �_resolvenarrowrevisioninfo� s E
rd Fc c s� � |sdS | j } |r
d}
n|rt||| |�}d}
nd}
|rhg }i }i }|D ]@}|�|�}||�}| �|�}|||< || v rB|||< n ||
vrGq%t| ||||||| |
� \}}}||f||< |||< |�|� q%|}d}|dur{| jj|td�t|�d�}| j� dd�}|dvr�td �}| j�
|| � t j }|r�t j
}n|d
kr�t j}n|d kr�t j}|j||
d | || d
�}t|�D ]:\} } |r�|�| d � |r�|| j }| j|v r�|| j \}}|| _|| _| jt jO _n|| j�}|| _| V q�|r�|�� dS dS )a6 Calculate deltas for a set of revisions.
Is a generator of ``revisiondelta`` instances.
If topic is not None, progress detail will be generated using this
topic name (e.g. changesets, manifests, etc).
See `revlogutil.sidedata.get_sidedata_helpers` for the doc on
`sidedata_helpers`.
Ns storages nodesr� r� s devels bundle.delta)r � p1� fulls0 config "devel.bundle.delta" as unknown value: %sre rf T)�
nodesorder� revisiondata�assumehaveparentrevisions� deltamoder� r )r� rF r� rd r� r5 r� r r) r� �warnr �CG_DELTAMODE_STD�CG_DELTAMODE_PREV�CG_DELTAMODE_P1�CG_DELTAMODE_FULL�
emitrevisions� enumerater� rc rb rc rh �REVISION_FLAG_ELLIPSISr` r ) r� rD rE rX r@ �forcedeltaparentprev�topic�ellipsesrV rW rY r� r� rg �filtered�adjustedparents� linknodesrc r� r` rN rb rc r � configtarget�msgrj � revisionsrU r7 r r r! �
deltagroup\ s� �
� ��
�
�r| c @ s\ e Zd Z ddd�Z ddd�Z ddd �Z dd
d �Zd d
� Z ddd�ZdS )�cgpackerFNc C s� |sJ �|sJ �|| _ || _|| _|| _|| _|| _| | _|du r$t� }|| _|
du r.t� }
|
| _ |
| _
| | _ | | _ || _
| j
jjrO| j
jjsO| j
jj| _dS dd� | _dS )a
Given a source repo, construct a bundler.
oldmatcher is a matcher that matches on files the client already has.
These will not be included in the changegroup.
matcher is a matcher that matches on files to include in the
changegroup. Used to facilitate sparse changegroups.
forcedeltaparentprev indicates whether delta parents must be against
the previous revision in a delta group. This should only be used for
compatibility with changegroup version 1.
builddeltaheader is a callable that constructs the header for a group
delta.
manifestsend is a chunk to send after manifests have been fully emitted.
ellipses indicates whether ellipsis serving mode is enabled.
bundlecaps is optional and can be used to specify the set of
capabilities which can be used to build the bundle. While bundlecaps is
unused in core Mercurial, extensions rely on this feature to communicate
capabilities to customize the changegroup packer.
shallow indicates whether shallow data might be sent. The packer may
need to pack file contents not introduced by the changes being packed.
fullnodes is the set of changelog nodes which should not be ellipsis
nodes. We store this rather than the set of nodes that should be
ellipsis because for very large histories we expect this to be
significantly smaller.
remote_sidedata is the set of sidedata categories wanted by the remote.
Nc S s d S rP r )�sr r r! r� ? s z#cgpacker.__init__.<locals>.<lambda>)� _oldmatcher�_matcherr� �_forcedeltaparentprev�_builddeltaheader�
_manifestsend� _ellipsesr� � _bundlecaps�_remote_sidedata�
_isshallow� _fullclnodes�_precomputedellipsis�_repor5 �verbose� debugflag�note� _verbosenote)rI r� �
oldmatcher�matcherr� �builddeltaheader� manifestsendrs �
bundlecapsru �shallow�
ellipsisroots� fullnodes�remote_sidedatar r r! rL � s, 2zcgpacker.__init__Tc
# s� � | j ��j� | �td�� d}d}| jdkr'| j}|dkr!�j}t��|�}| j � |||d�\} }
|
D ]} t
| j | | j �D ] } |t | �7 }| V q>q4t
� }
|t |
�7 }t
� V | �td�| � | d }| d }| d
}|ott��� }i }d}| j||||||| d |d �}|D ]A\}}
|r�| jd
v s�J �t|�} |t | �7 }| V |
D ]} t
| j | | j �}|D ] } |t | �7 }| V q�q�t
� }
|t |
�7 }|
V q�| �td�| � | jV d}| jr�| jr�fdd�t�|�D �}|�� � fdd�|D �}| j||||||||d �}|D ]D\}}
t|�}t |�}|V |
D ]} t
| j | | j �}|D ] } |t | �7 }| V �q&�qt
� }
|t |
�7 }|
V | �td�||f � �q t
� V |�rc�jdt|d �|d� dS dS )z�Yield a sequence of changegroup byte chunks.
If changelog is False, changelog data won't be added to changegroup
s% uncompressed size of bundle content:
r Nr� s strip)�generater� s %8.i (changelog)
�
clrevorderr� � changedfiles� clrevtomanifestrev)r� )r% r� s %8.i (manifests)
c s" g | ]
\}}� j | �� |f�qS r )r| rQ )r� r
�lr�r� r r! r� � s ��z%cgpacker.generate.<locals>.<listcomp>c s h | ]}� � |��qS r r� )r� r� r� r r! � <setcomp>� � z$cgpacker.generate.<locals>.<setcomp>s %8.i %s
s outgoing)rc �source)r� r� r� r r� r� �_wanted_sidedatar� r� �_generatechangelogr> r� r) r( r �istreemanifest�generatemanifestsr+ r� r� r� r
r� �clear�
generatefilesr� r )rI �
commonrevs�clnodes�fastpathlinkrevr� r� r r� r� �clstater� ro ru r3 �
clrevorder� manifests� changedfiles�fnodes�it�treer6 �mfdicts�clrevsr* r� r r� r! r� A s� �
�
�
� � �
� �
��
�
� �zcgpacker.generatec
s� i �i ��j j�t� � i ���� �d�}|sA�jsAt|�jd�}|D ]}��|�}t���|< ��|j |� � �
|j � q"|dfS � ������fdd�}t �j �|d|�j
�jtd�i �j�j|d� } || fS ) a� Generate data for changelog chunks.
Returns a 2-tuple of a dict containing state and an iterable of
byte chunks. The state will not be fully populated until the
chunk stream has been fully consumed.
if generate is False, the state will be fully populated and no chunk
stream will be yielded
See `revlogutil.sidedata.get_sidedata_helpers` for the doc on
`sidedata_helpers`.
)r� r� r� r� rA r c s� �� | �}t���| < �jrI| �jv s��| ��jv r0|j}��|| � ����| ���|�� �jrA� � �|j �
� � � � | S � � |j � | S ��|j| � � � |j � | S rP )
r� r) r� r� r� r� r� r� r� r� rQ �keysr� )r� r< � manifestnode�r� r� r� �clrevtomanifestrevr� �mflrI r r! �lookupcl s$
�
�� z-cgpacker._generatechangelog.<locals>.lookupclTr� �ru rt rV rW rY r� )r� r| r� r� rC r� r� r) r� r� r� r� r| r� r r� r� )
rI r� rE r� r� �staterc r< r� �genr r� r! r� � sD �
(�zcgpacker._generatechangelogc # s � | j } | j�d�i�� �����fdd�}
�r���� \} } | j�| dd� �}
| r-|
s-q��| �}|
s7g }n| �|| |�}| rC|sCq|
| | �}t| j ||d|| j| j t
d�|| j | j |d� }| j
�|jdd� �sp| |fV n |D ]}qr| s|| g fV �sdS dS ) a8 Returns an iterator of changegroup chunks containing manifests.
`source` is unused here, but is used by extensions like remotefilelog to
change what is sent based in pulls vs pushes, etc.
See `revlogutil.sidedata.get_sidedata_helpers` for the doc on
`sidedata_helpers`.
r c s, �r �rJ ��j S ���� ��fdd�}|S )Nc s� �| }�� �| �jdd�}|�� D ]D\}}}|dkr:�| d }��|i �}|�||�}� | � | k r9|||< q�| } ��| i �}
|
�||�} � | � | k rV||
|< q|S )aP Callback for looking up the linknode for manifests.
Returns the linkrev node for the specified manifest.
SIDE EFFECT:
1) fclnodes gets populated with the list of relevant
file nodes if we're not using fastpathlinkrev
2) When treemanifests are in use, collects treemanifest nodes
to send
Note that this means manifests must be completely sent to
the client before you can trust the list of files and
treemanifests to send.
T)r� � t� /)r� �readfast� iterentriesr� ) r� �clnode�mdatarT r
r �subtree�
tmfclnodes� tmfclnoder �fclnodes�fclnode)r� r� r� rE �tmfnodesr� r r! �lookupmflinknode^ s" � �zRcgpacker.generatemanifests.<locals>.makelookupmflinknode.<locals>.lookupmflinknode)� __getitem__)r� rE r� �r� r� r� r� r� r� )rE r� r! �makelookupmflinknodeV s
!z8cgpacker.generatemanifests.<locals>.makelookupmflinknodeNrK Fr� r� )r� r| �popitemr� �visitdirr} �_prunemanifestsr| r� r� r r� r� r r� )rI r� r� r� r� r� r� rV r� r� r� r� rE � should_visitrD � prunednodes�lookupfnr� r r r� r! r� < sL �+
�
�zcgpacker.generatemanifestsc s2 | j s|�� S |j|j��� ��fdd�|D �S )Nc � g | ] }��|��� vr|�qS r r �r� r
�r� �flr�frevr r! r� � s z,cgpacker._prunemanifests.<locals>.<listcomp>)r� r� r� rN )rI rD rE r� r r� r! r� � s zcgpacker._prunemanifestsc # sp � � fdd�|D �}|s�fdd�� n � j jj� � �fdd�� i �� jr=� fdd��D ��� j jj�����
� fdd�} n� } � j }
|
jjtd �td �t|�d
�} t t
|��D ]Z\} }
|
� |
�}|skt �
td �|
����� | ||
�� � fd d
�}|j|j�����fdd�� D �}|s�qW| j| d |
d� t� j ||d|� j� j�� j� j|d� }|
|fV qW| �� d S )Nc s$ g | ]}� � |�r� �|�s|�qS r )r� r )r� r rM r r! r� � s ���z*cgpacker.generatefiles.<locals>.<listcomp>c s � � |i �S rP r� )�unusedr^ )r� r r! �normallinknodes� rR z/cgpacker.generatefiles.<locals>.normallinknodesc s4 | j � | j�� fdd�| D �}���fdd�|D �S )Nc 3 s � | ] }|� |�fV qd S rP r )r� �r)�flinkrevr r! r^ � s � zBcgpacker.generatefiles.<locals>.normallinknodes.<locals>.<genexpr>c s& i | ]\}}|�v r�|�� |��qS r r )r� r� r� )�clnr� �fnoder r! �
<dictcomp>� s & zCcgpacker.generatefiles.<locals>.normallinknodes.<locals>.<dictcomp>)rN rc )rD r^ r� )r� r� )r� r� r! r� � s c s g | ]}� j | �qS r )r� r� rM r r! r� � r� c s� �D ] }z|� |�}| �|��|�� < W q tjy Y qw �| |�}t|�t��krO�D ] \}}|�|d �}||v rHt|| |� d�||< q.|rN|||< q.|S )NrA )�filenoder� r �ManifestLookupErrorr) r� �min)�flogr^ r< r� �linksr
r� )rR rV �
commonctxsr� r� r r! rx � s"
�
�z)cgpacker.generatefiles.<locals>.linknodes� filesr� s! empty or missing file data for %sc s � | S rP r r� )� linkrevnodesr r! �
lookupfilelog � z-cgpacker.generatefiles.<locals>.lookupfilelogc r� r r r� r� r r! r� s r )�itemF)ru rV rW rY r� )r� r� rc r� r� r5 r� r r) rq rC �filer r r� rN r� r| r� r� r� r� r )rI r� r� r� r� r� r� r� r� rx r� r rU r^ �
filerevlogr� � filenodesr� r )
r� rR r� rV r� r� r� r� r� r� r� r� rI r! r� � sb �
�
�
�
�� zcgpacker.generatefiles)FNFFNNN)T)TNrP ) r r r rL r� r� r� r� r� r r r r! r} � s&
�R
�
�o
�y�r} c
C s( dd� } t | ||d| dd|||||d� S )Nc S s t �| j| j| j| j�S rP )r r$ rc rb rc r` �r r r r! r� @ s �z _makecg1packer.<locals>.<lambda>r? r T)r� r� rs r� ru r� r� r� �r} �
r� r� r� r� ru r� r� r� r� r� r r r! �_makecg1packer5 s �r� c
C s& dd� } t | ||d| d|||||d� S )Nc S s t �| j| j| j| j| j�S rP )r# r$ rc rb rc r5 r` r� r r r! r� _ s �z _makecg2packer.<locals>.<lambda>r! r �r� r� r� ru r� r� r� r� r� r r r! �_makecg2packerT s �r� c
C s( dd� } t | ||d| t� |||||d� S )Nc S s t �| j| j| j| j| j| j�S rP )r* r$ rc rb rc r5 r` rh r� r r r! r� } s �z _makecg3packer.<locals>.<lambda>r% r� �r} r( r� r r r! �_makecg3packerr s �r� c
C s* dd� } t | ||d| t� ||||||d� S )Nc S s$ t �| j| j| j| j| j| j| j�S rP ) r1 r$ ri rc rb rc r5 r` rh r� r r r! r� � s �z(_makecg4packer.<locals>.builddeltaheaderr� )r� r� r� ru r� r� r� r� r� r� r r r! �_makecg4packer� s
�r� )r? r! r% r� c C s� t t�� �}d}| j�dd�s| j�dd�st�| �r d}|s$|�d� | j�dd�p6tj | jv p6tj
| jv }|s>|�d� |S ) NFs experimentals changegroup3s treemanifestTr% s changegroup4r� ) r� �
_packermapr� r5 r� r r� �discardr r� r� )r� �versions�needv03�want_v4r r r! �allsupportedversions� s& � ��
�
�
r� c C s t | �S rP )r� r� r r r! �supportedincomingversions� r� r� c C sh t | �}t�| �r|�d� |�d� tj| jv r#|�d� |�d� t| jv r2|�d� |�d� |S )Nr? r! )r� r r� r� r �NARROW_REQUIREMENT�LFS_REQUIREMENT�r� r� r r r! �supportedoutgoingversions� s
r� c C s t t| ��S rP )�maxr� r� r r r! � localversion� s r c C s. t | �}tj| jv r|�d� |sJ �t|�S )Nr? )r� r �GENERALDELTA_REQUIREMENTr� r� r� r r r! � safeversion s
r c
C s� | t |�v sJ �|d u rt�� }|d u rt�� }| dkr%|�� s%t�d��|r4| dv r4t�td�| ��|�|�}t | d }
|
||||||||| d� S )Nr? s; version 01 changegroups do not support sparse file matchers)r? r! sS ellipsis nodes require at least cg3 on client and server, but negotiated version %sr )ru r� r� r� r� )
r� �matchmod�always�neverr �ProgrammingErrorr r � narrowmatchr� ) r� r� r� r� r� ru r� r� r� r� �fnr r r! �
getbundler s< � ���
�r
c C s t | d |||d�S )Nr )rG )r� )r� r9 rJ rG r r r! � getunbundlerA s r c C sb | j js|dkr| j �td�t|� � | j jr-| j �d� |D ]}| j �dt|� � q d S d S )Ns bundles %d changesets found
s list of changesets:
s %s
)r5 r� r� r r) r� r� r )r� rE r� rc r r r! �_changegroupinfoE s �r c C s2 t | |||||d�}t|t�|�d dt|j�i�S )N)�fastpathr� s clcount)�
makestreamr r � chunkbufferr) �missing)r� �outgoingr� r� r
r� �cgstreamr r r! �makechangegroupN s � �r c
C s~ t || |||d�}| �� } |j} |j}
|j} | �� |p(| jd u o(| t| �� �k} | j dd|d� t
| |
|� |� | |
| |�S )N)r� r� r� s preoutgoingT)r� r� ) r
r� �commonr � ancestorsof�sort�
filternamerC r� r� r r� )
r� r r� r� r
r� r� r� �bundlerr� �csetsr� r� r r r! r a s$
�� r c C s� d}d}| j jtd�td�|d�} t|ji �D ]�}
|d7 }|
d } | j �d| � | �� | �| �} t| �}
z|� � }| j
||||d�}|sMt � td���W n t j
yd } z t � td �| ��d }~ww |t| �|
7 }| |v r�|| }t�|
t| ��D ]}| �|�}||v r�|�|� q}t � td
���|s�|| = q| �� t�|�D ]*\} }| �| �} |D ] }z| �|� W q� t jy� t � td �| t|�f ��w q�||fS ) Nr r� r� r r] r& rx s# received file revlog group is emptys# received delta base is censored: %ss# received spurious file revlog entrys+ missing file data for %s:%s - run hg verify)r5 r� r r r_ r� rz r� r) r{ r~ r r �CensoredBaseErrorr
r� rc �remover r� r� � LookupErrorr )r� r� r� r� �
expectedfilesr ry r{ r� r r r r �or� r� �e�needs�newr
r r r! r� � sj �
����
�
���� r� rP )NFNNNN)FFNNN)NNNFFNNN)FN)FNNN)F�
__future__r r0 r r� �i18nr rc r r r r
r � r r
r r r r r r �
interfacesr � revlogutilsr r� r r� �utilsr �Structr r# r* r1 r� r r" r& r( r+ r= �objectr> r r$ r, r2 r> rF rd r| r} r� r� r� r� r� r� r� r� r r r
r r r r r� r r r r! �<module> s� (
# g!2"f
� O
�$
�#
�#
�)�
�
3
�
�*�