1 files changed, 35 insertions, 16 deletions
diff --git a/internal/config/client.go b/internal/config/client.go
index 3d8c45a..d6d106f 100644
--- a/internal/config/client.go
+++ b/internal/config/client.go
@@ -1,28 +1,47 @@
 package config
 
-// ClientColorConfig allows to override the default terminal color codes.
-type ClientColorConfig struct {
-	OkBgColor Color
-}
+import "github.com/mimecast/dtail/internal/color"
 
-/*
-	splitted := strings.Split(line, "|")
-	if splitted[2] == "100" {
-		splitted[2] = Paint(BgGreen, splitted[2])
-	} else {
-		splitted[2] = Paint(BgRed, splitted[2])
-	}
-	info := strings.Join(splitted[0:5], "|")
-	log := strings.Join(splitted[5:], "|")
-*/
+// ClientColorConfig allows to override the default terminal color color.
+type termColors struct {
+	RemoteTextFg      color.Color
+	RemoteStatsOkBg   color.Color
+	RemoteStatsWarnBg color.Color
+	RemoteTraceBg     color.Color
+	RemoteDebugBg     color.Color
+	RemoteWarnBg      color.Color
+	RemoteErrorBg     color.Color
+	RemoteFatalBg     color.Color
+	RemoteFatalAttr   color.Attribute
+	ClientStatsBg     color.Color
+	ClientWarnFg      color.Color
+	ClientErrorFg     color.Color
+}
 
 // ClientConfig represents a DTail client configuration (empty as of now as there
 // are no available config options yet, but that may changes in the future).
 type ClientConfig struct {
-	TerminalColors ClientColorConfig `json:",omitempty"`
+	TermColorsEnabled bool
+	TermColors        termColors `json:",omitempty"`
 }
 
 // Create a new default client configuration.
 func newDefaultClientConfig() *ClientConfig {
-	return &ClientConfig{}
+	return &ClientConfig{
+		TermColorsEnabled: true,
+		TermColors: termColors{
+			RemoteTextFg:      color.LightGray,
+			RemoteStatsOkBg:   color.BgGreen,
+			RemoteStatsWarnBg: color.BgRed,
+			RemoteTraceBg:     color.BgYellow,
+			RemoteDebugBg:     color.BgYellow,
+			RemoteWarnBg:      color.BgYellow,
+			RemoteErrorBg:     color.BgRed,
+			RemoteFatalBg:     color.BgRed,
+			RemoteFatalAttr:   color.Bold,
+			ClientStatsBg:     color.BgBlue,
+			ClientWarnFg:      color.Purple,
+			ClientErrorFg:     color.BgRed,
+		},
+	}
 }