File: //lib/python3/dist-packages/hgext/lfs/__pycache__/wireprotolfsserver.cpython-310.pyc
o
�]Lb
/ � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlmZ d dl m
Z
m Z m Z m
Z
ddlmZ ejZejZejZejZejZejZejZe
�
� Ze�e
d�dd � �Zdd
d �Zd d
� Zdd� Z dd� Z dd� Z dS )� )�absolute_importN)�common)� exthelper�pycompat�util�wireprotoserver� )� blobstores handlewsgirequestc
s | ��|� �r dS �j j�dd�sdS t��j jd�s dS �js"dS z+�jdkr6� ��d� t�j �|�W S �j�d�rKt �j �|� ��fd d
��W S W dS t
j y� } z)|j D ] \}}||j |< qYt
�
|jt�|��|_|�d t�|� � W Y d }~dS d }~ww )
z�Wrap wireprotoserver.handlewsgirequest() to possibly process an LFS
request if it is left unprocessed by the wrapped method.
Ts experimentals lfs.serveF�lfslocalblobstores .git/info/lfs/objects/batch� pulls .hg/lfs/objectsc s � ��| �S �N� )�perm�� checkperm�rctx�reqr
�>/usr/lib/python3/dist-packages/hgext/lfs/wireprotolfsserver.py�<lambda>> s z#handlewsgirequest.<locals>.<lambda>s 0
%s
N)�repo�ui�
configboolr � safehasattr�svfs� dispatchpath�_processbatchrequest�
startswith�_processbasictransfer� hgwebcommon�
ErrorResponse�headers�
statusmessage�coder �bytestr�status� setbodybytes)�origr r �resr �e�k�vr
r r �handlewsgirequest% s2
� ��r+ c C s( t j||d�| _d| jd< | �d� d S )N��message� text/plain; charset=utf-8� Content-Type� )r r! r$ r r% )r' r" r- r
r
r �
_sethttperrorL s
r1 c C sP t �t�� �}| jd }d}| jr|| j7 }|d| j 7 }|�d||f � dS )z=Write information about the current exception to wsgi.errors.s wsgi.errorsr0 � /s4 Exception happened while processing request '%s':
%sN)r �sysbytes� traceback�
format_exc�rawenv�apppathr �write)r �tb�errorlog�urir
r
r �
_logexceptionR s
�r< c C s |j dkr t|t� dS |jd dkrt|t� dS |jd dkr(t|t� dS t�|j� � �}d|�
ddg�vrAt|t d� dS |�
d �}t� |�}|d
vrYt|t d | � dS | j
j}d d
� t||�
dg �||�D �}d|d�}t�t�|_d|jd< |�t� t�|��� dS )z�Handle a request for the Batch API, which is the gateway to granting file
access.
https://github.com/git-lfs/git-lfs/blob/master/docs/api/batch.md
s POSTTr/ s application/vnd.git-lfs+jsons Accept�basic� transferss0 Only the basic LFS transfer handler is supported� operation)� upload� downloads& Unsupported LFS transfer operation: %sc S s g | ]}|�qS r
r
)�.0�pr
r
r �
<listcomp>� s ��z(_processbatchrequest.<locals>.<listcomp>�objects)�transferrE )�methodr1 �HTTP_METHOD_NOT_ALLOWEDr �HTTP_UNSUPPORTED_MEDIA_TYPE�HTTP_NOT_ACCEPTABLEr �
json_loads�bodyfh�read�get�HTTP_BAD_REQUESTr# r r
�_batchresponseobjectsr r! �HTTP_OKr$ r% �json�dumps)r r r' �lfsreqr? �
localstorerE �rspr
r
r r a sN
�
����
r c
# sp � |D ]�}|� d�}|�d�}||� d�d�}d}d} z|�|�} | r,|dkr,|�|� W n1 ty^ }
z%|
jtjkrRt� � d|
jpDd d
�|d < |V W Y d }
~
qd}W Y d }
~
nd }
~
ww |d
kr~|spddd
�|d < |V q| s}ddd
�|d < |V qn| r�|V qt j �
� t j dd� } � fdd�} dt �
|� t �
d� j� j|f �| �d�| � d�i|d< |V qd S )a4 Yield one dictionary of attributes for the Batch API response for each
object in the list.
req: The parsedrequest for the Batch API request
objects: The list of objects in the Batch API object request list
action: 'upload' or 'download'
store: The local blob store for servicing requests�oid�ascii�size)rW rY TFr@ i� zInternal Server Server)r" r- �errorNrA i� zThe object does not exist� zThe object is corrupt�
)�minutesc s2 ddi} � j �dd�}|�d�rt�|�| d< | S )N�Acceptzapplication/vnd.git-lfss
Authorizationr0 s Basic �
Authorization)r rN r r �strurl)�hdr�auth�r r
r � _buildheader s
z+_batchresponseobjects.<locals>._buildheaderz%ss %s%s/.hg/lfs/objects/%sz%Y-%m-%dT%H:%M:%SZ)�href�
expires_at�header�actions)rN �encode�verify�linkfromusercache�IOError�errno�ENOENTr< �strerror�datetime�now� timedeltar r` �baseurlr7 �strftime)
r rE �action�store�obj�soidrW rV �exists�verifies�inst� expiresatrd r
rc r rP � sp �
�
�
� ��
�
��
���� �rP c C sL |j }|jd }| jj}t|j�dkrt|t� dS |dkra|d� |�|�}tj }z|�
||j |j d � ||r;t
nt�|_W n tjyT t|� |dd�|_Y nw d |j d
< |�d � dS |d kr�|d
� t� t
�|_d|j d
< z |�|�|�� W dS tjy� t|� t� dd�|_|�d � Y dS w t|td| d� dS )ap Handle a single file upload (PUT) or download (GET) action for the Basic
Transfer Adapter.
After determining if the request is for an upload or download, the access
must be checked by calling ``checkperm()`` with either 'pull' or 'upload'
before accessing the files.
https://github.com/git-lfs/git-lfs/blob/master/docs/api/basic-transfers.md
���� Ts PUTr@ s Content-Lengthr[ s corrupt blobr. r/ r0 s GETr s application/octet-streams# Unsupported LFS transfer method: %sr, )rG �
dispatchpartsr r
�lenr1 �HTTP_NOT_FOUND�hasr r! �downloadrL r rQ � HTTP_CREATEDr$ r �LfsCorruptionErrorr< r% rM rH ) r r r' r rG rW rU �existedr! r
r
r r * sN
�
� � �r r ) �
__future__r rp rm rR r4 �mercurial.hgwebr r � mercurialr r r r � r rQ r� rO r� rH rJ rI �eh� wrapfunctionr+ r1 r< r rP r r
r
r
r �<module> s.
&R w