schnutibox/vendor/github.com/cosmtrek/air/runner/logger.go

113 lines
2.1 KiB
Go

package runner
import (
"fmt"
"os"
"strings"
"time"
"github.com/fatih/color"
)
var (
rawColor = "raw"
// TODO: support more colors
colorMap = map[string]color.Attribute{
"red": color.FgRed,
"green": color.FgGreen,
"yellow": color.FgYellow,
"blue": color.FgBlue,
"magenta": color.FgMagenta,
"cyan": color.FgCyan,
"white": color.FgWhite,
}
)
type logFunc func(string, ...interface{})
type logger struct {
config *config
colors map[string]string
loggers map[string]logFunc
}
func newLogger(cfg *config) *logger {
if cfg == nil {
return nil
}
colors := cfg.colorInfo()
loggers := make(map[string]logFunc, len(colors))
for name, nameColor := range colors {
loggers[name] = newLogFunc(nameColor, cfg.Log)
}
loggers["default"] = defaultLogger()
return &logger{
config: cfg,
colors: colors,
loggers: loggers,
}
}
func newLogFunc(colorname string, cfg cfgLog) logFunc {
return func(msg string, v ...interface{}) {
// There are some escape sequences to format color in terminal, so cannot
// just trim new line from right.
msg = strings.Replace(msg, "\n", "", -1)
msg = strings.TrimSpace(msg)
if len(msg) == 0 {
return
}
// TODO: filter msg by regex
msg = msg + "\n"
if cfg.AddTime {
t := time.Now().Format("15:04:05")
msg = fmt.Sprintf("[%s] %s", t, msg)
}
if colorname == rawColor {
fmt.Fprintf(os.Stdout, msg, v...)
} else {
color.New(getColor(colorname)).Fprintf(color.Output, msg, v...)
}
}
}
func getColor(name string) color.Attribute {
if v, ok := colorMap[name]; ok {
return v
}
return color.FgWhite
}
func (l *logger) main() logFunc {
return l.getLogger("main")
}
func (l *logger) build() logFunc {
return l.getLogger("build")
}
func (l *logger) runner() logFunc {
return l.getLogger("runner")
}
func (l *logger) watcher() logFunc {
return l.getLogger("watcher")
}
func rawLogger() logFunc {
return newLogFunc("raw", defaultConfig().Log)
}
func defaultLogger() logFunc {
return newLogFunc("white", defaultConfig().Log)
}
func (l *logger) getLogger(name string) logFunc {
v, ok := l.loggers[name]
if !ok {
return rawLogger()
}
return v
}