|
|
# narrowrepo.py - repository which supports narrow revlogs, lazy loading
|
|
|
#
|
|
|
# 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 import (
|
|
|
changegroup,
|
|
|
hg,
|
|
|
narrowspec,
|
|
|
scmutil,
|
|
|
)
|
|
|
|
|
|
from . import (
|
|
|
narrowdirstate,
|
|
|
narrowrevlog,
|
|
|
)
|
|
|
|
|
|
def wrappostshare(orig, sourcerepo, destrepo, **kwargs):
|
|
|
orig(sourcerepo, destrepo, **kwargs)
|
|
|
if changegroup.NARROW_REQUIREMENT in sourcerepo.requirements:
|
|
|
with destrepo.wlock():
|
|
|
with destrepo.vfs('shared', 'a') as fp:
|
|
|
fp.write(narrowspec.FILENAME + '\n')
|
|
|
|
|
|
def unsharenarrowspec(orig, ui, repo, repopath):
|
|
|
if (changegroup.NARROW_REQUIREMENT in repo.requirements
|
|
|
and repo.path == repopath and repo.shared()):
|
|
|
srcrepo = hg.sharedreposource(repo)
|
|
|
with srcrepo.vfs(narrowspec.FILENAME) as f:
|
|
|
spec = f.read()
|
|
|
with repo.vfs(narrowspec.FILENAME, 'w') as f:
|
|
|
f.write(spec)
|
|
|
return orig(ui, repo, repopath)
|
|
|
|
|
|
def wraprepo(repo):
|
|
|
"""Enables narrow clone functionality on a single local repository."""
|
|
|
|
|
|
class narrowrepository(repo.__class__):
|
|
|
|
|
|
def file(self, f):
|
|
|
fl = super(narrowrepository, self).file(f)
|
|
|
narrowrevlog.makenarrowfilelog(fl, self.narrowmatch())
|
|
|
return fl
|
|
|
|
|
|
# I'm not sure this is the right place to do this filter.
|
|
|
# context._manifestmatches() would probably be better, or perhaps
|
|
|
# move it to a later place, in case some of the callers do want to know
|
|
|
# which directories changed. This seems to work for now, though.
|
|
|
def status(self, *args, **kwargs):
|
|
|
s = super(narrowrepository, self).status(*args, **kwargs)
|
|
|
narrowmatch = self.narrowmatch()
|
|
|
modified = list(filter(narrowmatch, s.modified))
|
|
|
added = list(filter(narrowmatch, s.added))
|
|
|
removed = list(filter(narrowmatch, s.removed))
|
|
|
deleted = list(filter(narrowmatch, s.deleted))
|
|
|
unknown = list(filter(narrowmatch, s.unknown))
|
|
|
ignored = list(filter(narrowmatch, s.ignored))
|
|
|
clean = list(filter(narrowmatch, s.clean))
|
|
|
return scmutil.status(modified, added, removed, deleted, unknown,
|
|
|
ignored, clean)
|
|
|
|
|
|
def _makedirstate(self):
|
|
|
dirstate = super(narrowrepository, self)._makedirstate()
|
|
|
return narrowdirstate.wrapdirstate(self, dirstate)
|
|
|
|
|
|
repo.__class__ = narrowrepository
|
|
|
|
