/
lib64
/
python3.9
/
site-packages
/
mercurial
/
__pycache__
/
Upload File
HOME
a �+�b� � @ s2 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*� Z3dPd,d-�Z4G d.d/� d/e+�Z5dQd0d1�Z6dRd2d3�Z7dSd4d5�Z8dTd6d7�Z9e6e,fe7e-fe8e.fe9e/fd8�Z:d9d:� Z;d;d<� Z<d=d>� Z=d?d@� Z>dAdB� Z?dUdCdD�Z@dVdEdF�ZAdGdH� ZBdWdIdJ�ZCdXdKdL�ZDdYdMdN�ZEdS )Z� )�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<|r8t�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 �;/usr/lib64/python3.9/site-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; s 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 s r* c C s d}d}z�|r.|r |� |d�}qLt |dd�}n tjddd�\}}t�|d�}|}|D ]}|�|� qTd}|W |dur||�� |dur�|r�|r�|�|� S t�|� S |dur�|r�|r�|�|� n t�|� n>|dur�|�� |du�r|� r�|� r�|�|� n t�|� 0 dS )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 Zmkstemp�os�fdopen�write�close�unlink)�ui�chunks�filenameZvfs�fhZcleanup�fd�cr r r! � writechunksJ sB � � r9 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 r d}|t jjvr*t�td�| ��|dkr6d}t j�|�}|�|�| _|| _ |pZi | _ d | _ d S )N� UNs# unknown stream compression type: %ss BZs _truncatedBZ) r Z compenginesZsupportedbundletypesr r r Z forbundletypeZdecompressorreader�_stream�_type�extras�callback)�selfr6 �algr? Z compenginer r r! �__init__� s zcg1unpacker.__init__c C s | j d uo| j dkS )Nr< )r> �rA r r r! � compressed� s zcg1unpacker.compressedc C s | j �|�S �N)r= �read)rA r r r r! rG � s zcg1unpacker.readc C s | j �|�S rF )r= �seek)rA �posr r r! rH � s zcg1unpacker.seekc C s | j �� S rF )r= �tellrD r r r! rJ � s zcg1unpacker.tellc C s | j �� S rF )r= r1 rD r r r! r1 � s zcg1unpacker.closec C sT t | jd�}t�d|�d }|dkr>|r:t�td�| ��dS | jrL| �� |d S )Nr r r r )r r= r r r r r r@ )rA r r r r r! � _chunklength� s zcg1unpacker._chunklengthc C s i S )z*v10 does not have a changelog header chunkr rD r r r! �changelogheader� s zcg1unpacker.changelogheaderc C s i S )z)v10 does not have a manifest header chunkr rD r r r! �manifestheader� s zcg1unpacker.manifestheaderc C s$ | � � }|si S t| j|�}d|iS )zBreturn the header of the filelogs chunk, v10 only has the filename� filename)rK r r= )rA r �fnamer r r! � filelogheader� s zcg1unpacker.filelogheaderc C s8 |\}}}}|d u r|}n|}d}d} ||||||| fS �Nr r ) rA � 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 rF )rK r r= �deltaheadersize� deltaheaderr r[ ) rA rS r Z headerdata�header�deltarT rU rV rX rW rY rZ r r r! � deltachunk� s zcg1unpacker.deltachunkc c s� d}|d| j k r�d}t| �}|sB|dk r4|d7 }q�|r�|d7 }q�d}tt|��V d}|t|�k r|d }|||� V |}qXqt� V qdS )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' )rA �partsZ noentries�chunkrI �nextr r r! � getchunks� s$ zcg1unpacker.getchunksc C sH |j | _| �� | �� }|j�d�}|j||||d� |�� d | _d S )Nr �� addrevisioncb)� incrementr@ rM � deltaiter� manifestlog� getstorage�addgroup�complete)rA �repo�revmap�trp�progrh �deltasZstorager r r! �_unpackmanifests s z cg1unpacker._unpackmanifestsc ( s� �� � �tj�jv s tj�jv o0| jdko0|dk}|rNtj�|pDt� 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��sf�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!��rt�j ��j"}t �#�|�D ]H}��$|�j%}|| �&� }t �'|�D ] \}}|� |t� ��(|� �qP�q*d} | �r�t j)| d v �r�� fd"d#�} �j�td$�� t*�| | | ||| d �\} } | �r>t j+| d v �r��,| | �|d � � �-� D ] \} \} }!| �,| | | |!� �q�� �-� 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�r�d*|jv�rzt2��3���|jd*< t2��3|d ��|jd+< t4|j��n2t4|j��t2��3����d*< t2��3|d ���d+< �j d5d dit � ���� t �#�|�� d}&|d-v �r ��5� �r�t6j7 }}&n t6j8 }}&� �rt6�9�||� � |&du�rN��r2��:� � n� �t6j;�||&g �d.� g �|d&k�r�� �����fd/d0��|�<d1� ��fd2d3�� W �j�=� n �j�=� 0 |#d&k �r�|#d }'n|#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)ZpullNc s �j �dt| � � t� �S )Ns add changeset %s )r3 �debugr r( ��x��clro r r! �csmap; s z cg1unpacker.apply.<locals>.csmapc s � � | �S rF ��revrw �rz r r! rp ? s z!cg1unpacker.apply.<locals>.revmaps sources url� prechangegroup�throws adding changesets � changesets� chunks�Zunit�totalc s |� k r�� |� d S rF )�append)rz r} )�clstart�duprevsr r! �ondupchangelog_ s z)cg1unpacker.apply.<locals>.ondupchangelogc s& | � |�}� �|j� ��||� d S rF )�changelogrevision�update�filesZregister_changeset)rz r} �ctx)� efilessetro r r! � onchangelogc s z&cg1unpacker.apply.<locals>.onchangelog)Z alwayscacherh Zduplicaterevisioncbs( applied empty changelog from changegroups warn-empty-changegroup)�configs adding manifests � manifestsr c s0 � � | �}|s ||f� | < n|d |f� | < d S rQ ��get)�manifestr} �range)�touched_manifestsr r! �on_manifest_rev� s z*cg1unpacker.apply.<locals>.on_manifest_revrg s servers validatec s0 � � | �}|s ||f� | < n|d |f� | < d S rQ r� )Zfilelogr} r� )�touched_filelogsr r! �on_filelog_rev� s 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>/ r 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� r8 r r r! r� 4 r )r� )r� ) r( �hookr � strkwargs�copyr rT �headsr3 �log�join)Zunused_successr} �argsZnewheads)�addedrz r� �hookargsr� ro r r! �runhooks s �z#cg1unpacker.apply.<locals>.runhookss changegroup-runhooks-%020ic s � � ��S rF )Z _afterlock)�tr)ro r� r r! �<lambda>9 r z#cg1unpacker.apply.<locals>.<lambda>)r )r� )>� unfilteredr �REVLOGV2_REQUIREMENT�CHANGELOGV2_REQUIREMENT�version� sidedatamod�get_sidedata_helpers�setr� � setdefaultr� r r� � changelogZ delayupdater� �weakref�proxyr3 �statusr r( � makeprogressri r@ rL rj rm Z develwarnrn �revlog_constantsZKIND_MANIFESTLOGrt � configboolrk �xranger� r� Z readdelta� iteritems�addZ KIND_FILELOG�_addchangegroupfilesZKIND_CHANGELOGZrewrite_sidedata�itemsZchanges�quietZ closesbranchZinvalidatevolatilesetsr rT �dictZ publishingr Zpublic�draftZ registernew�extendZadvanceboundaryZ addpostclose�flush)(rA ro r� Zsrctype�urlZ targetphaseZ expectedtotalZsidedata_categoriesZadding_sidedata�sidedata_helpersr{ rp rq �progressr� r� rs ZefilesZclendZ changesetsr� � needfiles�mlZcsetZmfnodeZmfest�f�nr� ZnewrevsZnewfiles�mfZstartrevZendrev�flZ deltaheadsr� r� Zphaseall�retr ) r� rz r� r� r� r� r� ro r� r� r� r! �apply sR ���� �� � ���� � � � ��� � zcg1unpacker.applyc # s8 d� t � �fdd�i �D ]}|dd� V |d � qdS )z� returns an iterator of the deltas in this changegroup Useful for passing to the underlying storage system to be stored. Nc s �� � �S rF )r` r ��chainrA r r! r� K r z'cg1unpacker.deltaiter.<locals>.<lambda>� r )�iter)rA � chunkdatar r� r! rj D s zcg1unpacker.deltaiter)N)N) �__name__� __module__� __qualname__�__doc__�_CHANGEGROUPV1_DELTA_HEADERr] �sizer\ r� rb rC rE rG rH rJ r1 rK rL rM rP r[ r` rf rt r r� r� rj r r r r! r: m s4 - � 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 rQ r � rA rR rS rT rU rV rX rW rY rZ r r r! r[ ] s zcg2unpacker._deltaheaderN) r� r� r� r� �_CHANGEGROUPV2_DELTA_HEADERr] r� r\ r� r[ 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. � 03ra c C s& |\}}}}}}d} ||||||| fS rQ r r� r r r! r[ q s zcg3unpacker._deltaheaderNc sx t t| �j|||||d� t| ji �D ]L}|d }|j�d| � | �� }|j� |�j ||||d�s&t � t d���q&d S )Nrg rN � adding %s revisions s" received dir revlog group is empty)�superr� rt r� rP r3 rv rj rk rl rm r r r ) rA ro rp rq rr rh r� r rs �� __class__r r! rt v s � �z cg3unpacker._unpackmanifests)N) r� r� r� r� �_CHANGEGROUPV3_DELTA_HEADERr] r� r\ r� rb r[ rt � __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. ru c C s$ |\}}}}}}} |||||| |fS rF r ) rA rR rS rZ rT rU rV rX rW rY r r r! r[ � s zcg4unpacker._deltaheaderc sf t t| ��|�}|s|S |\ }}}}}}} } } i } | tj@ rPt| j�} t�| �} ||||||| | | f S rF ) r� r� r` r �CG_FLAG_SIDEDATAr" r= r� Zdeserialize_sidedata) rA rS �resrT rU rV rW rX r_ rY r rZ Z sidedata_rawr� r r! r` � s8 � �zcg4unpacker.deltachunk) r� r� r� r� �_CHANGEGROUPV4_DELTA_HEADERr] r� r\ r� r[ r` r� r r r� r! r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�headerlessfixupc C s || _ || _d S rF )�_h�_fh)rA r6 r� r r r! rC � s zheaderlessfixup.__init__c C sZ | j rN| j d |� | j |d � }| _ t|�|k rJ|t| j|t|� �7 }|S t| j|�S rF )r� r( r r� )rA r� r r r r! rG � s zheaderlessfixup.readN)r� r� r� rC rG r r r r! r� � s r� c c s� |j durd|j }}n:|j| jkr:|j}t�t|��}n|j}t�|jt|��}||�}t t|�t|� t|� �V |V |r�|V |V |j t j @ r�|j }t t|��V |V dS )z0Serialize a revisiondelta to changegroup chunks.Nr )r_ �basenodeZnullid�revisionr Ztrivialdiffheaderr( ZreplacediffheaderZbaserevisionsizer& rZ r r� r )ro r_ Zheaderfnr+ �data�metar r r r! �_revisiondeltatochunks� s" r� c s � �fdd�}t ||d�S )z&Sort nodes for changegroup generation.c s � � �| ��S rF r| )r� �rz �lookupr r! r� � r z$_sortnodesellipsis.<locals>.<lambda>��key)�sorted)�store�nodesrz r� r� r r� r! �_sortnodesellipsis� s r� c s� �| } � ������fdd��| r6�� ��ttfkrBtt } } nDt| �dkrlt�fdd�| D ��\} t} nt�fdd�| D ��\} } ��| ���| � } } | | |fS )Nc s� | t kr t S �r| S | g}|r�|d }|dd� }|�v rB�| S |�v rf|�dd� � �|�D �� q|�v r�|�dd� �| D �� qt��dd�D ]}��|�| kr�| S q�t��d�rƈj}n�j j}t � d |�| f ��qt 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 s g | ]}|t kr|�qS r �r �r� �ppr r r! r� + r z=_resolvenarrowrevisioninfo.<locals>.local.<locals>.<listcomp>c S s g | ]}|t kr|�qS r r r r r r! r� . r ����targets? unable to resolve parent while packing '%s' %r for changeset %r) r r� � parentrevsr r� �linkrevr Z safehasattrZ display_idZ_revlogr r )�clrev�walk�p�ir )rz �clrevtolocalrev� fullclnodes� ischangelog�precomputedellipsisr} r� r r! �local s8 � ��z)_resolvenarrowrevisioninfo.<locals>.localr c 3 s | ]}� |�V qd S rF r �r� r �r r r! � <genexpr>R r z-_resolvenarrowrevisioninfo.<locals>.<genexpr>c 3 s | ]}� |�V qd S rF r r r r r! r U r )r r r( r� rT )rz r� r r} r �linknoder r r Z linkparentsrU rV �p1node�p2noder )rz r r r r r r} r� r! �_resolvenarrowrevisioninfo� s E r Fc c s |sdS | j } |rd} n |r0t||| |�}d} nd} |r�g }i }i }|D ]�}|�|�}||�}| �|�}|||< || v r�|||< n>|| vr�qHn2t| ||||||| | � \}}}||f||< |||< |�|� qH|}d}|dur�| jj|td�t|�d�}| j� dd�}|dv�r&td �}| j� || � t j }|�r:t j }n"|d k�rLt j}n|d k�r\t j}|j|| d | || d �}t|�D ]|\} } |�r�|�| d � |�r�|| j }| j|v �r�|| j \}}|| _|| _| jt jO _n || j�}|| _| V �q||�r|�� 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: %sr r T)� nodesorderZ revisiondataZassumehaveparentrevisions� deltamoder� r )r� r� r} r r� r3 r� r r( r� �warnr ZCG_DELTAMODE_STDZCG_DELTAMODE_PREVZCG_DELTAMODE_P1ZCG_DELTAMODE_FULLZ emitrevisions� enumerater� rT r r rY ZREVISION_FLAG_ELLIPSISr rn ) ro r� r� r r� �forcedeltaparentprev�topic�ellipsesr r r r� rz r ZfilteredZadjustedparents� linknodesrT r} r r r r r� Z configtarget�msgr � revisionsr r� r r r! � deltagroup\ s� � � � r# c @ sF e Zd Zddd�Zddd�Zddd �Zdd d �Zd d � Zddd�ZdS )�cgpackerFNc C s� || _ || _|| _|| _|| _|| _| | _|du r8t� }|| _| du rLt� } | | _ | | _ | | _ | | _ || _ | j jjr�| j jjs�| j jj| _n 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 rF r )�sr r r! r� ? r z#cgpacker.__init__.<locals>.<lambda>)� _oldmatcher�_matcherr� �_forcedeltaparentprev�_builddeltaheader� _manifestsend� _ellipsesr� Z _bundlecaps�_remote_sidedata� _isshallow� _fullclnodes�_precomputedellipsis�_repor3 �verbose� debugflagZnote� _verbosenote)rA ro � oldmatcher�matcherr� �builddeltaheader� manifestsendr � bundlecapsr �shallow� ellipsisroots� fullnodes�remote_sidedatar r r! rC � s( 4zcgpacker.__init__Tc # s� | j ��j� | �td�� d}d}| jdkrL| j}|dkr@�j}t��|�}| j � |||d�\} } | D ],} t | j | | j �D ]} |t | �7 }| V qzqft � } |t | �7 }t � V | �td�| � | d }| d }| d }|o�t��� }i }d}| j||||||| d |d �}|D ]|\}} |�r:t|�} |t | �7 }| V | D ]4} t | j | | j �}|D ]} |t | �7 }| V �qV�q>t � } |t | �7 }| V �q| �td �| � | jV d}| j�r�| j�rևfdd�t�|�D �}|�� � fdd�|D �}| j||||||||d �}|D ]�\}} t|�}t |�}|V | D ]4} t | j | | j �}|D ]} |t | �7 }| V �qH�q0t � } |t | �7 }| V | �td�||f � �qt � V |�r��jdt|d �|d� 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 Nru s strip)�generater� s %8.i (changelog) � clrevorderr� � changedfiles� clrevtomanifestrev)r� s %8.i (manifests) c s" g | ]\}}� j | �� |f�qS r )rk rG )r� r� �lr�ro r r! r� � s �z%cgpacker.generate.<locals>.<listcomp>c s h | ]}� � |��qS r r| )r� rx r~ r r! � <setcomp>� r z$cgpacker.generate.<locals>.<setcomp>s %8.i %s s outgoing)rT �source)r0 r� r3 r r� r, Z_wanted_sidedatar� r� �_generatechangelogr� r) r( r' r �istreemanifest�generatemanifestsr* r* r+ r- r r� �clear� generatefilesr� r )rA � commonrevsZclnodes�fastpathlinkrevrD r� r� r� r< Zclstaters r_ rd r1 � clrevorder� manifests� changedfiles�fnodes�it�treer4 �mfdicts�clrevsr) r� r ry r! r= A s� �� � � � �� � zcgpacker.generatec s� i �i ��j j�t� � i ���� �d�}|s��js�t|�jd�}|D ]4}��|�}t���|< ��|j |� � � |j � qD|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@ r� r c s� �� | �}t���| < �jr�| �jv s6��| ��jv r`|j}��|| � ����| ���|�� �jr�� � �|j � � � � � q�� � |j � n��|j| � � � |j � | S rF ) r� r( r+ r. r} r/ r� r� r- r� rG �keysr� )rx r8 Z manifestnode�rN rz rL ZclrevtomanifestrevrM �mflrA r r! �lookupcl s$ �� � z-cgpacker._generatechangelog.<locals>.lookupclTr� �r r r r r r� )r0 rk r� r+ r� r} r� r( r� r� r� r� r# r( r r. r/ ) rA rz r� r= r� �staterT r8 rW �genr rU r! rE � sD � (�zcgpacker._generatechangelogc # s� | j } | j�d�i�� �����fdd�} �r���� \} } | j�| dd� �} | rX| sXq*��| �}| slg }n| �|| |�}| r�|s�q*| | | �}t| j ||d|| j| j t d�|| j | j |d� }| j �|jdd� �s�| |fV q*|D ]}q�| s*| g fV q*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 �j S ���� ��fdd�}|S )Nc s� �| }�� �| �jdd�}|�� D ]�\}}}|dkrt�| d }��|i �}|�||�}� | � | k r�|||< q$�| } ��| i �} | �||�} � | � | k r$|| |< 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)r9 � t� /)r� ZreadfastZ iterentriesr� ) rx ZclnodeZmdatar r� r� ZsubtreeZ tmfclnodesZ tmfclnoder� ZfclnodesZfclnode)rL rO rV r� �tmfnodesrQ r r! �lookupmflinknode^ s zRcgpacker.generatemanifests.<locals>.makelookupmflinknode.<locals>.lookupmflinknode)� __getitem__)rQ r� r^ �rL rK rO rM rV r] )r� rQ r! �makelookupmflinknodeV s !z8cgpacker.generatemanifests.<locals>.makelookupmflinknodeNr Fr� rX )r0 rk �popitemr' Zvisitdirrl �_prunemanifestsr# r( r+ r r. r/ r& rQ )rA rJ rL rK rM rO rD r r� ro ra rQ r� Z should_visitr� Z prunednodesZlookupfnrs r r r` r! rG < sH + � zcgpacker.generatemanifestsc s2 | j s|�� S |j|j ��� ��fdd�|D �S )Nc s g | ]}��|��� vr|�qS r r �r� r� �rJ �flr�frevr r! r� � r z,cgpacker._prunemanifests.<locals>.<listcomp>)r+ r� r} r )rA r� r� rJ r re r! rc � s zcgpacker._prunemanifestsc # sp � fdd�|D �}|s$�fdd�� n� j jj� � �fdd�� i �� jrx� fdd��D ��� j jj����� � fdd�} n� } � j } | jjtd �td �t|�d �} t t |��D ]�\} } | � | �}|s�t � td �| ����� | || �� � fd d �}|j|j �����fdd�� D �}|�s q�| j| d | d� t� j ||d|� j� j�� j� j|d� }| |fV q�| �� d S )Nc s$ g | ] }� � |�r� �|�s|�qS r )r' r& )r� r� rD r r! r� � s �z*cgpacker.generatefiles.<locals>.<listcomp>c s � � |i �S rF r� )ZunusedrO )rO r r! �normallinknodes� s z/cgpacker.generatefiles.<locals>.normallinknodesc s4 | j � | j�� fdd�| D �}���fdd�|D �S )Nc 3 s | ]}|� |�fV qd S rF r )r� �r)�flinkrevr r! r � r zBcgpacker.generatefiles.<locals>.normallinknodes.<locals>.<genexpr>c s& i | ] \}}|�v r�|�� |��qS r r )r� ri rA )�clnrS �fnoder r! � <dictcomp>� r zCcgpacker.generatefiles.<locals>.normallinknodes.<locals>.<dictcomp>)r rT )r� rO r� )rk rS )rj rl r! rh � s c s g | ]}� j | �qS r )r0 r� rD r r! r� � r c s� �D ]:}z |� |�}| �|��|�� < W q tjy< Y q0 q�| |�}t|�t��kr��D ]@\}}|�|d �}||v r�t|| |� d�||< q^|r^|||< q^|S )Nr� )Zfilenoder} r ZManifestLookupErrorr( r� �min)ZflogrO r8 rl Zlinksr� rA )r r � commonctxsrR rh r r! r � s z)cgpacker.generatefiles.<locals>.linknodes� filesr� s! empty or missing file data for %sc s � | S rF r rw )� linkrevnodesr r! � lookupfilelog s z-cgpacker.generatefiles.<locals>.lookupfilelogc s g | ]}��|��� vr|�qS r r rd re r r! r� s r )�itemF)r r r r r� )r0 r� rT r- r} r3 r� r r( r r� �filer r rH r r� r# r( r+ r. r/ rn )rA rN rJ rD rR rK rO rS r� r ro r� r rO Z filerevlogrr Z filenodesrs r ) rk r rS r ro rJ rf rO rg rq rR rh rA r! rI � s` � � � �� zcgpacker.generatefiles)FNFFNNN)T)TN)N)N) r� r� r� rC r= rE rG rc rI 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 rF )r� r$ rT r r r �r r r r! r� @ s �z _makecg1packer.<locals>.<lambda>r; r T)r6 r7 r r8 r r9 r: r; �r$ � ro r4 r5 r8 r r9 r: r; r< r6 r r r! �_makecg1packer5 s �rx c C s& dd� } t | ||d| d|||||d� S )Nc S s t �| j| j| j| j| j�S rF )r� r$ rT r r r� r ru r r r! r� _ s �z _makecg2packer.<locals>.<lambda>r� r �r6 r7 r8 r r9 r: r; rv rw r r r! �_makecg2packerT s �rz c C s( dd� } t | ||d| t� |||||d� S )Nc S s t �| j| j| j| j| j| j�S rF )r� r$ rT r r r� r rY ru r r r! r� } s �z _makecg3packer.<locals>.<lambda>r� ry �r$ r' rw 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 rF ) r� r$ rZ rT r r r� r rY ru r r r! r6 � s �z(_makecg4packer.<locals>.builddeltaheaderru )r6 r7 r8 r r9 r: r; r< r{ rw r r r! �_makecg4packer� s �r} )r; r� r� ru c C s� t t�� �}d}| j�dd�s6| j�dd�s6t�| �r:d}|sH|�d� | j�dd�pltj | jv pltj | jv }|s||�d� |S ) NFs experimentals changegroup3s treemanifestTr� s changegroup4ru ) r� � _packermaprT r3 r� r rF �discardr r� r� )ro �versionsZneedv03Zwant_v4r r r! �allsupportedversions� s& � �� � � r� c C s t | �S rF )r� rB r r r! �supportedincomingversions� s r� c C sh t | �}t�| �r&|�d� |�d� tj| jv rF|�d� |�d� t| jv rd|�d� |�d� |S )Nr; r� )r� r rF r r ZNARROW_REQUIREMENT�LFS_REQUIREMENT�ro r� r r r! �supportedoutgoingversions� s r� c C s t t| ��S rF )�maxr� rB r r r! � localversion� s r� c C s& t | �}tj| jv r |�d� t|�S )Nr; )r� r ZGENERALDELTA_REQUIREMENTr rn r� r r r! � safeversion s r� c C s� |d u rt �� }|d u r t �� }| dkr:|�� s:t�d��|rX| dv rXt�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 )r r9 r: r; r< ) �matchmod�always�neverr ZProgrammingErrorr r Z narrowmatchr~ ) r� ro r8 r4 r5 r r9 r: r; r< �fnr r r! � getbundler s: � ��� �r� c C s t | d |||d�S )Nr )r? )r~ )r� r6 rB r? r r r! � getunbundlerA s r� c C s^ | j js|dkr(| j �td�t|� � | j jrZ| j �d� |D ]}| j �dt|� � q@d S )Ns bundles %d changesets found s list of changesets: s %s )r3 r1 r� r r( r2 rv r )ro r� rD rT r r r! �_changegroupinfoE s r� c C s2 t | |||||d�}t|t�|�d dt|j�i�S )N)�fastpathr8 s clcount)� makestreamr� r Z chunkbufferr( �missing)ro �outgoingr� rD r� r8 Zcgstreamr r r! �makechangegroupN s � �r� c C s~ t || |||d�}| �� } |j} |j} |j} | �� |pP| jd u oP| t| �� �k} | j dd|d� t | | |� |� | | | |�S )N)r8 r5 r< s preoutgoingT)r� rD ) r� r� �commonr� Z ancestorsof�sortZ filternamer� r� r� r� r= ) ro r� r� rD r� r8 r5 r< ZbundlerrJ Zcsetsr� rK r r r! r� a s$ �� r� c C s� d}d}| j jtd�td�|d�} t|ji �D �]} |d7 }| d } | j �d| � | �� | �| �} t| �} z0|� � }| j ||||d�}|s�t � td���W n8 t j y� } z t � td �| ��W Y d }~n d }~0 0 |t| �| 7 }| |v r.|| }t�| t| ��D ]4}| �|�}||v �r,|�|� nt � td ����q|s.|| = q.| �� t�|�D ]`\} }| �| �} |D ]F}z| �|� W n0 t j�y� t � td �| t|�f ��Y n0 �qr�q\||fS ) Nr rp r� r rN r� rg 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)r3 r� r r� rP rv ri rt r( rj rm r r ZCensoredBaseErrorr r� rT �removern r� r} � LookupErrorr )ro rD rp rq Z expectedfilesr� rh r" r� r� r� r� r� �ors r� �eZneeds�newr� r r r! r� � s^ � �( ��r� )N)NFNNNN)FFNNN)FFNNN)FFNNN)FFNNN)NNNFFNNN)N)FN)FNNN)N)FZ __future__r r. r r� Zi18nr rT r r r r r � r r r� r r r r r Z interfacesr Z revlogutilsr r� r r� Zutilsr �Structr� r� r� r� r� r r" r& r' r* r9 �objectr: r� r� r� r� r� r� r r# r$ rx rz 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 � � * �