formatter.py
176 lines
| 5.8 KiB
| text/x-python
|
PythonLexer
|
r39243 | # Copyright 2016-present Facebook. All Rights Reserved. | ||
# | ||||
# format: defines the format used to output annotate result | ||||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
# GNU General Public License version 2 or any later version. | ||||
|
r46729 | from mercurial.node import ( | ||
hex, | ||||
short, | ||||
) | ||||
|
r39243 | from mercurial import ( | ||
encoding, | ||||
pycompat, | ||||
templatefilters, | ||||
util, | ||||
) | ||||
|
r43346 | from mercurial.utils import dateutil | ||
|
r39243 | |||
# imitating mercurial.commands.annotate, not using the vanilla formatter since | ||||
# the data structures are a bit different, and we have some fast paths. | ||||
class defaultformatter(object): | ||||
"""the default formatter that does leftpad and support some common flags""" | ||||
def __init__(self, ui, repo, opts): | ||||
self.ui = ui | ||||
self.opts = opts | ||||
if ui.quiet: | ||||
datefunc = dateutil.shortdate | ||||
else: | ||||
datefunc = dateutil.datestr | ||||
datefunc = util.cachefunc(datefunc) | ||||
getctx = util.cachefunc(lambda x: repo[x[0]]) | ||||
hexfunc = self._hexfunc | ||||
# special handling working copy "changeset" and "rev" functions | ||||
|
r43347 | if self.opts.get(b'rev') == b'wdir()': | ||
|
r39243 | orig = hexfunc | ||
hexfunc = lambda x: None if x is None else orig(x) | ||||
|
r43347 | wnode = hexfunc(repo[b'.'].node()) + b'+' | ||
wrev = b'%d' % repo[b'.'].rev() | ||||
wrevpad = b'' | ||||
if not opts.get(b'changeset'): # only show + if changeset is hidden | ||||
wrev += b'+' | ||||
wrevpad = b' ' | ||||
revenc = lambda x: wrev if x is None else (b'%d' % x) + wrevpad | ||||
|
r43346 | |||
|
r40763 | def csetenc(x): | ||
if x is None: | ||||
return wnode | ||||
|
r43347 | return pycompat.bytestr(x) + b' ' | ||
|
r43346 | |||
|
r39243 | else: | ||
|
r40763 | revenc = csetenc = pycompat.bytestr | ||
|
r39243 | |||
# opt name, separator, raw value (for json/plain), encoder (for plain) | ||||
|
r43346 | opmap = [ | ||
|
r43347 | (b'user', b' ', lambda x: getctx(x).user(), ui.shortuser), | ||
(b'number', b' ', lambda x: getctx(x).rev(), revenc), | ||||
(b'changeset', b' ', lambda x: hexfunc(x[0]), csetenc), | ||||
(b'date', b' ', lambda x: getctx(x).date(), datefunc), | ||||
(b'file', b' ', lambda x: x[2], pycompat.bytestr), | ||||
(b'line_number', b':', lambda x: x[1] + 1, pycompat.bytestr), | ||||
|
r43346 | ] | ||
|
r43347 | fieldnamemap = {b'number': b'rev', b'changeset': b'node'} | ||
|
r43346 | funcmap = [ | ||
(get, sep, fieldnamemap.get(op, op), enc) | ||||
for op, sep, get, enc in opmap | ||||
if opts.get(op) | ||||
] | ||||
|
r39243 | # no separator for first column | ||
funcmap[0] = list(funcmap[0]) | ||||
|
r43347 | funcmap[0][1] = b'' | ||
|
r39243 | self.funcmap = funcmap | ||
def write(self, annotatedresult, lines=None, existinglines=None): | ||||
"""(annotateresult, [str], set([rev, linenum])) -> None. write output. | ||||
annotateresult can be [(node, linenum, path)], or [(node, linenum)] | ||||
""" | ||||
|
r43346 | pieces = [] # [[str]] | ||
maxwidths = [] # [int] | ||||
|
r39243 | |||
# calculate padding | ||||
for f, sep, name, enc in self.funcmap: | ||||
l = [enc(f(x)) for x in annotatedresult] | ||||
pieces.append(l) | ||||
|
r43347 | if name in [b'node', b'date']: # node and date has fixed size | ||
|
r39243 | l = l[:1] | ||
|
r39762 | widths = pycompat.maplist(encoding.colwidth, set(l)) | ||
|
r43346 | maxwidth = max(widths) if widths else 0 | ||
|
r39243 | maxwidths.append(maxwidth) | ||
# buffered output | ||||
|
r43347 | result = b'' | ||
|
r39243 | for i in pycompat.xrange(len(annotatedresult)): | ||
for j, p in enumerate(pieces): | ||||
sep = self.funcmap[j][1] | ||||
|
r43347 | padding = b' ' * (maxwidths[j] - len(p[i])) | ||
|
r39243 | result += sep + padding + p[i] | ||
if lines: | ||||
if existinglines is None: | ||||
|
r43347 | result += b': ' + lines[i] | ||
|
r43346 | else: # extra formatting showing whether a line exists | ||
|
r39243 | key = (annotatedresult[i][0], annotatedresult[i][1]) | ||
if key in existinglines: | ||||
|
r43347 | result += b': ' + lines[i] | ||
|
r39243 | else: | ||
|
r43347 | result += b': ' + self.ui.label( | ||
b'-' + lines[i], b'diff.deleted' | ||||
|
r43346 | ) | ||
|
r39243 | |||
|
r43347 | if result[-1:] != b'\n': | ||
result += b'\n' | ||||
|
r39243 | |||
self.ui.write(result) | ||||
@util.propertycache | ||||
def _hexfunc(self): | ||||
|
r43347 | if self.ui.debugflag or self.opts.get(b'long_hash'): | ||
|
r46729 | return hex | ||
|
r39243 | else: | ||
|
r46729 | return short | ||
|
r39243 | |||
def end(self): | ||||
pass | ||||
|
r43346 | |||
|
r39243 | class jsonformatter(defaultformatter): | ||
def __init__(self, ui, repo, opts): | ||||
super(jsonformatter, self).__init__(ui, repo, opts) | ||||
|
r43347 | self.ui.write(b'[') | ||
|
r39243 | self.needcomma = False | ||
def write(self, annotatedresult, lines=None, existinglines=None): | ||||
if annotatedresult: | ||||
self._writecomma() | ||||
|
r43346 | pieces = [ | ||
(name, pycompat.maplist(f, annotatedresult)) | ||||
for f, sep, name, enc in self.funcmap | ||||
] | ||||
|
r39243 | if lines is not None: | ||
|
r43347 | pieces.append((b'line', lines)) | ||
|
r39243 | pieces.sort() | ||
|
r43347 | seps = [b','] * len(pieces[:-1]) + [b''] | ||
|
r39243 | |||
|
r43347 | result = b'' | ||
|
r39243 | lasti = len(annotatedresult) - 1 | ||
for i in pycompat.xrange(len(annotatedresult)): | ||||
|
r43347 | result += b'\n {\n' | ||
|
r39243 | for j, p in enumerate(pieces): | ||
k, vs = p | ||||
|
r43347 | result += b' "%s": %s%s\n' % ( | ||
|
r43346 | k, | ||
templatefilters.json(vs[i], paranoid=False), | ||||
seps[j], | ||||
) | ||||
|
r43347 | result += b' }%s' % (b'' if i == lasti else b',') | ||
|
r39243 | if lasti >= 0: | ||
self.needcomma = True | ||||
self.ui.write(result) | ||||
def _writecomma(self): | ||||
if self.needcomma: | ||||
|
r43347 | self.ui.write(b',') | ||
|
r39243 | self.needcomma = False | ||
@util.propertycache | ||||
def _hexfunc(self): | ||||
|
r46729 | return hex | ||
|
r39243 | |||
def end(self): | ||||
|
r43347 | self.ui.write(b'\n]\n') | ||