From 785f10a79271e57b489f659104093946d5b6799c Mon Sep 17 00:00:00 2001
From: Rob Moore <giftiger.wunsch@xantoria.com>
Date: Sat, 5 Dec 2015 10:39:03 +0000
Subject: [PATCH] Support customisation of default colours in unix

Add a field to LogBackend which specifies a ColorConfig
If using colour, ensure that we use the value for the corresponding
level in the backend's ColorConfig if specified, falling back to the
default colors.

Expose the colour constants found in log_nix.go so that applications can
reference them in specifying their own colour config.

Add a new ConvertColors func to log_nix as a convenient way of
converting a list of numbers representing ansi codes into the strings
needed, without having to wrap each one in colorSeq or colorSeqBold
individually.

Also expose ColorSeq and ColorSeqBold to enable individual levels to
be easily set bold or otherwise if that behaviour is desired, however.
---
 log_nix.go | 67 +++++++++++++++++++++++++++++++++++-------------------
 1 file changed, 44 insertions(+), 23 deletions(-)

diff --git a/log_nix.go b/log_nix.go
index f06a871..4ff2ab1 100644
--- a/log_nix.go
+++ b/log_nix.go
@@ -16,37 +16,38 @@ import (
 type color int
 
 const (
-	colorBlack = iota + 30
-	colorRed
-	colorGreen
-	colorYellow
-	colorBlue
-	colorMagenta
-	colorCyan
-	colorWhite
+	ColorBlack = iota + 30
+	ColorRed
+	ColorGreen
+	ColorYellow
+	ColorBlue
+	ColorMagenta
+	ColorCyan
+	ColorWhite
 )
 
 var (
 	colors = []string{
-		CRITICAL: colorSeq(colorMagenta),
-		ERROR:    colorSeq(colorRed),
-		WARNING:  colorSeq(colorYellow),
-		NOTICE:   colorSeq(colorGreen),
-		DEBUG:    colorSeq(colorCyan),
+		CRITICAL: ColorSeq(ColorMagenta),
+		ERROR:    ColorSeq(ColorRed),
+		WARNING:  ColorSeq(ColorYellow),
+		NOTICE:   ColorSeq(ColorGreen),
+		DEBUG:    ColorSeq(ColorCyan),
 	}
 	boldcolors = []string{
-		CRITICAL: colorSeqBold(colorMagenta),
-		ERROR:    colorSeqBold(colorRed),
-		WARNING:  colorSeqBold(colorYellow),
-		NOTICE:   colorSeqBold(colorGreen),
-		DEBUG:    colorSeqBold(colorCyan),
+		CRITICAL: ColorSeqBold(ColorMagenta),
+		ERROR:    ColorSeqBold(ColorRed),
+		WARNING:  ColorSeqBold(ColorYellow),
+		NOTICE:   ColorSeqBold(ColorGreen),
+		DEBUG:    ColorSeqBold(ColorCyan),
 	}
 )
 
 // LogBackend utilizes the standard log module.
 type LogBackend struct {
-	Logger *log.Logger
-	Color  bool
+	Logger      *log.Logger
+	Color       bool
+	ColorConfig []string
 }
 
 // NewLogBackend creates a new LogBackend.
@@ -57,8 +58,13 @@ func NewLogBackend(out io.Writer, prefix string, flag int) *LogBackend {
 // Log implements the Backend interface.
 func (b *LogBackend) Log(level Level, calldepth int, rec *Record) error {
 	if b.Color {
+		col := colors[level]
+		if len(b.ColorConfig) > int(level) && b.ColorConfig[level] != "" {
+			col = b.ColorConfig[level]
+		}
+
 		buf := &bytes.Buffer{}
-		buf.Write([]byte(colors[level]))
+		buf.Write([]byte(col))
 		buf.Write([]byte(rec.Formatted(calldepth + 1)))
 		buf.Write([]byte("\033[0m"))
 		// For some reason, the Go logger arbitrarily decided "2" was the correct
@@ -69,11 +75,26 @@ func (b *LogBackend) Log(level Level, calldepth int, rec *Record) error {
 	return b.Logger.Output(calldepth+2, rec.Formatted(calldepth+1))
 }
 
-func colorSeq(color color) string {
+// ConvertColors takes a list of ints representing colors for log levels and
+// converts them into strings for ANSI color formatting
+func ConvertColors(colors []int, bold bool) []string {
+	converted := []string{}
+	for _, i := range colors {
+		if bold {
+			converted = append(converted, ColorSeqBold(color(i)))
+		} else {
+			converted = append(converted, ColorSeq(color(i)))
+		}
+	}
+
+	return converted
+}
+
+func ColorSeq(color color) string {
 	return fmt.Sprintf("\033[%dm", int(color))
 }
 
-func colorSeqBold(color color) string {
+func ColorSeqBold(color color) string {
 	return fmt.Sprintf("\033[%d;1m", int(color))
 }