File: //lib/python3/dist-packages/hgext/remotefilelog/__pycache__/contentstore.cpython-310.pyc
o
�]Lb�3 � @ s� d dl mZ d dlZd dlmZmZ d dlmZ d dlm Z m
Z
m Z ddl m
Z
mZmZ G dd � d e�ZG d
d � d e
j�ZG d d
� d
e
j
�ZG dd� de�ZG dd� de�ZdS )� )�absolute_importN)�hex�sha1nodeconstants)�getattr)�mdiff�pycompat�revlog� )� basestore� constants� shallowutilc @ s$ e Zd ZdZdZdZdZdZdZdS )�
ChainIndiciesz>A static class for easy reference to the delta chain indicies.r r � � � N) �__name__�
__module__� __qualname__�__doc__�NAME�NODE�BASENAME�BASENODE�DATA� r r �B/usr/lib/python3/dist-packages/hgext/remotefilelog/contentstore.pyr
s r
c s� e Zd Z� fdd�Zdd� Zejjdd� �Zdd� Z ejjd d
� �Z
d d � Z ejjd
d� �Z dd� Z
dd� Zdd� Zddd�Z� ZS )�unioncontentstorec s: t t| �j|i |�� || _|�d�| _|�dd�| _d S )N�
writestore�allowincompleteF)�superr �__init__�stores�getr r ��self�args�kwargs�� __class__r r r ) s zunioncontentstore.__init__c C sj | � ||�}|d tj tjkrt|t|�f��|�� tj }|}|r3|�� tj }t �
||g�}|s#|S )a Fetches the full text revision contents of the given name+node pair.
If the full text doesn't exist, throws a KeyError.
Under the hood, this uses getdeltachain() across all the stores to build
up a full chain to produce the full text.
���) �
getdeltachainr
r r �nullid�KeyErrorr �popr r �patches)r$ �name�node�chain�fulltext�text�deltar r r r" 4 s �zunioncontentstore.getc C �@ | j D ]}z |�||�W S ty Y qw t|t|�f��)z;Return the single delta entry for the given name/node pair.)r! �getdeltar, r �r$ r/ r0 �storer r r r6 L s
�zunioncontentstore.getdeltac C s� | � ||�}|d tj tjkr?|d \}}}}}z
| � ||�}|�|� W n ty4 | js1� Y |S w |d tj tjks|S )a
Returns the deltachain for the given name/node pair.
Returns an ordered list of:
[(name, node, deltabasename, deltabasenode, deltacontent),...]
where the chain is terminated by a full text entry with a nullid
deltabasenode.
r) )�_getpartialchainr
r r r+ �extendr, r )r$ r/ r0 r1 �x�
deltabasename�
deltabasenode� morechainr r r r* W s
�� z unioncontentstore.getdeltachainc C r5 )z)Returns the metadata dict for given node.)r! �getmetar, r r7 r r r r? o s
�zunioncontentstore.getmetac C s dd� | j D �}tj|� S )Nc S s g | ]}|� � �qS r )�
getmetrics)�.0�sr r r �
<listcomp>z s z0unioncontentstore.getmetrics.<locals>.<listcomp>)r! r �sumdicts)r$ �metricsr r r r@ y s
z unioncontentstore.getmetricsc C r5 )z�Returns a partial delta chain for the given name/node pair.
A partial chain is a chain that may not be terminated in a full-text.
)r! r* r, r r7 r r r r9 } s
�z"unioncontentstore._getpartialchainc C � t d���Ns5 cannot add content only to remotefilelog contentstore�� RuntimeError�r$ r/ r0 �datar r r �add� � �zunioncontentstore.addc C s" |}| j D ] }|r|�|�}q|S �N)r! �
getmissing)r$ �keys�missingr8 r r r rO � s
�z unioncontentstore.getmissingc C s" | j r
| j �|||� d S td��)Ns no writable store configured)r �addremotefilelognoderI rJ r r r rR � s z&unioncontentstore.addremotefilelognodeNc C s | j D ]}|�||� qd S rN )r! �
markledger)r$ �ledger�optionsr8 r r r rS � s
�z unioncontentstore.markledgerrN )r r r r r" r
�baseunionstore� retriabler6 r* r? r@ r9 rL rO rR rS �
__classcell__r r r' r r ( s
r c sT e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
� Z S )�remotefilelogcontentstorec s$ t t| �j|i |�� t�� | _d S rN )r rY r � threading�local� _threaddatar# r' r r r � s z"remotefilelogcontentstore.__init__c C s� | � ||�}t�|�\}}}|||| � }t�|�}|| \} }
} } d }
| r+t| �}
| �|||� |tj@ r;d }
} t�|| |
�}|S rN ) �_getdatar �parsesizeflags� ancestormapr �_updatemetacacher �REVIDX_EXTSTORED�createrevlogtext)r$ r/ r0 rK �offset�size�flags�contentr_ �p1�p2�linknode�copyfrom�copyrev�revisionr r r r" � s
z remotefilelogcontentstore.getc C s" | � ||�}||tj| �||�fS rN )r" r r+ r? �r$ r/ r0 rl r r r r6 � s
�z"remotefilelogcontentstore.getdeltac C � | � ||�}||d tj|fgS rN �r" r r+ rm r r r r* � � z'remotefilelogcontentstore.getdeltachainc C sN | � � || jjd kr!| �||�}t�|�\}}}| �|||� | jjd S �Nr r )�_sanitizemetacacher\ � metacacher] r r^ r` )r$ r/ r0 rK rc rd re r r r r? � s z!remotefilelogcontentstore.getmetac C rF rG rH rJ r r r rL � rM z remotefilelogcontentstore.addc C s& t | jdd �}|d u rd| j_d S d S )Nrs )NN)r r\ rs )r$ rs r r r rr � s �z,remotefilelogcontentstore._sanitizemetacachec C s< | � � || jjd krd S tj|tj|i}||f| j_d S )Nr )rr r\ rs r � METAKEYFLAG� METAKEYSIZE)r$ r0 rd re �metar r r r` � s
z*remotefilelogcontentstore._updatemetacache) r r r r r" r6 r* r? rL rr r` rX r r r' r rY � s rY c @ sN e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z ddd�Z
dS )�remotecontentstorec C s || _ || _d S rN )� _fileservice�_shared)r$ �ui� fileservice�sharedr r r r � s
zremotecontentstore.__init__c C �* | j j|t|�fgddd� | j�||�S �NT)�force� fetchdata)rx �prefetchr ry r" �r$ r/ r0 r r r r" � � �zremotecontentstore.getc C s$ | � ||�}||tj| j�||�fS rN )r" r r+ ry r? rm r r r r6 � s �zremotecontentstore.getdeltac C rn rN ro rm r r r r* � rp z remotecontentstore.getdeltachainc C r} r~ )rx r� r ry r? r� r r r r? r� zremotecontentstore.getmetac C rF )Ns cannot add to a remote storerH rJ r r r rL
� zremotecontentstore.addc C s |S rN r )r$ rP r r r rO � z remotecontentstore.getmissingNc C � d S rN r )r$ rT rU r r r rS r� z remotecontentstore.markledgerrN ) r r r r r" r6 r* r? rL rO rS r r r r rw � s rw c @ sx e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d d
�Zdd� Z dd� Z
dd� Z dd� Z dd� Z
d dd�Zdd� Zd S ) �manifestrevlogstorec C s4 |j | _|j| _t� | _tj| jdd�| _d| _d S )Ns
00changelog.i��radixr ) r8 �_store�svfs�_svfs�dict�_revlogsr �_cl�_repackstartlinkrev)r$ �repor r r r s
z manifestrevlogstore.__init__c C s | � |��|�S rN )�_revlog�rawdatar� r r r r" s zmanifestrevlogstore.getc C s$ | � ||�}||| jj| �||�fS rN )r" r� r+ r? rm r r r r6 " s z manifestrevlogstore.getdeltac C s | � ||�}||d | jj|fgS rN )r" r� r+ rm r r r r* & s z!manifestrevlogstore.getdeltachainc C s0 | � |�}|�|�}tj|�|�tj|�|�iS rN )r� �revr rt re ru �rawsize)r$ r/ r0 �rlr� r r r r? * s
�zmanifestrevlogstore.getmetaNc C s� |d u rt � }||v r
g S | �|�}i }|h}|j|�|�gdd�D ]J}|�|�}|�|� |�|�\} }
| | jjkrD| |vrD|� | � |
| jjkrS|
|vrS|� |
� | j�|�
|��} | |
| df||�|�< |sl |S q"|S )NT)� inclusive� ) �setr� � ancestorsr� r0 �discard�parentsr� r+ rL �linkrev) r$ r/ r0 �knownr� r� rQ �ancrev�ancnoderg rh ri r r r � getancestors2 s,
�z manifestrevlogstore.getancestorsc C sD | j }| �|�}|�|�}|�|�|��}|d |d |�|�d fS rq )r� r� r� r� r� r0 )r$ r/ r0 �clr� r� r� r r r � getnodeinfoK s
z manifestrevlogstore.getnodeinfoc G rF )Ns cannot add to a revlog storerH )r$ r% r r r rL R r� zmanifestrevlogstore.addc C sF | j �|�}|d u r!d}|dkrd| }tj| j|d�}|| j |< |S )N� 00manifesttreer� s meta/%s/00manifestr� )r� r" r r� )r$ r/ r� �
revlognamer r r r� U s
zmanifestrevlogstore._revlogc C s8 g }|D ]\}}| � |�}||jvr|�||f� q|S rN )r� �nodemap�append)r$ rP rQ r/ r0 �mfrevlogr r r rO _ s
�z manifestrevlogstore.getmissingc C s || _ || _d S rN )r� �_repackendlinkrev)r$ �startrev�endrevr r r �setrepacklinkrevrangeh s
z)manifestrevlogstore.setrepacklinkrevrangec
C s^ |r
|� tj�r
d S d}tj| jdd�}| j}| j}t�t |�d dd�D ]%}|�
|�}||k r2 n||kr7q%|� |�} |� | || � |�
| || � q%| j�� D ]\\}
} } | d d� dkse| dd � d krfqP| dt d
� � }tj| j| d d� d �}t�t |�d dd�D ]%}|�
|�}||k r� n||kr�q�|� |�} |� | || � |�
| || � q�qPd S ) Nr� r� r� r r) � s meta/���s .is /00manifest)� indexfile)r" r �OPTION_PACKSONLYr r� r� r� r �xrange�lenr� r0 �
markdataentry�markhistoryentryr� � datafiles)
r$ rT rU �treenamer� � startlinkrev�
endlinkrevr� r� r0 �t�pathrd r r r rS l s>
��z manifestrevlogstore.markledgerc C r� rN r )r$ rT r r r �cleanup� r� zmanifestrevlogstore.cleanuprN )r r r r r" r6 r* r? r� r� rL r� rO r� rS r� r r r r r� s
"r� )�
__future__r rZ �mercurial.noder r �mercurial.pycompatr � mercurialr r r � r
r r �objectr
rV r rY rw r� r r r r �<module> s zH-