|
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�z � @ s� d dl mZ d dlZd dlZddlmZ ddlmZmZ ddl m
Z m Z m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZm Z m Z m Z m Z ddl!m"Z"m#Z#m$Z$ ddl%m&Z& e j'Z'e j(Z(d Z)d�d
d �Z*d d
� Z+dd� Z,dd� Z-dd� Z.G dd� de/�Z0ed�ed�fed�ed�fed�ed�fd �Z1 d�d d �Z2g Z3i Z4d!d"� Z5d#d$� Z6d%d&� Z7e5d'�d(d)� �Z8e5d*�d+d,� �Z9e5d-�d.d/� �Z:e5d0�d1d2� �Z;d3d4� Z<d5d6� Z=g Z>i Z?d�d7d8�Z@d9d:� ZAd;d<� ZBe@d=�d>d?� �ZCe@d@�dAdB� �ZDe@d'�dCdD� �ZEe@d*�dEdF� �ZFdGdH� ZGdIdJ� ZHe@dK�dLdM� �ZIe@d0�dNdO� �ZJdPdQ� ZKdRdS� ZLdTdU� ZMdVdW� ZNe@dXd dY�dZd[� �ZOd\d]� ZPd^d_� ZQd`da� ZRejSfdbdc�ZTddde� ZUdfdg� ZVG dhdi� die/�ZWG djdk� dke jX�ZYdldm� ZZdndo� Z[dpdq� Z\ d�drds�Z]g Z^i Z_dtdu� Z`dvdw� Zae`dx�dydz� �Zbe`d{�d|d}� �Zcd~d� Zdd�d�� Zed�d�� Zfd�d�� Zgd�d�� Zhd�d�� Zid�d�� Zjd�d�� Zkd�d�d��Zld�d�� Zmg Zni Zod�d�d��Zpd�d�� Zq d�d�d��Zrepd��d�d�� �Zsepd{� d�d�d���Ztepd0� d�d�d���Zuepd�� d�d�d���ZvepdK� d�d�d���Zwepd�� d�d�d���Zxepd�� d�d�d���Zyepd�� d�d�d���Zzd�d�� Z{d�d�� Z|d�d�� Z}d�d�� Z~dS )�� )�absolute_importN� )�_)�hex�nullrev)� bookmarks�bundle2� bundlecaches� changegroup� discovery�error�lock� logexchange�
narrowspec�obsolete�obsutil�phases�pushkey�pycompat� requirements�scmutil� streamclone�url�util�wireprototypes)�hashutil�
stringutil�urlutil)�
repositorys narrowaclc C s� t �|d�}d }|s!d}|�d�s |�d�r t �||�}d}d}n|r(|�|�}|dd� |dd� }}|dkrBt�td �| ��|d
krV|d u rPt �|d�}t �||�S |�d �ret j
| ||| d �S |d
krnt � |�S t�td�||f ��)N� � streams HG� s HG10s UNr � s %s: not a Mercurial bundles 10� 2)� magicstrings S1s %s: unknown bundle version %s)
r
� readexactly�
startswith�headerlessfixup�joinr �Abortr � cg1unpackerr � getunbundlerr �streamcloneapplier)�ui�fh�fname�vfs�header�alg�magic�version� r5 �4/usr/lib/python3/dist-packages/mercurial/exchange.py�
readbundle5 s2 �
�r7 c C s� dd� }t | |d�}t|tj�r,|j}|dkrd}||�}|s(t�td�| ��d| S t|tj �r�d|j
v rJ||j
d �}|sIt�td�| ��nd }d}|� � D ]@}|j d
krr|j
d }|d v red
}qRtjtd�| td�d��|j dkr�|du r�t
�|j
d �}|�� } t�| �}
d|
S qR|s�t�td���d||f S t|tj�r�t�|�d }t�|�} d| S t�td�| ��)z�Infer the bundlespec from a bundle file handle.
The input file handle is seeked and the original seek position is not
restored.
c S s, z t j�| ��� d W S ty Y d S w �Nr )r � compengines�
forbundletype�
bundletype�KeyError)r2 r5 r5 r6 �speccompression[ s
�z&getbundlespec.<locals>.speccompressionNs _truncatedBZs BZs! unknown compression algorithm: %ss %s-v1s Compressions none� changegroup� version)� 01s 02s v2s7 changegroup version %s does not have a known bundlespecs# try upgrading your Mercurial client��hint� stream2s requirementss none-v2;stream=v2;%ss0 could not identify changegroup version in bundles %s-%sr" s none-packed1;%ss unknown bundle type: %s)r7 �
isinstancer
r* �_typer r) r r �
unbundle20�params� iterparts�type�urlreq�unquote�split�_formatrequirementsparamsr r, �readbundle1header) r- r. r= �br2 �compr4 �partr �splittedrG � formattedr5 r5 r6 �
getbundlespecT sb
��
���
��
rT c sF | j }|r|j� � fdd�|D �}n| jg}|s |�� }t�| ||�S )z�Computes which revs are outgoing given a set of common
and a set of heads.
This is a separate function so extensions can have access to
the logic.
Returns a discovery.outgoing object.
c s g | ]}� |�r|�qS r5 r5 ��.0�n��hasnoder5 r6 �
<listcomp>� � z$_computeoutgoing.<locals>.<listcomp>)� changelogrY �nullid�headsr �outgoing)�repor^ �common�clr5 rX r6 �_computeoutgoing� s rc c C s
| j }|j}|�dd�}| js|dvrd S t| jd�}|�dd�s#d S | jd u r1|�d��d�}n|�d | j�}||�d�M }|r�|d
krR|� t
d �t |� � d S |d krj|� t
d
�t |� �rht
�t
d���d S |dkr�t
d�t |� }t
d�}t
j||d��d S d S )N� experimentals auto-publish)� warn� confirm� abort� phases�
publishingFs serveds not public()s ::%ln - public()re s$ %i changesets about to be published
rf s2 push and publish %i changesets (yn)?$$ &Yes $$ &Nos user quitrg s push would publish %i changesetss: use --publish or adjust 'experimental.auto-publish' configrA )r` r- �config�publish�listkeys�remote�get�revs�filtered�warnr �len� promptchoicer �
CanceledErrorr) )�pushopr` r- �behavior� remotephases� published�msgrB r5 r5 r6 �
_checkpublish� sB
������ rz c C s6 | j j}|�dd�}d|vod|v }|p| j�d� S )zsreturn true if a pull/push must use bundle1
This function is used to allow testing of the older bundle version� devel� legacy.exchange� bundle2s bundle1)r` r- �
configlistrm �capable)�opr- �exchange� forcebundle1r5 r5 r6 �
_forcebundle1� s r� c @ sN e Zd ZdZ d
dd�Zejdd� �Zejd d
� �Ze d d � �Z
dS )�
pushoperationz�A object that represent a single push operation
Its purpose is to carry push related state and very common operations.
A new pushoperation should be created at the beginning of each push and
discarded afterward.
FNr5 c C s� || _ |j| _|| _|| _|| _|| _|| _t� | _d | _ d | _
d | _ d | _ d | _
d | _d | _d | _d | _t� | _g | _d | _i | _|| _|| _d S �N)r` r- rm �forcero r � newbranch�set� stepsdone�cgresult�bkresultr_ � remoteheads�
pushbranchmap�incomingrw �outdatedphases�fallbackoutdatedphases�
outobsmarkers� outbookmarks� trmanager�pkfailcb�pushvarsrk ) �selfr` rm r� ro r� r rk r� r5 r5 r6 �__init__� s.
zpushoperation.__init__c C s | j jS )z2future remote heads if the changeset push succeeds)r_ � ancestorsof�r� r5 r5 r6 � futureheads6 s zpushoperation.futureheadsc st | j du r | jjS | j�� }| jj� | jjjj�� �fdd�| j D �}|� d| jj| jj
�}|� dd� |D �� |S )z/future remote heads if the changeset push failsNc s g | ]
}�|�� v r|�qS r5 r5 )rV �node�ra �revr5 r6 rZ R � z/pushoperation.fallbackheads.<locals>.<listcomp>s %ln and parents(roots(%ln))c s � � | ]}|� � V qd S r� �r� �rV �cr5 r5 r6 � <genexpr>Z � � z.pushoperation.fallbackheads.<locals>.<genexpr>) ro r_ � commonheadsr` �
unfilteredra r\ �indexr� r� �missing�extend)r� �unfi�cheads�revsetr5 r� r6 �
fallbackheads; s
�zpushoperation.fallbackheadsc C s | j r| jS | jS )z3set of all common heads after changeset bundle push)r� r� r� r� r5 r5 r6 r� ] s zpushoperation.commonheads�FNFr5 FN) �__name__�
__module__� __qualname__�__doc__r� r �
propertycacher� r� �propertyr� r5 r5 r5 r6 r� � s
�K
!r� s updating bookmark %s
s updating bookmark %s failed
s exporting bookmark %s
s exporting bookmark %s failed
s deleting remote bookmark %s
s# deleting remote bookmark %s failed
)� update� export� deleteFr5 c
C s� |du ri }t | ||||||fi t�|���}|j�� r:t|jj�|j�� j } | r:t d�d�
t | �� }
t �
|
��|j�� sFt �
t d���|j�d�sSt �
t d���t t�|j��D ] } | j�� D ] \} }
|
�| �rm n
qbt �
t d�t�| � ��q[d }}z)|j�dd �}t|�s�|r�t�| �s�|j�� }|j� � }t |jd
|j� � �|_ W n t j y� } zd t!�"|� }
|j�#|
� W Y d}~nd}~ww |p�t$�%� �d |p�t$�%� �M |j p�t$�%� �5 |j�&|� t'|� t(|� |j)s�t*|� t|�s�t+|� t,|� t-|� t.|� t/|� W d � n 1 �sw Y W d � n 1 �s'w Y W d � n 1 �s7w Y | j�dd ��rJt0�1| |� |S )
aW Push outgoing changesets (limited by revs) from a local
repository to remote. Return an integer:
- None means nothing to push
- 0 means HTTP error
- 1 means we pushed and remote head count is unchanged *or*
we have outgoing changesets but refused to push
- other values as described by addchangegroup()
N�: required features are not supported in the destination: %s� , s! destination does not support push� unbundlesL cannot push: destination does not support the unbundle wire protocol commandsJ cannot push: required sidedata category not supported by this client: '%s'rd � bundle2.pushbacks
push-responses" cannot lock source repository: %s
� remotenames)2r� r � strkwargsrm �localr� r` r � supportedr r( �sortedr r) �canpushr r �read_remote_wanted_sidedata�_sidedata_computers�itemsrn �bytestrr- �
configboolr� �bookmod�bookmarksinstore�wlockr
�transactionmanagerr r� �LockUnavailabler � forcebytestr�debugr �nullcontextmanager� checkpushrz �_pushdiscoveryr� �_checksubrepostate� _pushbundle2�_pushchangeset�_pushsyncphase�
_pushobsolete�
_pushbookmarkr �pullremotenames)r` rm r� ro r� r rk �opargsru r� ry �category�kind� computersr� r
� maypushback�errr5 r5 r6 �pushw s� ��
�� �
��
�����
���
����
�� �� � r� c � � fdd�}|S )a| decorator for function performing discovery before push
The function is added to the step -> function mapping and appended to the
list of steps. Beware that decorated function will be added in order (this
may matter).
You can only use this decorator for a new step, if you want to wrap a step
from an extension, change the pushdiscovery dictionary directly.c �" � t vsJ �| t � < t�� � | S r� )�pushdiscoverymapping�pushdiscoveryorder�append��func��stepnamer5 r6 �dec� �
zpushdiscovery.<locals>.decr5 �r� r� r5 r� r6 �
pushdiscovery� �
r� c C � t D ]
}t| }|| � qdS �zRun all discovery stepsN)r� r� )ru r� �stepr5 r5 r6 r� �
�r� c C s~ | j }t|�d��rt|�d��sdS | jjD ]%}|| }d|�� v r<d|�� v r<t|j�D ]
}|� |�}|j
dd� q.qdS )zDEnsure all outgoing referenced subrepo revisions are present locallys .hgsubs .hgsubstateNT)�onpush) r` rr �filer_ r� �manifest�filesr� �substate�sub�verify)ru r` rW �ctx�subpathr� r5 r5 r6 r�
s
��r� s changesetc C sz t j}| jr|| j| j| j| jd�}n
|| j| j| jd�}|\}}}t j}|| j| j| j|| jd�}|| _|| _|| _ dS )z-discover the changeset that need to be pushed)r� r� )r� )� onlyheads� commonincr� N)
r �findcommonincomingro r` rm r� �findcommonoutgoingr_ r� r� )ru �fcir� ra �incr� �fcor_ r5 r5 r6 �_pushdiscoverychangeset s* �
�
r s phasec
C s | j }| j�� }t| jd�}| j�dd�r)|r)| j js)|�dd�r)g | _ g | _
dS t � | j| j
|�| _| jj}d}| jjs?d}d | }t|�||| j
��}| jjs_| jr_t|�d
| j|��}n!|jse|}nt|�d |j|��} d d
� | D �} t|�|| | j��}|| _ || _
dS )zpdiscover the phase that needs to be pushed
(computed for both success and failure case for changesets push)rh � ui� _usedassubrepori FN� s
and public()s heads((%%ln::%%ln) %s)s %ln and (not public() or %ln::)s roots(%ln + %ln::)c S � g | ]}|� � �qS r5 r� )rV �fr5 r5 r6 rZ q � z'_pushdiscoveryphase.<locals>.<listcomp>)r_ r` r� rl rm r- r� r� rn r� r� r �remotephasessummaryr� rw �
draftroots�
publishing�listr� rk r� )
ru r_ r� rw �droots� extracondr� �fallback�future�fdrootsr5 r5 r6 �_pushdiscoveryphase8 sL
���
�
����
r s obsmarkerc C sf t �| jt j�s
d S | jjsd S dt| jd�vrd S | j}dd� |�d| j�D �}| jj� |�| _
d S )N� obsoletes
namespacesc s r� r� r� r� r5 r5 r6 r� � r� z+_pushdiscoveryobsmarkers.<locals>.<genexpr>� ::%ln) r � isenabledr` � exchangeopt�obsstorerl rm r� r� �relevantmarkersr� )ru r` �nodesr5 r5 r6 �_pushdiscoveryobsmarkersw s r � bookmarksc s� | j }| j�� � | j}|�d� d}| jr&t�� jj | j�}� jj
|dd�}t � t
|d��}� fdd�| jD �}t �� � j|�}t| ||||�S )Ns checking for updated bookmarks
r5 T)� inclusiver c s h | ]}� j �|��qS r5 ��
_bookmarks�
expandname�rV �bookmark�r` r5 r6 � <setcomp>� � �z*_pushdiscoverybookmarks.<locals>.<setcomp>)r- r` r� rm r� ro r �maplistr\ r� � ancestorsr� �unhexlifybookmarksrl r �comparebookmarksr �_processcompared)ru r- rm r% �revnums�remotebookmark�explicitrP r5 r! r6 �_pushdiscoverybookmarks� s
�r, c C s� |\}}}}} }
} } | j }
|D ]!\}}}||v r |�|� |r'|
| �� |v r0| j�|||f� q|D ]*\}}}||v r]|�|� t�|�rT| j�t d�| � d| _
q3| j�|d|f� q3t |�t | � t |
� D ]\}}}||v r�|�|� | j�|||f� qj|D ]\}}}||v r�|�|� | j�||df� q�| D ]\}}}||v r�|�|� q�|r�t |�}| j�t d�|d � d| _
| j�
� dS )zntake decision on bookmarks to push to the remote repo
Exists to help extensions alter this behavior.
s# cannot push divergent bookmark %s!
r" r s> bookmark %s does not exist on the local or remote repository!
r N)r` �remover� r� r� r� � isdivergentr- rq r r� r r� �sort)ru �pushedr+ � remotebmsrP �addsrc�adddst�advsrc�advdst�diverge�differ�invalid�samer` rO �scid�dcidr5 r5 r6 r( � sR
�
�"
�
�
����r( c C s� | j }| j�� }|jst�|j||j� dS | js[|j rVt
d�}t
d�}t
d�}t
d�||d�}|j D ]"}|| }|� � rDt
�|| ��|�� rUt
�||�� d | ��q3t�| � dS ) NFs% push includes obsolete changeset: %s!s, push includes phase-divergent changeset: %s!s. push includes content-divergent changeset: %s!s# push includes orphan changeset: %s!)s orphans phase-divergents content-divergentr T)r_ r` r� r� r �nochangesfoundr- �excludedr� r r r� r r r) �
isunstable�
instabilitiesr �
checkheads) ru r_ r� �mso�mspd�mscd�mstr� r� r5 r5 r6 �_pushcheckoutgoing� s.
�
�
rE c � � �fdd�}|S )aw decorator for function generating bundle2 part
The function is added to the step -> function mapping and appended to the
list of steps. Beware that decorated functions will be added in order
(this may matter).
You can only use this decorator for new steps, if you want to wrap a step
from an extension, attack the b2partsgenmapping dictionary directly.c �: �t vsJ �| t �< � d u rt��� | S t�� �� | S r� )�b2partsgenmapping�b2partsgenorderr� �insertr� ��idxr� r5 r6 r�
�
�z b2partsgenerator.<locals>.decr5 �r� rL r� r5 rK r6 �b2partsgenerator �
rO c
C s� | j sb| jjrdd|j�dd�v }| jdu }|r|r%|jdt| j�d� dS t � }t
� | j�D ]"\}}|\}}} }
|durPt |�} |t |
�| @ O }|| t |� O }q.|rftt |��} |jd| d� dS dS dS dS )zdGenerate race condition checking parts
Exists as an independent function to aid extensions
s relateds
checkheadsr5 Ns check:heads��datas check:updated-heads)
r� r_ r� � capabilitiesrn r� �newpart�iterr� r� r � iteritemsr� )
ru �bundler�allowunrelated� emptyremote�affected�branchr^ r� �newheads�
unsyncedheads�discardedheadsrm rR r5 r5 r6 �_pushb2ctxcheckheads s( �
� �r_ c C s t | jjp
| jp
| jp
| j�S )z&return True if we are pushing anything)�boolr_ r� r� r� r� �ru r5 r5 r6 �_pushing4 s ���rb s check-bookmarksc C sx t | �r| jr dS t�| j�}d|v }| jr|sdS g }| jD ] \}}}|�||f� q t�| j |�}|j
d|d� dS )z insert bookmark move checkingNr s check:bookmarksrQ ) rb r� r � bundle2capsrm r� r� r� � binaryencoder` rT ) ru rW �b2caps�hasbookmarkcheckrR �book�old�new� checkdatar5 r5 r6 �_pushb2checkbookmarks> s
rk s check-phasesc C s� t | �r| jr dS t�| j�}d|�dd�v }| jdurZ|r\dd� tjD �}|tj �
| jj � |tj �
| jj
� tt�|��r^|D ]}|| �� qCt�|�}|jd|d� dS dS dS dS ) zinsert phase move checkingN� headsrh r5 c S � i | ]}|g �qS r5 r5 �rV �pr5 r5 r6 �
<dictcomp>W � z&_pushb2checkphases.<locals>.<dictcomp>s check:phasesrQ )rb r� r rc rm rn rw r � allphases�publicr� � publicheads�draftr �anyr �
itervaluesr/ rd rT )ru rW re �
hasphaseheads�checks�phaserj r5 r5 r6 �_pushb2checkphasesN s
�r{ c s d�j v rdS �j �d� t��sdS �j��� t�|� t��j�}d}|� d�}|rC�fdd�|D �}|s?t
� t d���t
|�}t��j�}tj�j�j|d||d �}|jd|d
�� |re� �d |� t��j�rq� �d d
� tj�jjv r~� �dd
� � �fdd�}|S )zwhandle changegroup push through bundle2
addchangegroup result is stored in the ``pushop.cgresult`` attribute.
�
changesetsNr@ r> c s g | ] }|t �� j�v r|�qS r5 )r
�supportedoutgoingversionsr` �rV �vra r5 r6 rZ u s
�z _pushb2ctx.<locals>.<listcomp>� no common changegroup version� push��
bundlecaps�remote_sidedatarQ r? � treemanifest� 1� exp-sidedatac s8 | j �� j�}t|d �dksJ �|d d d �_dS )z0extract addchangegroup returns from server replyr> r r � returnN)�records�
getreplies�idrr r� )r� � cgreplies��cgpartru r5 r6 � handlereply� s z _pushb2ctx.<locals>.handlereply)r� �addrE r` �prepushoutgoinghooksr_ r rc rm rn r r) r �maxr� r
�
makestreamr_ rT �addparamr �istreemanifestr �REPO_FEATURE_SIDE_DATA�features)ru rW re r4 �
cgversionsr� �cgstreamr� r5 r� r6 �
_pushb2ctxa sF
� � r� c C sn d| j v rdS t�| j�}| jj}d|�dd�v }d|v }d|�dd�v }|r.|s.t| |�S |r5t | |�S dS )z!handle phase push through bundle2rh Nr{ r| � pushkeyrl r5 )
r� r rc rm r` r- r~ rn �_pushb2phaseheads�_pushb2phasespushkey)ru rW re r- � legacyphase�
haspushkeyrx r5 r5 r6 �
_pushb2phases� s
�r� c C s\ | j �d� | jr,dd� tjD �}|d �dd� | jD �� t�|�}|jd|d� d S d S )
z6push phase information through a bundle2 - binary partrh c S rm r5 r5 rn r5 r5 r6 rp � rq z%_pushb2phaseheads.<locals>.<dictcomp>r c s r� r� r� �rV �hr5 r5 r6 r� � r� z$_pushb2phaseheads.<locals>.<genexpr>� phase-headsrQ N)r� r� r� r rr r� rd rT )ru rW �updates� phasedatar5 r5 r6 r� � s
�r� c s� �j �d� g � � fdd�}tj}�jD ]=}|�d�}|�d|d�� |�d||�� �� |�d|dtj �� |�d |dtj
�� � � |j |f� |�j
|j < q� �fd
d �}|S ) z7push phase information through a bundle2 - pushkey partrh c s6 t |j�}� D ]\}}||krt�td�| ��qd S )Ns updating %s to public failed)�int�partidr r) r )ru �exc�targetidr� r� )� part2noder5 r6 �
handlefailure� s
��z+_pushb2phasespushkey.<locals>.handlefailurer� � namespace� key� old� %d� newc s| � D ]9\}}| j �|�}|d }t|�dksJ �d }|s#td�| }nt|d d �s1td�| }|d ur;�j�|� qd S )Nr� r s' server ignored update of %s to public!
r r� � updating %s to public failed!
)r� r� rr r r� r- rq )r� r� r� �partrep�resultsry �r� ru r5 r6 r� � s ��z)_pushb2phasespushkey.<locals>.handlereply)r� r� r �encoder� rT r� r r ru rs r� r� r� )ru rW r� �enc�
newremoteheadrQ r� r5 r� r6 r� � s
r� �
obsmarkersc C s^ d| j v rd S t�|j�}t�|�d u rd S | j �d� | jr-t� | j�}t�
||� d S d S )Nr� ) r� r �obsmarkersversionrS r �
commonversionr� r� r �
sortedmarkers�buildobsmarkerspart)ru rW �remoteversions�markersr5 r5 r6 �_pushb2obsmarkers� s
�r� c C s^ d| j v rdS t�| j�}| jj�dd�}d|v }|s$d|v r$t| |�S d|v r-t| |�S dS )z$handle bookmark push through bundle2r Nr{ r| r� ) r� r rc rm r` r- r~ �_pushb2bookmarkspart�_pushb2bookmarkspushkey)ru rW re �legacy� legacybooksr5 r5 r6 �_pushb2bookmarks� s
�r� c C s | sdS |sdS dS )z"small utility for bookmark pushingr� r� r� r5 )rh ri r5 r5 r6 � _bmaction� s
r� c C s2 |r| j | �� tjkrt�td�| ��dS dS )z6abort if a given bookmark points to a secret changesets: cannot push bookmark %s as it points to a secret changesetN)r` rz r �secretr r) r )ru r� rO r5 r5 r6 �_abortonsecretctx� s
��r� c s� �j �d� �js d S g � g }�jD ] \}}}t�||� |�||f� � �|t||�f� qt��j|�}|j d|d� � �fdd�}|S )Nr rQ c s. �j }� D ]\}}|�t| d | � qd S r8 )r- �status�
bookmsgmap)r� r- rg �action��
allactionsru r5 r6 r� s �z)_pushb2bookmarkspart.<locals>.handlereply)
r� r� r� r� r� r� r� rd r` rT )ru rW rR rg rh ri rj r� r5 r� r6 r� s r� c
s� �j �d� g � tj}� fdd�}�jD ]N\}}}t�||� |�d�}|�d|d�� |�d||�� |�d|t|��� |�d|t|��� d }|sOd
}n|sSd }� � |j
||f� |�j |j
< q� �fd d
�} | S )Nr c s@ t |j�}� D ]\}}}||kr t�t| d �� | ��qJ �)Nr )r� r� r r) r� �rstrip)ru r� r� r� rg r� )� part2bookr5 r6 r� # s
�z._pushb2bookmarkspushkey.<locals>.handlefailurer� r� r� r� r� r� r� r� c s� �j }� D ]M\}}}| j�|�}|d }t|�dks J �|s)�j �td�| � qt|d d �}|r?|�t| d | � q|�t| d | � �j d urRd�_ qd S )Nr� r s" server ignored bookmark %s update
r r� )
r- r� r� rr rq r r� r� r� r� )r� r- r� rg r� r� r� �ret�r� ru r5 r6 r� : s
��z,_pushb2bookmarkspushkey.<locals>.handlereply) r� r� r r� r� r� rT r� r r� r� r� )
ru rW r� r� rg rh ri rQ r� r� r5 r� r6 r� s(
r� � pushvars)rL c C s� | j }|r<i }|D ]}d|vrd}t�|| ��|�dd�\}}|||< q |�d�}t�|�D ]\} }
|j| |
dd� q/dS dS )zsend shellvars via bundle2� =sI unable to parse variable '%s', should follow 'KEY=VALUE' or 'KEY=' formatr r� F�� mandatoryN)r� r r) rL rT r rV r� ) ru rW r� � shellvars�rawry �kr rQ �key�valuer5 r5 r6 �_getbundlesendvarsN s �
�r� c
C s6 t �| jt �| j��}| jo| j�dd�}t �t j| j |dd��}|j
d|d� g }t D ]}t | }|| |�}t
|�r@|�|� q,|jdkrHdS t�|�� �}z�z'| j�� �} | �d |d
g| j�� d ���� }
W d � n1 sqw Y W n tjy� } z t�td �| ��d} ~ ww zd} |r�| jj} t � | j |
| �}
W nE tjy� } z t�td �| ��d} ~ w t j y� } z#| j�td
�| � | j dur�| j�td
�d| j � t�td���d} ~ ww W n' tj �y } zt | j!�}|| j"vr�� | j"| | | � W Y d} ~ nd} ~ ww |D ]}||
� �qdS )z�push data to the remote using bundle2
The only currently supported type of data is changegroup but this will
evolve in the future.rd r� � client)�
allowpushback�roles replycapsrQ r Nr� � force)s bundlerl � url� missing support for %ss remote: %s
s (%s)s push failed on remote)#r �bundle20r- rc rm r� r� �
encodecaps� getrepocapsr` rT rI rH �callabler� �nbpartsr � chunkbuffer� getchunks�commandexecutor� callcommandr �resultr �BundleValueError� RemoteErrorr � transaction�
processbundle�
AbortFromPartrB �
PushkeyFailedr� r� r� )ru rW �pushback�capsblob�
replyhandlers� partgenname�partgenr� �stream�e�replyr� �trgetterr� r� �rephandr5 r5 r6 r� d s| ��
�
��� �� ���
���
�� �r� c C s� d| j v rdS | j �d� t| �sdS | j�d�sJ �| j�| � | j}d}| jdu r@|j s@| jj
j s@t j
| j|ddd|d�}n t j
| j|dd|d�}| jrRd g}n| j}| j�||| j�� �| _dS )
z7Make the actual push of changeset bundle to remote repor| Nr� r@ r� T)�fastpathr� �r� r� )r� r� rE rm r r` r� r_ ro r= r\ � filteredrevsr
�makechangegroupr� r� �unbundler r� )ru r_ r� �cgr� r5 r5 r6 r� � s:
��� � r� c
sV | j }t| jd�}| j�dd�r!|r!| jdu r!|�dd�r!ddi}|s*t| |� dS t� | j
||�}|\� }|�dd�rBt| |� n t| � � t| |tj � | jr\d| j v rXdS | j
}n| j}| j �d� � fdd �|D �}|D ]8}| j�� � }|�d
d|�� d tj d tj d ���� }W d � n1 s�w Y |s�| j�td
�| � qpdS )z2synchronise phase information locally and remotelyrh r r Nri Fs Truec s g | ]
}|� � � vr|�qS r5 r� r� ��pheadsr5 r6 rZ � r� z"_pushsyncphase.<locals>.<listcomp>r� r� �r� r� r� r� r� )r� rl rm r- r� r� rn �_localphasemover �analyzeremotephasesr` ru r� r� r� r� r� r� r rs r� rq r ) ru r� rw �anar �outdatedr� r� �rr5 r r6 r� � sX ��
�
��� �
���r� c s` �j rt��j�j �� � |� dS � �fdd�|D �}tj� }|r.�j�td�| � dS dS )z0move <nodes> to <phase> in the local source repoc s" g | ]
}� �j | �� k r|�qS r5 )r` rz rU �rz ru r5 r6 rZ � " z#_localphasemove.<locals>.<listcomp>s8 cannot lock source repo, skipping local %s phase update
N) r� r �advanceboundaryr` r� �
phasenamesr- r� r )ru r rz � actualmoves�phasestrr5 r r6 r s �
����r c C s� d| j v rdS | j}| j}| j �d� | jrT| j�d� g }t�| j�}t �
|�}t |dd�D ]}|| }|� |�
d|d|�� q/dd � |D �rVtd
�}|j�|� dS dS dS ) z5utility function to push obsolete markers to a remoter� Ns' try to push obsolete markers to remote
T��reverser r c S s g | ]}|s|�qS r5 r5 )rV r r5 r5 r6 rZ 7 r z!_pushobsolete.<locals>.<listcomp>s& failed to push some obsolete markers!
)r� r` rm r� r� r- r� r r� r �_pushkeyescaper� r� r r rq ) ru r` rm �rsltsr� �
remotedatar� rR ry r5 r5 r6 r� ' s$
� r� c
C s� | j dks
d| jv r dS | j�d� | j}| j}| jD ]V\}}}d}|s'd}n|s+d}|�� �}|�dd|t|�t|�d��� � }W d � n1 sKw Y |r^|�
t | d | � q|� t | d | � | j
durqd | _
qdS )
z"Update bookmark position on remoter r Nr� r� r� r� r r )r� r� r� r- rm r� r� r� r r� r� r� rq r� ) ru r- rm rO rh ri r� r� r r5 r5 r6 r� < s<
��� �
��r� c @ s^ e Zd ZdZ ddd�Zejdd� �Zejd d
� �Zejd d � �Z d
d� Z
dS )�
pulloperationz�A object that represent a single pull operation
It purpose is to carry pull related state and very common operation.
A new should be created at the beginning of each pull and discarded
afterward.
NFr5 c s� � | _ || _| | _|| _� fdd�|D �| _|| _|| _d | _d | _d | _ d | _
|| _ d | _ t
� | _d| _|| _| | _|
| _d S )Nc s g | ]}� j �|��qS r5 r r r! r5 r6 rZ � r# z*pulloperation.__init__.<locals>.<listcomp>F)r` rm � remote_pathr^ �explicitbookmarksr� �streamclonerequestedr� ra �rheads�fetch�remotebookmarksr� r� r� �clonebundleattempted� includepats� excludepats�depth) r� r` rm r^ r� r r r r r r �pathr5 r! r6 r� h s(
�
zpulloperation.__init__c C sF | j du r t| j�}t| j�}| jD ] }||vr |�|� q|S | j S )z0heads of the set of changeset target by the pullN)r^ r� ra r r r� )r� r� r� rW r5 r5 r6 � pulledsubset� s
�zpulloperation.pulledsubsetc C s
t | � S r� )r� r� r5 r5 r6 �
canusebundle2� �
zpulloperation.canusebundle2c C s t �| j�S r� )r rc rm r� r5 r5 r6 �remotebundle2caps� s z pulloperation.remotebundle2capsc C s
| j �� S r� )r� r� r� r5 r5 r6 �gettransaction� r" z pulloperation.gettransaction) NFr5 NNNNNN) r� r� r� r� r� r r� r r! r# r$ r5 r5 r5 r6 r _ s&
�7
r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) r� z�An object to manage the life cycle of a transaction
It creates the transaction on demand and calls the appropriate hooks when
closing the transaction.c C s || _ || _|| _d | _d S r� )r` �sourcer �_tr)r� r` r% r r5 r5 r6 r� � s
ztransactionmanager.__init__c C sL | j s#d| jt�| j�f }| j�|�| _ | j| j jd< | j| j jd< | j S )z<Return an open transaction object, constructing if necessarys %s
%s� sourcer� )r& r% r � hidepasswordr r` r� �hookargs)r� �trnamer5 r5 r6 r� � s z transactionmanager.transactionc C � | j dur | j �� dS dS )z close transaction if createdN)r& �closer� r5 r5 r6 r, � �
�ztransactionmanager.closec C r+ )z release transaction if createdN)r& �releaser� r5 r5 r6 r. � r- ztransactionmanager.releaseN)r� r� r� r� r� r� r, r. r5 r5 r5 r6 r� � s r� c C s@ | � � �}|�dd|i��� W d � S 1 sw Y d S )N� listkeysr� )r� r� r� )rm � namespacer� r5 r5 r6 rl � s
$�rl c s� | � � � � j�� fdd�}� fdd�} ��� }t��}t|� tj| jv r(d S |t��kr0d S t�fdd�|jD ��r>d S |��� |�}|||j �|_ t
|j�|j |_q)Nc � � � d| |�}dd� |D �S )zReturns heads(h1 % h2)s heads(%ln %% %ln)c S � h | ]}|� � �qS r5 r� �rV r� r5 r5 r6 r" � r z8_fullpullbundle2.<locals>.headsofdiff.<locals>.<setcomp>�r� ��h1�h2�res�r� r5 r6 � headsofdiff� � z%_fullpullbundle2.<locals>.headsofdiffc r1 )z Returns heads((h1 + h2) - null)s heads((%ln + %ln - null))c S r2 r5 r� r3 r5 r5 r6 r" � r z9_fullpullbundle2.<locals>.headsofunion.<locals>.<setcomp>r4 r5 r9 r5 r6 � headsofunion� r; z&_fullpullbundle2.<locals>.headsofunionTc 3 s � | ]}� � |�V qd S r� rX rU )�unficlr5 r6 r� � � z#_fullpullbundle2.<locals>.<genexpr>) r� r\ r^ rr � _pullbundle2r �NARROW_REQUIREMENT�allr ra r� )r` �pullopr: r< � old_heads�clstart� new_headsr5 )r� r= r6 �_fullpullbundle2� s$
�rF c sF |j �� }tj| |ddd� t�| �� �� � fdd�}|�d|� dS )z�adds a finalize callback to transaction which can be used to show stats
to user and confirm the pull before committing transaction� pullT)�txnname� as_validatorc s( � � }t d�}|j�|�rt�d��d S )Ns+ accept incoming changes (yn)?$$ &Yes $$ &Nos user aborted)r r- rs r r) )�trr` �cm��reporefr5 r6 �prompt s
�z$add_confirm_callback.<locals>.prompts 900-pull-promptN)r� r� r �registersummarycallback�weakref�refr� � addvalidator)r` rB rJ rN r5 rL r6 �add_confirm_callback
s
� rS c C s, |du ri }|s
| r|pt � }| pt � } n| j\}} t�|� t�| � t| |f||||||| |
d�t�|���} | j�� }
|
r[t |
j �| j
j }|r[t d�d�
t|�� }t�|��| jD ] }| j�� D ] \}}|�|�rp n
qet�t d�t�|� ��q^t| d|�� �| _t�� }t � | �s�| � � }|�g | � � �S | j�@ | s�| j �!dd�r�| j �"� s�t#| | � t$| � t%�&| � t'| � | j(r�t)| | � t*| � t+| � t,| � t-| � W d � n1 s�w Y W d � n1 s�w Y W d � n 1 �sw Y | j �!dd ��rt.�/| |� | S )
a� Fetch repository data from a remote.
This is the main function used to retrieve data from a remote repository.
``repo`` is the local repository to clone into.
``remote`` is a peer instance.
``heads`` is an iterable of revisions we want to pull. ``None`` (the
default) means to pull everything from the remote.
``bookmarks`` is an iterable of bookmarks requesting to be pulled. By
default, all remote bookmarks are pulled.
``opargs`` are additional keyword arguments to pass to ``pulloperation``
initialization.
``streamclonerequested`` is a boolean indicating whether a "streaming
clone" is requested. A "streaming clone" is essentially a raw file copy
of revlogs from the server. This only works when the local repository is
empty. The default value of ``None`` means to respect the server
configuration for preferring stream clones.
``includepats`` and ``excludepats`` define explicit file patterns to
include and exclude in storage, respectively. If not defined, narrow
patterns from the repo instance are used, if available.
``depth`` is an integer indicating the DAG depth of history we're
interested in. If defined, for each revision specified in ``heads``, we
will fetch up to this many of its ancestors and data associated with them.
``confirm`` is a boolean indicating whether the pull should be confirmed
before committing the transaction. This overrides HGPLAIN.
Returns the ``pulloperation`` created for this pull.
N)r r^ r� r r r r r r� r� sD sidedata category requested by local side without localsupport: '%s'rG rf rd r� )0r� �
narrowpatsr �validatepatternsr r r� rm r� r r` r� r r( r� r r) �_wanted_sidedatar� r� rn �ProgrammingErrorr� r� r r� r r� r� r� r� r
r- r� �plainrS �_maybeapplyclonebundler � maybeperformlegacystreamclone�_pulldiscoveryr! rF �_pullchangeset�
_pullphase�_pullbookmarks�
_pullobsoleter r� )r` rm r r^ r� r r� r r r r �confirmrB � peerlocalr� ry r� r� r� r� r5 r5 r6 �pull s� *
�� �
� �
�����
��
�� � rb c r� )a| decorator for function performing discovery before pull
The function is added to the step -> function mapping and appended to the
list of steps. Beware that decorated function will be added in order (this
may matter).
You can only use this decorator for a new step, if you want to wrap a step
from an extension, change the pulldiscovery dictionary directly.c r� r� )�pulldiscoverymapping�pulldiscoveryorderr� r� r� r5 r6 r� � r� zpulldiscovery.<locals>.decr5 r� r5 r� r6 �
pulldiscovery� r� re c C r� r� )rd rc )rB r� r� r5 r5 r6 r[ � r� r[ s b1:bookmarksc C s> | j durdS | jrd| jv rdS t| jd�}t�|�| _ dS )z�fetch bookmark data in bundle1 case
If not using bundle2, we have to fetch bookmarks before changeset
discovery to reduce the chance and impact of race conditions.Nr/ r )r r! r# rl rm r� r& )rB �booksr5 r5 r6 �_pullbookmarkbundle1� s
rg r> c C s� t j| j| j| j| jd�}|\}}}| j�� jjj }|r?|r?t
|�}|D ]}||�r3||vr3|� |� q$t
|�� t
|��r?g }|| _
|| _|| _dS )z~discovery phase for the pull
Current handle changeset discovery only, will change handle all discovery
at some point.)r^ r� N)r r� r` rm r^ r� r� r\ r� �has_noder� r� �issubsetra r r )rB �tmpra r r rh �scommonrW r5 r5 r6 �_pulldiscoverychangegroup� s" �
�
rl c C s` dt | jdd�i}| jj}tj| dd�d }| j|d< | jp | j|d< | j� t
j �}|r4| j r4| j |d < |r>| j
r>| j
|d
< |rUd |d < d|d
< | j�d� | j�d� n9| j�d� | j|d < d|�dd�v }d| j�dd�v }|s|rd|d< | j�d� d| jv r�d| jvr�dg|d< d }d|�dd�v }d| jv } | jdur�| j�d� d| jvr�| jdu r�|s�| r�d|d< d}d| jv r�d| jvr�| j�d� |�dg ��d� | j� d�r�| jdu r�t| j�| jjgkr�| j|d< |r�| jj�td�� n+| j�s | jj�td�� d| _ n| jdu �r&t| j�| jjgk�r&| jj�td�� t � | jt j ��rGt �!| j�}
t �"|
�du�rGd|d < | j�d � t#| |� t �$| j�} | �rY| |d < | j�%� �l} t&|�}
d |
d < | �'d |
��(� }zt j)| j| j*d d!�}d"|j+d< t j,| j||d#� W n9 t j-�y� } z| jj�.td$�| � t.j/td%�|j0d&��d}~w t.j1�y� } z t.�/td'�| ��d}~ww W d � n 1 �s�w Y | j�r�t �2|�| _ |j3d D ]\}}|dk�r�t4| |� �q�|�r i }|j3d D ] }|d( ||d) < �q�|| _n|j3d D ]\}}|dk�r t5�6|�| _�q| jdu�r.t7| � dS dS )*zNpull data using bundle2
For now, the only supported data are changegroup.s
bundlecapsr� �r� T)r r � commonrl s includepatss excludepatsF� cgr r> rh r{ r| r5 r/ r Ns request-bookmarks� clonebundless cbattempteds streaming all changes
� no changes found
� requesting all changes
r� s remote_sidedatarG r' � getbundle)r% s records�r� s remote: abort: %s
s pull failed on remoterA r� s nodes bookmark)8�
caps20to10r` r- r �canperformstreamclonera r^ r rm r r � NARROWCAPr r r� r� r r~ r# rn r �
setdefaultr� r r] r r� r r� r r r r r� r� �_pullbundle2extrapreparer� r� �dictr� r� �bundleoperationr$ �modesr� r� r r� rB r� �combinechangegroupresultsr� �_pullapplyphasesr� r& r^ )rB �kwargsr- � streaming� servernarrowr� �hasbinaryphase�bookmarksrequested�legacybookmark�
hasbinarybookr� r� r� �args�bundler� r� r0 r� rf �recordr5 r5 r6 r? � s�
��
�
"
�
���� �
�
� �r? c C s dS )z>hook function so that extensions can extend the getbundle callNr5 )rB r r5 r5 r6 ry � s ry c C s� d| j v rdS | j �d� | js | jj�td�� d| _dS | �� }| j du r;t
| j �| jj gkr;| jj�td�� n| j du rJ| j
�d�rJ| j| _ | j
�d�r_| j
jd| j | j p[| jd �}nW| j du r�| j
�� �}|�d| jdd
���� }W d � n1 s�w Y n0| j
�d�s�t�td ���| j
�� �}|�d| j| j dd ���� }W d � n1 s�w Y t�| j||d| j
�� �}t�|�| _dS )
z0pull changeset from unbundle into the local repor> Nrq r rr s changegroupsubsetrs rG �ra r^ )s nodesr' sW partial pull cannot be done because other repository doesn't support changegroupsubset.)s basesrl r' )r� r� r r` r- r� r r� r$ r^ r ra r] rm r r � getbundler� r� r� r r) r � applybundler r} )rB rJ r r� �bundleopr5 r5 r6 r\ � sb
�
��� �� �� ��� �
�r\ c C s( d| j v rd S t| jd�}t| |� d S )Nrh )r� rl rm r~ )rB rw r5 r5 r6 r] � s
r] c s� d| j v rdS | j �d� t|�dd��}|r(|s(t�| j| j|�\}}| j}n| j}g }| j�� ��j j
��j j j
�tj�tj� ����fdd�|D �}|r\| �� }t�| j|�|� � ���fdd�|D �}|ry| �� }t�| j|� |� dS dS )z/apply phase movement from observed remote staterh Nri Fc s" g | ]
}� ��|���kr|�qS r5 r5 �rV �pn)rz rs r� r� r5 r6 rZ � r
z$_pullapplyphases.<locals>.<listcomp>c s" g | ]
}���|��� kr|�qS r5 r5 r� )ru rz r� r� r5 r6 rZ � r
)r� r� r` rn r r r` r r� � _phasecacherz r\ r� �get_revrs ru r$ r )rB rw r
r �_dr�dheadsrJ r5 )ru rz rs r� r� r6 r~ � s2
�
�r~ c C sd d| j v rdS | j �d� | j}| j}d}| jdur | jj}tj|j||| j �
� | j | j |d� dS )z?process the remote bookmark information to update the local oner N)r+ �mode)
r� r� r` r r �bookmarks_moder� �updatefromremoter- rm r r$ r )rB r` r r� r5 r5 r6 r^ � s"
�r^ c C s� d| j v rdS | j �d� d}t�| jtj�r]| jj�d� t| j d�}d|v r]| �
� }g }t |dd�D ]}|� d�rMt
�|| �}t�|�\}}||7 }q4|rX| jj�||� | j�� |S ) aV utility function to pull obsolete markers from a remote
The `gettransaction` is function that return the pull transaction, creating
one if necessary. We return the transaction to inform the calling code that
a new transaction have been created (when applicable).
Exists mostly to allow overriding for experimentation purposer� Ns! fetching remote obsolete markers
r s dump0Tr s dump)r� r� r r r` r r- r� rl rm r$ r� r& r � b85decode� _readmarkersr �invalidatevolatilesets)rB rJ � remoteobsr� r� rR r4 �newmarksr5 r5 r6 r_ s(
�
r_ c
C s | j }|�|j�d�p|�� �}|�t|d |�td��}|�t|d |�td��}|s7t�t d�t|f ��dd� |D �}d d� |D �}t
|�d
g ��}t
|�d g ��}t � ||||�\}}}|rpt�t d �|t
�|�f ��i } | �|� d
| d< d
| d< || d
< |r�|| d < | S )z�Apply narrow fetch access control.
This massages the named arguments for getbundle wire protocol commands
so requested data is filtered through access control rules.
s REMOTE_USERs .includess default.includess .excludess default.excludess% %s configuration for user %s is emptyc S � g | ] }|d kr
dnd| �qS �� *s path:.s path:r5 rn r5 r5 r6 rZ 6 � �z"applynarrowacl.<locals>.<listcomp>c S r� r� r5 rn r5 r5 r6 rZ 9 r� r r s4 The following includes are not accessible for %s: %sT�narrow�
narrow_acl)r- � shortuser�environrn �usernamer~ �_NARROWACL_SECTIONr r) r r� r �restrictpatternsr �pprint�update)
r` r r- r� �
user_includes�
user_excludes� req_includes� req_excludes�invalid_includes�new_argsr5 r5 r6 �applynarrowacl sV
�
�����
� ��
r� c s� �j � �j}� j��fdd�|D �thB }�fdd�|D �}|r'dd� |D �} t�t�}
t�t�����fdd�} �fdd ����fd
d ��t� j||d ��} t | �}
t� }� fd
d�| D �}t|�|B }|
D ]�}� �
|��dd� � � |�D �}|dur�| | }|D ]}t |d | �
||d ��| |< q�d}|du p�| | |k}|r�|�j �� }|r�|� �
|d �j �� }t|�||��}|s�t|�dkr�|� �
|d �j �� }t|�||��}nt|�|��}|�r|
| D ]}| ||� q�|D ]}|�|� q�|�� �|�� qg|�s tg}||v �r+|
| D ]}| ||� �q|D ]
}|
| �|� �q qg|D ]
}|
| |
| O < �q-qg|D ]}|
| D ]}| ||� �qD�q>||�fS )a Compute the shape of a narrowed DAG.
Args:
repo: The repository we're transferring.
common: The roots of the DAG range we're transferring.
May be just [nullid], which means all ancestors of heads.
heads: The heads of the DAG range we're transferring.
match: The narrowmatcher that allows us to identify relevant changes.
depth: If not None, only consider nodes to be full nodes if they are at
most depth changesets away from one of heads.
Returns:
A tuple of (visitnodes, relevant_nodes, ellipsisroots) where:
visitnodes: The list of nodes (either full or ellipsis) which
need to be sent to the client.
relevant_nodes: The set of changelog nodes which change a file inside
the narrowspec. The client needs these as non-ellipsis nodes.
ellipsisroots: A dict of {rev: parents} that is used in
narrowchangegroup to produce ellipsis nodes with the
correct parents.
c � h | ]}� |��qS r5 r5 rU ��clrevr5 r6 r" q r z#_computeellipsis.<locals>.<setcomp>c r� r5 r5 rU r� r5 r6 r" r r c S s i | ]}|d �qS )r r5 r� r5 r5 r6 rp u rq z$_computeellipsis.<locals>.<dictcomp>c sR � | � |� t� | �dkr'�| �\}}�| ||� |} t� | �dksdS dS )z=Add a root to an ellipsis head, splitting heads with 3 roots.r" N)r� rr )�head� curchange�child�roots)�
ellipsisroots� splithead�
splitrootsr5 r6 �addrootz s �z!_computeellipsis.<locals>.addrootc s< � | � |� � | �|� � | �|� � | �|� d S r� )�difference_updater� r� �discard)r� r� r� )r� r5 r6 r� � s z$_computeellipsis.<locals>.splitrootsc s� t � | �\}}}||f||f||ffD ]6\}}��d|| || �}|D ]&}||kr3|||ff S |� vs?t� | �dk rI|||ff S q#qt�td�| |||f ��)Ns% sort(merge() & %d::%d & %d::%d, -rev)r" s; Failed to split up ellipsis node! head: %d, roots: %d %d %d)r� ro rr r r) r )r� �r1�r2�r3�nr1�nr2�mid�j)r� r` r5 r6 r� � s&
����
��z#_computeellipsis.<locals>.splitheadr� c s g | ]}� � |��qS r5 r� )rV �m)rb r5 r6 rZ � s z$_computeellipsis.<locals>.<listcomp>c S s g | ]}|t kr|�qS r5 )r )rV �prevr5 r5 r6 rZ � r[ Nr Fr )r\ � manifestlogr� r � collections� defaultdictr� r �findmissingrevs�reversed�changelogrevision�
parentrevs�minrn r� �readr` �diffrr rv �walkr� r� ) r` ra r^ �known�matchr �mfl�
commonrevs� headsrevs�revdepth�
ellipsisheadsr� r� �visit�relevant_nodes�
visitnodes�requiredr� �ps�curdepthro �needed�shallow_enough�curmf�p1mf�p2mfr� r� r5 )rb r� r� r` r� r� r6 �_computeellipsisU st
�
�� �
r� c C s2 dh}t �t j| |d��}|�dt�|� � |S )zFreturn a set with appropriate options to use bundle20 during getbundles HG20rm � bundle2=)r r� r� r� rJ �quote)r` r� �capsr� r5 r5 r6 ru � s ru c rF )a� decorator for function generating bundle2 part for getbundle
The function is added to the step -> function mapping and appended to the
list of steps. Beware that decorated functions will be added in order
(this may matter).
You can only use this decorator for new steps, if you want to wrap a step
from an extension, attack the getbundle2partsmapping dictionary directly.c rG r� )�getbundle2partsmapping�getbundle2partsorderr� rJ r� rK r5 r6 r� � rM z%getbundle2partsgenerator.<locals>.decr5 rN r5 rK r6 �getbundle2partsgenerator� rP r� c C s | d ur
t dd� | D ��S dS )Nc s s � | ]}|� d �V qdS )s HG2N)r& )rV �capr5 r5 r6 r� r> z#bundle2requested.<locals>.<genexpr>F)rv r� r5 r5 r6 �bundle2requested s r� c K s4 t �|�}i }t|�}|sC|r|�dd�sttd���|r,ttd�d�t|�� �� ��t | ||�} d|d< |t
j | | d|||d �fS d
|d< i }
|D ]} | � d �ret
�| td �d � �} |
�t�| �� qKt�| j|
�}
||d
< ||d< tD ]}t| }||
| |f||
|d�t �|��� qw|
j|d< ||
�� fS )a Return chunks constituting a bundle's raw data.
Could be a bundle HG10 or a bundle HG20 depending on bundlecaps
passed.
Returns a 2-tuple of a dict with metadata about the generated bundle
and an iterator over raw chunks (of varying sizes).
ro Ts- request for bundle10 must include changegroups# unsupported getbundle arguments: %sr� r s
bundleversionr@ r� r" r� Nrl rn )r� re r� s prefercompressed)r � byteskwargsr� rn �
ValueErrorr r( r� �keysrc r
r� r& rJ rK rr r� r �
decodecapsr� r- r� r� r� �prefercompressedr� )r` r% r^ ra r� r� r �info�
usebundle2r_ re �bcaps�blobrW �namer� r5 r5 r6 �getbundlechunks sf
��� ��
����
r� rC c O s t j| |fi |��S r� )r �addpartbundlestream2)rW r` r� r r5 r5 r6 �_getbundlestream2R s r� c s� |� dd�r|s
dS d} |� d�}
|
r)� fdd�|
D �}
|
s%t�td���t|
�} t� ||�} | js4dS |� d d
�rZttt |� d g ���} ttt |� d g ���}
t
j � j | |
d
�}nd}t
j� | | ||||d�}| jd|d�}|
rw|�d| � |jddt| j� d
d� t�� �r�|�dd� tj� jv r�|�dd� t�� �}|�d|� |� d d
�r�|� dd
�r�| s�|
r�| �d�}dd�| �d�|
�f |_dS dS dS dS ) z.add a changegroup part to the requested bundler TNr@ r> c s g | ] }|t �� �v r|�qS r5 )r
r} r~ r! r5 r6 rZ j s
�z-_getbundlechangegrouppart.<locals>.<listcomp>r� r� Fr r )�include�exclude)r� �matcherr� rQ r? s nbchangesr� r� r� r� r� s exp-wanted-sidedatar� s Narrow:responsespecs %s %s�
)rn r r) r r� rc r� r� �filterr` r r� �rootr
r� rT r� rr r r� r r� r� r � format_remote_wanted_sidedatar( rR )rW r` r% r� re r^ ra r� r r4 r� r_ r� r� r� r� rQ �sidedata�narrowspecpartr5 r! r6 �_getbundlechangegrouppartW sh
� �
�
���
� �r c K sZ |� dd�sdS |rd|vrt�td���t�|�}t�||�}|r+| jd|d� dS dS )z+add a bookmark part to the requested bundler FNr s# no common bookmarks exchange methodrQ )rn r r) r r� �listbinbookmarksrd rT )rW r` r% r� re r rf rR r5 r5 r6 �_getbundlebookmarkpart� s
�r r/ c
K sJ |� dd�}|D ]}| �d�}|�d|� |�|��� } t�| �|_qdS )z@add parts containing listkeys namespaces to the requested bundlerl r5 r/ r� N)rn rT r� rl r� r �
encodekeysrR )
rW r` r% r� re r rl r0 rQ r� r5 r5 r6 �_getbundlelistkeysparts� s
�r c K s\ |� dd�r,|du r|�� }dd� |�d|�D �}|j�|�}t�|�}t�| |� dS dS )z8add an obsolescence markers part to the requested bundle�
obsmarkersFNc S r r5 r� r� r5 r5 r6 rZ � r z+_getbundleobsmarkerpart.<locals>.<listcomp>r ) rn r^ r� r r r r� r r� ) rW r` r% r� re r^ r �subsetr� r5 r5 r6 �_getbundleobsmarkerpart� s
�r rh c s> |� dd�r�|rd|� d�vrt�td���|du r |�� }t�t�� |�� r-|� t j
< nX|j j }|j
j}|j
j} |D ]}
� ||| |
�� �|
� q;t� �� �} | r[t| �t jks[J �� � t jt� �} | r�� � t j
t� �}
d}|�|| |
�}|D ] }� t j
�||�� qx� fdd �t jD �}t �|�}| jd
|d � dS dS ) z,add phase heads part to the requested bundler Frl rh s no common phases exchange methodNs" heads(only(%ln, %ln) and public())c s i | ] }|t � | ��qS r5 )r� )rV rz �� headsbyphaser5 r6 rp � s �z(_getbundlephasespart.<locals>.<dictcomp>r� rQ )rn r r) r r^ r� r� r� r
r rs r� rz r\ r� r� r� r r� r� ru ro rr rd rT )rW r` r% r� re r^ r rz r� r� r� �
seenphases�
draftheadsrt r� �
extraheadsr � phasemappingr� r5 r r6 �_getbundlephasespart� s:
�
�r � hgtagsfnodesc K s: |r |� dd�r d|v sdS t|||�}t�|| |� dS )z�Transfer the .hgtags filenodes mapping.
Only values for heads in this bundle will be transferred.
The part data consists of pairs of 20 byte changeset node and .hgtags
filenodes raw values.
r Tr N)rn rc r �addparttagsfnodescache� rW r` r% r� re r^ ra r r_ r5 r5 r6 �_getbundletagsfnodes� s r s cache:rev-branch-cachec K sR |� dd�r|rd|vs|� dd�s|j�t�rdS t|||�}t�|| |� dS )z�Transfer the rev-branch-cache mapping
The payload is a series of data related to each branch
1) branch name length
2) number of open heads
3) number of closed heads
4) open heads nodes
5) closed heads nodes
r Ts rev-branch-cacher� FN)rn r- � has_sectionr� rc r �addpartrevbranchcacher r5 r5 r6 �_getbundlerevbranchcache
s
��
�
� r c C sX | � � }t�d�t|����� }|dgks&||ks(|d|gks*t�d| ��dS dS dS )zVcheck if the heads of a repo have been modified
Used by peer for unbundling.
r r� s hasheds. repository changed while %s - please try againN)r^ r �sha1r( r� �digestr � PushRaced)r` � their_heads�contextr^ �
heads_hashr5 r5 r6 � check_heads@
s
��r c sV d�g d�� d}�j �dd�}��d�s��d�rd}z�t�|d � t|tj�sjd
��t� ��g�}��
� �+ �� |��}t� �||���} t�
| ��W d � n1 sUw Y W d � n1 sdw Y n�d�zd� ���fd d �}
�j �dd
�s�|
� tj�|
|dd�} z!tj�|| d�} W | j�|r��dur��j jddd� �fdd�}n| j�|rdurj jddd� �fdd�}w � d durψ d �� W n# ty� } zd| _|r�dur��� �| _�fdd�}� d} ~ ww W t�� d � d � d � |du�r|�j �� � �S t�� d � d � d � |du�r*|�j �� � w w )a Apply a bundle to a repo.
this function makes sure the repo is locked during the application and have
mechanism to check that no push race occurred between the creation of the
bundle and its application.
If the push was raced as PushRaced exception is raised.r �NNNNrd s bundle2-output-captures remote:http:s
remote:https:Ts uploading changesr� c sj � d s1t ���s��� � d< ��� � d< ����� d< �� d jd< �� d jd< d� d jd< � d S )Nr" r r r' r� r� r} )r� r� r� r
r� r) r5 )� lockandtrr` r% r r5 r6 r$ u
s
z unbundle.<locals>.gettransactions bundle2lazylockingr� )�
captureoutputr% rt )r �subprocc s � j d| dd� d S �Ns outputF)rR r� )rT )�output)r r5 r6 � recordout�
s zunbundle.<locals>.recordoutr" c s t jd| dd�}� �|� d S r" )r �
bundlepartr� )r# rQ )�partsr5 r6 r$ �
s �r )r- r� r& r rD r rF r( r r( r
r� r� r} r{ r� r� �
pushbufferr, �
BaseException�duringunbundle2�
salvageoutput�_bundle2salvagedoutput�lockmodr. � popbuffer) r` r r^ r% r r$ r rH rJ r� r$ r� r5 )r r& r r` r% r r6 r� S
s~ � �� �
�� �� � ���
�
�r� c C sf | j }| j}|j�dd�sdS t|�rdS | jrdS |�d�s!dS |�� �}|�di �� � }W d � n1 s8w Y d| _
t � ||�}|sR|j�
td�� dS t j||| jd�}|so|j�td�� |j�td�� dS t �|j|�}|d d
}|j�td �| � t|j||�r�|j�td �� dS |j�dd
�r�|j�td�� dS tjtd�td�d��)z0Apply a clone bundle from a remote, if possible.r rp NTsD no clone bundles available on remote; falling back to regular clone
)r sO no compatible clone bundles available on server; falling back to regular clone
s5 (you may want to report this to the server operator)
r s URLs applying clone bundle from %s
s finished applying clone bundle
s clonebundlefallbacks falling back to normal clone
s error applying bundles} if this error persists, consider contacting the server operator or disable clone bundles via "--config ui.clonebundles=false"rA )r` rm r- r� rr r^ r r� r� r� r r �parseclonebundlesmanifest�noter �filterclonebundleentriesr rq �sortclonebundleentriesr� �trypullbundlefromurlr r) )rB r` rm r� r8 �entriesr r5 r5 r6 rY �
s`
� ������ ��rY c C sF |� � �� |�d��}z0t�| |�}t| |d�}t|tj�r$|�|� n t �
|||d|� W W d � W d � dS t j yZ } z| �
td�t�|� � W Y d}~n$d}~w t jyy } z| �
td�t�|j� � W Y d}~nd}~ww W d � W d � dS 1 s�w Y W d � dS 1 s�w Y dS ) z%Attempt to apply a bundle from a URL.s bundleurlr rp NTs HTTP error fetching bundle: %s
s error fetching bundle: %s
F)r
r� �urlmod�openr7 rD r r, �applyr r� �urlerr� httperrorrq r r r� �urlerror�reason)r- r` r rJ r. r r� r5 r5 r6 r2 �
s4 �
���
����P�r2 r� r� )
NNFr5 NNNNNN)NNNN)NNNNN)NNr )�
__future__r r� rP �i18nr r� r r � r r� r r r
r r r
r, r r r r r r r r r r r r4 r r �utilsr r r �
interfacesr r7 rJ r� r7 rT rc rz r� �objectr� r� r� r� r� r� r� r� r r r r, r( rE rI rH rO r_ rb rk r{ r� r� r� r� r� r� r� r� r� r� r� r� r� r� rs r r� r� r �
transactionalr� rl rF rS rb rd rc re r[ rg rl r? ry r\ r] r~ r^ r_ r� r� ru r� r� r� r� r� r� r r r r r r r r r� rY r2 r5 r5 r5 r6 �<module> sH X
D& ����
�q
>
6'
6
%
0
;0B#_ *
�
;$
8
�K
�K � � �
�0 � �%Y R