forked from cosmtrek/air
/
logger.go
107 lines (89 loc) · 1.97 KB
/
logger.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
package runner
import (
"fmt"
"time"
"github.com/fatih/color"
)
// TODO: support more colors
var 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 {
colors map[string]string
loggers map[string]logFunc
}
func newLogger(cfg *config) *logger {
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{
colors: colors,
loggers: loggers,
}
}
func newLogFunc(nameColor string, cfg cfgLog) logFunc {
return func(msg string, v ...interface{}) {
if msg[len(msg)-1:] != "\n" {
msg = msg + "\n"
}
if cfg.AddTime {
t := time.Now().Format("15:04:05.000")
msg = fmt.Sprintf("[%s] %s", t, msg)
}
color.New(getColor(nameColor)).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 (l *logger) app() logFunc {
return l.getLogger("app")
}
func defaultLogger() logFunc {
return newLogFunc("white", cfgLog{AddTime: true})
}
func (l *logger) getLogger(name string) logFunc {
v, ok := l.loggers[name]
if !ok {
dft, _ := l.loggers["default"]
return dft
}
return v
}
type appLogWriter struct {
l logFunc
}
func newAppLogWriter(l *logger) appLogWriter {
return appLogWriter{
l: l.app(),
}
}
func (w appLogWriter) Write(data []byte) (n int, err error) {
w.l(string(data))
return len(data), nil
}