File: //usr/lib/python3/dist-packages/mercurial/__pycache__/wireprotov1peer.cpython-310.pyc
o
�]LbV � @ s� d dl mZ d dlZd dlZddlmZ ddlmZ ddlm Z m
Z
ddl m Z m
ZmZmZmZmZmZmZ ddlmZmZ dd lmZ ejZd
d � Zd d
� Z G dd� dej j �Z e� ej!�G dd� de"��Z#e� ej$ej%�G dd� dej&��Z'dS )� )�absolute_importN� )�_)�bin)�getattr�setattr)�bundle2� changegroup�encoding�error�pushkey�pycompat�util�wireprototypes)�
repositoryr )�hashutilc s* � fdd�}t |d� � t |d� j� |S )a� annotation for batchable methods
Such methods must implement a coroutine as follows:
@batchable
def sample(self, one, two=None):
# Build list of encoded arguments suitable for your wire protocol:
encoded_args = [('one', encode(one),), ('two', encode(two),)]
# Return it, along with a function that will receive the result
# from the batched request.
return encoded_args, decode
The decorator returns a function which wraps this coroutine as a plain
method, but adds the original method as an attribute called "batchable",
which is used by remotebatch to split the call into separate encoding and
decoding phases.
c sB � | i |��\}}|s
|S | d }t �� j�}|�||�}||�S )Nr )r
�bytesurl�__name__�
_submitone)�args�opts�encoded_args_or_res�decode�self�cmd� encoded_res��f� �;/usr/lib/python3/dist-packages/mercurial/wireprotov1peer.py�plain9 s zbatchable.<locals>.plain� batchabler )r r )r r r r r r! & s r! c sl t j� g }| D ])\}}t� fdd�|D ��sJ �d�� fdd�t�|�D ��}|�d||f � qd�|�S )z;Return a ``cmds`` argument value for the ``batch`` command.c 3 s � | ] }� |�|kV qd S �Nr )�.0�k�� escapeargr r � <genexpr>P s � z"encodebatchcmds.<locals>.<genexpr>� ,c 3 s( � | ]\}}d � |�� |�f V qdS )s %s=%sNr )r# r$ �vr% r r r' R s
� �
�s %s %s� ;)r �escapebatcharg�all�joinr
� iteritems�append)�req�cmds�op�argsdictr r r% r �encodebatchcmdsG s �
r4 c @ s e Zd ZdZddd�ZdS )� unsentfuturea0 A Future variation to represent an unsent command.
Because we buffer commands and don't submit them immediately, calling
``result()`` on an unsent future could deadlock. Futures for buffered
commands are represented by this type, which wraps ``result()`` to
call ``sendcommands()``.
Nc C s, | � � r tjj�| |�S | j�� | �|�S r" )�doner
�futures�Future�result�
_peerexecutor� sendcommands)r �timeoutr r r r9 d s
zunsentfuture.resultr" )r �
__module__� __qualname__�__doc__r9 r r r r r5 [ s r5 c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )� peerexecutorc C s2 || _ d| _d| _g | _t�� | _d | _d | _d S )NF) �_peer�_sent�_closed�_calls�weakref�WeakSet�_futures�_responseexecutor�
_responsef)r �peerr r r �__init__r s
zpeerexecutor.__init__c C s | S r" r �r r r r � __enter__{ s zpeerexecutor.__enter__c C s | � � d S r" )�close)r �exctype�excvalee�exctbr r r �__exit__~ s zpeerexecutor.__exit__c s� �j rt�d���jrt�d��t�jt���d ���s#t�d� ��� ���fdd�}t�dd�r=|� }t|_ �|_
|S �j rGt�d� ��|� }�� � |S ) Ns4 callcommand() cannot be used after commands are sents* callcommand() cannot be used after close()sA cannot call command %s: method of same name not available on peerc s. t j�� } �j�| � �j��� �| f� | S r" )r
r7 r8 rG �addrD r/ r �r �command�fnr r r �addcall� s
z)peerexecutor.callcommand.<locals>.addcallr! FsX %s is not batchable and cannot be called on a command executor along with other commands)
rB r �ProgrammingErrorrC r rA r
�sysstrr5 � __class__r: rD r; )r rU r rW r r rT r � callcommand� s< ���� ���zpeerexecutor.callcommandc
C s� | j rd S | js
d S d| _ | jD ]}t|t�r tjj|_d |_ q| j}d | _t
|�dkrd|d \}}}}|� � s:d S z |di t� |���}W n t
y\ t�|t�� dd � � Y d S w |�|� d S g }g }|D ]L\}}}}|� � suqjz|j|jfi t� |���\} }
W n t
y� t�|t�� dd � � Y d S w |
s�|�| � qj|�|| f� |�||t|
f� qj|s�d S | j�|�} tj�d�| _| j�| j|| �| _d S )NTr r r ) rB rD rG �
isinstancer5 r
r7 r8 rZ r: �len� set_running_or_notify_cancel� strkwargs� Exception�future_set_exception_info�sys�exc_info�
set_resultr! �__self__r/ rA � _submitbatch�ThreadPoolExecutorrH �submit�_readbatchresponserI ) r r �callsrU r rV r9 �requests�statesr r � wireresultsr r r r; � sd
� �
�� �
�zpeerexecutor.sendcommandsc
C s� | � � | jr d S d| _| jsd S z-| j�� W | jjdd� d | _d | _| jD ]}|�� s9|�t �
t d�d �� q(d | _d S | jjdd� d | _d | _| jD ]}|�� s`|�t �
t d�d �� qOd | _w )NT)�wait�" unfulfilled batch command response) r; rC rI r9 rH �shutdownrG r6 �
set_exceptionr �
ResponseErrorr )r r r r r rN
s@
���
�
���zpeerexecutor.closec
C s� |D ]B\}}}}zt |�}W n ty" |�t�td�d �� Y qw z||�}W n ty> t�|t �
� dd � � Y qw |� |� qd S )Nro r ) �next�
StopIterationrq r rr r r` r
ra rb rc rd ) r rl rm rU r r! r � remoteresultr9 r r r ri * s" ���
� �z peerexecutor._readbatchresponseN)
r r= r> rK rM rR r[ r; rN ri r r r r r@ p s 7R r@ c @ s� e Zd ZdZdd� Zdd� Zedd� �Zedd � �Zed
d � �Z ed d
� �Z
edd� �Z edd� �Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd3d%d&�Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d$S )4�wirepeerz�Client-side interface for communicating with a peer repository.
Methods commonly call wire protocol commands of the same name.
See also httppeer.py and sshpeer.py for protocol-specific
implementations of this interface.
c C s t | �S r" )r@ rL r r r �commandexecutorP s zwirepeer.commandexecutorc C s | � dtd�� | �d�S )Ns clonebundless
clone bundles)�
requirecapr �_callrL r r r � clonebundlesU s
zwirepeer.clonebundlesc s. � � dtd�� � fdd�}dt�|�i|fS )Ns lookups look up remote revisionc s< | d d� � dd�\}}t|�rt|�S � �t�|�� d S )N���� r )�split�intr �_abortr � RepoError)�d�success�datarL r r r ] s z wirepeer.lookup.<locals>.decode� key)rx r r
� fromlocal)r �keyr r rL r �lookupY s zwirepeer.lookupc � � fdd�}i |fS )Nc s@ z
t �| d d� �W S ty � �t�td�| �� Y d S w )Nr{ � unexpected response:)r �
decodelist�
ValueErrorr r rr r �r� rL r r r h s
�z wirepeer.heads.<locals>.decoder �r r r rL r �headsf s zwirepeer.headsc s � fdd�}dt �|�i|fS )Nc sB z dd� t �| �D �W S ty � �t�td�| �� Y d S w )Nc S s g | ]}t t|���qS r )�boolr~ �r# �br r r �
<listcomp>t s z2wirepeer.known.<locals>.decode.<locals>.<listcomp>r� )r
� iterbytestrr� r r rr r r� rL r r r r s
�z wirepeer.known.<locals>.decode� nodes�r �
encodelist)r �nodesr r rL r �knownp s zwirepeer.knownc r� )Nc sv z%i }| � � D ]}|�dd�\}}t�t�|��}t�|�}|||< q|W S ty: � � t
� t d�| �� Y d S w )Nr| r r� )
�
splitlinesr} r
�tolocal�urlreq�unquoter r� � TypeErrorr r rr r )r� � branchmap�
branchpart�
branchname� branchheadsrL r r r | s
�z"wirepeer.branchmap.<locals>.decoder r� r rL r r� z s zwirepeer.branchmapc sB �� d�s i d fS �j�d� � � �fdd�}dt�� �i|fS )N� pushkeys preparing listkeys for "%s"
c s" �j �d� t| �f � t�| �S )Ns$ received listkey for "%s": %i bytes
)�ui�debugr] �
pushkeymod�
decodekeysr� �� namespacer r r r � s �
z!wirepeer.listkeys.<locals>.decode� namespace��capabler� r� r
r� )r r� r r r� r �listkeys� s
zwirepeer.listkeysc sX � � d�sdS � j�d||f � � fdd�}t�|�t�|�t�|�t�|�d�|fS )Nr� )FNs preparing pushkey for "%s:%s"
c sf | � dd�\} }ztt| ��} W n ty t�td�| ��w |�d�D ] }� j� td�|� q%| S )N�
r �" push failed (unexpected response):T� remote: )
r} r� r~ r� r rr r r� r� �status)r� �output�lrL r r r � s ��z wirepeer.pushkey.<locals>.decode)r� r� s olds newr� )r r� r� �old�newr r rL r r � s
��zwirepeer.pushkeyc C s
| � d�S )Ns
stream_out)� _callstreamrL r r r �
stream_out� s
zwirepeer.stream_outc K s" t �|�}| �dtd�� i }|�d�pt� }t �|�D ]S\}}|d u r%q tj�|�}|d u r6t �
d| ��|dkr@t� |�}n+|dkrJd� |�}n!|dkrVd� t
|��}n|d krad
t|� }n
|d krktd | ��|||< q | jdi t �|���}td
d� |D ��r�t�| j|�S t�|d�S )N� getbundle� look up remote changess
bundlecapss$ Unexpectedly None keytype for key %sr� s csvr( s scsvs booleans %is plains unknown getbundle option type %sc s s � | ]}|� d �V qdS )s HG2N)�
startswith)r# �capr r r r' � s � z%wirepeer.getbundle.<locals>.<genexpr>� UN)r� )r
� byteskwargsrx r �get�setr. r �GETBUNDLE_ARGUMENTSr rX r� r- �sortedr� �KeyError�_callcompressabler_ �anyr � getunbundlerr� �changegroupmod� cg1unpacker) r �source�kwargsr �
bundlecapsr� �value�keytyper r r r � getbundle� s6
�
zwirepeer.getbundlec C s� |dgkr | � d�r t�dt�d�t|����� g�}nt�|�}t� |d�rf| j
d||d�\}}|dkr=t � t
d�|��zt|�}W n tyR t � t
d �|��w |�d
�D ] }| j�t
d �|� qX|S | jd||d�}t�| j|�}|S ) a� Send cg (a readable file-like object representing the
changegroup to push, typically a chunkbuffer object) to the
remote server as a bundle.
When pushing a bundle10 stream, return an integer indicating the
result of the push (see changegroup.apply()).
When pushing a bundle20 stream, return a bundle20 stream.
`url` is the url the client thinks it's pushing to, which is
visible to hooks.
s forces unbundlehashs hashed� s deltaheaders unbundle)r� s push failed:r� Tr� )r� r r� r �sha1r- r� �digestr � safehasattr� _callpushr rr r r~ r� r� r� r� �_calltwowaystreamr r� )r �bundler� �url�retr� r� �streamr r r �unbundle� s, �
���zwirepeer.unbundlec C s\ t �|�}| jd|d�}z dd� |�� D �}|W S ty- | �t�td�|�� Y d S w )Ns branches)r� c S s g | ] }t t�|���qS r )�tupler r� r� r r r r� s z%wirepeer.branches.<locals>.<listcomp>r� ) r r� ry r� r� r r rr r )r r� �nr� �brr r r �branches� s
�zwirepeer.branchesc
C s� d}g }t �dt|�|�D ];}d�dd� |||| � D ��}| jd|d�}z|�dd � |�� D �� W q
tyH | �t �
t d
�|�� Y q
w |S ) N� r r| c S s g | ]}t �|d ��qS )� -r� )r# �pr r r r�
s
��z$wirepeer.between.<locals>.<listcomp>s between)�pairsc s s" � | ] }|r t �|�p g V qd S r" )r r� )r# r� r r r r' s
� �
�z#wirepeer.between.<locals>.<genexpr>r� ) r
�xranger] r- ry �extendr� r� r r rr r )r r� �batch�r�ir� r� r r r �between s" ��
� �zwirepeer.betweenc C s$ t �|�}| jd|d�}t�|d�S )Ns changegroup)�rootsr� )r r� r� r� r� )r r� r� r� r r r r r s
zwirepeer.changegroupc C s@ | � dtd�� t�|�}t�|�}| jd||d�}t�|d�S )Ns changegroupsubsetr� )�basesr� r� )rx r r r� r� r� r� )r r� r� r� r r r r �changegroupsubset! s
� zwirepeer.changegroupsubsetc c s � | j }|jr&|�dd�r&|�d� |D ]\}}d}|�||t|�f � qtj}| jdt|�d�}|� d�}|g} |rwd|vrR|rR|� d�}| �
|� d|vrR|sBd � | �}
d|
v rl|
� dd
�\} }
|| �V d|
v s[|� d�}|
|g} |s<|d � | ��V d S ) zqrun batch request <req> on the server
Returns an iterator of the raw responses from the server.
s devels debug.peer-requests$ devel-peer-request: batched-content
s+ devel-peer-request: - %s (%d arguments)
s batch)r1 i r* r� r N)
r� � debugflag�
configboolr� r] r �unescapebatchargr� r4 �readr/ r- r} ) r r0 r� r2 r �msg� unescapearg�rsp�chunk�work�merged�oner r r rf , s2 �
�
�
�
zwirepeer._submitbatchc C s | j |fi t�|���S r" )ry r
r_ )r r2 r r r r r I s zwirepeer._submitoneNc C s: i }|d ur
||d< |d ur||d< | j d||d�|��S )N�three�four�
debugwireargs)r� �two)r� )ry )r r� r� r� r� �fiver r r r �
debugwireargsL s zwirepeer.debugwireargsc K � t � �)z�execute <cmd> on the server
The command is expected to return a simple string.
returns the server reply as a string.��NotImplementedError�r r r r r r ry U s zwirepeer._callc K r� )a execute <cmd> on the server
The command is expected to return a stream. Note that if the
command doesn't return a stream, _callstream behaves
differently for ssh and http peers.
returns the server reply as a file like object.
r� r� r r r r� ] � zwirepeer._callstreamc K r� )aE execute <cmd> on the server
The command is expected to return a stream.
The stream may have been compressed in some implementations. This
function takes care of the decompression. This is the only difference
with _callstream.
returns the server reply as a file like object.
r� r� r r r r� h s zwirepeer._callcompressablec K r� )z�execute a <cmd> on server
The command is expected to be related to a push. Push has a special
return method.
returns the server reply as a (ret, output) tuple. ret is either
empty (error) or a stringified int.
r� �r r �fpr r r r r� u r� zwirepeer._callpushc K r� )zqexecute <cmd> on server
The command will send a stream to the server and get a stream in reply.
r� r� r r r r� � s zwirepeer._calltwowaystreamc C r� )zBclearly abort the wire protocol connection and raise the exceptionr� )r � exceptionr r r r � s zwirepeer._abort)NNN) r r= r> r? rw rz r! r� r� r� r� r� r r� r� r� r� r� r r� rf r r� ry r� r� r� r� r r r r r rv D s@
-
rv )(�
__future__r rb rE �i18nr �noder r
r r � r r r� r
r r r� r r �
interfacesr �
interfaceutil�utilsr r� r! r4 r7 r8 r5 � implementer�ipeercommandexecutor�objectr@ �
ipeercommands�ipeerlegacycommandsrJ rv r r r r �<module> s( (
!
T�