File: //usr/lib/python3/dist-packages/hgext/largefiles/__pycache__/proto.cpython-310.pyc
o
�]Lb> � @ s� d dl mZ d dlZd dlmZ d dlmZ d dlmZm Z m
Z
m Z m Z m
Z
mZ ddlmZ e jZe jZdZe � � Zd d
� Zd d � Zd
d� Zdd� Ze�ed�dd� �Zdd� ZdS )� )�absolute_importN)�_)�open)�error� exthelper�httppeer�util�wireprototypes�wireprotov1peer�wireprotov1server� )�lfutilsa
This repository uses the largefiles extension.
Please enable it in your Mercurial config file.
c C sT |� � ��}t�| |�}t�tj�|�� tj|| j j
d�}zkz-|� � D ]}|� |� q#|j
�d� |t�|j
�kr@tdtd���|�� t�| |� W n7 ty� } z+| j�td�||jf � t�d|ri|�� nd�W Y d}~W |�� W d � S d}~ww W |�� n|�� w W d � n1 s�w Y t�d|r�|�� �S d�S )zcServer command for putting a largefile into a repository's local store
and into the user cache.)�
createmoder s$ largefile contents do not match hashs, largefiles: failed to put %s into store: %s
r � N)�mayberedirectstdior
� storepathr �makedirs�os�path�dirname�atomictempfile�storer �
getpayload�write�_fp�seek�hexsha1�IOErrorr �close�linktousercache�ui�warn�strerrorr �pushres�getvalue�discard)�repo�proto�sha�outputr �tmpfp�p�e� r- �8/usr/lib/python3/dist-packages/hgext/largefiles/proto.py�putlfile$ s:
�����
�
�r/ c sZ t �| |�}|st�td�| ��t|d�� t�� �� �d �� �fdd�}t j
|� d�S )z\Server command for retrieving a largefile from the repository-local
cache or user cache.s+ requested largefile %s not present in caches rb� c 3 s& � d� V t �� �D ]} | V q d S )Ns %d
)r �
filechunkiter)�chunk��f�lengthr- r. � generatorR s
�
�zgetlfile.<locals>.generator)�gen) r
�findfiler �Abortr r r �fstat�filenor �streamreslegacy)r& r' r( �filenamer6 r- r3 r. �getlfileB s
�
r> c C s$ t �| |�}|s
t�d�S t�d�S )a; Server command for checking if a largefile is present - returns '2
' if
the largefile is missing, '0
' if it seems to be in good condition.
The value 1 is reserved for mismatched checksum, but that is too expensive
to be verified on every stat and must be caught be running 'hg verify'
server side.s 2
s 0
)r
r8 r �
bytesresponse)r& r' r( r= r- r- r. � statlfileZ s
r@ c s& |j � G � fdd�d|j�}||_d S )Nc sD e Zd Z� fdd�Zejdd� �Zdd� Zdd� Zejd d
� �Z d S ) z+wirereposetup.<locals>.lfileswirerepositoryc s6 �� }| � d�r|j� G � fdd�d|j�}||_|S )Ns
largefilesc s e Zd Z� fdd�ZdS )zWwirereposetup.<locals>.lfileswirerepository.commandexecutor.<locals>.lfscommandexecutorc s |dkrd}� ||�S )Ns headss lheadsr- )�self�command�args��orig_callcommandr- r. � callcommandq s
zcwirereposetup.<locals>.lfileswirerepository.commandexecutor.<locals>.lfscommandexecutor.callcommandN)�__name__�
__module__� __qualname__rF r- rD r- r. �lfscommandexecutorp s rJ )�capablerF � __class__)rA �executorrJ ��orig_commandexecutorrD r. �commandexecutork s
z;wirereposetup.<locals>.lfileswirerepository.commandexecutorc S s | j �| �S )N)�heads� batchable)rA r- r- r. �lheadsy s z2wirereposetup.<locals>.lfileswirerepository.lheadsc S s� t | jtj�rE| jd||ddid�}z |�dd�\}}|�d�D ] }| j�td�|� q t |�W S t
yD | j�td �| � Y dS w z| j d||d
�\}}|d kr\t �
td �|��t |�W S tyj Y dS t
yx t �
td
�|��w )Ns putlfilez content-typezapplication/mercurial-0.1)�datar( �headers�
r Ts remote: s! unexpected putlfile response: %r
�r( r s putlfile failed:s& putlfile failed (unexpected response):)�
issubclassrL r �_call�split�
splitlinesr r! r �int�
ValueError� _callpushr �
ResponseErrorr )rA r( �fd�res�dr) �l�retr- r- r. r/ } s< �
��
��z4wirereposetup.<locals>.lfileswirerepository.putlfilec s s� � | j d|d�}|�� }zt|�}W n ty& | �t�td�|�� Y nw tj ||d�D ]}|V q.t
| j t j �rO|�
d�}|rQ| �t�td�|�� dS dS dS )z<returns an iterable with the chunks of the file with sha shas getlfilerW s unexpected response:)�limitr N)� _callstream�readliner\ r] �_abortr r_ r r r1 rX rL r �read)rA r( �streamr5 r2 r- r- r. r> � s( � ��
��z4wirereposetup.<locals>.lfileswirerepository.getlfilec S s dd� }d|i}||fS )Nc S s& zt | �W S ttjfy Y dS w )N� )r\ r] �urlerr� httperror)rb r- r- r. �decode� s
�zEwirereposetup.<locals>.lfileswirerepository.statlfile.<locals>.decodes shar- )rA r( rn �resultr- r- r. r@ � s
z5wirereposetup.<locals>.lfileswirerepository.statlfileN)
rG rH rI rP r
rR rS r/ r> r@ r- rN r- r. �lfileswirerepositoryj s
%rp )rP rL )r r& rp r- rN r. �
wirereposetupg s
^rq s
_capabilitiesc C s | ||�}|� d� |S )z$announce largefile server capabilitys largefiles=serve)�append)�origr& r' �capsr- r- r. �
_capabilities� s
ru c C s t �|�r
t�t�S | ||�S )zTWrap server command - largefile capable clients will know to call
lheads instead)r
� islfilesrepor �ooberror�LARGEFILES_REQUIRED_MSG)rs r& r' r- r- r. rQ � s
rQ ) �
__future__r r �mercurial.i18nr �mercurial.pycompatr � mercurialr r r r r r
r � r
rl �urlreqrx �ehr/ r> r@ rq � wrapfunctionru rQ r- r- r- r. �<module> s$ $
�
e