15 narrowspec, |
15 narrowspec, |
16 node, |
16 node, |
17 wireprotov1server, |
17 wireprotov1server, |
18 ) |
18 ) |
19 |
19 |
20 from . import narrowbundle2 |
20 NARROWCAP = 'narrow' |
|
21 ELLIPSESCAP = 'ellipses' |
21 |
22 |
22 def uisetup(): |
23 def uisetup(): |
23 def peersetup(ui, peer): |
24 def peersetup(ui, peer): |
24 # We must set up the expansion before reposetup below, since it's used |
25 # We must set up the expansion before reposetup below, since it's used |
25 # at clone time before we have a repo. |
26 # at clone time before we have a repo. |
44 extensions.wrapfunction(wireprotov1server, '_capabilities', addnarrowcap) |
45 extensions.wrapfunction(wireprotov1server, '_capabilities', addnarrowcap) |
45 |
46 |
46 def addnarrowcap(orig, repo, proto): |
47 def addnarrowcap(orig, repo, proto): |
47 """add the narrow capability to the server""" |
48 """add the narrow capability to the server""" |
48 caps = orig(repo, proto) |
49 caps = orig(repo, proto) |
49 caps.append(narrowbundle2.NARROWCAP) |
50 caps.append(NARROWCAP) |
50 if repo.ui.configbool('experimental', 'narrowservebrokenellipses'): |
51 if repo.ui.configbool('experimental', 'narrowservebrokenellipses'): |
51 caps.append(narrowbundle2.ELLIPSESCAP) |
52 caps.append(ELLIPSESCAP) |
52 return caps |
53 return caps |
53 |
54 |
54 def reposetup(repo): |
55 def reposetup(repo): |
55 def wirereposetup(ui, peer): |
56 def wirereposetup(ui, peer): |
56 def wrapped(orig, cmd, *args, **kwargs): |
57 def wrapped(orig, cmd, *args, **kwargs): |