diff --git a/pkg/logger/logger.go b/pkg/logger/logger.go index f9541cbb..c33d5196 100644 --- a/pkg/logger/logger.go +++ b/pkg/logger/logger.go @@ -2,6 +2,7 @@ package logger import ( "fmt" + "os" "sync" "time" @@ -34,49 +35,49 @@ func SpinnerStop() { s = nil } -func Warning(msg ...interface{}) { - if s != nil { - SpinnerText(Sprintf(msg), Bold(Yellow("Warn")).BgBlack().String()) - return +func msg(level string, msg ...interface{}) { + var levelMsg string + switch level { + case "warning": + levelMsg = Bold(Yellow("Warn")).BgBlack().String() + case "debug": + levelMsg = Bold(White("Debug")).BgBlack().String() + case "info": + levelMsg = Bold(Blue("Info")).BgBlack().String() + case "error": + levelMsg = Bold(Red("Error")).BgBlack().String() } - cmd := []interface{}{Bold(Yellow("Warn")).BgBlack().String()} - for _, f := range msg { - cmd = append(cmd, f) - } - fmt.Println(cmd...) -} -func Debug(msg ...interface{}) { - if s != nil { - SpinnerText(Sprintf(msg), Bold(White("Debug")).BgBlack().String()) - return - } - cmd := []interface{}{Bold(White("Debug")).String()} + + if s != nil { + SpinnerText(Sprintf(msg), levelMsg) + // return + } + + cmd := []interface{}{levelMsg} for _, f := range msg { cmd = append(cmd, f) } + fmt.Println(cmd...) } -func Info(msg ...interface{}) { - if s != nil { - SpinnerText(Sprintf(msg), Bold(Blue("Info")).BgBlack().String()) - return - } - cmd := []interface{}{Bold(Green("Info")).String()} - for _, f := range msg { - cmd = append(cmd, f) - } - fmt.Println(cmd...) +func Warning(mess ...interface{}) { + msg("warning", mess) } -func Error(msg ...interface{}) { - if s != nil { - SpinnerText(Sprintf(msg), Bold(Red("Error")).BgBlack().String()) - return - } - cmd := []interface{}{Bold(Red("Error")).String()} - for _, f := range msg { - cmd = append(cmd, f) - } - fmt.Println(cmd...) +func Debug(mess ...interface{}) { + msg("debug", mess) +} + +func Info(mess ...interface{}) { + msg("info", mess) +} + +func Error(mess ...interface{}) { + msg("error", mess) +} + +func Fatal(mess ...interface{}) { + Error(mess) + os.Exit(1) }