ruạṛ
a �+�b / � @ 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 | ��|� �rdS �j j�dd�s&dS t��j jd�s:dS �jsDdS zV�jdkrl� ��d� t�j �|�W S �j�d�r�t �j �|� ��fd d ��W S W dS t j �y } zR|j D ]\}}||j |< q�t � |jt�|��|_|�d t�|� � W Y d }~dS d }~0 0 d S ) 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� )Zperm�� checkperm�rctx�reqr �B/usr/lib64/python3.9/site-packages/hgext/lfs/wireprotolfsserver.py�<lambda>> � z#handlewsgirequest.<locals>.<lambda>s 0 %s N)�repoZuiZ configboolr Z safehasattr�svfs� dispatchpath�_processbatchrequest� startswith�_processbasictransfer� hgwebcommonZ ErrorResponse�headers� statusmessage�coder �bytestr�status� setbodybytes)Zorigr r �resr �e�k�vr r r �handlewsgirequest% s. � r% c C s( t j||d�| _d| jd< | �d� d S )N��message� text/plain; charset=utf-8� Content-Typer )r r r r r )r! r r' r r r � _sethttperrorL s r* 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.errorsr � /s4 Exception happened while processing request '%s': %sN)r Zsysbytes� traceback� format_excZrawenv�apppathr �write)r �tbZerrorlogZurir r r � _logexceptionR s �r1 c C s |j dkrt|t� dS |jd dkr4t|t� dS |jd dkrPt|t� dS t�|j� � �}d|� ddg�vr�t|t d� dS |� d �}t� |�}|d vr�t|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 AcceptZbasicZ 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)Ztransferr8 )�methodr* �HTTP_METHOD_NOT_ALLOWEDr �HTTP_UNSUPPORTED_MEDIA_TYPE�HTTP_NOT_ACCEPTABLEr Z json_loads�bodyfh�read�get�HTTP_BAD_REQUESTr r r �_batchresponseobjectsr r �HTTP_OKr r �json�dumps)r r r! Zlfsreqr2 � localstorer8 �rspr r r r a sN � ���� r c # st |D �]h}|� d�}|�d�}||� d�d�}d}d} z$|�|�} | rX|dkrX|�|� W nb ty� } zJ| jtjkr�t� � d| jp�d d �|d < |V W Y d } ~ qd}W Y d } ~ n d } ~ 0 0 |d kr�|s�ddd �|d < |V qn| 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)rG rI TFr3 i� zInternal Server Server)r r' �errorNr4 i� zThe object does not exist� zThe object is corrupt� )Zminutesc s2 ddi} � j �dd�}|�d�r.t�|�| d< | S )NZAcceptzapplication/vnd.git-lfss Authorizationr s Basic Z Authorization)r r? r r �strurl)ZhdrZauth�r r r � _buildheader s z+_batchresponseobjects.<locals>._buildheaderz%ss %s%s/.hg/lfs/objects/%sz%Y-%m-%dT%H:%M:%SZ)ZhrefZ expires_at�header�actions)r? �encodeZverifyZlinkfromusercache�IOError�errno�ENOENTr1 �strerror�datetimeZnowZ timedeltar rM Zbaseurlr. �strftime) r r8 �action�store�objZsoidrG rF �existsZverifies�instZ expiresatrO r rN r rA � sd � � � � ���� rA c C sT |j }|jd }| jj}t|j�dkr4t|t� dS |dkr�|d� |�|�}tj }z,|� ||j |j d � ||rvt nt�|_W n( tjy� t|� |dd�|_Y n0 d |j d < |�d � dS |d k�r:|d � t� t �|_d|j d < z|�|�|�� W n6 tj�y4 t|� t� dd�|_|�d � Y n0 dS t|td| d� dS 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 PUTr3 s Content-LengthrK s corrupt blobr( r) r s GETr s application/octet-streams# Unsupported LFS transfer method: %sr& N)r9 Z dispatchpartsr r �lenr* �HTTP_NOT_FOUNDZhasr r Zdownloadr= r rB � HTTP_CREATEDr r ZLfsCorruptionErrorr1 r r> r: ) r r r! r r9 rG rE Zexistedr r r r r * sH �r )N) Z __future__r rW rT rC r, Zmercurial.hgwebr r Z mercurialr r r r � r rB rb r@ ra r: r< r; ZehZ wrapfunctionr% r* r1 r rA r r r r r �<module> s, & Rw
cải xoăn