89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
// Copyright © 2022 Roberto Hidalgo <chinampa@un.rob.mx>
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
package logger
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
"git.rob.mx/nidito/chinampa/pkg/runtime"
|
|
"github.com/fatih/color"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var bold *color.Color
|
|
var boldRedBG *color.Color
|
|
var boldRed *color.Color
|
|
var boldYellowBG *color.Color
|
|
var boldYellow *color.Color
|
|
var dimmed *color.Color
|
|
|
|
func init() {
|
|
bold = color.New(color.Bold)
|
|
bold.EnableColor()
|
|
boldRedBG = color.New(color.Bold, color.BgRed, color.FgWhite)
|
|
boldRedBG.EnableColor()
|
|
boldRed = color.New(color.Bold, color.FgHiRed)
|
|
boldRed.EnableColor()
|
|
boldYellowBG = color.New(color.Bold, color.BgYellow, color.FgBlack)
|
|
boldYellowBG.EnableColor()
|
|
boldYellow = color.New(color.Bold, color.FgHiYellow)
|
|
boldYellow.EnableColor()
|
|
dimmed = color.New(color.Faint)
|
|
dimmed.EnableColor()
|
|
}
|
|
|
|
type ttyFormatter struct {
|
|
}
|
|
|
|
func (f *ttyFormatter) Format(entry *logrus.Entry) ([]byte, error) {
|
|
prefix := ""
|
|
colorEnabled := runtime.ColorEnabled()
|
|
message := entry.Message
|
|
switch {
|
|
case runtime.VerboseEnabled():
|
|
date := strings.Replace(entry.Time.Local().Format(time.DateTime), " ", "T", 1)
|
|
component := ""
|
|
if c, ok := entry.Data[componentKey]; ok {
|
|
component = " " + c.(string)
|
|
}
|
|
level := entry.Level.String()
|
|
if colorEnabled {
|
|
switch {
|
|
case entry.Level <= logrus.ErrorLevel:
|
|
level = boldRed.Sprint(level)
|
|
case entry.Level == logrus.WarnLevel:
|
|
level = boldYellow.Sprint(level)
|
|
case entry.Level >= logrus.DebugLevel:
|
|
level = dimmed.Sprint(level)
|
|
message = dimmed.Sprint(message)
|
|
default:
|
|
level = dimmed.Sprint(level)
|
|
}
|
|
}
|
|
|
|
prefix = dimmed.Sprint(date) + " " + level + dimmed.Sprint(component) + "\t"
|
|
case entry.Level == logrus.ErrorLevel:
|
|
if colorEnabled {
|
|
prefix = boldRedBG.Sprint(" ERROR ") + " "
|
|
} else {
|
|
prefix = "ERROR: "
|
|
}
|
|
case entry.Level == logrus.WarnLevel:
|
|
if colorEnabled {
|
|
prefix = boldYellowBG.Sprint(" WARNING ") + " "
|
|
} else {
|
|
prefix = "WARNING: "
|
|
}
|
|
case entry.Level >= logrus.DebugLevel:
|
|
if colorEnabled {
|
|
prefix = dimmed.Sprintf("%s: ", strings.ToUpper(entry.Level.String()))
|
|
message = dimmed.Sprint(message)
|
|
} else {
|
|
prefix = strings.ToUpper(entry.Level.String()) + ": "
|
|
}
|
|
}
|
|
|
|
return []byte(prefix + message + "\n"), nil
|
|
}
|