|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/hgext/fastannotate/__pycache__/ |
Upload File : |
o
�]Lb� � @ s� d dl mZ d dlZd dlZd dlmZ d dlmZ d dlm Z m
Z
m Z m Z m
Z
mZmZ d dlmZ ddlmZ d d
� Zd d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zejdd� �Z d!dd�Z dd � Z d d � Z d d � Z dS )"� )�absolute_importN)�_)�open)�error�
extensions�hg�pycompat�util�wireprotov1peer�wireprotov1server)�urlutil� )�contextc C s" | � dd�}|st�td���|S )z)get the mainbranch, and enforce it is set� fastannotates
mainbranchsF fastannotate.mainbranch is required for both the client and the server)�configr �Abortr )�ui�master� r �=/usr/lib/python3/dist-packages/hgext/fastannotate/protocol.py�
_getmaster s ��r c C s | ||�}|� d� |S )N� getannotate)�append)�orig�repo�proto�resultr r r �
_capabilities/ s
r c
C sl d}| j �ddd�}t�| |���}|rUt| j �}z7z
|�|�s%|�||� W n# tyI |�� z|�||� W n
tyF |�� Y nw Y nw W |� � n|� � w |j
|kr�|j |j fD ]C}t
j�|�siq`t|d�� } | �� }
W d � n1 s}w Y t| jjd �} || d � } |d| t|
�|
f 7 }q`W d � |S W d � |S 1 s�w Y |S )N� r s serverbuildondemandTs rb� /s %s %d %s)r �
configboolr �annotatecontextr �
isuptodate�annotate� Exception�rebuild�close�lastnode�
revmappath� linelogpath�os�path�existsr �read�len�vfs�base)
r r r+ r' r �
buildondemand�actxr �p�f�content�
vfsbaselen�relpathr r r � _getannotate5 sP �
� ����
� �
� �
� � r8 c C s" dt jv rd S t �dd�t� d S )Nr s
path lastnode)r �commands�wireprotocommandr8 r r r r �_registerwireprotocommand` s
�r; c C s t � t�tdt� d S )Ns
_capabilities)r; r � wrapfunctionr r �r r r r �
serveruisetuph s r> c C s� i }d}t | �d }d}d }}||k ra| ||d � }|dkrH|dkrA| |d |d t|� � ||< |t|�7 }d}d }}n|dkrGd}n|dkrQ||7 }n|dkrY||7 }|d7 }||k s|S )Nr r
r � )r. �int)�payloadr �i�l�state�vfspath�size�chr r r �_parseresponsep s.
�
�rH c s G � fdd�d|j �}||_ d S )Nc s e Zd Zejd� fdd� �ZdS )z#peersetup.<locals>.fastannotatepeerNc s2 | � d�s� �td�� dS ||pdd�}|tfS )Nr s* remote peer cannot provide annotate cache
�NNr �s paths lastnode)�capable�warnr rH )�selfr+ r' �argsr= r r � getannotate� s
z/peersetup.<locals>.fastannotatepeer.getannotate�N)�__name__�
__module__� __qualname__r
� batchablerO r r= r r �fastannotatepeer� s rU �� __class__)r �peerrU r r= r � peersetup� s
rY c c s\ � | j }|�ddd�}t�d| ||�}|d }t�|i |�}z
|V W |�� d S |�� w )Nr s
remotepaths defaultr )r r r �get_unique_pull_pathr rX r& )r r �
remotedest�r�
remotepathrX r r r � annotatepeer� s �r^ c
C s� |sdS |du r#t | ��}t| |||�W d � S 1 s w Y |du r)i }| j}g }|�� ��}|�dt|� � |D ]}|�|�d||�|�d��� q>|D ]a}|� � } dd� t
� | �D �} t | �D ]L}
|
�
d�rsd|
d v r{|�d
|
� qf| |
} |jr�|�d t| �|
f � | j�tj�|
�� | �|
d ��
} | �| � W d � n1 s�w Y qfqRW d � dS 1 s�w Y dS )
z1download annotate cache from the server for pathsNs" fastannotate: requesting %d files
r rJ c S s i | ]
\}}t �|�|�qS r )r �pconvert)�.0r3 �vr r r �
<dictcomp>� s z clientfetch.<locals>.<dictcomp>s
fastannotate/s /../r s( fastannotate: ignored malicious path %s
s% fastannotate: writing %d bytes to %s
s wb)r^ � clientfetchr �commandexecutor�debugr. r � callcommand�getr r � iteritems�sorted�
startswith� debugflagr/ �makedirsr* r+ �dirname�write)
r �paths� lastnodemaprX r �results�batcherr3 r r\ r+ r5 r4 r r r rc � sZ
�
�� ��
�� ����"�rc c C sj | j �ddd�}|dkr|S g }|D ] }zt| �|��|kr#|�|� W q ty2 |�|� Y qw |S )z�return a subset of paths whose history is long and need to fetch linelog
from the server. works with remotefilelog and non-remotefilelog repos.
r s clientfetchthreshold�
r )r � configintr. �filer r$ )r ro � thresholdr r+ r r r �_filterfetchpaths� s
� �rw c C s G dd� d|j �}||_ d S )Nc @ s e Zd Zddd�ZdS )z(localreposetup.<locals>.fastannotaterepoNc
S s� t | j�}g }i }z>t| |�D ])}t�| |��}|j|dd�s)|�|� |j||< W d � n1 s3w Y q|rEt| |||� W d S W d S t yb } z| j�
d| � W Y d }~d S d }~ww )NF)�stricts" fastannotate: prefetch failed: %r
) r r rw r r! r" r r' rc r$ re ) rM ro rX r �needupdatepathsrp r+ r2 �exr r r �prefetchfastannotate� s&
� ��� ��z=localreposetup.<locals>.fastannotaterepo.prefetchfastannotaterP )rQ rR rS r{ r r r r �fastannotaterepo� s r| rV )r r r| r r r �localreposetup� s
r} c C s6 t � |�� r t| |� ttjvrtj�t� d S d S rP )r; �localr} rY r �wirepeersetupfuncsr )r r r r r �clientreposetup� s
�r� rI )!�
__future__r �
contextlibr* �mercurial.i18nr �mercurial.pycompatr � mercurialr r r r r r
r �mercurial.utilsr � r r r r8 r; r> rH rY �contextmanagerr^ rc rw r} r� r r r r �<module> s* $ +
0