narrowdirstate.py
75 lines
| 2.7 KiB
| text/x-python
|
PythonLexer
|
r36096 | # narrowdirstate.py - extensions to mercurial dirstate to support narrow clones | ||
# | ||||
# 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, | ||||
match as matchmod, | ||||
) | ||||
|
r38142 | def wrapdirstate(repo, dirstate): | ||
|
r36096 | """Add narrow spec dirstate ignore, block changes outside narrow spec.""" | ||
|
r38142 | def _editfunc(fn): | ||
def _wrapper(self, *args): | ||||
|
r36096 | narrowmatch = repo.narrowmatch() | ||
for f in args: | ||||
|
r39996 | if f is not None and not narrowmatch(f) and f not in self: | ||
|
r36096 | raise error.Abort(_("cannot track '%s' - it is outside " + | ||
"the narrow clone") % f) | ||||
|
r38142 | return fn(self, *args) | ||
return _wrapper | ||||
|
r36096 | |||
|
r38142 | class narrowdirstate(dirstate.__class__): | ||
def walk(self, match, subrepos, unknown, ignored, full=True, | ||||
narrowonly=True): | ||||
if narrowonly: | ||||
# hack to not exclude explicitly-specified paths so that they | ||||
# can be warned later on e.g. dirstate.add() | ||||
em = matchmod.exact(match._root, match._cwd, match.files()) | ||||
nm = matchmod.unionmatcher([repo.narrowmatch(), em]) | ||||
match = matchmod.intersectmatchers(match, nm) | ||||
return super(narrowdirstate, self).walk(match, subrepos, unknown, | ||||
ignored, full) | ||||
|
r36096 | |||
|
r38142 | # Prevent adding/editing/copying/deleting files that are outside the | ||
# sparse checkout | ||||
@_editfunc | ||||
def normal(self, *args): | ||||
return super(narrowdirstate, self).normal(*args) | ||||
|
r36096 | |||
|
r38142 | @_editfunc | ||
def add(self, *args): | ||||
return super(narrowdirstate, self).add(*args) | ||||
@_editfunc | ||||
def normallookup(self, *args): | ||||
return super(narrowdirstate, self).normallookup(*args) | ||||
@_editfunc | ||||
def copy(self, *args): | ||||
return super(narrowdirstate, self).copy(*args) | ||||
|
r36096 | |||
|
r38142 | @_editfunc | ||
def remove(self, *args): | ||||
return super(narrowdirstate, self).remove(*args) | ||||
@_editfunc | ||||
def merge(self, *args): | ||||
return super(narrowdirstate, self).merge(*args) | ||||
def rebuild(self, parent, allfiles, changedfiles=None): | ||||
if changedfiles is None: | ||||
# Rebuilding entire dirstate, let's filter allfiles to match the | ||||
# narrowspec. | ||||
allfiles = [f for f in allfiles if repo.narrowmatch()(f)] | ||||
super(narrowdirstate, self).rebuild(parent, allfiles, changedfiles) | ||||
|
r36096 | |||
|
r38142 | dirstate.__class__ = narrowdirstate | ||
return dirstate | ||||