comparison hgext/color.py @ 10046:0c23b0b3516b

color: Add support for bookmarks

10045:d1f9640e9a67

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

import os, sys

from mercurial import cmdutil, commands, extensions, error

                    'unknown': ['magenta', 'bold', 'underline'],
                    'ignored': ['black', 'bold'],
                    'clean': ['none'],
                    'copied': ['none'], }

def colorqseries(orig, ui, repo, *dummy, **opts):
    '''run the qseries command with colored output'''
    ui.pushbuffer()
    retval = orig(ui, repo, **opts)
    patchlines = ui.popbuffer().splitlines()

        churn = extensions.find('churn')
        _setupcmd(ui, 'churn', churn.cmdtable, colorchurn, _diff_effects)
    except KeyError:
        churn = None

def _setupcmd(ui, cmd, table, func, effectsmap):
    '''patch in command to command table and load effect map'''
    def nocolor(orig, *args, **opts):

        if (opts['no_color'] or opts['color'] == 'never' or

10046:0c23b0b3516b

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

  bookmarks.current = green
'''

import os, sys

from mercurial import cmdutil, commands, extensions, error

                    'unknown': ['magenta', 'bold', 'underline'],
                    'ignored': ['black', 'bold'],
                    'clean': ['none'],
                    'copied': ['none'], }

_bookmark_effects = { 'current': ['green'] }

def colorbookmarks(orig, ui, repo, *pats, **opts):
    def colorize(orig, s):
        lines = s.split('\n')
        for i, line in enumerate(lines):
            if line.startswith(" *"):
                lines[i] = render_effects(line, _bookmark_effects['current'])
        orig('\n'.join(lines))
    oldwrite = extensions.wrapfunction(ui, 'write', colorize)
    try:
        orig(ui, repo, *pats, **opts)
    finally:
        ui.write = oldwrite

def colorqseries(orig, ui, repo, *dummy, **opts):
    '''run the qseries command with colored output'''
    ui.pushbuffer()
    retval = orig(ui, repo, **opts)
    patchlines = ui.popbuffer().splitlines()

        churn = extensions.find('churn')
        _setupcmd(ui, 'churn', churn.cmdtable, colorchurn, _diff_effects)
    except KeyError:
        churn = None

    try:
        bookmarks = extensions.find('bookmarks')
        _setupcmd(ui, 'bookmarks', bookmarks.cmdtable, colorbookmarks,
                  _bookmark_effects)
    except KeyError:
        # The bookmarks extension is not enabled
        pass

def _setupcmd(ui, cmd, table, func, effectsmap):
    '''patch in command to command table and load effect map'''
    def nocolor(orig, *args, **opts):

        if (opts['no_color'] or opts['color'] == 'never' or