color.py
332 lines
| 11.7 KiB
| text/x-python
|
PythonLexer
/ mercurial / color.py
Pierre-Yves David
|
r30652 | # utility for color output for Mercurial commands | ||
# | ||||
# Copyright (C) 2007 Kevin Christen <kevin.christen@gmail.com> and other | ||||
# | ||||
# 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 | ||||
Pierre-Yves David
|
r30971 | from .i18n import _ | ||
Pierre-Yves David
|
r31067 | from . import pycompat | ||
Pierre-Yves David
|
r30968 | try: | ||
import curses | ||||
# Mapping from effect name to terminfo attribute name (or raw code) or | ||||
# color number. This will also force-load the curses module. | ||||
_terminfo_params = {'none': (True, 'sgr0', ''), | ||||
'standout': (True, 'smso', ''), | ||||
'underline': (True, 'smul', ''), | ||||
'reverse': (True, 'rev', ''), | ||||
'inverse': (True, 'rev', ''), | ||||
'blink': (True, 'blink', ''), | ||||
'dim': (True, 'dim', ''), | ||||
'bold': (True, 'bold', ''), | ||||
'invisible': (True, 'invis', ''), | ||||
'italic': (True, 'sitm', ''), | ||||
'black': (False, curses.COLOR_BLACK, ''), | ||||
'red': (False, curses.COLOR_RED, ''), | ||||
'green': (False, curses.COLOR_GREEN, ''), | ||||
'yellow': (False, curses.COLOR_YELLOW, ''), | ||||
'blue': (False, curses.COLOR_BLUE, ''), | ||||
'magenta': (False, curses.COLOR_MAGENTA, ''), | ||||
'cyan': (False, curses.COLOR_CYAN, ''), | ||||
'white': (False, curses.COLOR_WHITE, '')} | ||||
except ImportError: | ||||
curses = None | ||||
_terminfo_params = {} | ||||
Pierre-Yves David
|
r30967 | # start and stop parameters for effects | ||
_effects = {'none': 0, | ||||
'black': 30, | ||||
'red': 31, | ||||
'green': 32, | ||||
'yellow': 33, | ||||
'blue': 34, | ||||
'magenta': 35, | ||||
'cyan': 36, | ||||
'white': 37, | ||||
'bold': 1, | ||||
'italic': 3, | ||||
'underline': 4, | ||||
'inverse': 7, | ||||
'dim': 2, | ||||
'black_background': 40, | ||||
'red_background': 41, | ||||
'green_background': 42, | ||||
'yellow_background': 43, | ||||
'blue_background': 44, | ||||
'purple_background': 45, | ||||
'cyan_background': 46, | ||||
'white_background': 47} | ||||
Pierre-Yves David
|
r30652 | _styles = {'grep.match': 'red bold', | ||
'grep.linenumber': 'green', | ||||
'grep.rev': 'green', | ||||
'grep.change': 'green', | ||||
'grep.sep': 'cyan', | ||||
'grep.filename': 'magenta', | ||||
'grep.user': 'magenta', | ||||
'grep.date': 'magenta', | ||||
'bookmarks.active': 'green', | ||||
'branches.active': 'none', | ||||
'branches.closed': 'black bold', | ||||
'branches.current': 'green', | ||||
'branches.inactive': 'none', | ||||
'diff.changed': 'white', | ||||
'diff.deleted': 'red', | ||||
'diff.diffline': 'bold', | ||||
'diff.extended': 'cyan bold', | ||||
'diff.file_a': 'red bold', | ||||
'diff.file_b': 'green bold', | ||||
'diff.hunk': 'magenta', | ||||
'diff.inserted': 'green', | ||||
'diff.tab': '', | ||||
'diff.trailingwhitespace': 'bold red_background', | ||||
'changeset.public' : '', | ||||
'changeset.draft' : '', | ||||
'changeset.secret' : '', | ||||
'diffstat.deleted': 'red', | ||||
'diffstat.inserted': 'green', | ||||
'histedit.remaining': 'red bold', | ||||
'ui.prompt': 'yellow', | ||||
'log.changeset': 'yellow', | ||||
'patchbomb.finalsummary': '', | ||||
'patchbomb.from': 'magenta', | ||||
'patchbomb.to': 'cyan', | ||||
'patchbomb.subject': 'green', | ||||
'patchbomb.diffstats': '', | ||||
'rebase.rebased': 'blue', | ||||
'rebase.remaining': 'red bold', | ||||
'resolve.resolved': 'green bold', | ||||
'resolve.unresolved': 'red bold', | ||||
'shelve.age': 'cyan', | ||||
'shelve.newest': 'green bold', | ||||
'shelve.name': 'blue bold', | ||||
'status.added': 'green bold', | ||||
'status.clean': 'none', | ||||
'status.copied': 'none', | ||||
'status.deleted': 'cyan bold underline', | ||||
'status.ignored': 'black bold', | ||||
'status.modified': 'blue bold', | ||||
'status.removed': 'red bold', | ||||
'status.unknown': 'magenta bold underline', | ||||
'tags.normal': 'green', | ||||
'tags.local': 'black bold'} | ||||
Pierre-Yves David
|
r30653 | |||
def loadcolortable(ui, extname, colortable): | ||||
_styles.update(colortable) | ||||
Pierre-Yves David
|
r30969 | |||
Pierre-Yves David
|
r30971 | def configstyles(ui): | ||
for status, cfgeffects in ui.configitems('color'): | ||||
if '.' not in status or status.startswith(('color.', 'terminfo.')): | ||||
continue | ||||
cfgeffects = ui.configlist('color', status) | ||||
if cfgeffects: | ||||
good = [] | ||||
for e in cfgeffects: | ||||
if valideffect(e): | ||||
good.append(e) | ||||
else: | ||||
ui.warn(_("ignoring unknown color/effect %r " | ||||
"(configured in color.%s)\n") | ||||
% (e, status)) | ||||
_styles[status] = ' '.join(good) | ||||
Pierre-Yves David
|
r30969 | def valideffect(effect): | ||
'Determine if the effect is valid or not.' | ||||
Pierre-Yves David
|
r30970 | return ((not _terminfo_params and effect in _effects) | ||
or (effect in _terminfo_params | ||||
or effect[:-11] in _terminfo_params)) | ||||
Pierre-Yves David
|
r30972 | |||
def _effect_str(effect): | ||||
'''Helper function for render_effects().''' | ||||
bg = False | ||||
if effect.endswith('_background'): | ||||
bg = True | ||||
effect = effect[:-11] | ||||
try: | ||||
attr, val, termcode = _terminfo_params[effect] | ||||
except KeyError: | ||||
return '' | ||||
if attr: | ||||
if termcode: | ||||
return termcode | ||||
else: | ||||
return curses.tigetstr(val) | ||||
elif bg: | ||||
return curses.tparm(curses.tigetstr('setab'), val) | ||||
else: | ||||
return curses.tparm(curses.tigetstr('setaf'), val) | ||||
Pierre-Yves David
|
r30973 | |||
def _render_effects(text, effects): | ||||
'Wrap text in commands to turn on each effect.' | ||||
if not text: | ||||
return text | ||||
Pierre-Yves David
|
r31071 | if _terminfo_params: | ||
start = ''.join(_effect_str(effect) | ||||
for effect in ['none'] + effects.split()) | ||||
stop = _effect_str('none') | ||||
else: | ||||
Pierre-Yves David
|
r30973 | start = [str(_effects[e]) for e in ['none'] + effects.split()] | ||
start = '\033[' + ';'.join(start) + 'm' | ||||
stop = '\033[' + str(_effects['none']) + 'm' | ||||
return ''.join([start, text, stop]) | ||||
Pierre-Yves David
|
r31067 | |||
Pierre-Yves David
|
r31086 | def colorlabel(ui, msg, label): | ||
"""add color control code according to the mode""" | ||||
if ui._colormode == 'debug': | ||||
if label and msg: | ||||
if msg[-1] == '\n': | ||||
msg = "[%s|%s]\n" % (label, msg[:-1]) | ||||
else: | ||||
msg = "[%s|%s]" % (label, msg) | ||||
elif ui._colormode is not None: | ||||
effects = [] | ||||
for l in label.split(): | ||||
s = _styles.get(l, '') | ||||
if s: | ||||
effects.append(s) | ||||
elif valideffect(l): | ||||
effects.append(l) | ||||
effects = ' '.join(effects) | ||||
if effects: | ||||
msg = '\n'.join([_render_effects(line, effects) | ||||
for line in msg.split('\n')]) | ||||
return msg | ||||
Pierre-Yves David
|
r31067 | w32effects = None | ||
if pycompat.osname == 'nt': | ||||
import ctypes | ||||
import re | ||||
_kernel32 = ctypes.windll.kernel32 | ||||
_WORD = ctypes.c_ushort | ||||
_INVALID_HANDLE_VALUE = -1 | ||||
class _COORD(ctypes.Structure): | ||||
_fields_ = [('X', ctypes.c_short), | ||||
('Y', ctypes.c_short)] | ||||
class _SMALL_RECT(ctypes.Structure): | ||||
_fields_ = [('Left', ctypes.c_short), | ||||
('Top', ctypes.c_short), | ||||
('Right', ctypes.c_short), | ||||
('Bottom', ctypes.c_short)] | ||||
class _CONSOLE_SCREEN_BUFFER_INFO(ctypes.Structure): | ||||
_fields_ = [('dwSize', _COORD), | ||||
('dwCursorPosition', _COORD), | ||||
('wAttributes', _WORD), | ||||
('srWindow', _SMALL_RECT), | ||||
('dwMaximumWindowSize', _COORD)] | ||||
_STD_OUTPUT_HANDLE = 0xfffffff5 # (DWORD)-11 | ||||
_STD_ERROR_HANDLE = 0xfffffff4 # (DWORD)-12 | ||||
_FOREGROUND_BLUE = 0x0001 | ||||
_FOREGROUND_GREEN = 0x0002 | ||||
_FOREGROUND_RED = 0x0004 | ||||
_FOREGROUND_INTENSITY = 0x0008 | ||||
_BACKGROUND_BLUE = 0x0010 | ||||
_BACKGROUND_GREEN = 0x0020 | ||||
_BACKGROUND_RED = 0x0040 | ||||
_BACKGROUND_INTENSITY = 0x0080 | ||||
_COMMON_LVB_REVERSE_VIDEO = 0x4000 | ||||
_COMMON_LVB_UNDERSCORE = 0x8000 | ||||
# http://msdn.microsoft.com/en-us/library/ms682088%28VS.85%29.aspx | ||||
w32effects = { | ||||
'none': -1, | ||||
'black': 0, | ||||
'red': _FOREGROUND_RED, | ||||
'green': _FOREGROUND_GREEN, | ||||
'yellow': _FOREGROUND_RED | _FOREGROUND_GREEN, | ||||
'blue': _FOREGROUND_BLUE, | ||||
'magenta': _FOREGROUND_BLUE | _FOREGROUND_RED, | ||||
'cyan': _FOREGROUND_BLUE | _FOREGROUND_GREEN, | ||||
'white': _FOREGROUND_RED | _FOREGROUND_GREEN | _FOREGROUND_BLUE, | ||||
'bold': _FOREGROUND_INTENSITY, | ||||
'black_background': 0x100, # unused value > 0x0f | ||||
'red_background': _BACKGROUND_RED, | ||||
'green_background': _BACKGROUND_GREEN, | ||||
'yellow_background': _BACKGROUND_RED | _BACKGROUND_GREEN, | ||||
'blue_background': _BACKGROUND_BLUE, | ||||
'purple_background': _BACKGROUND_BLUE | _BACKGROUND_RED, | ||||
'cyan_background': _BACKGROUND_BLUE | _BACKGROUND_GREEN, | ||||
'white_background': (_BACKGROUND_RED | _BACKGROUND_GREEN | | ||||
_BACKGROUND_BLUE), | ||||
'bold_background': _BACKGROUND_INTENSITY, | ||||
'underline': _COMMON_LVB_UNDERSCORE, # double-byte charsets only | ||||
'inverse': _COMMON_LVB_REVERSE_VIDEO, # double-byte charsets only | ||||
} | ||||
passthrough = set([_FOREGROUND_INTENSITY, | ||||
_BACKGROUND_INTENSITY, | ||||
_COMMON_LVB_UNDERSCORE, | ||||
_COMMON_LVB_REVERSE_VIDEO]) | ||||
stdout = _kernel32.GetStdHandle( | ||||
_STD_OUTPUT_HANDLE) # don't close the handle returned | ||||
if stdout is None or stdout == _INVALID_HANDLE_VALUE: | ||||
w32effects = None | ||||
else: | ||||
csbi = _CONSOLE_SCREEN_BUFFER_INFO() | ||||
if not _kernel32.GetConsoleScreenBufferInfo( | ||||
stdout, ctypes.byref(csbi)): | ||||
# stdout may not support GetConsoleScreenBufferInfo() | ||||
# when called from subprocess or redirected | ||||
w32effects = None | ||||
else: | ||||
origattr = csbi.wAttributes | ||||
ansire = re.compile('\033\[([^m]*)m([^\033]*)(.*)', | ||||
re.MULTILINE | re.DOTALL) | ||||
Pierre-Yves David
|
r31088 | def win32print(text, writefunc, **opts): | ||
Pierre-Yves David
|
r31067 | label = opts.get('label', '') | ||
attr = origattr | ||||
def mapcolor(val, attr): | ||||
if val == -1: | ||||
return origattr | ||||
elif val in passthrough: | ||||
return attr | val | ||||
elif val > 0x0f: | ||||
return (val & 0x70) | (attr & 0x8f) | ||||
else: | ||||
return (val & 0x07) | (attr & 0xf8) | ||||
# determine console attributes based on labels | ||||
for l in label.split(): | ||||
style = _styles.get(l, '') | ||||
for effect in style.split(): | ||||
try: | ||||
attr = mapcolor(w32effects[effect], attr) | ||||
except KeyError: | ||||
# w32effects could not have certain attributes so we skip | ||||
# them if not found | ||||
pass | ||||
# hack to ensure regexp finds data | ||||
if not text.startswith('\033['): | ||||
text = '\033[m' + text | ||||
# Look for ANSI-like codes embedded in text | ||||
m = re.match(ansire, text) | ||||
try: | ||||
while m: | ||||
for sattr in m.group(1).split(';'): | ||||
if sattr: | ||||
attr = mapcolor(int(sattr), attr) | ||||
_kernel32.SetConsoleTextAttribute(stdout, attr) | ||||
Pierre-Yves David
|
r31088 | writefunc(m.group(2), **opts) | ||
Pierre-Yves David
|
r31067 | m = re.match(ansire, m.group(3)) | ||
finally: | ||||
# Explicitly reset original attributes | ||||
_kernel32.SetConsoleTextAttribute(stdout, origattr) | ||||