90 lines
1.5 KiB
Go
90 lines
1.5 KiB
Go
package log
|
|
|
|
import (
|
|
"fmt"
|
|
"runtime"
|
|
|
|
"github.com/apex/log"
|
|
acli "github.com/apex/log/handlers/cli"
|
|
"github.com/davecgh/go-spew/spew"
|
|
"github.com/pterm/pterm"
|
|
)
|
|
|
|
type Level = log.Level
|
|
|
|
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 Debugf(format string, args ...interface{}) {
|
|
DebugReal(fmt.Sprintf(format, args...), 2)
|
|
}
|
|
|
|
func Debug(arg string) {
|
|
DebugReal(arg, 2)
|
|
}
|
|
|
|
func DebugReal(arg string, cs int) {
|
|
_, callerFile, callerLine, ok := runtime.Caller(cs)
|
|
if !ok {
|
|
return
|
|
}
|
|
tag := fmt.Sprintf("%s:%d: ", callerFile, callerLine)
|
|
log.Debug(tag + arg)
|
|
}
|
|
|
|
func Dump(args ...interface{}) {
|
|
DebugReal(spew.Sdump(args...), 2)
|
|
}
|
|
|
|
func EnableDebugLogging() {
|
|
SetLevel(log.DebugLevel)
|
|
}
|
|
|
|
func VerbosityStepsToLogLevel(l int) log.Level {
|
|
switch l {
|
|
case 1:
|
|
return log.WarnLevel
|
|
case 2:
|
|
return log.InfoLevel
|
|
case 3:
|
|
return log.DebugLevel
|
|
}
|
|
return log.ErrorLevel
|
|
}
|
|
|
|
func SetLevelFromVerbosity(l int) {
|
|
SetLevel(VerbosityStepsToLogLevel(l))
|
|
}
|
|
|
|
func SetLevel(arg log.Level) {
|
|
log.SetLevel(arg)
|
|
}
|
|
|
|
func GetLogger() *log.Logger {
|
|
if logger, ok := log.Log.(*log.Logger); ok {
|
|
return logger
|
|
}
|
|
panic("unable to get logger")
|
|
}
|
|
|
|
func GetLevel() log.Level {
|
|
return GetLogger().Level
|
|
}
|
|
|
|
func WithError(e error) *log.Entry {
|
|
return GetLogger().WithError(e)
|
|
}
|