adding custom colors for logging

Author: codemodify

Logging/Persisters/console.go

@@ -8,20 +8,41 @@ import (
 
 type consoleLogger struct {
 	logUntil loggingC.LogType
+	colors   map[loggingC.LogType]ConsoleLoggerColorDelegate
 }
 
 // NewConsoleLogger -
-func NewConsoleLogger(logUntil loggingC.LogType) loggingC.Logger {
+func NewConsoleLogger(logUntil loggingC.LogType, colors map[loggingC.LogType]ConsoleLoggerColorDelegate) loggingC.Logger {
 	return &consoleLogger{
 		logUntil: logUntil,
+		colors:   colors,
 	}
 }
 
+// NewConsoleLoggerDefaultColors -
+func NewConsoleLoggerDefaultColors() map[loggingC.LogType]ConsoleLoggerColorDelegate {
+	return map[loggingC.LogType]ConsoleLoggerColorDelegate{
+		loggingC.TypeDisable: WhiteString,
+		loggingC.TypeTrace:   BlackStringYellowBG,
+		loggingC.TypePanic:   RedString,
+		loggingC.TypeFatal:   RedString,
+		loggingC.TypeError:   RedString,
+		loggingC.TypeWarning: YellowString,
+		loggingC.TypeInfo:    WhiteString,
+		loggingC.TypeDebug:   CyanString,
+	}
+}
+
+// ConsoleLoggerColorDelegate -
+type ConsoleLoggerColorDelegate func(string, ...interface{}) string
+
+// BlackStringYellowBG -
 func BlackStringYellowBG(format string, a ...interface{}) string {
 	c := New(FgBlack, BgYellow)
 	return c.Sprintf(format, a...)
 }
 
+// BlackStringWhiteBG -
 func BlackStringWhiteBG(format string, a ...interface{}) string {
 	c := New(FgBlack, BgWhite)
 	return c.Sprintf(format, a...)
@@ -38,14 +59,18 @@ func (thisRef consoleLogger) Log(logEntry loggingC.LogEntry) {
 	}
 
 	if logEntry.Type == loggingC.TypeTrace {
-		fmt.Println(BlackStringYellowBG(logEntry.Message))
+		fmt.Println(thisRef.colors[loggingC.TypeTrace](logEntry.Message))
+
 	} else if logEntry.Type < loggingC.TypeWarning {
-		fmt.Println(RedString(logEntry.Message))
+		fmt.Println(thisRef.colors[loggingC.TypeError](logEntry.Message))
+
 	} else if logEntry.Type == loggingC.TypeWarning {
-		fmt.Println(YellowString(logEntry.Message))
+		fmt.Println(thisRef.colors[loggingC.TypeWarning](logEntry.Message))
+
 	} else if logEntry.Type == loggingC.TypeInfo {
-		fmt.Println(WhiteString(logEntry.Message))
+		fmt.Println(thisRef.colors[loggingC.TypeInfo](logEntry.Message))
+
 	} else if logEntry.Type == loggingC.TypeDebug {
-		fmt.Println(CyanString(logEntry.Message))
+		fmt.Println(thisRef.colors[loggingC.TypeDebug](logEntry.Message))
 	}
 }

Logging/logging.go

@@ -20,8 +20,13 @@ func Init(logger loggingC.EasyLogger) {
 
 // NewConsoleLogger -
 func NewConsoleLogger() loggingC.EasyLogger {
+	return NewConsoleLoggerCustomColors(loggingP.NewConsoleLoggerDefaultColors())
+}
+
+// NewConsoleLoggerCustomColors -
+func NewConsoleLoggerCustomColors(colors map[loggingC.LogType]loggingP.ConsoleLoggerColorDelegate) loggingC.EasyLogger {
 	return housekeeping.NewDefaultHelperImplmentation(
-		loggingP.NewConsoleLogger(loggingC.TypeDebug),
+		loggingP.NewConsoleLogger(loggingC.TypeDebug, colors),
 	)
 }
 
@@ -32,6 +37,7 @@ func NewFileLogger() loggingC.EasyLogger {
 	)
 }
 
+// NewEasyLoggerForLogger -
 func NewEasyLoggerForLogger(logger loggingC.Logger) loggingC.EasyLogger {
 	return housekeeping.NewDefaultHelperImplmentation(logger)
 }

Logging/tests/02-log-smart_test.go

@@ -15,7 +15,7 @@ func Test_02(t *testing.T) {
 		loggingF.NewSimpleFormatterLogger(
 			loggingM.NewMultiLogger(
 				[]loggingC.Logger{
-					loggingP.NewConsoleLogger(loggingC.TypeDebug),
+					loggingP.NewConsoleLogger(loggingC.TypeDebug, loggingP.NewConsoleLoggerDefaultColors()),
 					loggingP.NewFileLogger(loggingC.TypeDebug, "log.log"),
 				},
 			),

Logging/tests/03-log-async-and_or-buffered_test.go

@@ -17,7 +17,7 @@ func Test_03(t *testing.T) {
 				loggingM.NewBufferedLogger(
 					loggingM.NewMultiLogger(
 						[]loggingC.Logger{
-							loggingP.NewConsoleLogger(loggingC.TypeDebug),
+							loggingP.NewConsoleLogger(loggingC.TypeDebug, loggingP.NewConsoleLoggerDefaultColors()),
 							loggingP.NewFileLogger(loggingC.TypeDebug, "log.log"),
 						},
 					),

Service/system-service.go

@@ -1,6 +1,9 @@
 package Service
 
-import "fmt"
+import (
+	"encoding/json"
+	"fmt"
+)
 
 // SystemService - represents a generic system service configuration
 type SystemService interface {
@@ -38,6 +41,16 @@ type ServiceCommand struct {
 	RunAsGroup          string
 }
 
+func (thisRef ServiceCommand) String() string {
+	bytes, err := json.Marshal(thisRef)
+	if err != nil {
+		// INFO: in normal app you could log this
+		return ""
+	}
+
+	return string(bytes)
+}
+
 // ServiceStatus is a generic representation of the service running on the system
 type ServiceStatus struct {
 	Running bool
@@ -54,8 +67,8 @@ const (
 	ServiceErrorOther                         = iota
 )
 
-func (e ServiceErrorType) String() string {
-	switch e {
+func (thisRef ServiceErrorType) String() string {
+	switch thisRef {
 	case ServiceErrorSuccess:
 		return "Success"
 
@@ -66,7 +79,7 @@ func (e ServiceErrorType) String() string {
 		return "Other error occured"
 
 	default:
-		return fmt.Sprintf("%d", int(e))
+		return fmt.Sprintf("%d", int(thisRef))
 	}
 }