|
|
# Copyright 2016-present Facebook. All Rights Reserved.
|
|
|
#
|
|
|
# support: fastannotate support for hgweb, and filectx
|
|
|
#
|
|
|
# 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 (
|
|
|
context as hgcontext,
|
|
|
dagop,
|
|
|
extensions,
|
|
|
hgweb,
|
|
|
patch,
|
|
|
util,
|
|
|
)
|
|
|
|
|
|
from . import (
|
|
|
context,
|
|
|
revmap,
|
|
|
)
|
|
|
|
|
|
|
|
|
class _lazyfctx(object):
|
|
|
"""delegates to fctx but do not construct fctx when unnecessary"""
|
|
|
|
|
|
def __init__(self, repo, node, path):
|
|
|
self._node = node
|
|
|
self._path = path
|
|
|
self._repo = repo
|
|
|
|
|
|
def node(self):
|
|
|
return self._node
|
|
|
|
|
|
def path(self):
|
|
|
return self._path
|
|
|
|
|
|
@util.propertycache
|
|
|
def _fctx(self):
|
|
|
return context.resolvefctx(self._repo, self._node, self._path)
|
|
|
|
|
|
def __getattr__(self, name):
|
|
|
return getattr(self._fctx, name)
|
|
|
|
|
|
|
|
|
def _convertoutputs(repo, annotated, contents):
|
|
|
"""convert fastannotate outputs to vanilla annotate format"""
|
|
|
# fastannotate returns: [(nodeid, linenum, path)], [linecontent]
|
|
|
# convert to what fctx.annotate returns: [annotateline]
|
|
|
results = []
|
|
|
fctxmap = {}
|
|
|
annotateline = dagop.annotateline
|
|
|
for i, (hsh, linenum, path) in enumerate(annotated):
|
|
|
if (hsh, path) not in fctxmap:
|
|
|
fctxmap[(hsh, path)] = _lazyfctx(repo, hsh, path)
|
|
|
# linenum: the user wants 1-based, we have 0-based.
|
|
|
lineno = linenum + 1
|
|
|
fctx = fctxmap[(hsh, path)]
|
|
|
line = contents[i]
|
|
|
results.append(annotateline(fctx=fctx, lineno=lineno, text=line))
|
|
|
return results
|
|
|
|
|
|
|
|
|
def _getmaster(fctx):
|
|
|
"""(fctx) -> str"""
|
|
|
return fctx._repo.ui.config(b'fastannotate', b'mainbranch') or b'default'
|
|
|
|
|
|
|
|
|
def _doannotate(fctx, follow=True, diffopts=None):
|
|
|
"""like the vanilla fctx.annotate, but do it via fastannotate, and make
|
|
|
the output format compatible with the vanilla fctx.annotate.
|
|
|
may raise Exception, and always return line numbers.
|
|
|
"""
|
|
|
master = _getmaster(fctx)
|
|
|
annotated = contents = None
|
|
|
|
|
|
with context.fctxannotatecontext(fctx, follow, diffopts) as ac:
|
|
|
try:
|
|
|
annotated, contents = ac.annotate(
|
|
|
fctx.rev(), master=master, showpath=True, showlines=True
|
|
|
)
|
|
|
except Exception:
|
|
|
ac.rebuild() # try rebuild once
|
|
|
fctx._repo.ui.debug(
|
|
|
b'fastannotate: %s: rebuilding broken cache\n' % fctx._path
|
|
|
)
|
|
|
try:
|
|
|
annotated, contents = ac.annotate(
|
|
|
fctx.rev(), master=master, showpath=True, showlines=True
|
|
|
)
|
|
|
except Exception:
|
|
|
raise
|
|
|
|
|
|
assert annotated and contents
|
|
|
return _convertoutputs(fctx._repo, annotated, contents)
|
|
|
|
|
|
|
|
|
def _hgwebannotate(orig, fctx, ui):
|
|
|
diffopts = patch.difffeatureopts(
|
|
|
ui, untrusted=True, section=b'annotate', whitespace=True
|
|
|
)
|
|
|
return _doannotate(fctx, diffopts=diffopts)
|
|
|
|
|
|
|
|
|
def _fctxannotate(
|
|
|
orig, self, follow=False, linenumber=False, skiprevs=None, diffopts=None
|
|
|
):
|
|
|
if skiprevs:
|
|
|
# skiprevs is not supported yet
|
|
|
return orig(
|
|
|
self, follow, linenumber, skiprevs=skiprevs, diffopts=diffopts
|
|
|
)
|
|
|
try:
|
|
|
return _doannotate(self, follow, diffopts)
|
|
|
except Exception as ex:
|
|
|
self._repo.ui.debug(
|
|
|
b'fastannotate: falling back to the vanilla ' b'annotate: %r\n' % ex
|
|
|
)
|
|
|
return orig(self, follow=follow, skiprevs=skiprevs, diffopts=diffopts)
|
|
|
|
|
|
|
|
|
def _remotefctxannotate(orig, self, follow=False, skiprevs=None, diffopts=None):
|
|
|
# skipset: a set-like used to test if a fctx needs to be downloaded
|
|
|
with context.fctxannotatecontext(self, follow, diffopts) as ac:
|
|
|
skipset = revmap.revmap(ac.revmappath)
|
|
|
return orig(
|
|
|
self, follow, skiprevs=skiprevs, diffopts=diffopts, prefetchskip=skipset
|
|
|
)
|
|
|
|
|
|
|
|
|
def replacehgwebannotate():
|
|
|
extensions.wrapfunction(hgweb.webutil, b'annotate', _hgwebannotate)
|
|
|
|
|
|
|
|
|
def replacefctxannotate():
|
|
|
extensions.wrapfunction(hgcontext.basefilectx, b'annotate', _fctxannotate)
|
|
|
|