#!/usr/bin/python

# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

''' A mini class for terminal color '''

import re


CSI = '\033['                       # ANSI Control Sequence Introducer
NORMAL = CSI + '0m'
CODE = {'black': CSI + '1;30m',     # color code for foreground color
        'red': CSI + '1;31m',
        'green': CSI + '1;32m',
        'yellow': CSI + '1;33m',
        'blue': CSI + '1;34m',
        'magenta': CSI + '1;35m',
        'cyan': CSI + '1;36m',
        'white': CSI + '1;37m',
        'default': CSI + '1;39m',
       }


def color_string(string, bgn_sym, end_sym, color):
    ''' Insert color code for a bracketed substring in a given string '''
    cstring = re.sub(bgn_sym, CODE[color] + bgn_sym, string)
    cstring = re.sub(end_sym, end_sym + NORMAL, cstring)
    return cstring