80 lines
1.8 KiB
Go
80 lines
1.8 KiB
Go
package loggo
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/fatih/color"
|
|
)
|
|
|
|
// struct for log levels
|
|
type logLevel struct {
|
|
level int
|
|
name string
|
|
color color.Color
|
|
}
|
|
|
|
// debug log level (level = 0, name = "debug")
|
|
var DebugLevel = logLevel{0, "debug", *color.New(color.FgCyan)}
|
|
|
|
// info log level (level = 1, name = "info")
|
|
var InfoLevel = logLevel{1, "info", *color.New(color.FgGreen)}
|
|
|
|
// warning log level (level = 2, name = "warning")
|
|
var WarningLevel = logLevel{2, "warning", *color.New(color.FgYellow)}
|
|
|
|
// error log level (level = 3, name = "error")
|
|
var ErrorLevel = logLevel{3, "error", *color.New(color.FgRed)}
|
|
|
|
// fatal log level (level = 4, name = "fatal")
|
|
var FatalLevel = logLevel{4, "fatal", *color.New(color.FgRed).Add(color.Bold)}
|
|
|
|
// current log level (defaults to: "infoLevel")
|
|
var currentLevel = InfoLevel
|
|
|
|
// return current set log level
|
|
func GetLogLevel() logLevel {
|
|
return currentLevel
|
|
}
|
|
|
|
// set current log level
|
|
func SetLogLevel(level logLevel) {
|
|
currentLevel = level
|
|
}
|
|
|
|
// set current log level by name
|
|
func SetLogLevelByName(name string) {
|
|
switch strings.ToLower(name) {
|
|
case DebugLevel.name:
|
|
SetLogLevel(DebugLevel)
|
|
case InfoLevel.name:
|
|
SetLogLevel(InfoLevel)
|
|
case WarningLevel.name:
|
|
SetLogLevel(WarningLevel)
|
|
case ErrorLevel.name:
|
|
SetLogLevel(ErrorLevel)
|
|
case FatalLevel.name:
|
|
SetLogLevel(FatalLevel)
|
|
}
|
|
}
|
|
|
|
// set current log level by level
|
|
func SetLogLevelByLevel(level int) {
|
|
switch level {
|
|
case DebugLevel.level:
|
|
SetLogLevel(DebugLevel)
|
|
case InfoLevel.level:
|
|
SetLogLevel(InfoLevel)
|
|
case WarningLevel.level:
|
|
SetLogLevel(WarningLevel)
|
|
case ErrorLevel.level:
|
|
SetLogLevel(ErrorLevel)
|
|
case FatalLevel.level:
|
|
SetLogLevel(FatalLevel)
|
|
}
|
|
}
|
|
|
|
// set the color of a log level
|
|
func (level *logLevel) SetColor(color color.Color) {
|
|
level.color = color
|
|
}
|