HEX

File: //usr/lib/python3/dist-packages/hgext/lfs/__pycache__/wrapper.cpython-310.pyc
o

�]Lb3C�@s�ddlmZddlZddlmZddlmZmZmZddl	m
Z
mZddlm
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZddlmZmZ ddl!m"Z"dd	l#m$Z$m%Z%d
dl&m'Z'dd
l(m)Z)m*Z*e��Z+e+�,ed�dd��Z-e+�,ed�dd��Z.e+�,ed�dd��Z/dd�Z0dd�Z1dd�Z2dYdd�Z3ddej4fdd �Z5d!d"�Z6d#d$�Z7e+�,ed%�d&d'��Z8e+�,ej9d(�d)d*��Z:e+�,ej9d+�d,d-��Z;d.d/�Z<e+�,ed0�d1d2��Z=e+�,ed3�d4d5��Z>e+�,ej?d6�d7d8��Z@d9d:�ZAd;d<�ZBd=d>�ZCd?d@�ZDdAdB�ZEe+�,edC�dDdE��ZFe+�,e
dF�dGdH��ZGdIdJ�ZHdZdLdM�ZIdZdNdO�ZJdPdQ�ZKe+�,e dR�dSdT��ZLe+�,edU�e+�,edV�dWdX���ZMdS)[�)�absolute_importN)�_)�bin�hex�short)�getattr�setattr)�bundle2�changegroup�cmdutil�context�error�exchange�	exthelper�	localrepo�pycompat�revlog�scmutil�util�vfs�wireprotov1server)�actions�engine)�
repository)�storageutil�
stringutil�)�lfutil�)�	blobstore�pointersmakefilestoragecKs(d|vr
|�tj�|d||d�|��S)N�lfs)�requirements�features�)�addr�REPO_FEATURE_LFS)�origr"r#�kwargsr$r$�3/usr/lib/python3/dist-packages/hgext/lfs/wrapper.py�localrepomakefilestorage:sr*sallsupportedversionscCs||�}|�d�|S)Ns03)r%)r'�ui�versionsr$r$r)�allsupportedversionsBs
r-s
_capabilitiescCs:|||�}t�|jd�rd|jvr|�d�|�d�|S)z5Wrap server command to announce lfs server capability�lfslocalblobstorer!s	lfs-serve)r�safehasattr�svfsr"�append)r'�repo�proto�capsr$r$r)�
_capabilitiesIs



r5cC�dS�NFr$)�self�textr$r$r)�bypasscheckhashYsr:cCs�t�|�}|��}|jj}|�|�s|j|_|jj�|g|�|j	|dd�}i}|�
�D]}|�d�rA|td�d�}||||<q,|sI|�d�rOt
�||�}|dfS)a%Read filelog content from local blobstore transform for flagprocessor.

    Default tranform for flagprocessor, returning contents from blobstore.
    Returns a 2-typle (text, validatehash) where validatehash is True as the
    contents of the blobstore should be checked using checkhash.
    F)�verifysx-hg-Ns
T)r �deserialize�oid�opener�lfslocalblobstore�has�filename�lfsremoteblobstore�	readbatch�read�keys�
startswith�lenr�packmeta)r8r9�pr=�store�hgmeta�k�namer$r$r)�
readfromstore]s 


�rNc
Cs�t�|�\}}|r|dkr||d�}tt�|����}|jj�||�d|}t	j
|dt|�d�}t�
|�s<d|d<|durPt�|�D]
\}}||d|<qE|��}	|	dfS)	Nrs	sha256:%ss%d)r=�size�0�x-is-binarysx-hg-%sF)r�	parsemetar�hashlib�sha256�digestr>r?�writer �
gitlfspointerrGr�binaryr�	iteritems�	serialize)
r8r9rK�offsetr=�longoid�metadatarL�v�rawtextr$r$r)�writetostorezs
r`cCsP|dur|dur
dS|�|�}n|�|�}||jkrdS|�|�}t|tj@�Sr7)�rev�node�nullid�flags�boolr�REVIDX_EXTSTORED)�rlogrbrardr$r$r)�_islfs�s


rhc
Ksp|jjj�d�}|r't|�}t�|�\}
}|r||8}||jj|�r'|	tj	O}	|||||||f|||	d�|
��S)Nslfstrack)�
cachedeltarbrd)
�_revlogr>�options�getrGrrRrArrf)r'r8r9�transaction�link�p1�p2rirbrd�kwds�lfstrack�textlen�metar[r$r$r)�filelogaddrevision�s,
��
�rucCs\t|j|�r)|j�|�}|sdSt�|�}d|vr'd|vr'|dt|d�fSdS|||�S)NFs	x-hg-copysx-hg-copyrev)rhrj�rawdatar r<r)r'r8rbr_r]r$r$r)�filelogrenamed�s

rwcCs:t|j|d�r|j�|�}t�|�}t|d�S|||�S)N)rassize)rhrjrvr r<�int)r'r8rar_r]r$r$r)�filelogsize�s


rys_verify_revisioncCsjt||d�r,|�|�}t�|�}|jj�|���r |tj	M}n|tj	@r,|d�
|�|||||�dS)N)rbssafe_renamed)rhrvr r<r>r?r@r=rrfr%)r'�rl�	skipflags�staterbr_r]r$r$r)�_verify_revision�s


r}scmpcCsP|��r#t|ddd���r#t�|���}t�|���}|��|��kS|||�S)z+returns True if text is different than fctx�islfscSr6r7r$r$r$r$r)�<lambda>�szfilectxcmp.<locals>.<lambda>)r~rr r<rvr=)r'r8�fctxrorpr$r$r)�
filectxcmp�s

r�sisbinarycCs2|��rt�|���}tt|�dd���S||�S)NrQr)r~r r<rvrerxrl)r'r8r]r$r$r)�filectxisbinarysr�cCst|��j|���S�N)rh�filelogrj�filenode)r8r$r$r)�filectxislfssr�s_updatecatformattercCs(||||||�|j||��d�dS)N)rv)�datarv)r'�fm�ctx�matcher�path�decoder$r$r)�_updatecatformattersr�swrapconvertsinkcs2||�}|jdkrG�fdd�d|j���|_|S)NshgcseZdZ��fdd�Z�ZS)zconvertsink.<locals>.lfssinkc		slt�|�j}	|	||||||||�}
d|jjvr4|j|
�t�fdd�|D��r4|jj�d�t�|j�|
S)Nr!c3s(�|]\}}|�vo�|��VqdSr�)r~)�.0�f�n�r�r$r)�	<genexpr>=s�&z9convertsink.<locals>.lfssink.putcommit.<locals>.<genexpr>)�super�	putcommitr2r"�anyr%r�writereporequirements)r8�files�copies�parents�commit�source�revmap�full�cleanp2�pcrb)�	__class__�lfssinkr�r)r�!s"�
z&convertsink.<locals>.lfssink.putcommit)�__name__�
__module__�__qualname__r��
__classcell__r$�r�)r�r)r� sr�)�repotyper�)r'�sinkr$r�r)�convertsinks

#r�s__init__cCs^|||�|j��D]\}}|�d�r||j|<q
dD]}t�||�r,t||t||��qdS)Nr!)r.�lfsremoteblobstore)rk�itemsrFrr/rr)r'r8�othervfsrLr^rMr$r$r)�vfsinitJs


���r�c
Cs�t�|jd�s	dSg}t�}|jj}|D]3\}}||}|�|�D]%}t||�}	|	rF|	��|vrF|�|	���sF||	_	|�
|	�|�|	���q!q|rUt�
|��||�dSdS)zSEnsure that required LFS blobs are present, fetching them as a group if
    needed.r.N)rr/r0�setr?�walk�pointerfromctxr=r@rAr1r%r�remoterC)
r2�
revmatches�pointers�oids�
localstorera�matchr�r�rIr$r$r)�_prefetchfilesXs$

���r�cCs"t�|jd�s	dSt|jjtj�S)Nr�T�rr/r0�
isinstancerBr�_nullremote�r2r$r$r)�_canskipuploadqsr�cCs$t�|jd�s	dSt|jjtj�S)Nr�Fr�r�r$r$r)�candownloadzsr�cCs$t|�rdSt||�}t||�dS)zuupload lfs blobs introduced by revs

    Note: also used by other extensions e. g. infinitepush. avoid renaming.
    N)r��extractpointers�uploadblobs)r2�revsr�r$r$r)�uploadblobsfromrevs�s
r�cCst|j|jj�S)z�Prepush hook.

    Read through the revisions to push, looking for filelog entries that can be
    deserialized into metadata so that we can block the push on their upload to
    the remote blobstore.
    )r�r2�outgoing�missing)�pushopr$r$r)�prepush�sr�spushcOs�d|jvrB|��s|�d�std�}tj|dtd�d��|jj}zt�	||�
��|j_|||g|�Ri|��W||j_S||j_w|||g|�Ri|��S)z�bail on push if the extension isn't enabled on remote when needed, and
    update the remote store based on the destination path.r!s:required features are not supported in the destination: %ss&enable the lfs extension on the server)�hint)r"�local�capablerr
�Abortr0rBrr��url)r'r2r��argsr(�m�remotestorer$r$r)�push�s
�r�swritenewbundlec	Os,t||j�|||||||g|�Ri|��S)z;upload LFS blobs added by outgoing revisions on 'hg bundle')r�r�)	r'r+r2r�rA�
bundletyper�r�r(r$r$r)�writenewbundle�s���r�cCs�|j�d�i}|jj}|td�td�t|���.}|D]}||}t|���D]}|||��<q'|��qt	|��dd�d�Wd�S1sIwYdS)z1return a list of lfs pointers added by given revss&lfs: computing set of blobs to upload
s
lfs searchs
changesetscSs|��Sr�)r=)rIr$r$r)r�sz!extractpointers.<locals>.<lambda>)�keyN)
r+�debug�makeprogressrrG�pointersfromctx�valuesr=�	increment�sorted)r2r�r�r��progress�rr�rIr$r$r)r��s�
$�r�Fc
Cs�|}||vr"|s
dS||��vr|��}n
||��vr |��}ndS||}t|��j|���s2dSzt�|���}||krA|WSiWStj	y`}zt
�td�|t
|���|f��d}~ww)a?return a pointer for the named file from the given changectx, or None if
    the file isn't LFS.

    Optionally, the pointer for a file deleted from the context can be returned.
    Since no such pointer is actually stored, and to distinguish from a non LFS
    file, this pointer is represented by an empty dict.
    Ns#lfs: corrupted pointer (%s@%s): %s
)rorprhr�rjr�r r<rv�InvalidPointerr
r�rrrb)r�r��removed�_ctxr�rI�exr$r$r)r��s2

����r�cCsJi}|����}|��D]}||�sqt|||d�}|dur"|||<q|S)z�return a dict {path: pointer} for given single changectx.

    If ``removed`` == True and the LFS file was removed from ``ctx``, the value
    stored for the path is an empty dict.
    )r�N)r2�narrowmatchr�r�)r�r��resultr�r�rIr$r$r)r��s�r�cCs$|sdS|jj}|�||jj�dS)z*upload given pointers from local blobstoreN)r0rB�
writebatchr?)r2r��
remoteblobr$r$r)r�sr�sfinishdatamigrationcCs�|||||�t�|jd�rCt�|jd�rE|jjj}|jjj}|��D]#\}}}	|	D]}
|�td�|
�t�	|�
|
�|�
|
��q*q#dSdSdS)Nr.scopying lfs blob %s
)rr/r0r?rr�rVrrrn�join)r'r+�srcrepo�dstrepor"�	srclfsvfs�	dstlfsvfs�dirpath�dirsr�r=r$r$r)�upgradefinishdatamigrations��

��r�spreservedrequirementsssupporteddestrequirementscCs ||�}d|jvr|�d�|S)Nr!)r"r%)r'r2�reqsr$r$r)�upgraderequirements s

r�)NN)F)N�
__future__rrS�mercurial.i18nr�mercurial.noderrr�mercurial.pycompatrr�	mercurialr	r
rrr
rrrrrrrr�vfsmodr�mercurial.upgrade_utilsr�upgrade_actionsr�upgrade_engine�mercurial.interfacesr�mercurial.utilsrr�
largefilesr�rr �eh�wrapfunctionr*r-r5r:rNr`rh�REVIDX_DEFAULT_FLAGSrurwryr}�basefilectxr�r�r�r�r��readonlyvfsr�r�r�r�r�r�r�r�r�r�r�r�r�r�r$r$r$r)�<module>sv@







�)
	


	



.

		







!