-
Notifications
You must be signed in to change notification settings - Fork 3
/
logger.go
106 lines (87 loc) · 2.27 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
package io
import (
"bytes"
"fmt"
"io"
"os/exec"
"strings"
)
type Logger struct {
out io.Writer
enableColors bool
debug bool
}
func NewLogger(out io.Writer, enableColors bool, debug bool) *Logger {
return &Logger{
out: out,
enableColors: enableColors,
debug: debug,
}
}
func NewTestLogger() (*Logger, *bytes.Buffer) {
out := new(bytes.Buffer)
return NewLogger(out, false, false), out
}
func (logger *Logger) Debug(a ...interface{}) (n int, err error) {
if logger.debug {
logger.Write([]byte("DEBUG: "))
return logger.Println(a...)
}
return 0, nil
}
func (logger *Logger) Debugf(format string, a ...interface{}) (n int, err error) {
if logger.debug {
logger.Write([]byte("DEBUG: "))
return logger.Printf(format, a...)
}
return 0, nil
}
func (logger *Logger) Write(p []byte) (n int, err error) {
return logger.out.Write(p)
}
func (logger *Logger) Println(a ...interface{}) (n int, err error) {
return fmt.Fprintln(logger.out, a...)
}
func (logger *Logger) Printf(format string, a ...interface{}) (n int, err error) {
return fmt.Fprintf(logger.out, format, a...)
}
func (logger *Logger) Printfln(format string, a ...interface{}) (n int, err error) {
return fmt.Fprintf(logger.out, format+"\n", a...)
}
func (logger *Logger) PrintStatusErrors(errors []string, firstLineOnly bool) {
for _, err := range errors {
message := err
if firstLineOnly {
message = strings.Split(message, "\n")[0]
}
logger.Printfln(" ► %s", message)
}
}
// PrettyPrintDiff - prints a diff using syntax highlighting if bat is available
func (logger *Logger) PrettyPrintDiff(patch string) {
logger.prettyPrint(patch, "diff")
}
func (logger *Logger) prettyPrint(code string, language string) {
if _, err := exec.LookPath("bat"); err != nil {
logger.Println(code)
return
}
color := "never"
if logger.enableColors {
color = "always"
}
cmd := exec.Command("bat", "--language", language, "--color", color, "--style", "plain")
cmd.Stdin = strings.NewReader(code)
var out bytes.Buffer
cmd.Stdout = &out
var stderr bytes.Buffer
cmd.Stderr = &stderr
if err := cmd.Run(); err != nil {
logger.Println("error running cmd")
logger.Println(stderr.String())
logger.Println(err)
logger.Println(code)
return
}
logger.Println(out.String())
}