|
|
# -*- coding: UTF-8 -*-
|
|
|
# beautifygraph.py - improve graph output by using Unicode characters
|
|
|
#
|
|
|
# Copyright 2018 John Stiles <johnstiles@gmail.com>
|
|
|
#
|
|
|
# This software may be used and distributed according to the terms of the
|
|
|
# GNU General Public License version 2 or any later version.
|
|
|
|
|
|
'''beautify log -G output by using Unicode characters (EXPERIMENTAL)
|
|
|
|
|
|
A terminal with UTF-8 support and monospace narrow text are required.
|
|
|
'''
|
|
|
|
|
|
from __future__ import absolute_import
|
|
|
|
|
|
from mercurial.i18n import _
|
|
|
from mercurial import (
|
|
|
encoding,
|
|
|
extensions,
|
|
|
graphmod,
|
|
|
pycompat,
|
|
|
templatekw,
|
|
|
)
|
|
|
|
|
|
# Note for extension authors: ONLY specify testedwith = 'ships-with-hg-core' for
|
|
|
# extensions which SHIP WITH MERCURIAL. Non-mainline extensions should
|
|
|
# be specifying the version(s) of Mercurial they are tested with, or
|
|
|
# leave the attribute unspecified.
|
|
|
testedwith = 'ships-with-hg-core'
|
|
|
|
|
|
def prettyedge(before, edge, after):
|
|
|
if edge == '~':
|
|
|
return '\xE2\x95\xA7' # U+2567 ╧
|
|
|
if edge == 'X':
|
|
|
return '\xE2\x95\xB3' # U+2573 ╳
|
|
|
if edge == '/':
|
|
|
return '\xE2\x95\xB1' # U+2571 ╱
|
|
|
if edge == '-':
|
|
|
return '\xE2\x94\x80' # U+2500 ─
|
|
|
if edge == '|':
|
|
|
return '\xE2\x94\x82' # U+2502 │
|
|
|
if edge == ':':
|
|
|
return '\xE2\x94\x86' # U+2506 ┆
|
|
|
if edge == '\\':
|
|
|
return '\xE2\x95\xB2' # U+2572 ╲
|
|
|
if edge == '+':
|
|
|
if before == ' ' and not after == ' ':
|
|
|
return '\xE2\x94\x9C' # U+251C ├
|
|
|
if after == ' ' and not before == ' ':
|
|
|
return '\xE2\x94\xA4' # U+2524 ┤
|
|
|
return '\xE2\x94\xBC' # U+253C ┼
|
|
|
return edge
|
|
|
|
|
|
def convertedges(line):
|
|
|
line = ' %s ' % line
|
|
|
pretty = []
|
|
|
for idx in pycompat.xrange(len(line) - 2):
|
|
|
pretty.append(prettyedge(line[idx], line[idx + 1], line[idx + 2]))
|
|
|
return ''.join(pretty)
|
|
|
|
|
|
def getprettygraphnode(orig, *args, **kwargs):
|
|
|
node = orig(*args, **kwargs)
|
|
|
if node == 'o':
|
|
|
return '\xE2\x97\x8B' # U+25CB ○
|
|
|
if node == '@':
|
|
|
return '\xE2\x97\x8D' # U+25CD ◍
|
|
|
if node == '*':
|
|
|
return '\xE2\x88\x97' # U+2217 ∗
|
|
|
if node == 'x':
|
|
|
return '\xE2\x97\x8C' # U+25CC ◌
|
|
|
if node == '_':
|
|
|
return '\xE2\x95\xA4' # U+2564 ╤
|
|
|
return node
|
|
|
|
|
|
def outputprettygraph(orig, ui, graph, *args, **kwargs):
|
|
|
(edges, text) = zip(*graph)
|
|
|
graph = zip([convertedges(e) for e in edges], text)
|
|
|
return orig(ui, graph, *args, **kwargs)
|
|
|
|
|
|
def extsetup(ui):
|
|
|
if encoding.encoding != 'UTF-8':
|
|
|
ui.warn(_('beautifygraph: unsupported encoding, UTF-8 required\n'))
|
|
|
return
|
|
|
|
|
|
if 'A' in encoding._wide:
|
|
|
ui.warn(_('beautifygraph: unsupported terminal settings, '
|
|
|
'monospace narrow text required\n'))
|
|
|
return
|
|
|
|
|
|
if ui.plain('graph'):
|
|
|
return
|
|
|
|
|
|
extensions.wrapfunction(graphmod, 'outputgraph', outputprettygraph)
|
|
|
extensions.wrapfunction(templatekw, 'getgraphnode', getprettygraphnode)
|
|
|
|
