mercurial-scm/hg-stable: mercurial/wireprototypes.py annotate

annotate mercurial/wireprototypes.py @ 51788:278af66e6595

typing: induce pytype to use the standard `attr` instead of the vendored copy What was previously happening with the vendored copy was that pytype would stub out all(?) classes that were decorated with `@attr.s` as `Any`. After this, we get a ton of classes defined, and numerous fields and methods now have proper types.

author	Matt Harbison <matt_harbison@yahoo.com>
date	Tue, 23 Jul 2024 19:20:22 -0400
parents	18c8c18993f0
children	f4733654f144

rev	line source
36111 cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1 # Copyright 2018 Gregory Szorc <gregory.szorc@gmail.com>
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	2 #
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	3 # This software may be used and distributed according to the terms of the
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	4 # GNU General Public License version 2 or any later version.
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	5
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	6
51788 278af66e6595 typing: induce pytype to use the standard `attr` instead of the vendored copy Matt Harbison <matt_harbison@yahoo.com> parents: 50952 diff changeset	7 import typing
278af66e6595 typing: induce pytype to use the standard `attr` instead of the vendored copy Matt Harbison <matt_harbison@yahoo.com> parents: 50952 diff changeset	8
37612 5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	9 from .node import (
5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	10 bin,
5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	11 hex,
5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	12 )
37783 9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	13 from .i18n import _
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	14 from .thirdparty import attr
51788 278af66e6595 typing: induce pytype to use the standard `attr` instead of the vendored copy Matt Harbison <matt_harbison@yahoo.com> parents: 50952 diff changeset	15
278af66e6595 typing: induce pytype to use the standard `attr` instead of the vendored copy Matt Harbison <matt_harbison@yahoo.com> parents: 50952 diff changeset	16 # Force pytype to use the non-vendored package
278af66e6595 typing: induce pytype to use the standard `attr` instead of the vendored copy Matt Harbison <matt_harbison@yahoo.com> parents: 50952 diff changeset	17 if typing.TYPE_CHECKING:
278af66e6595 typing: induce pytype to use the standard `attr` instead of the vendored copy Matt Harbison <matt_harbison@yahoo.com> parents: 50952 diff changeset	18 # noinspection PyPackageRequirements
278af66e6595 typing: induce pytype to use the standard `attr` instead of the vendored copy Matt Harbison <matt_harbison@yahoo.com> parents: 50952 diff changeset	19 import attr
278af66e6595 typing: induce pytype to use the standard `attr` instead of the vendored copy Matt Harbison <matt_harbison@yahoo.com> parents: 50952 diff changeset	20
37783 9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	21 from . import (
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	22 error,
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	23 util,
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	24 )
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	25 from .interfaces import util as interfaceutil
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	26 from .utils import compression
36402 0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	27
36565 3cd245945ef3 wireprotoserver: move SSHV1 and SSHV2 constants to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 36402 diff changeset	28 # Names of the SSH protocol implementations.
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	29 SSHV1 = b'ssh-v1'
36565 3cd245945ef3 wireprotoserver: move SSHV1 and SSHV2 constants to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 36402 diff changeset	30
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	31 NARROWCAP = b'exp-narrow-1'
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	32 ELLIPSESCAP1 = b'exp-ellipses-1'
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	33 ELLIPSESCAP = b'exp-ellipses-2'
42432 c767e655ffda narrow: use narrow_widen wireproto command to widen in case of ellipses Pulkit Goyal <7895pulkit@gmail.com> parents: 42041 diff changeset	34 SUPPORTED_ELLIPSESCAP = (ELLIPSESCAP1, ELLIPSESCAP)
40075 f7011b44d205 wireprotoserver: move narrow capabilities to wireprototypes.py Pulkit Goyal <pulkit@yandex-team.ru> parents: 40026 diff changeset	35
36638 abc3b9801563 wireproto: allow wire protocol commands to declare transport support Gregory Szorc <gregory.szorc@gmail.com> parents: 36565 diff changeset	36 # All available wire protocol transports.
abc3b9801563 wireproto: allow wire protocol commands to declare transport support Gregory Szorc <gregory.szorc@gmail.com> parents: 36565 diff changeset	37 TRANSPORTS = {
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 44470 diff changeset	38 SSHV1: {
89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 44470 diff changeset	39 b'transport': b'ssh',
89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 44470 diff changeset	40 b'version': 1,
89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 44470 diff changeset	41 },
89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 44470 diff changeset	42 b'http-v1': {
89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 44470 diff changeset	43 b'transport': b'http',
89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 44470 diff changeset	44 b'version': 1,
89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 44470 diff changeset	45 },
36638 abc3b9801563 wireproto: allow wire protocol commands to declare transport support Gregory Szorc <gregory.szorc@gmail.com> parents: 36565 diff changeset	46 }
abc3b9801563 wireproto: allow wire protocol commands to declare transport support Gregory Szorc <gregory.szorc@gmail.com> parents: 36565 diff changeset	47
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	48
49037 642e31cb55f0 py3: use class X: instead of class X(object): Gregory Szorc <gregory.szorc@gmail.com> parents: 48966 diff changeset	49 class bytesresponse:
36112 2f7290555c96 wireproto: introduce type for raw byte responses (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 36111 diff changeset	50 """A wire protocol response consisting of raw bytes."""
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	51
36112 2f7290555c96 wireproto: introduce type for raw byte responses (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 36111 diff changeset	52 def __init__(self, data):
2f7290555c96 wireproto: introduce type for raw byte responses (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 36111 diff changeset	53 self.data = data
2f7290555c96 wireproto: introduce type for raw byte responses (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 36111 diff changeset	54
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	55
49037 642e31cb55f0 py3: use class X: instead of class X(object): Gregory Szorc <gregory.szorc@gmail.com> parents: 48966 diff changeset	56 class ooberror:
36111 cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	57 """wireproto reply: failure of a batch of operation
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	58
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	59 Something failed during a batch call. The error message is stored in
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	60 `self.message`.
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	61 """
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	62
36111 cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	63 def __init__(self, message):
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	64 self.message = message
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	65
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	66
49037 642e31cb55f0 py3: use class X: instead of class X(object): Gregory Szorc <gregory.szorc@gmail.com> parents: 48966 diff changeset	67 class pushres:
36111 cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	68 """wireproto reply: success with simple integer return
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	69
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	70 The call was successful and returned an integer contained in `self.res`.
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	71 """
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	72
36111 cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	73 def __init__(self, res, output):
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	74 self.res = res
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	75 self.output = output
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	76
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	77
49037 642e31cb55f0 py3: use class X: instead of class X(object): Gregory Szorc <gregory.szorc@gmail.com> parents: 48966 diff changeset	78 class pusherr:
36111 cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	79 """wireproto reply: failure
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	80
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	81 The call failed. The `self.res` attribute contains the error message.
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	82 """
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	83
36111 cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	84 def __init__(self, res, output):
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	85 self.res = res
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	86 self.output = output
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	87
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	88
49037 642e31cb55f0 py3: use class X: instead of class X(object): Gregory Szorc <gregory.szorc@gmail.com> parents: 48966 diff changeset	89 class streamres:
36111 cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	90 """wireproto reply: binary stream
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	91
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	92 The call was successful and the result is a stream.
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	93
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	94 Accepts a generator containing chunks of data to be sent to the client.
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	95
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	96 ``prefer_uncompressed`` indicates that the data is expected to be
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	97 uncompressable and that the stream should therefore use the ``none``
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	98 engine.
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	99 """
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	100
36111 cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	101 def __init__(self, gen=None, prefer_uncompressed=False):
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	102 self.gen = gen
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	103 self.prefer_uncompressed = prefer_uncompressed
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	104
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	105
49037 642e31cb55f0 py3: use class X: instead of class X(object): Gregory Szorc <gregory.szorc@gmail.com> parents: 48966 diff changeset	106 class streamreslegacy:
36111 cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	107 """wireproto reply: uncompressed binary stream
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	108
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	109 The call was successful and the result is a stream.
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	110
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	111 Accepts a generator containing chunks of data to be sent to the client.
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	112
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	113 Like ``streamres``, but sends an uncompressed data for "version 1" clients
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	114 using the application/mercurial-0.1 media type.
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	115 """
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	116
36111 cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	117 def __init__(self, gen=None):
cd6ab329c5c7 wireprototypes: move wire protocol response types to new module Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	118 self.gen = gen
36402 0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	119
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	120
37612 5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	121 # list of nodes encoding / decoding
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	122 def decodelist(l, sep=b' '):
37612 5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	123 if l:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	124 return [bin(v) for v in l.split(sep)]
37612 5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	125 return []
5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	126
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	127
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	128 def encodelist(l, sep=b' '):
37612 5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	129 try:
5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	130 return sep.join(map(hex, l))
5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	131 except TypeError:
5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	132 raise
5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	133
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	134
37612 5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	135 # batched call argument encoding
5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	136
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	137
37612 5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	138 def escapebatcharg(plain):
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	139 return (
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	140 plain.replace(b':', b':c')
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	141 .replace(b',', b':o')
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	142 .replace(b';', b':s')
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	143 .replace(b'=', b':e')
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	144 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	145
37612 5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	146
5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	147 def unescapebatcharg(escaped):
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	148 return (
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	149 escaped.replace(b':e', b'=')
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	150 .replace(b':s', b';')
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	151 .replace(b':o', b',')
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	152 .replace(b':c', b':')
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	153 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	154
37612 5e71dea79aae wireproto: move value encoding functions to wireprototypes (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37485 diff changeset	155
37613 96d735601ca1 wireproto: move gboptsmap to wireprototypes and rename (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37612 diff changeset	156 # mapping of options accepted by getbundle and their types
96d735601ca1 wireproto: move gboptsmap to wireprototypes and rename (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37612 diff changeset	157 #
42958 9668744c9122 wireprototypes: clarify documentation of getbundle argument types Martin von Zweigbergk <martinvonz@google.com> parents: 42824 diff changeset	158 # Meant to be extended by extensions. It is the extension's responsibility to
9668744c9122 wireprototypes: clarify documentation of getbundle argument types Martin von Zweigbergk <martinvonz@google.com> parents: 42824 diff changeset	159 # ensure such options are properly processed in exchange.getbundle.
37613 96d735601ca1 wireproto: move gboptsmap to wireprototypes and rename (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37612 diff changeset	160 #
96d735601ca1 wireproto: move gboptsmap to wireprototypes and rename (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37612 diff changeset	161 # supported types are:
96d735601ca1 wireproto: move gboptsmap to wireprototypes and rename (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37612 diff changeset	162 #
42958 9668744c9122 wireprototypes: clarify documentation of getbundle argument types Martin von Zweigbergk <martinvonz@google.com> parents: 42824 diff changeset	163 # :nodes: list of binary nodes, transmitted as space-separated hex nodes
9668744c9122 wireprototypes: clarify documentation of getbundle argument types Martin von Zweigbergk <martinvonz@google.com> parents: 42824 diff changeset	164 # :csv: list of values, transmitted as comma-separated values
9668744c9122 wireprototypes: clarify documentation of getbundle argument types Martin von Zweigbergk <martinvonz@google.com> parents: 42824 diff changeset	165 # :scsv: set of values, transmitted as comma-separated values
37613 96d735601ca1 wireproto: move gboptsmap to wireprototypes and rename (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37612 diff changeset	166 # :plain: string with no transformation needed.
96d735601ca1 wireproto: move gboptsmap to wireprototypes and rename (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37612 diff changeset	167 GETBUNDLE_ARGUMENTS = {
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	168 b'heads': b'nodes',
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	169 b'bookmarks': b'boolean',
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	170 b'common': b'nodes',
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	171 b'obsmarkers': b'boolean',
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	172 b'phases': b'boolean',
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	173 b'bundlecaps': b'scsv',
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	174 b'listkeys': b'csv',
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	175 b'cg': b'boolean',
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	176 b'cbattempted': b'boolean',
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	177 b'stream': b'boolean',
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	178 b'includepats': b'csv',
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	179 b'excludepats': b'csv',
37613 96d735601ca1 wireproto: move gboptsmap to wireprototypes and rename (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37612 diff changeset	180 }
96d735601ca1 wireproto: move gboptsmap to wireprototypes and rename (API) Gregory Szorc <gregory.szorc@gmail.com> parents: 37612 diff changeset	181
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	182
37810 856f381ad74b interfaceutil: module to stub out zope.interface Gregory Szorc <gregory.szorc@gmail.com> parents: 37783 diff changeset	183 class baseprotocolhandler(interfaceutil.Interface):
36402 0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	184 """Abstract base class for wire protocol handlers.
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	185
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	186 A wire protocol handler serves as an interface between protocol command
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	187 handlers and the wire protocol transport layer. Protocol handlers provide
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	188 methods to read command arguments, redirect stdio for the duration of
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	189 the request, handle response types, etc.
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	190 """
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	191
37810 856f381ad74b interfaceutil: module to stub out zope.interface Gregory Szorc <gregory.szorc@gmail.com> parents: 37783 diff changeset	192 name = interfaceutil.Attribute(
36402 0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	193 """The name of the protocol implementation.
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	194
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	195 Used for uniquely identifying the transport type.
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	196 """
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	197 )
36402 0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	198
37296 78103e4138b1 wireproto: port protocol handler to zope.interface Gregory Szorc <gregory.szorc@gmail.com> parents: 37294 diff changeset	199 def getargs(args):
36402 0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	200 """return the value for arguments in <args>
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	201
37485 0b7475ea38cf wireproto: port heads command to wire protocol v2 Gregory Szorc <gregory.szorc@gmail.com> parents: 37414 diff changeset	202 For version 1 transports, returns a list of values in the same
0b7475ea38cf wireproto: port heads command to wire protocol v2 Gregory Szorc <gregory.szorc@gmail.com> parents: 37414 diff changeset	203 order they appear in ``args``. For version 2 transports, returns
0b7475ea38cf wireproto: port heads command to wire protocol v2 Gregory Szorc <gregory.szorc@gmail.com> parents: 37414 diff changeset	204 a dict mapping argument name to value.
0b7475ea38cf wireproto: port heads command to wire protocol v2 Gregory Szorc <gregory.szorc@gmail.com> parents: 37414 diff changeset	205 """
36402 0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	206
37393 afcfdf53e4b5 wireproto: provide accessors for client capabilities Joerg Sonnenberger <joerg@bec.de> parents: 37296 diff changeset	207 def getprotocaps():
afcfdf53e4b5 wireproto: provide accessors for client capabilities Joerg Sonnenberger <joerg@bec.de> parents: 37296 diff changeset	208 """Returns the list of protocol-level capabilities of client
afcfdf53e4b5 wireproto: provide accessors for client capabilities Joerg Sonnenberger <joerg@bec.de> parents: 37296 diff changeset	209
afcfdf53e4b5 wireproto: provide accessors for client capabilities Joerg Sonnenberger <joerg@bec.de> parents: 37296 diff changeset	210 Returns a list of capabilities as declared by the client for
afcfdf53e4b5 wireproto: provide accessors for client capabilities Joerg Sonnenberger <joerg@bec.de> parents: 37296 diff changeset	211 the current request (or connection for stateful protocol handlers)."""
afcfdf53e4b5 wireproto: provide accessors for client capabilities Joerg Sonnenberger <joerg@bec.de> parents: 37296 diff changeset	212
37414 2d965bfeb8f6 wireproto: allow direct stream processing for unbundle Joerg Sonnenberger <joerg@bec.de> parents: 37393 diff changeset	213 def getpayload():
2d965bfeb8f6 wireproto: allow direct stream processing for unbundle Joerg Sonnenberger <joerg@bec.de> parents: 37393 diff changeset	214 """Provide a generator for the raw payload.
36402 0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	215
37414 2d965bfeb8f6 wireproto: allow direct stream processing for unbundle Joerg Sonnenberger <joerg@bec.de> parents: 37393 diff changeset	216 The caller is responsible for ensuring that the full payload is
2d965bfeb8f6 wireproto: allow direct stream processing for unbundle Joerg Sonnenberger <joerg@bec.de> parents: 37393 diff changeset	217 processed.
36402 0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	218 """
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	219
37296 78103e4138b1 wireproto: port protocol handler to zope.interface Gregory Szorc <gregory.szorc@gmail.com> parents: 37294 diff changeset	220 def mayberedirectstdio():
36402 0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	221 """Context manager to possibly redirect stdio.
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	222
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	223 The context manager yields a file-object like object that receives
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	224 stdout and stderr output when the context manager is active. Or it
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	225 yields ``None`` if no I/O redirection occurs.
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	226
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	227 The intent of this context manager is to capture stdio output
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	228 so it may be sent in the response. Some transports support streaming
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	229 stdio to the client in real time. For these transports, stdio output
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	230 won't be captured.
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	231 """
0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	232
37296 78103e4138b1 wireproto: port protocol handler to zope.interface Gregory Szorc <gregory.szorc@gmail.com> parents: 37294 diff changeset	233 def client():
36402 0c231df1ffdc wireprototypes: move baseprotocolhandler from wireprotoserver Gregory Szorc <gregory.szorc@gmail.com> parents: 36112 diff changeset	234 """Returns a string representation of this client (as bytes)."""
36642 6e585bca962e wireproto: add transport specific capabilities in the transport Gregory Szorc <gregory.szorc@gmail.com> parents: 36638 diff changeset	235
37296 78103e4138b1 wireproto: port protocol handler to zope.interface Gregory Szorc <gregory.szorc@gmail.com> parents: 37294 diff changeset	236 def addcapabilities(repo, caps):
36642 6e585bca962e wireproto: add transport specific capabilities in the transport Gregory Szorc <gregory.szorc@gmail.com> parents: 36638 diff changeset	237 """Adds advertised capabilities specific to this protocol.
6e585bca962e wireproto: add transport specific capabilities in the transport Gregory Szorc <gregory.szorc@gmail.com> parents: 36638 diff changeset	238
6e585bca962e wireproto: add transport specific capabilities in the transport Gregory Szorc <gregory.szorc@gmail.com> parents: 36638 diff changeset	239 Receives the list of capabilities collected so far.
6e585bca962e wireproto: add transport specific capabilities in the transport Gregory Szorc <gregory.szorc@gmail.com> parents: 36638 diff changeset	240
6e585bca962e wireproto: add transport specific capabilities in the transport Gregory Szorc <gregory.szorc@gmail.com> parents: 36638 diff changeset	241 Returns a list of capabilities. The passed in argument can be returned.
6e585bca962e wireproto: add transport specific capabilities in the transport Gregory Szorc <gregory.szorc@gmail.com> parents: 36638 diff changeset	242 """
36809 66de4555cefd wireproto: formalize permissions checking as part of protocol interface Gregory Szorc <gregory.szorc@gmail.com> parents: 36642 diff changeset	243
37296 78103e4138b1 wireproto: port protocol handler to zope.interface Gregory Szorc <gregory.szorc@gmail.com> parents: 37294 diff changeset	244 def checkperm(perm):
36809 66de4555cefd wireproto: formalize permissions checking as part of protocol interface Gregory Szorc <gregory.szorc@gmail.com> parents: 36642 diff changeset	245 """Validate that the client has permissions to perform a request.
66de4555cefd wireproto: formalize permissions checking as part of protocol interface Gregory Szorc <gregory.szorc@gmail.com> parents: 36642 diff changeset	246
66de4555cefd wireproto: formalize permissions checking as part of protocol interface Gregory Szorc <gregory.szorc@gmail.com> parents: 36642 diff changeset	247 The argument is the permission required to proceed. If the client
66de4555cefd wireproto: formalize permissions checking as part of protocol interface Gregory Szorc <gregory.szorc@gmail.com> parents: 36642 diff changeset	248 doesn't have that permission, the exception should raise or abort
66de4555cefd wireproto: formalize permissions checking as part of protocol interface Gregory Szorc <gregory.szorc@gmail.com> parents: 36642 diff changeset	249 in a protocol specific manner.
66de4555cefd wireproto: formalize permissions checking as part of protocol interface Gregory Szorc <gregory.szorc@gmail.com> parents: 36642 diff changeset	250 """
37781 352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	251
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	252
49037 642e31cb55f0 py3: use class X: instead of class X(object): Gregory Szorc <gregory.szorc@gmail.com> parents: 48966 diff changeset	253 class commandentry:
37781 352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	254 """Represents a declared wire protocol command."""
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	255
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	256 def __init__(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	257 self,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	258 func,
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	259 args=b'',
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	260 transports=None,
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	261 permission=b'push',
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	262 cachekeyfn=None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	263 extracapabilitiesfn=None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	264 ):
37781 352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	265 self.func = func
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	266 self.args = args
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	267 self.transports = transports or set()
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	268 self.permission = permission
40022 c537144fdbef wireprotov2: support response caching Gregory Szorc <gregory.szorc@gmail.com> parents: 40021 diff changeset	269 self.cachekeyfn = cachekeyfn
40172 30f70d11c224 wireprotov2: advertise recommended batch size for requests Gregory Szorc <gregory.szorc@gmail.com> parents: 40075 diff changeset	270 self.extracapabilitiesfn = extracapabilitiesfn
37781 352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	271
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	272 def _merge(self, func, args):
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	273 """Merge this instance with an incoming 2-tuple.
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	274
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	275 This is called when a caller using the old 2-tuple API attempts
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	276 to replace an instance. The incoming values are merged with
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	277 data not captured by the 2-tuple and a new instance containing
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	278 the union of the two objects is returned.
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	279 """
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	280 return commandentry(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	281 func,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	282 args=args,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	283 transports=set(self.transports),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	284 permission=self.permission,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	285 )
37781 352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	286
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	287 # Old code treats instances as 2-tuples. So expose that interface.
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	288 def __iter__(self):
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	289 yield self.func
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	290 yield self.args
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	291
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	292 def __getitem__(self, i):
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	293 if i == 0:
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	294 return self.func
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	295 elif i == 1:
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	296 return self.args
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	297 else:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	298 raise IndexError(b'can only access elements 0 and 1')
37781 352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	299
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	300
37781 352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	301 class commanddict(dict):
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	302 """Container for registered wire protocol commands.
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	303
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	304 It behaves like a dict. But __setitem__ is overwritten to allow silent
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	305 coercion of values from 2-tuples for API compatibility.
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	306 """
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	307
37781 352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	308 def __setitem__(self, k, v):
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	309 if isinstance(v, commandentry):
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	310 pass
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	311 # Cast 2-tuples to commandentry instances.
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	312 elif isinstance(v, tuple):
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	313 if len(v) != 2:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	314 raise ValueError(b'command tuples must have exactly 2 elements')
37781 352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	315
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	316 # It is common for extensions to wrap wire protocol commands via
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	317 # e.g. ``wireproto.commands[x] = (newfn, args)``. Because callers
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	318 # doing this aren't aware of the new API that uses objects to store
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	319 # command entries, we automatically merge old state with new.
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	320 if k in self:
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	321 v = self[k]._merge(v[0], v[1])
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	322 else:
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	323 # Use default values from @wireprotocommand.
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	324 v = commandentry(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	325 v[0],
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	326 args=v[1],
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	327 transports=set(TRANSPORTS),
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	328 permission=b'push',
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	329 )
37781 352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	330 else:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	331 raise ValueError(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	332 b'command entries must be commandentry instances '
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	333 b'or 2-tuples'
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	334 )
37781 352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	335
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	336 return super(commanddict, self).__setitem__(k, v)
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	337
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	338 def commandavailable(self, command, proto):
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	339 """Determine if a command is available for the requested protocol."""
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	340 assert proto.name in TRANSPORTS
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	341
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	342 entry = self.get(command)
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	343
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	344 if not entry:
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	345 return False
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	346
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	347 if proto.name not in entry.transports:
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	348 return False
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	349
352932a11905 wireproto: move command registration types to wireprototypes Gregory Szorc <gregory.szorc@gmail.com> parents: 37728 diff changeset	350 return True
37783 9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	351
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	352
37783 9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	353 def supportedcompengines(ui, role):
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	354 """Obtain the list of supported compression engines for a request."""
42041 3e47d1ec9da5 util: extract compression code in `mercurial.utils.compression` Pierre-Yves David <pierre-yves.david@octobus.net> parents: 40542 diff changeset	355 assert role in (compression.CLIENTROLE, compression.SERVERROLE)
37783 9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	356
42041 3e47d1ec9da5 util: extract compression code in `mercurial.utils.compression` Pierre-Yves David <pierre-yves.david@octobus.net> parents: 40542 diff changeset	357 compengines = compression.compengines.supportedwireengines(role)
37783 9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	358
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	359 # Allow config to override default list and ordering.
42041 3e47d1ec9da5 util: extract compression code in `mercurial.utils.compression` Pierre-Yves David <pierre-yves.david@octobus.net> parents: 40542 diff changeset	360 if role == compression.SERVERROLE:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	361 configengines = ui.configlist(b'server', b'compressionengines')
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	362 config = b'server.compressionengines'
37783 9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	363 else:
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	364 # This is currently implemented mainly to facilitate testing. In most
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	365 # cases, the server should be in charge of choosing a compression engine
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	366 # because a server has the most to lose from a sub-optimal choice. (e.g.
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	367 # CPU DoS due to an expensive engine or a network DoS due to poor
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	368 # compression ratio).
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	369 configengines = ui.configlist(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	370 b'experimental', b'clientcompressionengines'
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	371 )
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	372 config = b'experimental.clientcompressionengines'
37783 9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	373
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	374 # No explicit config. Filter out the ones that aren't supposed to be
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	375 # advertised and return default ordering.
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	376 if not configengines:
50936 93b0de7f13ca compression: use sysstr to specify attribute to fetch for priority Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49037 diff changeset	377 attr = 'serverpriority' if role == util.SERVERROLE else 'clientpriority'
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	378 return [
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	379 e for e in compengines if getattr(e.wireprotosupport(), attr) > 0
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	380 ]
37783 9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	381
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	382 # If compression engines are listed in the config, assume there is a good
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	383 # reason for it (like server operators wanting to achieve specific
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	384 # performance characteristics). So fail fast if the config references
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	385 # unusable compression engines.
44470 9d2b2df2c2ba cleanup: run pyupgrade on our source tree to clean up varying things Augie Fackler <augie@google.com> parents: 43529 diff changeset	386 validnames = {e.name() for e in compengines}
9d2b2df2c2ba cleanup: run pyupgrade on our source tree to clean up varying things Augie Fackler <augie@google.com> parents: 43529 diff changeset	387 invalidnames = {e for e in configengines if e not in validnames}
37783 9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	388 if invalidnames:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	389 raise error.Abort(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	390 _(b'invalid compression engine defined in %s: %s')
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	391 % (config, b', '.join(sorted(invalidnames)))
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	392 )
37783 9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	393
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	394 compengines = [e for e in compengines if e.name() in configengines]
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	395 compengines = sorted(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	396 compengines, key=lambda e: configengines.index(e.name())
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	397 )
37783 9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	398
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	399 if not compengines:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	400 raise error.Abort(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	401 _(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	402 b'%s config option does not specify any known '
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	403 b'compression engines'
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	404 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	405 % config,
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	406 hint=_(b'usable compression engines: %s')
43529 f5fcf7123a92 wireprototypes: disable pytype where it's just confused Augie Fackler <augie@google.com> parents: 43089 diff changeset	407 % b', '.sorted(validnames), # pytype: disable=attribute-error
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	408 )
37783 9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	409
9d818539abfa wireproto: move supportedcompengines out of wireproto Gregory Szorc <gregory.szorc@gmail.com> parents: 37781 diff changeset	410 return compengines
40021 ed919b90acda wireprotov2: define type to represent pre-encoded object Gregory Szorc <gregory.szorc@gmail.com> parents: 39819 diff changeset	411
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	412
40021 ed919b90acda wireprotov2: define type to represent pre-encoded object Gregory Szorc <gregory.szorc@gmail.com> parents: 39819 diff changeset	413 @attr.s
49037 642e31cb55f0 py3: use class X: instead of class X(object): Gregory Szorc <gregory.szorc@gmail.com> parents: 48966 diff changeset	414 class encodedresponse:
40021 ed919b90acda wireprotov2: define type to represent pre-encoded object Gregory Szorc <gregory.szorc@gmail.com> parents: 39819 diff changeset	415 """Represents response data that is already content encoded.
ed919b90acda wireprotov2: define type to represent pre-encoded object Gregory Szorc <gregory.szorc@gmail.com> parents: 39819 diff changeset	416
ed919b90acda wireprotov2: define type to represent pre-encoded object Gregory Szorc <gregory.szorc@gmail.com> parents: 39819 diff changeset	417 Wire protocol version 2 only.
ed919b90acda wireprotov2: define type to represent pre-encoded object Gregory Szorc <gregory.szorc@gmail.com> parents: 39819 diff changeset	418
ed919b90acda wireprotov2: define type to represent pre-encoded object Gregory Szorc <gregory.szorc@gmail.com> parents: 39819 diff changeset	419 Commands typically emit Python objects that are encoded and sent over the
ed919b90acda wireprotov2: define type to represent pre-encoded object Gregory Szorc <gregory.szorc@gmail.com> parents: 39819 diff changeset	420 wire. If commands emit an object of this type, the encoding step is bypassed
ed919b90acda wireprotov2: define type to represent pre-encoded object Gregory Szorc <gregory.szorc@gmail.com> parents: 39819 diff changeset	421 and the content from this object is used instead.
ed919b90acda wireprotov2: define type to represent pre-encoded object Gregory Szorc <gregory.szorc@gmail.com> parents: 39819 diff changeset	422 """
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	423
40021 ed919b90acda wireprotov2: define type to represent pre-encoded object Gregory Szorc <gregory.szorc@gmail.com> parents: 39819 diff changeset	424 data = attr.ib()
40026 b099e6032f38 wireprotov2: server support for sending content redirects Gregory Szorc <gregory.szorc@gmail.com> parents: 40022 diff changeset	425
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	426
40026 b099e6032f38 wireprotov2: server support for sending content redirects Gregory Szorc <gregory.szorc@gmail.com> parents: 40022 diff changeset	427 @attr.s
49037 642e31cb55f0 py3: use class X: instead of class X(object): Gregory Szorc <gregory.szorc@gmail.com> parents: 48966 diff changeset	428 class alternatelocationresponse:
40026 b099e6032f38 wireprotov2: server support for sending content redirects Gregory Szorc <gregory.szorc@gmail.com> parents: 40022 diff changeset	429 """Represents a response available at an alternate location.
b099e6032f38 wireprotov2: server support for sending content redirects Gregory Szorc <gregory.szorc@gmail.com> parents: 40022 diff changeset	430
b099e6032f38 wireprotov2: server support for sending content redirects Gregory Szorc <gregory.szorc@gmail.com> parents: 40022 diff changeset	431 Instances are sent in place of actual response objects when the server
b099e6032f38 wireprotov2: server support for sending content redirects Gregory Szorc <gregory.szorc@gmail.com> parents: 40022 diff changeset	432 is sending a "content redirect" response.
b099e6032f38 wireprotov2: server support for sending content redirects Gregory Szorc <gregory.szorc@gmail.com> parents: 40022 diff changeset	433
b099e6032f38 wireprotov2: server support for sending content redirects Gregory Szorc <gregory.szorc@gmail.com> parents: 40022 diff changeset	434 Only compatible with wire protocol version 2.
b099e6032f38 wireprotov2: server support for sending content redirects Gregory Szorc <gregory.szorc@gmail.com> parents: 40022 diff changeset	435 """
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	436
40026 b099e6032f38 wireprotov2: server support for sending content redirects Gregory Szorc <gregory.szorc@gmail.com> parents: 40022 diff changeset	437 url = attr.ib()
b099e6032f38 wireprotov2: server support for sending content redirects Gregory Szorc <gregory.szorc@gmail.com> parents: 40022 diff changeset	438 mediatype = attr.ib()
b099e6032f38 wireprotov2: server support for sending content redirects Gregory Szorc <gregory.szorc@gmail.com> parents: 40022 diff changeset	439 size = attr.ib(default=None)
b099e6032f38 wireprotov2: server support for sending content redirects Gregory Szorc <gregory.szorc@gmail.com> parents: 40022 diff changeset	440 fullhashes = attr.ib(default=None)
b099e6032f38 wireprotov2: server support for sending content redirects Gregory Szorc <gregory.szorc@gmail.com> parents: 40022 diff changeset	441 fullhashseed = attr.ib(default=None)
b099e6032f38 wireprotov2: server support for sending content redirects Gregory Szorc <gregory.szorc@gmail.com> parents: 40022 diff changeset	442 serverdercerts = attr.ib(default=None)
b099e6032f38 wireprotov2: server support for sending content redirects Gregory Szorc <gregory.szorc@gmail.com> parents: 40022 diff changeset	443 servercadercerts = attr.ib(default=None)
40328 2c55716f8a1c wireprotov2: add response type that serializes to indefinite length bytestring Gregory Szorc <gregory.szorc@gmail.com> parents: 40176 diff changeset	444
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	445
40328 2c55716f8a1c wireprotov2: add response type that serializes to indefinite length bytestring Gregory Szorc <gregory.szorc@gmail.com> parents: 40176 diff changeset	446 @attr.s
49037 642e31cb55f0 py3: use class X: instead of class X(object): Gregory Szorc <gregory.szorc@gmail.com> parents: 48966 diff changeset	447 class indefinitebytestringresponse:
40328 2c55716f8a1c wireprotov2: add response type that serializes to indefinite length bytestring Gregory Szorc <gregory.szorc@gmail.com> parents: 40176 diff changeset	448 """Represents an object to be encoded to an indefinite length bytestring.
2c55716f8a1c wireprotov2: add response type that serializes to indefinite length bytestring Gregory Szorc <gregory.szorc@gmail.com> parents: 40176 diff changeset	449
2c55716f8a1c wireprotov2: add response type that serializes to indefinite length bytestring Gregory Szorc <gregory.szorc@gmail.com> parents: 40176 diff changeset	450 Instances are initialized from an iterable of chunks, with each chunk being
2c55716f8a1c wireprotov2: add response type that serializes to indefinite length bytestring Gregory Szorc <gregory.szorc@gmail.com> parents: 40176 diff changeset	451 a bytes instance.
2c55716f8a1c wireprotov2: add response type that serializes to indefinite length bytestring Gregory Szorc <gregory.szorc@gmail.com> parents: 40176 diff changeset	452 """
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42958 diff changeset	453
40328 2c55716f8a1c wireprotov2: add response type that serializes to indefinite length bytestring Gregory Szorc <gregory.szorc@gmail.com> parents: 40176 diff changeset	454 chunks = attr.ib()

Mercurial > public > mercurial-scm > hg-stable

annotate mercurial/wireprototypes.py @ 51788:278af66e6595