9 

     9 

    10 import itertools

    10 import itertools

    11 import os

    11 import os

    12 import sys

    12 import sys

    13 import tempfile

    13 import tempfile

    15 

    14 

    16 from .i18n import _

    15 from .i18n import _

    17 from .node import (

    16 from .node import (

    18     bin,

    17     bin,

    19     hex,

    18     hex,

    28     peer,

    27     peer,

    29     pushkey as pushkeymod,

    28     pushkey as pushkeymod,

    30     streamclone,

    29     streamclone,

    31     util,

    30     util,

    32 )

    31 )

    33 

    35 

    34 bundle2required = _(

    36 bundle2required = _(

    35     'incompatible Mercurial client; bundle2 required\n'

    37     'incompatible Mercurial client; bundle2 required\n'

    36     '(see https://www.mercurial-scm.org/wiki/IncompatibleClient)\n')

    38     '(see https://www.mercurial-scm.org/wiki/IncompatibleClient)\n')

    37 

    39 

   285         d = f.value

   287         d = f.value

   286         try:

   288         try:

   287             branchmap = {}

   289             branchmap = {}

   288             for branchpart in d.splitlines():

   290             for branchpart in d.splitlines():

   289                 branchname, branchheads = branchpart.split(' ', 1)

   291                 branchname, branchheads = branchpart.split(' ', 1)

   291                 branchheads = decodelist(branchheads)

   293                 branchheads = decodelist(branchheads)

   292                 branchmap[branchname] = branchheads

   294                 branchmap[branchname] = branchheads

   293             yield branchmap

   295             yield branchmap

   294         except TypeError:

   296         except TypeError:

   295             self._abort(error.ResponseError(_("unexpected response:"), d))

   297             self._abort(error.ResponseError(_("unexpected response:"), d))

   630 @wireprotocommand('branchmap')

   632 @wireprotocommand('branchmap')

   631 def branchmap(repo, proto):

   633 def branchmap(repo, proto):

   632     branchmap = repo.branchmap()

   634     branchmap = repo.branchmap()

   633     heads = []

   635     heads = []

   634     for branch, nodes in branchmap.iteritems():

   636     for branch, nodes in branchmap.iteritems():

   636         branchnodes = encodelist(nodes)

   638         branchnodes = encodelist(nodes)

   637         heads.append('%s %s' % (branchname, branchnodes))

   639         heads.append('%s %s' % (branchname, branchnodes))

   638     return '\n'.join(heads)

   640     return '\n'.join(heads)

   639 

   641 

   640 @wireprotocommand('branches', 'nodes')

   642 @wireprotocommand('branches', 'nodes')

   682         # otherwise, add 'streamreqs' detailing our local revlog format

   684         # otherwise, add 'streamreqs' detailing our local revlog format

   683         else:

   685         else:

   684             caps.append('streamreqs=%s' % ','.join(sorted(requiredformats)))

   686             caps.append('streamreqs=%s' % ','.join(sorted(requiredformats)))

   685     if repo.ui.configbool('experimental', 'bundle2-advertise', True):

   687     if repo.ui.configbool('experimental', 'bundle2-advertise', True):

   686         capsblob = bundle2.encodecaps(bundle2.getrepocaps(repo))

   688         capsblob = bundle2.encodecaps(bundle2.getrepocaps(repo))

   688     caps.append('unbundle=%s' % ','.join(bundle2.bundlepriority))

   690     caps.append('unbundle=%s' % ','.join(bundle2.bundlepriority))

   689     caps.append(

   691     caps.append(

   690         'httpheader=%d' % repo.ui.configint('server', 'maxhttpheaderlen', 1024))

   692         'httpheader=%d' % repo.ui.configint('server', 'maxhttpheaderlen', 1024))

   691     if repo.ui.configbool('experimental', 'httppostargs', False):

   693     if repo.ui.configbool('experimental', 'httppostargs', False):

   692         caps.append('httppostargs')

   694         caps.append('httppostargs')