narrowdirstate.py
76 lines
| 2.5 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 _ | ||||
|
r43346 | from mercurial import error | ||
|
r36096 | |||
|
r38142 | def wrapdirstate(repo, dirstate): | ||
|
r36096 | """Add narrow spec dirstate ignore, block changes outside narrow spec.""" | ||
|
r38142 | def _editfunc(fn): | ||
|
r42656 | def _wrapper(self, *args, **kwargs): | ||
|
r36096 | narrowmatch = repo.narrowmatch() | ||
for f in args: | ||||
|
r39996 | if f is not None and not narrowmatch(f) and f not in self: | ||
|
r43346 | raise error.Abort( | ||
_( | ||||
|
r43347 | b"cannot track '%s' - it is outside " | ||
+ b"the narrow clone" | ||||
|
r43346 | ) | ||
% f | ||||
) | ||||
|
r42656 | return fn(self, *args, **kwargs) | ||
|
r43346 | |||
|
r38142 | return _wrapper | ||
|
r36096 | |||
|
r38142 | class narrowdirstate(dirstate.__class__): | ||
# Prevent adding/editing/copying/deleting files that are outside the | ||||
# sparse checkout | ||||
@_editfunc | ||||
|
r42656 | def normal(self, *args, **kwargs): | ||
return super(narrowdirstate, self).normal(*args, **kwargs) | ||||
|
r36096 | |||
|
r38142 | @_editfunc | ||
r49211 | def set_tracked(self, *args, **kwargs): | |||
return super(narrowdirstate, self).set_tracked(*args, **kwargs) | ||||
r48393 | ||||
@_editfunc | ||||
r48399 | def set_untracked(self, *args): | |||
return super(narrowdirstate, self).set_untracked(*args) | ||||
@_editfunc | ||||
|
r38142 | 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 | ||||