equal
deleted
inserted
replaced
6 # GNU General Public License version 2 or any later version. |
6 # GNU General Public License version 2 or any later version. |
7 |
7 |
8 from __future__ import absolute_import |
8 from __future__ import absolute_import |
9 |
9 |
10 from . import ( |
10 from . import ( |
11 narrowbundle2, |
|
12 narrowdirstate, |
11 narrowdirstate, |
13 narrowrevlog, |
12 narrowrevlog, |
|
13 narrowwirepeer, |
14 ) |
14 ) |
15 |
15 |
16 def wraprepo(repo): |
16 def wraprepo(repo): |
17 """Enables narrow clone functionality on a single local repository.""" |
17 """Enables narrow clone functionality on a single local repository.""" |
18 |
18 |
27 dirstate = super(narrowrepository, self)._makedirstate() |
27 dirstate = super(narrowrepository, self)._makedirstate() |
28 return narrowdirstate.wrapdirstate(self, dirstate) |
28 return narrowdirstate.wrapdirstate(self, dirstate) |
29 |
29 |
30 def peer(self): |
30 def peer(self): |
31 peer = super(narrowrepository, self).peer() |
31 peer = super(narrowrepository, self).peer() |
32 peer._caps.add(narrowbundle2.NARROWCAP) |
32 peer._caps.add(narrowwirepeer.NARROWCAP) |
33 peer._caps.add(narrowbundle2.ELLIPSESCAP) |
33 peer._caps.add(narrowwirepeer.ELLIPSESCAP) |
34 return peer |
34 return peer |
35 |
35 |
36 repo.__class__ = narrowrepository |
36 repo.__class__ = narrowrepository |