From 96f30ef8cb676b4be3cddf8f992cb79fe20cb9df Mon Sep 17 00:00:00 2001
From: Kjeld Schouten <info@kjeldschouten.nl>
Date: Tue, 22 Oct 2024 13:53:20 +0200
Subject: [PATCH] feat(clustertool): fancy easy colors and color toggle for
 logger output

---
 clustertool/main.go | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/clustertool/main.go b/clustertool/main.go
index effe7275e36..e7c008630ac 100644
--- a/clustertool/main.go
+++ b/clustertool/main.go
@@ -18,6 +18,7 @@ import (
 )
 
 var version = "dev"
+var noColor = false
 
 func main() {
     // Configure zerolog
@@ -51,18 +52,16 @@ func main() {
         zapLevel = zapcore.InfoLevel
     }
 
+    if os.Getenv("DEBUG") != "" {
+        noColor = true
+    }
+
     // Set zerolog level
     zerolog.SetGlobalLevel(zerologLevel)
     log.Logger = log.Output(zerolog.ConsoleWriter{
         Out:        os.Stdout,
         TimeFormat: time.RFC3339, // Keep this for the timestamp format
-        NoColor:    false,        // Set to true if you prefer no color
-        FormatLevel: func(i interface{}) string {
-            return fmt.Sprintf("\033[1;34m[%s]\033[0m", i) // Blue color for level
-        },
-        FormatMessage: func(i interface{}) string {
-            return fmt.Sprintf("\033[1;32m%s\033[0m", i) // Green color for message
-        },
+        NoColor:    noColor,      // Set to true if you prefer no color
     })
 
     // Configure zap logger