45 lines
770 B
Go
45 lines
770 B
Go
package d2term
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"io"
|
|
"strings"
|
|
)
|
|
|
|
type terminalLogger struct {
|
|
terminal *terminal
|
|
buffer bytes.Buffer
|
|
writer io.Writer
|
|
}
|
|
|
|
func (tl *terminalLogger) Write(p []byte) (int, error) {
|
|
n, err := tl.buffer.Write(p)
|
|
if err != nil {
|
|
return n, err
|
|
}
|
|
|
|
reader := bufio.NewReader(&tl.buffer)
|
|
termBytes, _, err := reader.ReadLine()
|
|
if err != nil {
|
|
return n, err
|
|
}
|
|
|
|
line := string(termBytes[:])
|
|
lineLower := strings.ToLower(line)
|
|
|
|
if strings.Index(lineLower, "error") > 0 {
|
|
tl.terminal.OutputError(line)
|
|
} else if strings.Index(lineLower, "warning") > 0 {
|
|
tl.terminal.OutputWarning(line)
|
|
} else {
|
|
tl.terminal.Output(line)
|
|
}
|
|
|
|
return tl.writer.Write(p)
|
|
}
|
|
|
|
func (tl *terminalLogger) BindToTerminal(t *terminal) {
|
|
tl.terminal = t
|
|
}
|