narrowwirepeer.py
66 lines
| 2.3 KiB
| text/x-python
|
PythonLexer
|
r36096 | # narrowwirepeer.py - passes narrow spec with unbundle command | ||
# | ||||
# Copyright 2017 Google, Inc. | ||||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
# GNU General Public License version 2 or any later version. | ||||
from __future__ import absolute_import | ||||
from mercurial.i18n import _ | ||||
from mercurial import ( | ||||
error, | ||||
extensions, | ||||
hg, | ||||
|
r36178 | narrowspec, | ||
|
r36096 | node, | ||
|
r39559 | wireprotov1server, | ||
|
r36096 | ) | ||
|
r39566 | NARROWCAP = 'exp-narrow-1' | ||
ELLIPSESCAP = 'exp-ellipses-1' | ||||
|
r39559 | |||
|
r36096 | def uisetup(): | ||
def peersetup(ui, peer): | ||||
# We must set up the expansion before reposetup below, since it's used | ||||
# at clone time before we have a repo. | ||||
class expandingpeer(peer.__class__): | ||||
def expandnarrow(self, narrow_include, narrow_exclude, nodes): | ||||
ui.status(_("expanding narrowspec\n")) | ||||
|
r36118 | if not self.capable('exp-expandnarrow'): | ||
|
r36096 | raise error.Abort( | ||
'peer does not support expanding narrowspecs') | ||||
hex_nodes = (node.hex(n) for n in nodes) | ||||
new_narrowspec = self._call( | ||||
'expandnarrow', | ||||
includepats=','.join(narrow_include), | ||||
excludepats=','.join(narrow_exclude), | ||||
nodes=','.join(hex_nodes)) | ||||
return narrowspec.parseserverpatterns(new_narrowspec) | ||||
peer.__class__ = expandingpeer | ||||
hg.wirepeersetupfuncs.append(peersetup) | ||||
|
r39559 | extensions.wrapfunction(wireprotov1server, '_capabilities', addnarrowcap) | ||
def addnarrowcap(orig, repo, proto): | ||||
"""add the narrow capability to the server""" | ||||
caps = orig(repo, proto) | ||||
|
r39565 | caps.append(NARROWCAP) | ||
|
r39559 | if repo.ui.configbool('experimental', 'narrowservebrokenellipses'): | ||
|
r39565 | caps.append(ELLIPSESCAP) | ||
|
r39559 | return caps | ||
|
r36096 | def reposetup(repo): | ||
def wirereposetup(ui, peer): | ||||
def wrapped(orig, cmd, *args, **kwargs): | ||||
if cmd == 'unbundle': | ||||
|
r36119 | # TODO: don't blindly add include/exclude wireproto | ||
# arguments to unbundle. | ||||
|
r36096 | include, exclude = repo.narrowpats | ||
|
r36369 | kwargs[r"includepats"] = ','.join(include) | ||
kwargs[r"excludepats"] = ','.join(exclude) | ||||
|
r36096 | return orig(cmd, *args, **kwargs) | ||
extensions.wrapfunction(peer, '_calltwowaystream', wrapped) | ||||
hg.wirepeersetupfuncs.append(wirereposetup) | ||||