| Server IP : 162.144.4.212 / Your IP : 216.73.216.108 Web Server : Apache System : Linux gator2125.hostgator.com 5.14.0-162.23.1.9991722448259.nf.el9.x86_64 #1 SMP PREEMPT_DYNAMIC Wed Jul 31 18:11:45 UTC 2024 x86_64 User : cozeellc ( 2980) PHP Version : 8.3.31 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : ON Directory : /lib64/python3.9/site-packages/mercurial/__pycache__/ |
Upload File : |
a
�+�b�z �
@ sD 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 �Z1d�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�� Zqd�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 }|sBd}|�d�sP|�d�rPt �||�}d}d}n|rP|�|�}|dd� |dd� }}|dkr�t�td �| ��|d
kr�|d u r�t �|d�}t �||�S |�d �r�t j
| ||| d �S |d
kr�t � |�S t�td�||f ��d S )N� � streams HG� s HG10s UNr � s %s: not a Mercurial bundles 10� 2)Z magicstrings S1s %s: unknown bundle version %s)
r
Z readexactly�
startswithZheaderlessfixup�joinr �Abortr � cg1unpackerr Z getunbundlerr �streamcloneapplier)�ui�fh�fnameZvfs�header�alg�magic�version� r0 �8/usr/lib64/python3.9/site-packages/mercurial/exchange.py�
readbundle5 s0
�r2 c C s� dd� }t | |d�}t|tj�rX|j}|dkr2d}||�}|sPt�td�| ��d| S t|tj ��rJd|j
v r�||j
d �}|s�t�td�| ��nd }d}|� � D ]�}|j d
kr�|j
d }|d v r�d
}ntjtd�| td�d��q�|j dkr�|du r�t
�|j
d �}|�� } t�| �}
d|
S q�|�s>t�td���d||f S t|tj��rxt�|�d }t�|�} d| S t�td�| ��dS )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 s0 zt j�| ��� d W S ty* Y d S 0 d S �Nr )r Z compenginesZ
forbundletypeZ
bundletype�KeyError)r- r0 r0 r1 �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)r2 �
isinstancer
r' Z_typer r& r r �
unbundle20�paramsZ iterparts�type�urlreq�unquote�splitZ_formatrequirementsparamsr r( Zreadbundle1header) r) r* r5 �br- �compr/ �partr Zsplittedr> � formattedr0 r0 r1 �
getbundlespecT s^
�
���
�
rG c sF | j }|r$|j� � fdd�|D �}n| jg}|s8|�� }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 r0 r0 ��.0�n��hasnoder0 r1 �
<listcomp>� � z$_computeoutgoing.<locals>.<listcomp>)� changelogrL �nullid�headsr �outgoing)�reporQ �common�clr0 rK r1 �_computeoutgoing� s rV c C s | j }|j}|�dd�}| js&|dvr*d S t| jd�}|�dd�sFd S | jd u rb|�d��d�}n |�d | j�}||�d�M }|r�|d
kr�|� t
d �t |� � n\|d kr�|� t
d
�t |� �r�t
�t
d���n.|dkr�t
d�t |� }t
d�}t
j||d��d S )N� experimentals auto-publish)� warn� confirm� abort� phases�
publishingFs serveds not public()s ::%ln - public()rX s$ %i changesets about to be published
rY s2 push and publish %i changesets (yn)?$$ &Yes $$ &Nos user quitrZ s push would publish %i changesetss: use --publish or adjust 'experimental.auto-publish' configr9 )rS r) Zconfig�publish�listkeys�remote�get�revsZfiltered�warnr �len� promptchoicer Z
CanceledErrorr& )�pushoprS r) Zbehavior� remotephasesZ published�msgr: r0 r0 r1 �
_checkpublish� s<
����rh c C s6 | j j}|�dd�}d|vo"d|v }|p4| 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)rS r) �
configlistr_ �capable)�opr) ZexchangeZ forcebundle1r0 r0 r1 �
_forcebundle1� s ro c @ sB e Zd ZdZd
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.
FNr0 c C s� || _ |j| _|| _|| _|| _|| _|| _t� | _d | _ d | _
d | _ d | _ d | _
d | _d | _d | _d | _t� | _g | _d | _i | _|| _|| _d S �N)rS r) r_ �forcera r � newbranch�set� stepsdone�cgresult�bkresultrR � remoteheads�
pushbranchmap�incomingrf �outdatedphases�fallbackoutdatedphases�
outobsmarkers� outbookmarks� trmanager�pkfailcb�pushvarsr] ) �selfrS r_ rr ra rs r r] r� r0 r0 r1 �__init__� s.
zpushoperation.__init__c C s | j jS )z2future remote heads if the changeset push succeeds)rR � ancestorsof�r� r0 r0 r1 � 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 r0 r0 )rI �node�rT �revr0 r1 rM R rN z/pushoperation.fallbackheads.<locals>.<listcomp>s %ln and parents(roots(%ln))c s s | ]}|� � V qd S rq �r� �rI �cr0 r0 r1 � <genexpr>Z rN z.pushoperation.fallbackheads.<locals>.<genexpr>) ra rR � commonheadsrS �
unfilteredrT rO �indexr� rt �missing�extend)r� �unfi�cheads�revsetr0 r� r1 �
fallbackheads; s
�zpushoperation.fallbackheadsc C s | j r | jS | jS dS )z3set of all common heads after changeset bundle pushN)rv r� r� r� r0 r0 r1 r� ] s zpushoperation.commonheads)FNFr0 FN) �__name__�
__module__� __qualname__�__doc__r� r �
propertycacher� r� �propertyr� r0 r0 r0 r1 rp � s �
K
!rp 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� deleteFr0 c
C s� |du r i }t | ||||||fi t�|���}|j�� rtt|jj�|j�� j } | rtt d�d�
t | �� }
t �
|
��|j�� s�t �
t d���|j�d�s�t �
t d���t t�|j��D ]>} | j�� D ]\} }
|
�| �r� q�q�t �
t d�t�| � ��q�d }}zX|j�dd �}t|��s4|�r4t�| ��s4|j�� }|j� � }t |jd
|j� � �|_ W nB t j �y� } z&d t!�"|� }
|j�#|
� W Y d}~n
d}~0 0 |�p�t$�%� �� |�p�t$�%� �� |j �p�t$�%� �n |j�&|� t'|� t(|� |j)�s�t*|� t|��s
t+|� t,|� t-|� t.|� t/|� W d � n1 �s@0 Y W d � n1 �s`0 Y W d � n1 �s�0 Y | j�dd ��r�t0�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'rW � bundle2.pushbacks
push-responses" cannot lock source repository: %s
� remotenames)2rp r � strkwargsr_ �localrt rS r � supportedr r% �sortedr r& Zcanpushrm r �read_remote_wanted_sidedata�_sidedata_computers�itemsr` �bytestrr) �
configboolro �bookmod�bookmarksinstore�wlockr
�transactionmanagerr r ZLockUnavailabler � forcebytestr�debugr �nullcontextmanagerZ checkpushrh �_pushdiscoveryrr �_checksubrepostate� _pushbundle2�_pushchangeset�_pushsyncphase�
_pushobsolete�
_pushbookmarkr �pullremotenames)rS r_ rr ra rs r r] �opargsre r� rg �category�kind� computersr� r
Z maypushback�errr0 r0 r1 �pushw s� ��
�� �
��
������
�
�"
h r� c s � 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 s | t � < t�� � | S rq )�pushdiscoverymapping�pushdiscoveryorder�append��func��stepnamer0 r1 �dec� s
zpushdiscovery.<locals>.decr0 �r� r� r0 r� r1 �
pushdiscovery� s
r� c C s t D ]}t| }|| � qdS �zRun all discovery stepsN)r� r� )re r� �stepr0 r0 r1 r� s r� c C s~ | j }t|�d��r"t|�d��s&dS | jjD ]J}|| }d|�� v r.d|�� v r.t|j�D ]}|� |�}|j
dd� q\q.dS )zDEnsure all outgoing referenced subrepo revisions are present locallys .hgsubs .hgsubstateNT)Zonpush) rS rc �filerR r� �manifest�filesr� Zsubstate�subZverify)re rS rJ �ctxZsubpathr� r0 r0 r1 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)rr r� )rr )Z onlyheads� commonincrr N)
r �findcommonincomingra rS r_ rr ZfindcommonoutgoingrR rx rz )re Zfcir� rT Zincrx ZfcorR r0 r0 r1 �_pushdiscoverychangeset s* �
�r� s phasec
C s | j }| j�� }t| jd�}| j�dd�rR|rR| j jsR|�dd�rRg | _ g | _
dS t � | j| j
|�| _| jj}d}| jjs~d}d | }t|�||| j
��}| jjs�| jr�t|�d
| j|��}nB|js�|}n6t|�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)r[ � ui� _usedassubrepor\ FNrN s
and public()s heads((%%ln::%%ln) %s)s %ln and (not public() or %ln::)s roots(%ln + %ln::)c S s g | ] }|� � �qS r0 r� )rI �fr0 r0 r1 rM q rN z'_pushdiscoveryphase.<locals>.<listcomp>)rR rS r� r^ r_ r) r� r� r` r{ r| r Zremotephasessummaryr� rf �
draftroots�
publishing�listrt r] r� )
re rR r� rf �drootsZ extracondr� �fallbackZfutureZfdrootsr0 r0 r1 �_pushdiscoveryphase8 sL
���
�
����r� s obsmarkerc C sf t �| jt j�sd S | jjs d S dt| jd�vr4d S | j}dd� |�d| j�D �}| jj� |�| _
d S )N� obsoletes
namespacesc s s | ]}|� � V qd S rq r� r� r0 r0 r1 r� � rN z+_pushdiscoveryobsmarkers.<locals>.<genexpr>� ::%ln) r � isenabledrS � exchangeopt�obsstorer^ r_ rt r� �relevantmarkersr} )re rS �nodesr0 r0 r1 �_pushdiscoveryobsmarkersw s r� � bookmarksc s� | j }| j�� � | j}|�d� d}| jrLt�� jj | j�}� jj
|dd�}t � t
|d��}� fdd�| jD �}t �� � j|�}t| ||||�S )Ns checking for updated bookmarks
r0 T)Z inclusiver� c s h | ]}� j �|��qS r0 ��
_bookmarksZ
expandname�rI Zbookmark�rS r0 r1 � <setcomp>� s z*_pushdiscoverybookmarks.<locals>.<setcomp>)r) rS r� r_ r� ra r ZmaplistrO r� � ancestorsr� �unhexlifybookmarksr^ r Zcomparebookmarksr� �_processcompared)re r) r_ r� ZrevnumsZremotebookmark�explicitrD r0 r� r1 �_pushdiscoverybookmarks� s
�r� c C s� |\}}}}} }
} } | j }
|D ]B\}}}||v r:|�|� |rN|
| �� |v r | j�|||f� q |D ]T\}}}||v rf|�|� t�|�r�| j�t d�| � d| _
qf| j�|d|f� qft |�t | � t |
� D ].\}}}||v r�|�|� | j�|||f� q�|D ]2\}}}||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" rN s> bookmark %s does not exist on the local or remote repository!
r N)rS �remover� r~ r� r� Z isdivergentr) rb r rw r� r� �sort)re Zpushedr� Z remotebmsrD ZaddsrcZadddstZadvsrcZadvdstZdivergeZdifferZinvalidZsamerS rC ZscidZdcidr0 r0 r1 r� � sH
"
���r� c C s� | j }| j�� }|js,t�|j||j� dS | js�|j r�t
d�}t
d�}t
d�}t
d�||d�}|j D ]F}|| }|� � r�t
�|| ��qf|�� rft
�||�� d | ��qft�| � 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)rR rS r� r� r Znochangesfoundr) �excludedrr r� r r� r r r& Z
isunstableZ
instabilitiesr Z
checkheads) re rR r� ZmsoZmspdZmscdZmstr� r� r0 r0 r1 �_pushcheckoutgoing� s,
�
r� c s � �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 s, | t �< � d u r t��� n t�� �� | S rq )�b2partsgenmapping�b2partsgenorderr� �insertr� ��idxr� r0 r1 r�
s
z b2partsgenerator.<locals>.decr0 �r� r� r� r0 r� r1 �b2partsgenerator s
r� c
C s� | j s�| jjr�d|j�dd�v }| jdu }|r2|rH|jdt| j�d� nvt � }t
� | j�D ]D\}}|\}}} }
|durZt |�} |t |
�| @ O }|| t |� O }qZ|r�tt |��} |jd| d� dS )zdGenerate race condition checking parts
Exists as an independent function to aid extensions
s relateds
checkheadsr0 Ns check:heads��datas check:updated-heads)
rr rR r� � capabilitiesr` ry �newpart�iterrx rt r � iteritemsr� )
re �bundlerZallowunrelatedZ emptyremoteZaffected�branchrQ rx ZnewheadsZ
unsyncedheadsZdiscardedheadsr_ r� r0 r0 r1 �_pushb2ctxcheckheads s" �
r c C s t | jjp| jp| jp| j�S )z&return True if we are pushing anything)�boolrR r� r{ r} r~ �re r0 r0 r1 �_pushing4 s ���r s check-bookmarksc C sx t | �r| jrdS t�| j�}d|v }| jr0|s4dS g }| jD ]\}}}|�||f� q>t�| j |�}|j
d|d� dS )z insert bookmark move checkingNr� s check:bookmarksr� ) r rr r � bundle2capsr_ r~ r� r� � binaryencoderS r ) re r �b2capsZhasbookmarkcheckr� �book�old�new� checkdatar0 r0 r1 �_pushb2checkbookmarks> s
r s check-phasesc C s� t | �r| jrdS t�| j�}d|�dd�v }| jdur�|r�dd� tjD �}|tj �
| jj � |tj �
| jj
� tt�|��r�|D ]}|| �� q�t�|�}|jd|d� dS ) zinsert phase move checkingN� headsr[ r0 c S s i | ]
}|g �qS r0 r0 �rI �pr0 r0 r1 �
<dictcomp>W rN z&_pushb2checkphases.<locals>.<dictcomp>s check:phasesr� )r rr r r
r_ r` rf r � allphases�publicr� � publicheads�draftr� �anyr Z
itervaluesr� r r )re r r �
hasphaseheadsZchecks�phaser r0 r0 r1 �_pushb2checkphasesN s
r c s d�j v rdS �j �d� t��s&dS �j��� t�|� t��j�}d}|� d�}|r��fdd�|D �}|s~t
� t d���t
|�}t��j�}tj�j�j|d||d �}|jd|d
�� |rʈ �d |� t��j�r� �d d
� tj�jjv r�� �dd
� � �fdd�}|S )zwhandle changegroup push through bundle2
addchangegroup result is stored in the ``pushop.cgresult`` attribute.
�
changesetsNr8 r6 c s g | ]}|t �� j�v r|�qS r0 )r
�supportedoutgoingversionsrS �rI �vr r0 r1 rM u s �z _pushb2ctx.<locals>.<listcomp>� no common changegroup version� push��
bundlecaps�remote_sidedatar� r7 � treemanifest� 1� exp-sidedatac s$ | j �� j�}|d d d �_dS )z0extract addchangegroup returns from server replyr6 r � returnN)�records�
getreplies�idrv )rn Z cgreplies�Zcgpartre r0 r1 � handlereply� s z _pushb2ctx.<locals>.handlereply)ru �addr� rS �prepushoutgoinghooksr r r
r_ r` r r& r �maxr� r
�
makestreamrR r �addparamr �istreemanifestr �REPO_FEATURE_SIDE_DATA�features)re r r r/ �
cgversionsr& �cgstreamr/ r0 r. r1 �
_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 |rjt | |�S dS )z!handle phase push through bundle2r[ Nri rj � pushkeyr r0 )
ru r r
r_ rS r) rl r` �_pushb2phaseheads�_pushb2phasespushkey)re r r r) � legacyphaseZ
haspushkeyr r0 r0 r1 �
_pushb2phases� s
r? c C sX | j �d� | jrTdd� tjD �}|d �dd� | jD �� t�|�}|jd|d� d S )
z6push phase information through a bundle2 - binary partr[ c S s i | ]
}|g �qS r0 r0 r r0 r0 r1 r � rN z%_pushb2phaseheads.<locals>.<dictcomp>r c s s | ]}|� � V qd S rq r� �rI �hr0 r0 r1 r� � rN z$_pushb2phaseheads.<locals>.<genexpr>� phase-headsr� N)ru r0 r{ r r r� r r )re r Zupdates� phasedatar0 r0 r1 r< � s
r< c s� �j �d� g � � fdd�}tj}�jD ]z}|�d�}|�d|d�� |�d||�� �� |�d|dtj �� |�d |dtj
�� � � |j |f� |�j
|j < q(� �fd
d �}|S ) z7push phase information through a bundle2 - pushkey partr[ c s6 t |j�}� D ]"\}}||krt�td�| ��qd S )Ns updating %s to public failed)�int�partidr r& r )re �exc�targetidrE r� )� part2noder0 r1 �
handlefailure� s
z+_pushb2phasespushkey.<locals>.handlefailurer; � namespace� key� old� %d� newc sl � D ]b\}}| j �|�}|d }d }|s6td�| }n t|d d �sRtd�| }|d ur�j�|� qd S )Nr; s' server ignored update of %s to public!
r r* � updating %s to public failed!
)r+ r, r rD r) rb )rn rE r� �partrep�resultsrg �rH re r0 r1 r/ � s z)_pushb2phasespushkey.<locals>.handlereply)ru r0 r �encoder{ r r4 r r r r r� r- r� )re r rI �enc�
newremoteheadrE r/ r0 rR r1 r= � s
r= �
obsmarkersc C sZ d| j v rd S t�|j�}t�|�d u r,d S | j �d� | jrVt� | j�}t�
||� d S )NrV ) ru r �obsmarkersversionr r �
commonversionr0 r} r �
sortedmarkers�buildobsmarkerspart)re r �remoteversions�markersr0 r0 r1 �_pushb2obsmarkers� s
r] c C s^ d| j v rdS t�| j�}| jj�dd�}d|v }|sHd|v rHt| |�S d|v rZt| |�S dS )z$handle bookmark push through bundle2r� Nri rj r; ) ru r r
r_ rS r) rl �_pushb2bookmarkspart�_pushb2bookmarkspushkey)re r r �legacyZ legacybooksr0 r0 r1 �_pushb2bookmarks� s
ra c C s | sdS |sdS dS )z"small utility for bookmark pushingr� r� r� r0 )r r r0 r0 r1 � _bmaction� s
rb c C s. |r*| j | �� tjkr*t�td�| ��dS )z6abort if a given bookmark points to a secret changesets: cannot push bookmark %s as it points to a secret changesetN)rS r r Zsecretr r& r )re r� rC r0 r0 r1 �_abortonsecretctx� s
�rc c s� �j �d� �jsd S g � g }�jD ]8\}}}t�||� |�||f� � �|t||�f� q$t��j|�}|j d|d� � �fdd�}|S )Nr� r� c s. �j }� D ] \}}|�t| d | � q
d S r3 )r) �status�
bookmsgmap)rn r) r
�action�Z
allactionsre r0 r1 r/ s z)_pushb2bookmarkspart.<locals>.handlereply)
ru r0 r~ rc r� rb r� r rS r )re r r� r
r r r r/ r0 rg r1 r^ s r^ c
s� �j �d� g � tj}� fdd�}�jD ]�\}}}t�||� |�d�}|�d|d�� |�d||�� |�d|t|��� |�d|t|��� d }|s�d
}n|s�d }� � |j
||f� |�j |j
< q(� �fd d
�} | S )Nr� c s@ t |j�}� D ],\}}}||krt�t| d �� | ��qd S )Nr )rD rE r r& re �rstrip)re rF rG rE r
rf )� part2bookr0 r1 rI # s
z._pushb2bookmarkspushkey.<locals>.handlefailurer; rJ rK rL rN r� r� r� c s� �j }� D ]�\}}}| j�|�}|d }|sB�j �td�| � q
t|d d �}|rn|�t| d | � q
|�t| d | � �jd ur
d�_q
d S )Nr; s" server ignored bookmark %s update
r r* r ) r) r+ r, rb r rD rd re rw )rn r) rE r
rf rP rQ �ret�ri re r0 r1 r/ : s
z,_pushb2bookmarkspushkey.<locals>.handlereply) ru r0 r rS r~ rc r r4 r r� r- r� )
re r rT rI r
r r rE rf r/ r0 rk r1 r_ s(
r_ � pushvars)r� c C s| | j }|rxi }|D ]6}d|vr0d}t�|| ��|�dd�\}}|||< q|�d�}t�|�D ]\} }
|j| |
dd� q^dS )zsend shellvars via bundle2� =sI unable to parse variable '%s', should follow 'KEY=VALUE' or 'KEY=' formatr rl F�� mandatoryN)r� r r& rB r r r r4 ) re r r� Z shellvars�rawrg �kr! rE �key�valuer0 r0 r1 �_getbundlesendvarsN s �
rt c
C sh t �| jt �| j��}| jo(| j�dd�}t �t j| j |dd��}|j
d|d� g }t D ](}t | }|| |�}t
|�rX|�|� qX|jdkr�dS t�|�� �}�z^zN| j�� �0} | �d |d
g| j�� d ���� }
W d � n1 s�0 Y W n: tj�y* } z t�td �| ��W Y d} ~ n
d} ~ 0 0 z&d} |�r@| jj} t � | j |
| �}
W n� tj�y� } z t�td �| ��W Y d} ~ nxd} ~ 0 t j �y� } zT| j�td
�| � | j du�r�| j�td
�d| j � t�td���W Y d} ~ n
d} ~ 0 0 W nP tj �yN } z4t | j!�}|| j"v�r*� | j"| | | � W Y d} ~ n
d} ~ 0 0 |D ]}||
� �qTdS )z�push data to the remote using bundle2
The only currently supported type of data is changegroup but this will
evolve in the future.rW r� � client)Z
allowpushback�roles replycapsr� r Nr� � force)s bundler � url� missing support for %ss remote: %s
s (%s)s push failed on remote)#r �bundle20r) r
r_ r r� �
encodecaps� getrepocapsrS r r� r� �callabler� Znbpartsr Z chunkbuffer� getchunks�commandexecutor� callcommandr �resultr �BundleValueError� RemoteErrorr � transaction�
processbundle�
AbortFromPartr: Z
PushkeyFailedrD rE r� )re r Zpushback�capsblobZ
replyhandlersZ partgennameZpartgenrj �stream�e�replyrF Ztrgetterrn rE Zrephandr0 r0 r1 r� d s` ��
��*(& (
&r� c C s� d| j v rdS | j �d� t| �s&dS | j�| � | j}d}| jdu rp|jsp| jjj spt
j | j|ddd|d�}nt
j | j|dd|d�}| j r�dg}n| j
}| j�||| j�� �| _dS ) z7Make the actual push of changeset bundle to remote repor Nr8 r# T)Zfastpathr% �r% rw )ru r0 r� rS r1 rR ra r� rO Z filteredrevsr
Zmakechangegrouprr rx r_ �unbundler rv )re rR r% �cgrx r0 r0 r1 r� � s8
��� � r� c
sX | j }t| jd�}| j�dd�rB|rB| jdu rB|�dd�rBddi}|sTt| |� �n t� | j
||�}|\� }|�dd�r�t| |� nt| � � t| |tj � | jr�d| j v r�dS | j
}n| j}| j �d� � fdd �|D �}|D ]r}| j�� �:}|�d
d|�� d tj d tj d ���� }W d � n1 �s00 Y |s�| j�td
�| � q�dS )z2synchronise phase information locally and remotelyr[ r� r� Nr\ Fs Truec s g | ]}|� � � vr|�qS r0 r� r� ��pheadsr0 r1 rM � rN z"_pushsyncphase.<locals>.<listcomp>r; rM �rJ rK rL rN rO )r� r^ r_ r) r� rv r` �_localphasemover �analyzeremotephasesrS r ru r{ r| r0 r r� r r r� rb r ) re r� rf Zanar� ZoutdatedrU r� �rr0 r� r1 r� � sP ���
�
��(
�r� c sZ �j r t��j�j �� � |� n6� �fdd�|D �}tj� }|rV�j�td�| � dS )z0move <nodes> to <phase> in the local source repoc s" g | ]}� �j | �� k r|�qS r0 )rS r rH �r re r0 r1 rM rN z#_localphasemove.<locals>.<listcomp>s8 cannot lock source repo, skipping local %s phase update
N) r r �advanceboundaryrS r� Z
phasenamesr) rd r )re r� r Z actualmovesZphasestrr0 r� r1 r� s �
���r� c C s� d| j v rdS | j}| j}| j �d� | jr�| j�d� g }t�| j�}t �
|�}t |dd�D ]"}|| }|� |�
d|d|�� q^dd � |D �r�td
�}|j�|� dS ) z5utility function to push obsolete markers to a remoterV Ns' try to push obsolete markers to remote
T��reverser� rN c S s g | ] }|s|�qS r0 r0 )rI r� r0 r0 r1 rM 7 rN z!_pushobsolete.<locals>.<listcomp>s& failed to push some obsolete markers!
)ru rS r_ r0 r} r) r� r rY r Z_pushkeyescaper� r� r r rb ) re rS r_ Zrsltsr\ Z
remotedatarr r� rg r0 r0 r1 r� ' s
r� c
C s� | j dksd| jv rdS | j�d� | j}| j}| jD ]�\}}}d}|sNd}n|sVd}|�� �2}|�dd|t|�t|�d��� � }W d � n1 s�0 Y |r�|�
t | d | � q6|� t | d | � | j
dur6d | _
q6dS )
z"Update bookmark position on remoter r� Nr� r� r� r; r� r )rv ru r0 r) r_ r~ r r� r r� rd re rb rw ) re r) r_ rC r r rf r� r� r0 r0 r1 r� < s2
��&
r� c @ sL e Zd ZdZddd�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.
NFr0 c s� � | _ || _| | _|| _� fdd�|D �| _|| _|| _d | _d | _d | _ d | _
|| _ d | _ t
� | _d| _|| _| | _|
| _d S )Nc s g | ]}� j �|��qS r0 r� r� r� r0 r1 rM � s z*pulloperation.__init__.<locals>.<listcomp>F)rS r_ � remote_pathrQ �explicitbookmarksrr �streamclonerequestedr rT �rheads�fetch�remotebookmarksrv rt ru �clonebundleattempted� includepats� excludepats�depth) r� rS r_ rQ rr r r� r� r� r� r� �pathr0 r� r1 r� h s(
�zpulloperation.__init__c C sJ | j du r@t| j�}t| j�}| jD ]}||vr$|�|� q$|S | j S dS )z0heads of the set of changeset target by the pullN)rQ rt rT r� r� r� )r� r� rj rJ r0 r0 r1 � pulledsubset� s
zpulloperation.pulledsubsetc C s
t | � S rq )ro r� r0 r0 r1 �
canusebundle2� s zpulloperation.canusebundle2c C s t �| j�S rq )r r
r_ r� r0 r0 r1 �remotebundle2caps� s z pulloperation.remotebundle2capsc C s
| j �� S rq )r r� r� r0 r0 r1 �gettransaction� s z pulloperation.gettransaction) NFr0 NNNNNN) r� r� r� r� r� r r� r� r� r� r� r0 r0 r0 r1 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 rq )rS �sourcer �_tr)r� rS r� r r0 r0 r1 r� � s ztransactionmanager.__init__c C sL | j sFd| jt�| j�f }| j�|�| _ | j| j jd< | j| j jd< | j S )z<Return an open transaction object, constructing if necessarys %s
%s� sourcerx )r� r� r � hidepasswordr rS r� �hookargs)r� Ztrnamer0 r0 r1 r� � s z transactionmanager.transactionc C s | j dur| j �� dS )z close transaction if createdN)r� �closer� r0 r0 r1 r� � s
ztransactionmanager.closec C s | j dur| j �� dS )z release transaction if createdN)r� �releaser� r0 r0 r1 r� � s
ztransactionmanager.releaseN)r� r� r� r� r� r� r� r� r0 r0 r0 r1 r� � s
r� c C s@ | � � �$}|�dd|i��� W d � S 1 s20 Y d S )N� listkeysrJ )r r� r� )r_ � namespacer� r0 r0 r1 r^ � s
r^ c s� | � � � � j�� fdd�}� fdd�}��� }t��}t|� tj| jv rLq�|t��krZq�t�fdd�|jD ��rtq�|��� |�}|||j �|_ t
|j�|j |_q&d S )Nc s � � d| |�}dd� |D �S )zReturns heads(h1 % h2)s heads(%ln %% %ln)c S s h | ] }|� � �qS r0 r� �rI r� r0 r0 r1 r� � rN z8_fullpullbundle2.<locals>.headsofdiff.<locals>.<setcomp>�rt �Zh1Zh2�res�r� r0 r1 � headsofdiff� s z%_fullpullbundle2.<locals>.headsofdiffc s � � d| |�}dd� |D �S )z Returns heads((h1 + h2) - null)s heads((%ln + %ln - null))c S s h | ] }|� � �qS r0 r� r� r0 r0 r1 r� � rN z9_fullpullbundle2.<locals>.headsofunion.<locals>.<setcomp>r� r� r� r0 r1 � headsofunion� s z&_fullpullbundle2.<locals>.headsofunionc 3 s | ]}� � |�V qd S rq rK rH )�unficlr0 r1 r� rN z#_fullpullbundle2.<locals>.<genexpr>) r� rO rQ rc � _pullbundle2r ZNARROW_REQUIREMENT�allr� rT rt )rS �pullopr� r� Z old_headsZclstartZ new_headsr0 )r� r� r1 �_fullpullbundle2� s
r� 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)�txnnameZ as_validatorc s( � � }t d�}|j�|�r$t�d��d S )Ns+ accept incoming changes (yn)?$$ &Yes $$ &Nos user aborted)r r) rd r r& )�trrS �cm�Zreporefr0 r1 �prompt s z$add_confirm_callback.<locals>.prompts 900-pull-promptN)r r� r Zregistersummarycallback�weakref�refr� Z addvalidator)rS r� r� r� r0 r� r1 �add_confirm_callback
s
� r� c C s: |du r i }|s| r*|p t � }| p&t � } n
| j\}} t�|� t�| � t| |f||||||| |
d�t�|���} | j�� }
|
r�t |
j �| j
j }|r�t d�d�
t|�� }t�|��| jD ]>}| j�� D ]\}}|�|�r� q�q�t�t d�t�|� ��q�t| d|�� �| _t�� }t � | ��s*| � � }|�� | � � �� | j�� | �sd| j �!dd��rn| j �"� �snt#| | � t$| � t%�&| � t'| � | j(�r�t)| | � t*| � t+| � t,| � t-| � W d � n1 �s�0 Y W d � n1 �s�0 Y W d � n1 �s0 Y | j �!dd ��r6t.�/| |� | 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� rQ rr r r� r� r� r� r� r� sD sidedata category requested by local side without localsupport: '%s'r� rY rW r� )0rt Z
narrowpatsr Zvalidatepatternsr� r r� r_ r� r rS r� r r% r� r r& Z_wanted_sidedatar� r� r` ZProgrammingErrorr� r� r r r r� r� r� r� r
r) r� Zplainr� �_maybeapplyclonebundler Z maybeperformlegacystreamclone�_pulldiscoveryr� r� �_pullchangeset�
_pullphase�_pullbookmarks�
_pullobsoleter r� )rS r_ r� rQ rr r r� r� r� r� r� Zconfirmr� Z peerlocalr� rg r� r� r� r� r0 r0 r1 �pull s� *
�� �
� �
��� ��
h r� c s � fdd�}|S )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 s | t � < t�� � | S rq )�pulldiscoverymapping�pulldiscoveryorderr� r� r� r0 r1 r� � s
zpulldiscovery.<locals>.decr0 r� r0 r� r1 �
pulldiscovery� s
r� c C s t D ]}t| }|| � qdS r� )r� r� )r� r� r� r0 r0 r1 r� � s r� s b1:bookmarksc C s> | j durdS | jr"d| jv r"dS 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� r^ r_ r� r� )r� �booksr0 r0 r1 �_pullbookmarkbundle1� s
r� r6 c C s� t j| j| j| j| jd�}|\}}}| j�� jjj }|r~|r~t
|�}|D ] }||�rH||vrH|� |� qHt
|�� t
|��r~g }|| _
|| _|| _dS )z~discovery phase for the pull
Current handle changeset discovery only, will change handle all discovery
at some point.)rQ rr N)r r� rS r_ rQ rr r� rO r� �has_nodert r� �issubsetrT r� r� )r� �tmprT r� r� r� ZscommonrJ r0 r0 r1 �_pulldiscoverychangegroup� s �
r� c C s� dt | jdd�i}| jj}tj| dd�d }| j|d< | jp@| j|d< | j� t
j �}|rh| j rh| j |d < |r|| j
r|| j
|d
< |r�d |d < d|d
< | j�d� | j�d� nv| j�d� | j|d < d|�dd�v }d| j�dd�v }|s�|r�d|d< | j�d� d| jv �r d| jv�r dg|d< d }d|�dd�v }d| jv } | jdu�rV| j�d� d| jv�r�| jdu �r�|�s�| �r�d|d< d}d| jv �r�d| jv�r�| j�d� |�dg ��d� | j� d��r�| jdu �r�t| j�| jjgk�r�| j|d< |�r| jj�td�� nV| j�s2| jj�td�� d| _ n4| jdu �rft| j�| jjgk�rf| jj�td�� t � | jt j ��r�t �!| j�}
t �"|
�du�r�d|d < | j�d � t#| |� t �$| j�} | �r�| |d < | j�%� ��} t&|�}
d |
d < | �'d |
��(� }z4t j)| j| j*d d!�}d"|j+d< t j,| j||d#� W n� t j-�y~ } z6| jj�.td$�| � t.j/td%�|j0d&��W Y d}~nBd}~0 t.j1�y� } z t.�/td'�| ��W Y d}~n
d}~0 0 W d � n1 �s�0 Y | j�r�t �2|�| _ |j3d D ] \}}|dk�r�t4| |� �q�|�rJi }|j3d D ]}|d( ||d) < �q*|| _n,|j3d D ] \}}|dk�rTt5�6|�| _�qT| jdu�r�t7| � dS )*zNpull data using bundle2
For now, the only supported data are changegroup.s
bundlecapsru �rv T)r r � commonr s includepatss excludepatsF� cgr r6 r[ ri rj r0 r� r� Ns request-bookmarks� clonebundless cbattempteds streaming all changes
� no changes found
� requesting all changes
rV s remote_sidedatar� r� � getbundle)r� s records�rn s remote: abort: %s
s pull failed on remoter9 ry s nodes bookmark)8�
caps20to10rS r) r ZcanperformstreamclonerT rQ r� r_ rm r Z NARROWCAPr� r� ru r0 r� rl r� r` r� �
setdefaultr� r� rP r� rd r rv r r� r� r rW rX �_pullbundle2extrapreparer� r �dictr� r� �bundleoperationr� Zmodesr� r� r r� r: r� �combinechangegroupresultsr+ �_pullapplyphasesr� r� r� )r� �kwargsr) Z streamingZ servernarrowr> ZhasbinaryphaseZbookmarksrequestedZlegacybookmarkZ
hasbinarybookr[ r& r� �argsZbundlern rF r� rs r� �recordr0 r0 r1 r� � s�
����
���
"
�
(H
r� c C s dS )z>hook function so that extensions can extend the getbundle callNr0 )r� r� r0 r0 r1 r� � s r� c C s� d| j v rdS | j �d� | js<| jj�td�� d| _dS | �� }| j du rvt
| j �| jj gkrv| jj�td�� n | j du r�| j
�d�r�| j| _ | j
�d�r�| j
jd| j | j p�| jd �}n�| j du �r| j
�� �(}|�d| jdd
���� }W d � n1 �s0 Y nf| j
�d��s.t�td ���nH| j
�� �,}|�d| j| j dd ���� }W d � n1 �sl0 Y t�| j||d| j
�� �}t�|�| _dS )
z0pull changeset from unbundle into the local repor6 Nr� r r� s changegroupsubsetr� r� �rT rQ )s nodesr� sW partial pull cannot be done because other repository doesn't support changegroupsubset.)s basesr r� )ru r0 r� rS r) rd r rv r� rQ r� rT rP r_ rm r� Z getbundler r� r� r r& r � applybundler r� )r� r� r� r� Zbundleopr0 r0 r1 r� � sT
� ��*�� ��( �r� c C s( d| j v rd S t| jd�}t| |� d S )Nr[ )ru r^ r_ r� )r� rf r0 r0 r1 r� � s
r� c s� d| j v rdS | j �d� t|�dd��}|rP|sPt�| j| j|�\}}| j}n
| j}g }| j�� ��j j
��j j j
�tj�tj� ����fdd�|D �}|r�| �� }t�| j|�|� � ���fdd�|D �}|r�| �� }t�| j|� |� dS )z/apply phase movement from observed remote stater[ Nr\ Fc s" g | ]}� ��|���kr|�qS r0 r0 �rI Zpn)r r r� r� r0 r1 rM � rN z$_pullapplyphases.<locals>.<listcomp>c s" g | ]}���|��� kr|�qS r0 r0 r� )r r r� r� r0 r1 rM � rN )ru r0 r r` r r� rS r� r� � _phasecacher rO r� Zget_revr r r� r� )r� rf r� r� Z_drZdheadsr� r0 )r r r r� r� r1 r� � s0
�
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)
ru r0 rS r� r� �bookmarks_moder� Zupdatefromremoter) r_ r r� r� )r� rS r� r� r0 r0 r1 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 ]2}|� d�rht
�|| �}t�|�\}}||7 }qh|r�| 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 purposerV Ns! fetching remote obsolete markers
r� s dump0Tr� s dump)ru r0 r r� rS r� r) r� r^ r_ r� r� r$ r Z b85decodeZ _readmarkersr� Zinvalidatevolatilesets)r� r� Z remoteobsr\ rr r� r/ Znewmarksr0 r0 r1 r� s&
r� c
C s | j }|�|j�d�p |�� �}|�t|d |�td��}|�t|d |�td��}|snt�t d�t|f ��dd� |D �}d d� |D �}t
|�d
g ��}t
|�d g ��}t � ||||�\}}}|r�t�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 s g | ]}|d krdnd| �qS �� *s path:.s path:r0 r r0 r0 r1 rM 6 s z"applynarrowacl.<locals>.<listcomp>c S s g | ]}|d krdnd| �qS r� r0 r r0 r0 r1 rM 9 s r� r� s4 The following includes are not accessible for %s: %sT�narrow�
narrow_acl)r) Z shortuser�environr` �usernamerl �_NARROWACL_SECTIONr r& r rt r Zrestrictpatternsr Zpprint�update)
rS r� r) r� Z
user_includesZ
user_excludesZ req_includesZ req_excludesZinvalid_includes�new_argsr0 r0 r1 �applynarrowacl sV
�
������
��
r� c s� �j � �j}� j��fdd�|D �thB }�fdd�|D �}|rNdd� |D �} t�t�}
t�t�����fdd�} �fdd ����fd
d ��t� j||d ��} t | �}
t� }� fd
d�| D �}t|�|B }|
D �]�}� �
|��dd� � � |�D �}|du�r.| | }|D ]$}t |d | �
||d ��| |< �qd}|du �pF| | |k}|�r�|�j �� }|�r�|� �
|d �j �� }t|�||��}|�s�t|�dk�r�|� �
|d �j �� }t|�||��}nt|�|��}|�r |
| D ]}| ||� �q�|D ]}|�|� �q�|�� �|�� q�|�s*tg}||v �rj|
| D ]}| ||� �q<|D ]}|
| �|� �qRq�|D ]}|
| |
| O < �qnq�|D ] }|
| D ]}| ||� �q��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 s h | ] }� |��qS r0 r0 rH ��clrevr0 r1 r� q rN z#_computeellipsis.<locals>.<setcomp>c s h | ] }� |��qS r0 r0 rH r� r0 r1 r� r rN c S s i | ]
}|d �qS )r r0 r@ r0 r0 r1 r u rN z$_computeellipsis.<locals>.<dictcomp>c s@ � | � |� t� | �dkr<�| �\}}�| ||� |} qdS )z=Add a root to an ellipsis head, splitting heads with 3 roots.r" N)r0 rc )�headZ curchange�child�roots)�
ellipsisroots� splithead�
splitrootsr0 r1 �addrootz s
z!_computeellipsis.<locals>.addrootc s< � | � |� � | �|� � | �|� � | �|� d S rq )�difference_updater0 r� �discard)r� r� r� )r� r0 r1 r� � s z$_computeellipsis.<locals>.splitrootsc s� t � | �\}}}||f||f||ffD ]l\}}��d|| || �}|D ]L}||krf|||ff S |� vs~t� | �dk rF|||ff S qFq(t�td�| |||f ��d S )Ns% sort(merge() & %d::%d & %d::%d, -rev)r" s; Failed to split up ellipsis node! head: %d, roots: %d %d %d)r� ra rc r r& r )r� Zr1Zr2Zr3Znr1Znr2Zmid�j)r� rS r0 r1 r� � s"
��
��z#_computeellipsis.<locals>.splitheadr� c s g | ]}� � |��qS r0 r� )rI �m)rU r0 r1 rM � rN z$_computeellipsis.<locals>.<listcomp>c S s g | ]}|t kr|�qS r0 )r )rI �prevr0 r0 r1 rM � rN Nr Fr )rO Z manifestlogr� r � collections� defaultdictrt r� Zfindmissingrevs�reversedZchangelogrevisionZ
parentrevs�minr` r� �readr �diffrc r �walkr0 r� ) rS rT rQ Zknown�matchr� ZmflZ
commonrevsZ headsrevsZrevdepthZ
ellipsisheadsr r� ZvisitZrelevant_nodesZ
visitnodes�requiredr� ZpsZcurdepthr ZneededZshallow_enoughZcurmfZp1mfZp2mfr� r� r0 )rU r� r� rS r� r� r1 �_computeellipsisU sl
"
r c C s2 dh}t �t j| |d��}|�dt�|� � |S )zFreturn a set with appropriate options to use bundle20 during getbundles HG20r� � bundle2=)r r{ r| r0 r@ �quote)rS rv Zcapsr� r0 r0 r1 r� � s r� c s � �fdd�}|S )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 s, | t �< � d u r t��� n t�� �� | S rq )�getbundle2partsmapping�getbundle2partsorderr� r� r� r� r0 r1 r� � s
z%getbundle2partsgenerator.<locals>.decr0 r� r0 r� r1 �getbundle2partsgenerator� s
r c C s | d urt dd� | D ��S dS )Nc s s | ]}|� d �V qdS )s HG2N)r$ )rI Zcapr0 r0 r1 r� rN z#bundle2requested.<locals>.<genexpr>F)r r� r0 r0 r1 �bundle2requested s r c K s4 t �|�}i }t|�}|s�|r6|�dd�s6ttd���|rXttd�d�t|�� �� ��t | ||�} d|d< |t
j | | d|||d �fS d
|d< i }
|D ]4} | � d �r�t
�| td �d � �} |
�t�| �� q�t�| j|
�}
||d
< ||d< tD ].}t| }||
| |f||
|d�t �|��� q�|
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).
r� Ts- request for bundle10 must include changegroups# unsupported getbundle arguments: %sr� r s
bundleversionr8 r$ r" r Nr r� )r% r r& s prefercompressed)r Z byteskwargsr r` �
ValueErrorr r% r� �keysrV r
r3 r$ r@ rA rc r� r Z
decodecapsrz r) r r r� Zprefercompressedr~ )rS r� rQ rT r% r& r� �infoZ
usebundle2rR r ZbcapsZblobr �namer� r0 r0 r1 �getbundlechunks sd
��� ��
���
r r; c O s t j| |fi |��S rq )r Zaddpartbundlestream2)r rS r� r� r0 r0 r1 �_getbundlestream2R s r c s� |� dd�r|sdS d} |� d�}
|
rR� fdd�|
D �}
|
sJt�td���t|
�} t� ||�} | jshdS |� d d
�r�ttt |� d g ���} ttt |� d g ���}
t
j � j | |
d
�}nd}t
j� | | ||||d�}| jd|d�}|
r�|�d| � |jddt| j� d
d� t�� ��r |�dd� tj� jv �rP|�dd� t�� �}|�d|� |� d d
��r�|� dd
��r�| �sx|
�r�| �d�}dd�| �d�|
�f |_dS ) z.add a changegroup part to the requested bundler� TNr8 r6 c s g | ]}|t �� �v r|�qS r0 )r
r r r� r0 r1 rM j s �z-_getbundlechangegrouppart.<locals>.<listcomp>r" r� Fr� r� )�include�exclude)r% �matcherr& r� r7 s nbchangesrM rn r' r( r) s exp-wanted-sidedatar� s Narrow:responsespecs %s %s�
)r` r r& r r2 rV r� r� �filterr r r
�rootr
r3 r r4 rc r r5 r r6 r7 r Z format_remote_wanted_sidedatar% r� )r rS r� r% r rQ rT r& r� r/ r8 rR r r r r9 rE ZsidedataZnarrowspecpartr0 r� r1 �_getbundlechangegrouppartW sf
� �
�
���
�r" c K sV |� dd�sdS |r d|vr*t�td���t�|�}t�||�}|rR| jd|d� dS )z+add a bookmark part to the requested bundler FNr� s# no common bookmarks exchange methodr� )r` r r& r r� Zlistbinbookmarksr r )r rS r� r% r r� r� r� r0 r0 r1 �_getbundlebookmarkpart� s
r# r� c
K sJ |� dd�}|D ]4}| �d�}|�d|� |�|��� } t�| �|_qdS )z@add parts containing listkeys namespaces to the requested bundler^ r0 r� rJ N)r` r r4 r^ r� r Z
encodekeysr� )
r rS r� r% r r� r^ r� rE r r0 r0 r1 �_getbundlelistkeysparts� s
r$ c K sX |� dd�rT|du r |�� }dd� |�d|�D �}|j�|�}t�|�}t�| |� dS )z8add an obsolescence markers part to the requested bundleZ
obsmarkersFNc S s g | ] }|� � �qS r0 r� r� r0 r0 r1 rM � rN z+_getbundleobsmarkerpart.<locals>.<listcomp>r� ) r` rQ rt r� r� r rY r rZ ) r rS r� r% r rQ r� Zsubsetr\ r0 r0 r1 �_getbundleobsmarkerpart� s
r% r[ c s* |� dd��r&|r d|� d�vr.t�td���|du r>|�� }t�t�� |�� r\|� t j
< n�|j j }|j
j}|j
j} |D ] }
� ||| |
�� �|
� qxt� �� �} | r�� � t jt� �} | r�� � t j
t� �}
d}|�|| |
�}|D ]}� t j
�||�� q� fdd �t jD �}t �|�}| jd
|d � dS ) z,add phase heads part to the requested bundler Fr r[ s no common phases exchange methodNs" heads(only(%ln, %ln) and public())c s i | ]}|t � | ��qS r0 )r� )rI r �Z headsbyphaser0 r1 r � s z(_getbundlephasespart.<locals>.<dictcomp>rB r� )r` r r& r rQ r r rt r� r r r� r rO r� r� r0 r� r r ra r r r )r rS r� r% r rQ r� r r� r� rA Z
seenphasesZ
draftheadsr r� Z
extraheadsr� Z phasemappingrC r0 r& r1 �_getbundlephasespart� s6
�
r' � hgtagsfnodesc K s: |r|� dd�rd|v s dS 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)r` rV r Zaddparttagsfnodescache� r rS r� r% r rQ rT r� rR r0 r0 r1 �_getbundletagsfnodes� s r* s cache:rev-branch-cachec K sR |� dd�r0|r0d|vs0|� dd�s0|j�t�r4dS 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)r` r) Z has_sectionr� rV r Zaddpartrevbranchcacher) r0 r0 r1 �_getbundlerevbranchcache
s
���
�
� r+ c C sP | � � }t�d�t|����� }|dgksL||ksL|d|gksLt�d| ��dS )zVcheck if the heads of a repo have been modified
Used by peer for unbundling.
rN rw s hasheds. repository changed while %s - please try againN)rQ r Zsha1r% r� �digestr Z PushRaced)rS Z their_heads�contextrQ Z
heads_hashr0 r0 r1 � check_heads@
s ��
��r. c sx d�g d�� d}�j �dd�}��d�s2��d�r6d}�zt�|d � t|tj�s�d
��t� ��g�}��
� �V �� |��,}t� �||���} t�
| ��W d � n1 s�0 Y W d � n1 s�0 Y �n2d�zԇ ���fd d �}
�j �dd
��s|
� tj�|
|dd�} zFtj�|| d�} W | j�|�r��du�r��j jddd� �fdd�}n4| j�|�r��du�r��j jddd� �fdd�}0 � d du�r�� d �� W nX t�y } z>d| _|�r�du�r��� �| _�fdd�}� W Y d} ~ n
d} ~ 0 0 W t�� d � d � d � |du�rt|�j �� � n4t�� d � d � d � |du�rr|�j �� � 0 �S )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 )NNNNrW s bundle2-output-captures remote:http:s
remote:https:Ts uploading changesr c sj � d sbt ���s ��� � d< ��� � d< ����� d< �� d jd< �� d jd< d� d jd< � d S )Nr" r r r� rx r( rk )r� r� r� r
r� r� r0 )� lockandtrrS r� r r0 r1 r� u
s
z unbundle.<locals>.gettransactions bundle2lazylockingr# )�
captureoutputr� r� )r Zsubprocc s � j d| dd� d S �Ns outputF)r� ro )r )�output)r� r0 r1 � recordout�
s zunbundle.<locals>.recordoutr" c s t jd| dd�}� �|� d S r1 )r Z
bundlepartr� )r2 rE )�partsr0 r1 r3 �
s �r )r) r� r$ r. r< r r= r% r r� r
r� r� r� r� r� r� Z
pushbufferr� �
BaseExceptionZduringunbundle2Z
salvageoutputZ_bundle2salvagedoutput�lockmodr� Z popbuffer) rS r� rQ r� r r3 r0 r� r� rn r� rF r0 )r/ r4 r� rS r� r r1 r� S
sj � J
���
�
r� c C sj | j }| j}|j�dd�s dS t|�r*dS | jr4dS |�d�sBdS |�� � }|�di �� � }W d � n1 sp0 Y d| _
t � ||�}|s�|j�
td�� dS t j||| jd�}|s�|j�td�� |j�td�� dS t �|j|�}|d d
}|j�td �| � t|j||��r.|j�td �� n8|j�dd
��rP|j�td�� ntjtd�td�d��dS )z0Apply a clone bundle from a remote, if possible.r� r� 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"r9 )rS r_ r) r� rc rQ rm r r� r� r� r ZparseclonebundlesmanifestZnoter Zfilterclonebundleentriesr� rb Zsortclonebundleentriesrd �trypullbundlefromurlr r& )r� rS r_ r� r� �entriesr r0 r0 r1 r� �
s^
. ������ ��r� c C sJ |� � ��* |�d�� ��}z`t�| |�}t| |d�}t|tj�rL|�|� nt �
|||d|� W W d � W d � dS t j y� } z$| �
td�t�|� � W Y d}~nHd}~0 t jy� } z&| �
td�t�|j� � W Y d}~n
d}~0 0 W d � W d � dS 1 �s 0 Y W d � n1 �s<0 Y dS ) z%Attempt to apply a bundle from a URL.s bundleurlr r� NTs HTTP error fetching bundle: %s
s error fetching bundle: %s
F)r
r� �urlmod�openr2 r< r r( Zapplyr r� �urlerrZ httperrorrb r r r� Zurlerror�reason)r) rS r r� r* r� r� r0 r0 r1 r7 �
s* ��
��r7 )N)FNFr0 FN)N)
NNFr0 NNNNNN)N)N)NNNN)NNNNN)NN)NN)NNN)NNN)NNNN)NNNN)Z
__future__r r r� Zi18nr r� r r � r r� r r r
r r r
r6 r r r r r r r r r r r r9 r r Zutilsr r r Z
interfacesr r; r@ r� r2 rG rV rh ro �objectrp re r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r: r? r< r= r] ra rb rc r^ r_ rt r� r� r� r r� r� r� r� Z
transactionalr� r^ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r� r r r r r r r" r# r$ r% r' r* r+ r. r� r� r7 r0 r0 r0 r1 �<module> sF X
D& ���� �
q
>
6'
6
%
0
;0B#_ * �
;$ 8
�
K
� K � � �
� 0 � � %YR