latest

internal/log/log.go

@@ -0,0 +1,45 @@
+package log
+
+import (
+	"github.com/apex/log"
+	acli "github.com/apex/log/handlers/cli"
+	"github.com/pterm/pterm"
+)
+
+func DisableStyling() {
+	pterm.DisableColor()
+	pterm.DisableStyling()
+	pterm.Debug.Prefix.Text = ""
+	pterm.Info.Prefix.Text = ""
+	pterm.Success.Prefix.Text = ""
+	pterm.Warning.Prefix.Text = ""
+	pterm.Error.Prefix.Text = ""
+	pterm.Fatal.Prefix.Text = ""
+}
+
+func Init() {
+	log.SetHandler(acli.Default)
+	log.SetLevel(log.InfoLevel)
+}
+
+func SetLevel(arg log.Level) {
+	log.SetLevel(arg)
+}
+
+func GetLevel() log.Level {
+	if logger, ok := log.Log.(*log.Logger); ok {
+		return logger.Level
+	}
+	return 0
+}
+
+func IncreaseLevel() {
+	SetLevel(GetLevel() + 1)
+}
+
+func WithError(e error) *log.Entry {
+	if logger, ok := log.Log.(*log.Logger); ok {
+		return logger.WithError(e)
+	}
+	return nil
+}