OpenDiablo2/d2core/d2term/terminal_logger.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
}