61 lines
1.4 KiB
Go
61 lines
1.4 KiB
Go
// +build windows
|
|
|
|
package windows
|
|
|
|
import (
|
|
"os"
|
|
|
|
ansiterm "github.com/Azure/go-ansiterm"
|
|
"github.com/Azure/go-ansiterm/winterm"
|
|
)
|
|
|
|
// ansiWriter wraps a standard output file (e.g., os.Stdout) providing ANSI sequence translation.
|
|
type ansiWriter struct {
|
|
file *os.File
|
|
fd uintptr
|
|
infoReset *winterm.CONSOLE_SCREEN_BUFFER_INFO
|
|
command []byte
|
|
escapeSequence []byte
|
|
inAnsiSequence bool
|
|
parser *ansiterm.AnsiParser
|
|
}
|
|
|
|
func newAnsiWriter(nFile int) *ansiWriter {
|
|
initLogger()
|
|
file, fd := winterm.GetStdFile(nFile)
|
|
info, err := winterm.GetConsoleScreenBufferInfo(fd)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
parser := ansiterm.CreateParser("Ground", winterm.CreateWinEventHandler(fd, file))
|
|
logger.Infof("newAnsiWriter: parser %p", parser)
|
|
|
|
aw := &ansiWriter{
|
|
file: file,
|
|
fd: fd,
|
|
infoReset: info,
|
|
command: make([]byte, 0, ansiterm.ANSI_MAX_CMD_LENGTH),
|
|
escapeSequence: []byte(ansiterm.KEY_ESC_CSI),
|
|
parser: parser,
|
|
}
|
|
|
|
logger.Infof("newAnsiWriter: aw.parser %p", aw.parser)
|
|
logger.Infof("newAnsiWriter: %v", aw)
|
|
return aw
|
|
}
|
|
|
|
func (aw *ansiWriter) Fd() uintptr {
|
|
return aw.fd
|
|
}
|
|
|
|
// Write writes len(p) bytes from p to the underlying data stream.
|
|
func (aw *ansiWriter) Write(p []byte) (total int, err error) {
|
|
if len(p) == 0 {
|
|
return 0, nil
|
|
}
|
|
|
|
logger.Infof("Write: % x", p)
|
|
logger.Infof("Write: %s", string(p))
|
|
return aw.parser.Parse(p)
|
|
}
|