chore: replace slog with pterm

radiofrance · Dec 14, 2023 · d6b4c57 · d6b4c57
1 parent ccb7dc7
commit d6b4c57
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 30 deletions.
diff --git a/cmd/logger.go b/cmd/logger.go
@@ -1,11 +1,26 @@
 package main
 
 import (
-	"github.com/radiofrance/dib/internal/logging"
+	"github.com/pterm/pterm"
 	"github.com/spf13/viper"
 )
 
 func initLogLevel() {
 	_ = viper.BindPFlag("log_level", rootCmd.PersistentFlags().Lookup("log-level"))
-	_ = logging.Init(viper.GetString("log_level"))
+	logLevel := viper.GetString("log_level")
+
+	switch logLevel {
+	case "debug":
+		pterm.EnableDebugMessages()
+		pterm.Debug.Println("debug mode enabled")
+	case "info", "warning", "error", "fatal":
+	default:
+		pterm.Error.Printfln("%q is not a valid loglevel", logLevel)
+	}
+
+	// Customize default info message.
+	pterm.Info.Prefix = pterm.Prefix{
+		Text:  "INFO",
+		Style: pterm.NewStyle(pterm.FgBlack, pterm.BgCyan),
+	}
 }
diff --git a/internal/logging/logging.go b/internal/logging/logging.go
diff --git a/pkg/report/utils.go b/pkg/report/utils.go
@@ -114,7 +114,7 @@ func RemoveTerminalColors(input []byte) []byte {
 	return results
 }
 
-// StripKanikoBuildLogs Improve readability of kaniko builds logs by removing unwanted stuff from a logrus
+// StripKanikoBuildLogs Improve readability of kaniko builds logs by removing unwanted stuff from a
 // standard logs message.
 func StripKanikoBuildLogs(input []byte) string {
 	results := patternKanikoLogs.ReplaceAll(input, []byte("$message"))