comparison hgext/color.py @ 7457:a70fb83cbb9e

diff colorization: finish highlighting trailing whitespace

7456:79eb16db5e4a

'''add color output to status, qseries, and diff-related commands

This extension modifies the status command to add color to its output to
reflect file status, the qseries command to add color to reflect patch status
(applied, unapplied, missing), and to diff-related commands to highlight
additions, removals, diff headers, and trailing whitespace.  Other effects in
addition to color, like bold and underlined text, are also available.
Effects are rendered with the ECMA-48 SGR control function (aka ANSI escape
codes).  This module also provides the render_text function, which can be
used to add effects to any text.

To enable this extension, add this to your .hgrc file:
[extensions]
color =

diff.file_b = green bold
diff.hunk = magenta
diff.deleted = red
diff.inserted = green
diff.changed = white
diff.whitespace = bold red_background
'''

import os, re, sys

from mercurial import cmdutil, commands, extensions

def colorwrap(orig, s):
    '''wrap ui.write for colored diff output'''
    lines = s.split('\n')
    for i, line in enumerate(lines):
        for prefix, style in _diff_prefixes:
            if line.startswith(prefix):
                effects = _diff_effects[style]
                lines[i] = render_effects(line, *_diff_effects[style])
                break
    orig('\n'.join(lines))

def colorshowpatch(orig, self, node):
    '''wrap cmdutil.changeset_printer.showpatch with colored output'''
    oldwrite = extensions.wrapfunction(self.ui, 'write', colorwrap)

                 'file_a': ('red', 'bold'),
                 'file_b': ('green', 'bold'),
                 'hunk': ('magenta',),
                 'deleted': ('red',),
                 'inserted': ('green',),
                 'changed': ('white',)}

def uisetup(ui):
    '''Initialize the extension.'''
    _setupcmd(ui, 'diff', commands.table, colordiff, _diff_effects)
    _setupcmd(ui, 'incoming', commands.table, None, _diff_effects)

7457:a70fb83cbb9e

'''add color output to status, qseries, and diff-related commands

This extension modifies the status command to add color to its output to
reflect file status, the qseries command to add color to reflect patch status
(applied, unapplied, missing), and to diff-related commands to highlight
additions, removals, diff headers, and trailing whitespace.

Other effects in addition to color, like bold and underlined text, are also
available.  Effects are rendered with the ECMA-48 SGR control function (aka
ANSI escape codes).  This module also provides the render_text function,
which can be used to add effects to any text.

To enable this extension, add this to your .hgrc file:
[extensions]
color =

diff.file_b = green bold
diff.hunk = magenta
diff.deleted = red
diff.inserted = green
diff.changed = white
diff.trailingwhitespace = bold red_background
'''

import os, re, sys

from mercurial import cmdutil, commands, extensions

def colorwrap(orig, s):
    '''wrap ui.write for colored diff output'''
    lines = s.split('\n')
    for i, line in enumerate(lines):
        stripline = line
        if line and line[0] in '+-':
            # highlight trailing whitespace, but only in changed lines
            stripline = line.rstrip()
        for prefix, style in _diff_prefixes:
            if stripline.startswith(prefix):
                lines[i] = render_effects(stripline, *_diff_effects[style])
                break
        if line != stripline:
            lines[i] += render_effects(
                line[len(stripline):], *_diff_effects['trailingwhitespace'])
    orig('\n'.join(lines))

def colorshowpatch(orig, self, node):
    '''wrap cmdutil.changeset_printer.showpatch with colored output'''
    oldwrite = extensions.wrapfunction(self.ui, 'write', colorwrap)

                 'file_a': ('red', 'bold'),
                 'file_b': ('green', 'bold'),
                 'hunk': ('magenta',),
                 'deleted': ('red',),
                 'inserted': ('green',),
                 'changed': ('white',),
                 'trailingwhitespace': ('bold', 'red_background'),}

def uisetup(ui):
    '''Initialize the extension.'''
    _setupcmd(ui, 'diff', commands.table, colordiff, _diff_effects)
    _setupcmd(ui, 'incoming', commands.table, None, _diff_effects)