40 lines
930 B
Go
40 lines
930 B
Go
|
package logger
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"git.rob.mx/nidito/chinampa/pkg/runtime"
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Formatter struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *Formatter) Format(entry *logrus.Entry) ([]byte, error) {
|
||
|
|
prefix := ""
|
||
|
|
colorEnabled := runtime.ColorEnabled()
|
||
|
|
if runtime.VerboseEnabled() {
|
||
|
|
date := strings.Replace(entry.Time.Local().Format(time.DateTime), " ", "T", 1)
|
||
|
|
component := ""
|
||
|
|
if c, ok := entry.Data[componentKey]; ok {
|
||
|
|
component = " " + c.(string)
|
||
|
|
}
|
||
|
|
prefix = fmt.Sprintf("\033[2m%s %s%s\033[0m\t", date, entry.Level.String(), component)
|
||
|
|
} else if entry.Level == logrus.ErrorLevel {
|
||
|
|
if colorEnabled {
|
||
|
|
prefix = "\033[41m\033[1m ERROR \033[0m "
|
||
|
|
} else {
|
||
|
|
prefix = "ERROR: "
|
||
|
|
}
|
||
|
|
} else if entry.Level == logrus.WarnLevel {
|
||
|
|
if colorEnabled {
|
||
|
|
prefix = "\033[43m\033[31m warning \033[0m "
|
||
|
|
} else {
|
||
|
|
prefix = "WARNING: "
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return []byte(prefix + entry.Message + "\n"), nil
|
||
|
|
}
|