File: //usr/lib/python3/dist-packages/hgext/narrow/__pycache__/narrowbundle2.cpython-310.pyc
o
�]Lby- � @ s d dl mZ d dlZd dlZd dlmZ d dlmZmZm Z m
Z
m Z m Z m
Z
mZmZmZmZ d dlmZ dZdZdZd Zd
Zd Zd Zd
ZdZ dZ e� e �Z e� e �Z d dd�Z!dd� Z"dd� Z#e�$eeef�dd� �Z%e�$e�dd� �Z&e�$e�dd� �Z'd d � Z(d d � Z)dS )!� )�absolute_importN)�_) �bundle2� changegroup�error�exchange� localrepo�
narrowspec�repair� requirements�scmutil�util�wireprototypes)�
stringutils narrowacls narrow:changespecs narrow:responsespecs narrow:specs includes excludes KILLs DONEs >20s20s20sls >20s20s20s20slc s� � j �dd�s J �|�d�}� fdd�|D �}|s ttd���t|�} ttt|�dg ���}
ttt|�dg ���} t | � |
| | |||�d d �� d S )
N� experimental� narrowservebrokenellipses� changegroupc s g | ] }|t �� �v r|�qS � )r �supportedoutgoingversions)�.0�v��repor �</usr/lib/python3/dist-packages/hgext/narrow/narrowbundle2.py�
<listcomp>8 s
�z3getbundlechangegrouppart_narrow.<locals>.<listcomp>s no common changegroup version� includepats� excludepats�depth)
�ui�
configbool�get�
ValueErrorr �max�sorted�filter�bool�generateellipsesbundle2) �bundlerr �source�
bundlecaps�b2caps�heads�common�kwargs�
cgversions�version�include�excluder r r � getbundlechangegrouppart_narrow+ s(
�
�r2 c C s� t j|j||d�}|d ur t|�}|dk r t�td�| ��t|p$|�� �}t|p,|j g�}t
j |||t� ||d�\} }
} |j �
dt|
� � | rytj|||d|d u| |
d�} | �|| dd �}
| jd
|
d �}|�d |� t�|�r{|�d
d� d S d S d S )N)r0 r1 � s depth must be positive, got %d)r s Found %d relevant revs
T��matcher�ellipses�shallow�
ellipsisroots� fullnodesF� narrow_widenr ��data� version� treemanifest� 1)r �match�root�intr �Abortr �setr+ �nullidr �_computeellipsisr �debug�lenr �
getbundler�generate�newpart�addparamr �istreemanifest)r' r r0 r1 r/ r, r+ r r@ �
visitnodes�relevant_nodesr8 �packer�cgdata�partr r r r& O s8
��
�r& c s� t |p�jg�}�fdd�|D �}|� � �fdd�}| jt|� d� t��t � |||�\} }
} | r_tj|�|dd| |
d�} | �|| dd �}
| jd
|
d�}|� d |� t
� ��ra|� d d
� d S d S d S )Nc s h | ]}� j �|��qS r )� changelog�rev)r �nr r r � <setcomp>� s z9generate_ellipses_bundle2_for_widening.<locals>.<setcomp>c 3 s* � � D ] } t V �j�| �V qtV d S �N)�_KILLNODESIGNALrS �node� _DONESIGNAL)�r��deadrevsr r r �genkills� s
�
z8generate_ellipses_bundle2_for_widening.<locals>.genkillsr; TFr4 r: r r= r> r? ) rD rE rK �_CHANGESPECPARTr rF r rI rJ rL r rM )r' r �oldmatch�newmatchr/ r, �known� knownrevsr^ �newvisit�newfull� newellipsisrP rQ rR r r\ r �&generate_ellipses_bundle2_for_wideningy s2
��
�rg c C s� t |j�td��� �}t |j�td��� �}t�|� t�|� tj | j
jvr5| j
j� tj � t �
| j
� | j
�||� t�| j
� d S )N� )rD �paramsr �_SPECPART_INCLUDE�
splitlines�_SPECPART_EXCLUDEr �validatepatternsr �NARROW_REQUIREMENTr �addr �writereporequirements�
setnarrowpats�copytoworkingcopy)�op�inpartr r r r r �_handlechangespec_2� s
ru c C s� |� � }|�d�\}}t|�� �}t|�� �}t�|� t�|� tj| jjvr6| jj� tj� t
� | j� | j� ||� t�
| j� d S )N� )�read�splitrD rk r rm r rn r ro r rp rq rr )rs rt r< �inc�excr r r r r �_handlenarrowspecs� s
r{ c C s� | j }|j}t� }t�|d�}|tkr;|tkr(t�|d�}|�|�r'|�|� n t �
t d�| ��t�|d�}|tks|ro|j | _
G dd� dt�}tjj �||� � tj| j|t|�ddd�}|ro|j�� | _| j�� || _t�|d �r{|�� d S d S )
N� � s) unexpected changespec node chunk type: %sc @ s e Zd Zdd� ZdS )z'_handlechangespec.<locals>.dummybmstorec S s d S rW r )�selfr �tr�changesr r r � applychanges� s z4_handlechangespec.<locals>.dummybmstore.applychangesN)�__name__�
__module__� __qualname__r� r r r r � dummybmstore� s r� Ts widen)�backup�topic�setnewnarrowpats) r rS rD r � readexactlyrZ rX �hasnodero r rC r �
_bookmarks�_bookmarksbackup�dictr �localrepositoryr
�stripr �list�uninterruptible�_widen_uninterr� __enter__�
_widen_bundler
� safehasattrr� ) rs rt r �cl�clkills� chunksignal�ckr� � chgrpfiler r r �_handlechangespec� s:
�
� � �
�r� c s� | j }| j}| j}| `|j}|�td�� |�|d�}z_t�||||�}|j s+|�
� nt � � }|�< t
|tj�rY|�d��� t�||� fdd�� W d � n1 sSw Y n
|�|dd|�|� d� W d � n1 spw Y W |�� n|�� w |�� D ]4\} }
z| �|
� W q� ty� } z | jtjkr�|�td �| �|
�t� | �f � W Y d} ~ q�d} ~ ww | j � ddd� |�|� dS )
zFChangegroup exchange handler which restores temporarily-stripped nodess adding branch
s rbs stripc s � S rW r r �r r r �<lambda>" s z)handlechangegroup_widen.<locals>.<lambda>Ns bundle:Ts error removing %s: %s
) r r r� �vfs�noter �openr �
readbundle�verbose�silentr
�nullcontextmanager�
isinstancer �
unbundle20� transaction�
processbundle�apply�join�close� undofiles�unlink�OSError�errno�ENOENT�warnr � forcebytestrr� �__exit__) rs rt r r r� r� �f�gen� maybe_silent�undovfs�undofile�er r� r �handlechangegroup_widen sL � ���� �� ����r� c sz t j} d| d< d| d< d| d< d| d< d| d< tjd � � fd
d �}|tjd < tjd ��fd d
�}�j|_|tjd < dS )z?Enable narrow repo support in bundle2-related extension points.s booleans narrows plains depths csvs oldincludepatss oldexcludepatss knownr c s^ | d }|j �t�rt�||�}|�dd�r&|j �dd�r&t| i |�� d S � | i |�� d S )Nr3 �narrowFr r )r � has_section�_NARROWACL_SECTIONr �applynarrowaclr r r2 )�argsr- r )�origcgfnr r � wrappedcgfnG s �zsetup.<locals>.wrappedcgfnc sL � | |� t �| d�rt| |� t �| d�r$tjj�| j| j� | `d S d S )Nr� r� ) r
r� r� r r� r� rD r r� )rs rt )�
origcghandlerr r �wrappedcghandlerX s
��z setup.<locals>.wrappedcghandlerN)r �GETBUNDLE_ARGUMENTSr �getbundle2partsmappingr �parthandlermappingri )�
getbundleargsr� r� r )r� r� r �setup: s
r� )NNNN)*�
__future__r r� �struct�mercurial.i18nr � mercurialr r r r r r r
r r r
r �mercurial.utilsr r� r_ � _RESSPECS� _SPECPARTrj rl rX rZ �_ELIDEDCSHEADER�_ELIDEDMFHEADER�calcsize�
_CSHEADERSIZE�
_MFHEADERSIZEr2 r& rg � parthandlerru r{ r� r� r� r r r r �<module> sB 4
�$*=
3 .