paulo@0: #!/usr/bin/env python
paulo@0: 
paulo@0: import sys
paulo@0: import re
paulo@0: import getopt
paulo@0: 
paulo@0: #x = "\x1B[1;31;49m" + x
paulo@0: #x = START + c["white"][0] + ";" + c["blue"][1] + END + x + CLEAR
paulo@0: 
paulo@0: START = "\x1B["
paulo@0: BOLD = "1;"
paulo@0: END = "m"
paulo@0: CLEAR = START + "0" + END
paulo@0: 
paulo@0: c = {
paulo@0: 	# color : (fg, bg)
paulo@0: 	"black" : ("30", "40"),
paulo@0: 	"red" : ("31", "41"),
paulo@0: 	"green" : ("32", "42"),
paulo@0: 	"brown" : ("33", "43"),
paulo@0: 	"blue" : ("34", "44"),
paulo@0: 	"magenta" : ("35", "45"),
paulo@0: 	"cyan" : ("36", "46"),
paulo@0: 	"white" : ("37", "47"),
paulo@0: 	"default" : ("39", "49"),
paulo@0: }
paulo@0: 
paulo@0: bold = False
paulo@0: fg = None
paulo@0: bg = None
paulo@11: colorLine = False
paulo@0: 
paulo@11: (optvals, args) = getopt.getopt(sys.argv[1:], "f:b:Bl")
paulo@0: 
paulo@0: for (opt, val) in optvals:
paulo@0: 	if opt == '-B':
paulo@0: 		bold = True
paulo@0: 	elif opt == '-f':
paulo@0: 		fg = val
paulo@0: 	elif opt == '-b':
paulo@0: 		bg = val
paulo@11: 	elif opt == '-l':
paulo@11: 		colorLine = True
paulo@0: 
paulo@0: if len(args) > 0:
paulo@0: 	for line in sys.stdin:
paulo@0: 		repl = START
paulo@0: 		if bold:
paulo@0: 			repl += BOLD
paulo@0: 		if fg is not None:
paulo@0: 			repl += c[fg][0]
paulo@0: 		if bg is not None:
paulo@0: 			repl += ';' + c[bg][1]
paulo@0: 		repl += END + r'\1' + CLEAR
paulo@11: 
paulo@11: 		x = line
paulo@11: 
paulo@11: 		if colorLine == False:
paulo@11: 			x = re.sub(r'(' + args[0] + r')', repl, line)
paulo@11: 		else:
paulo@11: 			if re.search(args[0], line) != None:
paulo@11: 				last_char = line[-1]
paulo@11: 				x = repl.replace(r'\1', line[:-1])
paulo@11: 				x += last_char
paulo@0: 		
paulo@0: 		sys.stdout.write(x)
paulo@0: 		sys.stdout.flush()