File: //lib/python3/dist-packages/hgext/remotefilelog/__pycache__/remotefilelog.cpython-310.pyc
o
�]Lb�= � @ s� d dl mZ d dlZd dlZd dlmZ d dlmZ d dlm Z m
Z
m Z m Z m
Z
d dlmZ d dlmZ dd lmZmZmZ G d
d � d e�ZG d d
� d
e�ZdS )� )�absolute_importN)�bin)�_)�ancestor�error�mdiff�pycompat�revlog)� storageutil)�flagutil� )� constants�fileserverclient� shallowutilc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�remotefilelognodemapc C s || _ || _d S �N)� _filename�_store)�self�filename�store� r �C/usr/lib/python3/dist-packages/hgext/remotefilelog/remotefilelog.py�__init__! s
z remotefilelognodemap.__init__c C s | j �| j|fg�}t|� S r )r �
getmissingr �bool)r �node�missingr r r � __contains__% s
z!remotefilelognodemap.__contains__c C s || vrt |��|S r )�KeyError�r r r r r �__get__) s z remotefilelognodemap.__get__N)�__name__�
__module__� __qualname__r r r! r r r r r s r c @ sJ e Zd ZdZejZdd� Zdd� ZdFdd�Z d d
� Z
dde j dfd d �Z
dFd
d�Zdd� Zdd� ZeZdd� Zdd� ZeZdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Z ' ' ' dGd(d)�Z d*d+� Z d,d-� Z d.d/� Z d0d1� Z!dHd2d3�Z"d4d5� Z#d6d7� Z$d8d9� Z%d:d;� Z&d<d=� Z'd>d?� Z(d@dA� Z)dBdC� Z*dDdE� Z+dS )I�
remotefilelogTc C s8 || _ || _|| _t| j|j�| _d| _ttj �| _
d S )Nr ) �openerr �repor � contentstore�nodemap�version�dictr �flagprocessors�_flagprocessors)r r& �pathr' r r r r 4 s zremotefilelog.__init__c C s4 | � |�}|�d�s |S |�dd�}||d d� S )z&returns the file contents at this nodes
� N)�revision�
startswith�index)r r �t�sr r r �read> s
zremotefilelog.readNc C s> t �||�d�|�d��}t�|||�}| j||||||d�S )N� copy� copyrev)r )r �createrevlogtext�getr
�hashrevisionsha1� addrevision) r �text�meta� transaction�linknode�p1�p2�hashtextr r r r �addF s � �zremotefilelog.addc C s� t �t|�|�}d||f } |}
d} |r"d|v r"|d } t|d �}
| d||
||| f 7 } t� } i }
g }|
| jjkrX| }| rFt| j| | j�}|
� |�
|
�� |� |
� | � |
� || jjkrp|
� | �
|�� |� |� | � |� d}|r�|�
d�}|
| \}}}}|p�d}|d|||||f 7 }|| jjkr�|| vr�|� |� | � |� || jjkr�|| vr�|� |� | � |� |st| |7 } | S )Ns %s %s� r6 r7 s %s%s%s%s%s r )r �buildfileblobheader�lenr �setr' �nullidr% r& �update� ancestormap�appendrC �pop)r r< r= �flagsr@ rA r r? �header�data�realp1�copyfrom�visited�
pancestors�queue�p1flog� ancestortext�c�pa1�pa2�ancestorlinknode�
pacopyfromr r r �_createfileblobP sV
�
�z remotefilelog._createfileblobc
C sV |d u r t �|||�}t �|�\}
} t�| ||�\} }
| j| ||||||||
| fd� S )N)�
_metatuple)r
r: � parsemetar �processflagswrite�addrawrevision)r r< r> r? r@ rA �
cachedeltar rM �sidedatar= �
metaoffset�rawtext� validatehashr r r r; � s&
��zremotefilelog.addrevisionc
C sl | r| \}
} nt �|�\}
} |dkrd} | r || d � } n|} | �| |
|||||�}
| jj�| j||
� |S )Nr )r
r^ r\ r' r( �addremotefilelognoder )r rd r> r? r@ rA r rM ra r] r= rc �blobtextrO r r r r` � s
�z remotefilelog.addrawrevisionc C s2 | j j�| j|�}|| \}}}}|r||fS dS �NF�r' �
metadatastore� getancestorsr )r r � ancestorsr@ rA r? rQ r r r �renamed� s
zremotefilelog.renamedc C s t | �|��S )z#return the size of a given revision)rF r5 r r r r �size� s zremotefilelog.sizec C s" || j jkrdS | �|�}||kS )zpcompare text with a given file revision
returns True if text is different than what is stored.
T)r' rH r5 )r r r< �nodetextr r r �cmp� s
zremotefilelog.cmpc C � dS )NTr �r r r r � __nonzero__� � zremotefilelog.__nonzero__c C s | j dv rdS td��)N)s .hgtagss .hgsubs .hgsubstater s len not supported)r � RuntimeErrorrr r r r �__len__� s
zremotefilelog.__len__c C � g S r r rr r r r �heads � zremotefilelog.headsc C rq rh r rr r r r �empty rt zremotefilelog.emptyc C s4 t |t�r
t�d��| jj}|�| j|��t j
d�S )Ns3 remotefilelog does not accept integer rev for flagsr ) �
isinstance�intr �ProgrammingErrorr' r( �getmetar r9 r
� METAKEYFLAG)r r r r r r rM s
�zremotefilelog.flagsc C sR || j jkr| j j| j jfS | j j�| j|�}|| \}}}}|r%| j j}||fS r )r' rH rj rk r �r r rJ r@ rA r? rQ r r r �parents s zremotefilelog.parentsc C s
| � |�S r )r� �r �revr r r �
parentrevs s
zremotefilelog.parentrevsc C s& | j j�| j|�}|| \}}}}|S r ri r� r r r r? s zremotefilelog.linknodec C s | j �� j�| �|��S r )r' �
unfiltered� changelogr� r? r r r r �linkrev$ � zremotefilelog.linkrevFc c s� � ~~~~~d }|D ]F} | � | �\}
} |d u r|
} }| | kr#| jj} | | jjkr2d }
| �| | �}n| �| �}
d }tj| |
| | �| �| | �| �d |
|d dd� V q
d S )Nr ) r �p1node�p2noder? �basenoderM �baserevisionsizer0 �deltarb �protocol_flags) r� r' rH �revdiff�rawdatar �revlogrevisiondeltar? rM )r �nodes�
nodesorder� revisiondata�assumehaveparentrevisions�
deltaprevious� deltamode�sidedata_helpers�prevnoder r@ rA r� r0 r� r r r �
emitrevisions' s: �
��zremotefilelog.emitrevisionsc C s t �| �|�| �|��S r )r �textdiffr� )r �node1�node2r r r r� Q r� zremotefilelog.revdiffc C s8 t |�dkr
t|�}t |�dkrt�|| jtd���|S )N�( � s invalid lookup input)rF r r � LookupErrorr r r r r r �lookupT s �zremotefilelog.lookupc C s |S r r r r r r r� ^ ry zremotefilelog.revc C s t |t�r
t�d��|S )Ns2 remotefilelog does not convert integer rev to node)r{ r| r r} r� r r r r b s
�zremotefilelog.nodec C s� || j jkrdS t|�dkrt�|| jtd���|| j jjks&|| j jj v r)tj
�| j j }|� | j|�}|r8|S |�
| j|�� tjd�}|dkrJ|S t�| ||�d S )z�returns the revlog contents at this node.
this includes the meta data traditionally included in file revlogs.
this is generally only used for bundling and communicating with vanilla
hg clients.
rD r� s invalid revision inputr )r' rH rF r r� r r �
nodeconstants�wdirid�wdirfilenodeids�WdirUnsupportedr( r9 r~ r
r r �processflagsread)r r �rawr rd rM r r r r0 j s" �zremotefilelog.revisionc C s | j |dd�S )NF)r� )r0 r r r r r� � s zremotefilelog.rawdatac C s� | j j}|j}t�| j j| j|�}z|�|�W S ty Y nw t� | j|�}t
j � | j
|�}zt�|�W S ty? Y nw |�| j|fg� z|�|�W S tyX Y nw t�|| jtd���)z3reads the raw file blob from disk, cache, or servers no node)r' � fileservice�
localcacher � getcachekey�namer r5 r � getlocalkey�osr. �join� localpathr �readfile�IOError�prefetchr r� r )r �idr� r� �cachekey�localkeyr� r r r �_read� s0 � � � �zremotefilelog._readc C s | j j�| j|�S r ri r r r r rJ � s zremotefilelog.ancestormapc C sv || j jks || j jkr| j jS | �||�\}}dd� t�|�D �}t�||| || �}|r7tt|j |��S | j jS )Nc S � i | ]\}}||�qS r r ��.0�k�vr r r �
<dictcomp>� � z*remotefilelog.ancestor.<locals>.<dictcomp>)
r' rH �_buildrevgraphr � iteritemsr rl �min�map� __getitem__�r �a�b�revmap�
parentfuncr) �ancsr r r r � s zremotefilelog.ancestorc C sf || j jks || j jkr| j jS | �||�\}}dd� t�|�D �}t�||| || �}t|j|�S )z@calculate all the heads of the common ancestors of nodes a and bc S r� r r r� r r r r� � r� z6remotefilelog.commonancestorsheads.<locals>.<dictcomp>) r' rH r� r r� r �commonancestorsheadsr� r� r� r r r r� � s z"remotefilelog.commonancestorsheadsc sN | � |�}| � |�}t�t�}t� � ||fD ]8}t�|�D ]0\}}|| } |\}
} } }
|
| jjkr=|
s=| � |
� � �
|
� | | jjkrM| � | � � �
| � q qt�t�}i }t� � fdd�|D ��}|r�|� � \}}||v rz|ry|| � || � qbt
|�t
|� d }|||< |r�|| � |� |�|�D ] }|�||f� q�|sd||jfS )z�Builds a numeric revision graph for the given two nodes.
Returns a node->rev map and a rev->[revs] parent function.
c 3 s � | ] }|� vrd |fV qd S r r )r� �n��
allparentsr r � <genexpr>� s � z/remotefilelog._buildrevgraph.<locals>.<genexpr>r )rJ � collections� defaultdict�listrG r r� r' rH rK rC �dequerL rF r9 �
appendleftr� )r r� r� �amap�bmap�
parentsmap�mappingr �pdatar� r@ rA r? rQ r� r� rT �prevrev�current�
currentrev�parentr r� r r� � sF
��
� �
z remotefilelog._buildrevgraphc C s d S r r )r �minlinkr> r r r �strip� rt zremotefilelog.stripc C rw r r rr r r r �files� rt zremotefilelog.filesc C rq )N)r r r rr r r r � checksize� rt zremotefilelog.checksize)NN)NFFFNN)F),r" r# r$ �
_generaldeltar � RevlogError�_flagserrorclassr r5 rC r\ r �REVIDX_DEFAULT_FLAGSr; r` rm rn �rawsizerp rs �__bool__rv rx rz rM r� r� r? r� r� r� r� r� r r0 r� r� rJ r r� r� r� r� r� r r r r r% / sd
@
�+
�,
�*
1 r% )�
__future__r r� r� �mercurial.noder �mercurial.i18nr � mercurialr r r r r �mercurial.utilsr
�mercurial.revlogutilsr � r
r r �objectr r% r r r r �<module> s