File: //usr/lib/python3/dist-packages/mercurial/__pycache__/bundlerepo.cpython-310.pyc
o
�]Lb,] � @ sb d Z ddlmZ ddlZddlZddlmZ ddlmZm Z ddl
m Z m Z m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m Z ddl m Z dd lm!Z" G d
d � d ej�Z#G d d
� d
e#e
j
�Z$G dd� de#ej%�Z&G dd� dej�Z'G dd� dej(�Z)G dd� dej*�Z+dd� Z,G dd� de-�Z.d#dd�Z/d d � Z0G d d � d e-�Z1 d$d!d"�Z2dS )%z�Repository class for viewing uncompressed bundles.
This provides a read-only repository interface to bundles as if they
were part of the actual repository.
� )�absolute_importN� )�_)�hex�nullrev)�bundle2� changegroup� changelog�cmdutil� discovery�encoding�error�exchange�filelog� localrepo�manifest�mdiff�pathutil�phases�pycompat�revlog� revlogutils�util�vfs)�urlutil)� constantsc sX e Zd Zdd� Zddd�Zdd� Zd� fdd � Zd
d � Zd d
� Zdd� Z dd� Z
� Z S )� bundlerevlogc C sL t �|�}tjj| |||d� || _t| �}|d | _t� | _|� � D ]�}|\}} }
} } }
}}t|
�}|�
� | }| j � |�rJ| j�
| j �|�� q#| |krQt}n|| �}| |
fD ]}| j � |�skt�|| jtd���qY| j � | �s{t| | jtd���| �| �}tj|||||| �| �| �|
�|d�}| j �|� | j�
|� |d7 }q#d S )N)�target�radixr s unknown parents unknown delta base)�flags� data_offset�data_compressed_length�data_delta_base�link_rev� parent_rev_1� parent_rev_2�node_id)�vfsmod� readonlyvfsr �__init__�bundle�len�
repotiprev�set�
bundlerevs� deltaiter�tell�index�has_node�add�revr r
� LookupError�
display_idr r �entry�append)�self�openerr r �
cgunpacker�
linkmapper�n� deltadata�node�p1�p2�cs� deltabase�deltar �sidedata�size�start�linkrev�p�baserev�e� rL �6/usr/lib/python3/dist-packages/mercurial/bundlerepo.pyr) 8 sR
�� �
�
�zbundlerevlog.__init__Nc C s< || j kr tj�| |�S | j�| �|�� | j�| �|��S �N)r, r �_chunkr* �seekrG �read�length)r9 r4 �dfrL rL rM rO p s
zbundlerevlog._chunkc C sr || j kr|| j kr| j| d }||kr| �|�S n|| j kr-|| j kr-tj�| ||�S t�| �|�| �|��S )z1return or calculate a delta between two revisions� )r, r1 rO r �revdiffr �textdiff�rawdata)r9 �rev1�rev2�revbrL rL rM rU y s
�zbundlerevlog.revdiffc s� |d u r | � |�}d}d }g }|}|| jkr7| jr&| jd |kr&| jd }n|�|� | j| d }|| jks|tkr>d}n|d u rUtt| �j| � |�||d�}|\} }}|rYd}|rk| �
|� � �}
t �
||
g�}|s[|||fS )NFr � rT � )�_df)r4 r, �_revisioncacher8 r1 r �superr �_rawtextr? rO �popr �patches) r9 r? r4 r] � validated�rawtext�chain�iterrev�r�__rD �� __class__rL rM r` � s6
�
�
�
zbundlerevlog._rawtextc O � t �rN ��NotImplementedError�r9 �args�kwargsrL rL rM � addrevision� � zbundlerevlog.addrevisionc O rk rN rl rn rL rL rM �addgroup� rr zbundlerevlog.addgroupc O rk rN rl rn rL rL rM �strip� rr zbundlerevlog.stripc C rk rN rl �r9 rL rL rM � checksize� rr zbundlerevlog.checksizerN ) �__name__�
__module__� __qualname__r) rO rU r` rq rs rt rv �
__classcell__rL rL ri rM r 7 s
8 r c @ � e Zd Zdd� ZdS )�bundlechangelogc C s6 t j �| |� dd� }t�| |tjd f| j||� d S )Nc S s | S rN rL )�xrL rL rM �<lambda>� s z*bundlechangelog.__init__.<locals>.<lambda>)r r) r �revlog_constants�KIND_CHANGELOGr )r9 r: r; r< rL rL rM r) � s �zbundlechangelog.__init__N�rw rx ry r) rL rL rL rM r| � � r| c s* e Zd Z ddd�Z� fdd�Z� ZS )�bundlemanifestNr\ c C sd t jj| |||d� t�| |tj|f| jj||� |d u r*i }| jj dkr*t
| j�}|| _ || _ d S )N)�trees 03)
r �manifestrevlogr) r r �KIND_MANIFESTLOG�_revlogr r* �version�_getfilestarts�
_dirlogstarts� _linkmapper)r9 �
nodeconstantsr: r; r< � dirlogstarts�dirrL rL rM r) � s �
zbundlemanifest.__init__c sJ || j v r | j�| j | � t| j| j| j| j| j |d�S tt| ��|�S )N)r� ) r� r* rP r� r� r: r� r_ �dirlog)r9 �dri rL rM r� � s
�zbundlemanifest.dirlog)Nr\ )rw rx ry r) r� rz rL rL ri rM r� � s
�r� c @ r{ )�
bundlefilelogc C s2 t j �| ||� t|tj|f| jj||d�| _d S )N)r r r; r< )r r) r r � KIND_FILELOGr� r )r9 r: �pathr; r< rL rL rM r) � s �zbundlefilelog.__init__Nr� rL rL rL rM r� � r� r� c @ r{ )�
bundlepeerc C � dS �NFrL ru rL rL rM �canpush� rr zbundlepeer.canpushN)rw rx ry r� rL rL rL rM r� � r� r� c s4 e Zd Z� fdd�Zdd� Zdd� Zdd� Z� ZS ) �bundlephasecachec s8 t t| �j|i |�� t�| d�rt�| j�| _d S d S )Nr: )r_ r� r) r � safehasattrr' r( r: rn ri rL rM r) � s �zbundlephasecache.__init__c C rk rN rl ru rL rL rM �write� rr zbundlephasecache.writec C rk rN rl )r9 �fprL rL rM �_write� rr zbundlephasecache._writec C s || j |< | �� d| _d S )NT)�
phaseroots�
invalidate�dirty)r9 �phase�newroots�trrL rL rM � _updateroots s
z bundlephasecache._updateroots)rw rx ry r) r� r� r� rz rL rL ri rM r� � s
r� c sF i }t � ji �D ]}|d }� �� ||< t � fdd�i �D ]}q q|S )Ns filenamec s
� � d �S rN )�
deltachunkrL �r; rL rM r~
s
z _getfilestarts.<locals>.<lambda>)�iter�
filelogheaderr0 )r; �filespos� chunkdata�fname�chunkrL r� rM r� s �r� c s� e Zd ZdZdd� Zdd� Zd&dd�Zejd d
� �Z ejd d � �Z
d
d� Z ejdd� �Z dd� Z
ejdd� �Zejdd� �Zdd� Z� fdd�Zdd � Zd d � Zd d � Zd!d"� Zd'� fd$d%� Z� ZS )(�bundlerepositorya% A repository instance that is a union of a local repo and a bundle.
Instances represent a read-only repository composed of a local repository
with the contents of a bundle file applied. The repository instance is
conceptually similar to the state of a repository after an
``hg unbundle`` operation. However, the contents of the bundle are never
applied to the actual base repository.
Instances constructed directly are not usable as repository objects.
Use instance() or makebundlerepository() to create instances.
c C sR || _ || _| j�dddd� d | _t�|d�}t�| j||�}t |t
j �r[|| _ d | _
d }|jdd�D ]}|jdkrC|rAtd ��|}| �||� q4|sSt�td
���|�d tj� n/t |tj�r|�� rx| j|j d d
d�}t�| j||| j �}|| _ || _
n t�td�t|� ��i | _ | j j d | _!t"�#| d t"j$dd� | | j!d � D �� d S )N� phasess publishFs
bundlerepo� rbT)�seekable� changegroups# can't process multiple changegroupss No changegroups foundr s .hg10un� HG10UN)�headers bundle type %s cannot be readr c S � g | ]}|� � �qS rL �r? ��.0�ctxrL rL rM �
<listcomp>X � z-bundlerepository.__init__.<locals>.<listcomp>)%� _tempparent�_url�ui� setconfig�tempfiler � posixfiler �
readbundle�
isinstancer �
unbundle20� _bundlefile� _cgunpacker� iterparts�typerm �_handlebundle2partr
�Abortr rP �os�SEEK_SETr � cg1unpacker�
compressed�_writetempbundlerQ r � _cgfilesposr r, � firstnewrevr �retractboundary�draft)r9 �
bundlepath�url�
tempparent�fr* �cgpart�partrL rL rM r) sP
� ���zbundlerepository.__init__c C sv |j dkrd S |}|j�dd�}t�| �}||vr$td�}t�|| ��|�� r1| � |j
d| �}t� ||d�| _ d S )Nr� s versions 01s# Unsupported changegroup version: %ss .cg%suns UN)
r� �params�getr �supportedincomingversionsr r
r� r� r� rQ � getunbundlerr� )r9 r* r� �cgstreamr� � legalcgvers�msgrL rL rM r� [ s
z#bundlerepository._handlebundle2partr\ c C s~ | j jd|d�\}}|| _t�|d��}|�|� |d�}|s!n|�|� qW d � n1 s1w Y | j j| jdd�S ) z Write a temporary file to disks
hg-bundle-)�prefix�suffix�wbTi Nr� )�mode)r �mkstempr� r� �fdopenr� �open)r9 �readfnr� r� �fdtemp�temp�fptempr� rL rL rM r� j s
� �z!bundlerepository._writetempbundlec C s t | | j�S rN )r� �_phasedefaultsru rL rL rM � _phasecachey s z bundlerepository._phasecachec C s( | j �� t| j| j �}| j �� | _|S rN )r� �changelogheaderr| �svfsr0 �manstart)r9 �crL rL rM r } s
zbundlerepository.changelogc C � d S rN rL ru rL rL rM �_refreshchangelog� s z"bundlerepository._refreshchangelogc C sZ | j �| j� | j �� | �� jj}t| j| j | j |�}| j �
� | _ t �
| j | || �� �S rN )r� rP r� �manifestheader�
unfilteredr r4 r� r� r� r0 � filestartr � manifestlog� narrowmatch)r9 r< � rootstorerL rL rM r� � s
� �z bundlerepository.manifestlogc C s8 | j �| j� | j �� | j �� D ]}q| j �� | _dS )zgConsumes the manifest portion of the bundle, setting filestart so the
file portion can be read.N)r� rP r� r� r/ r0 r� )r9 rD rL rL rM �_consumemanifest� s
z!bundlerepository._consumemanifestc C s | j | jS rN )r r� ru rL rL rM r� � s zbundlerepository.manstartc C s | j d| jvr | �� | jS )Nr� )r� �__dict__r� r� ru rL rL rM r� � s
zbundlerepository.filestartc C s | j S rN )r� ru rL rL rM r� � s zbundlerepository.urlc sj | j s| j�| j� t| j�| _ || j v r-| j�| j | � | �� jj}t| j || j|�S t
t | �� |�S rN )
r� r� rP r� r� r� r r4 r� r� r_ r� �file)r9 r� r< ri rL rM r� � s
zbundlerepository.filec C s> | j �� | jdur| j�| j� | jr t�| jd� dS dS )z'Close assigned bundle file immediately.NT)r� �closer� r �unlinkr� �shutil�rmtreeru rL rL rM r� � s
�zbundlerepository.closec C r� r� rL ru rL rL rM �cancopy� rr zbundlerepository.cancopyc C s t | �S rN )r� ru rL rL rM �peer� � zbundlerepository.peerc C s t �� S rN )r �getcwdru rL rL rM r � r zbundlerepository.getcwdNc s� |d u r| j }| j�|�}| j�|�}td�}| jj|k r'| j�|t|� � | jj|k r7| j�|t|� � tt | ��
||�S )Ns9 setting parent to node %s that only exists in the bundle
) �nullidr r4 r r, r� �warnr r_ r� �
setparents)r9 r@ rA �p1rev�p2revr� ri rL rM r � s zbundlerepository.setparents)r\ rN )rw rx ry �__doc__r) r� r� r �unfilteredpropertycacher� r r� r� r� r� r� r� r� r� r� r r r rz rL rL ri rM r� s0 <
r� c C s� |r t �td���| �dd�}|s t�t�� �}|d u r d}|r=t�� }||kr+d}nt� |�}|�
|�r=|t |�d � }t �
|�}|�� }|jdkrd|�dd�}t |�dkr_||d } }
n
|\} }
n||} }
t| | |
�S )Ns# cannot create new bundle repositorys bundles mainreporootr\ � +r r )r
r� r �configr
�findrepor r r � normasprefix�
startswithr+ r r� � localpath�scheme�split�makebundlerepository) r� r� �create�intents�
createopts�
parentpath�cwd�u�s�repopath�
bundlenamerL rL rM �instance� s.
r c C s� |r dt �|�|f }nd| }z tj| |dd�}d}W n& tjyB t�� }z
tj| |dd�}W n
ty? t �
|� � w Y nw G dd� dt |j �}||_ t �
||||� |S ) z?Make a bundle repository object based on repo and bundle paths.s bundle:%s+%ss bundle:%sF)r NTc @ s e Zd ZdS )z5makebundlerepository.<locals>.derivedbundlerepositoryN)rw rx ry rL rL rL rM �derivedbundlerepository! s r )r �
expandpathr r r
� RepoErrorr �mkdtemp� Exceptionr� r� r� rj r) )r� r r� r� �repor� r rL rL rM r s(
���r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�bundletransactionmanagerc C r� rN rL ru rL rL rM � transaction+ rr z$bundletransactionmanager.transactionc C rk rN rl ru rL rL rM r� . rr z bundletransactionmanager.closec C rk rN rl ru rL rL rM �release1 rr z bundletransactionmanager.releaseN)rw rx ry r$ r� r% rL rL rL rM r# * s r# Fc s t j|�||d�}|\}}} |s)z |rt�|� W n ty" Y nw |g �jfS t|���fdd�| D �} d� d���� }
|sC|
�s,| �dd�} d| voPd| v } | o]�� d �o]�� d�}
|
r���
� �&}|� d d
|| t j
|d d �d
d���� }t�| |�� |� }� W d � n1 s�w Y n��� d �r���
� �}|� d d
|| d���� }W d � n1 s�w Y nK|du r�� d�s��
� �}|� d|d
d���� }W d � n1 s�w Y d} n ��
� �}|� d|| d
d���� }W d � n1 s�w Y |
�r
d}nd}t�| |||� }� |�rd� |
�s,t|j|j|� }
�|�� }|
�r3|
�� }
|
j�|| �}��r{dd� ��jd� D �}��
� �}|� dddi��� }W d � n 1 �scw Y t j ��d|d�}t � |_ t � ||� � ��fdd �}|
||fS ) aD obtains a bundle of changes incoming from peer
"onlyheads" restricts the returned changes to those reachable from the
specified heads.
"bundlename", if given, stores the bundle to this file path permanently;
otherwise it's stored to a temp file and gets deleted again when you call
the returned "cleanupfn".
"force" indicates whether to proceed on unrelated repos.
Returns a tuple (local, csets, cleanupfn):
"local" is a local repo from which to obtain the actual incoming
changesets; it is a bundlerepo for the obtained bundle when the
original "peer" is remote.
"csets" lists the incoming changeset node ids.
"cleanupfn" must be called without arguments when you're done processing
the changes; it closes both the original "peer" and the one returned
here.
)�heads�forcec s g | ]}|� vr|�qS rL rL )r� r} )� commonsetrL rM r� V s z$getremotechanges.<locals>.<listcomp>Ns devels legacy.exchanges bundle2s bundle1s getbundles incomings client)�roleT)� source� common� headss
bundlecapss cg)r* r+ r, s changegroupsubsetr� )s nodesr* )s basesr, r* s HG10BZr� c S r� rL r� r� rL rL rM r� � r� s listkeyss namespacer� )r� r& c s& �r�� � � r
t�� � �� � d S rN )r� r� r� rL )r* �
bundlerepor rL rM �cleanup� s
z!getremotechanges.<locals>.cleanup) r �findcommonincomingr� r� �OSErrorr� r- �local�
configlist�capable�commandexecutor� callcommandr �
caps20to10�resultr � writechunks�_forwardchunksr � writebundler �baseui�rootr& r� r � findmissingr� �
pulloperationr# � trmanager�_pullapplyphases)r� r" r � onlyheadsr r' �tmp�common�incoming�rheadsr �legexc� forcebundle1�
canbundle2rK �b2r� �cg�
bundletype�csets� reponodes� remotephases�pullopr. rL )r* r- r( r rM �getremotechanges5 s�
� �
��
��� �
� ��
��� ��
��� �
��� �
�
��� ��
rP )NN)NNF)3r �
__future__r r� r� �i18nr r? r r � r r r r
r r r
r r r r r r r r r r r r r' �utilsr r r r r| r� r� r� � localpeerr� �
phasecacher� r� �objectr� r r r# rP rL rL rL rM �<module> s. T w(
Q"& �