cmd: Make --progress work in git bash on Windows - fixes #3531

This detects the presence of a VT100 terminal by using the TERM
environment variable and switches to using VT100 codes directly under
windows if it is found.

This makes --progress work correctly with git bash.
This commit is contained in:
Nick Craig-Wood 2019-09-13 15:24:47 +01:00
parent 48e23d8c85
commit e09a4ff019
3 changed files with 31 additions and 10 deletions

View File

@ -23,10 +23,29 @@ const (
logTimeFormat = "2006-01-02 15:04:05" logTimeFormat = "2006-01-02 15:04:05"
) )
var (
initTerminal func() error
writeToTerminal func([]byte)
)
// Initialise the VT100 terminal
func initTerminalVT100() error {
return nil
}
// Write to the VT100 terminal
func writeToTerminalVT100(b []byte) {
_, _ = os.Stdout.Write(b)
}
// startProgress starts the progress bar printing // startProgress starts the progress bar printing
// //
// It returns a func which should be called to stop the stats. // It returns a func which should be called to stop the stats.
func startProgress() func() { func startProgress() func() {
if os.Getenv("TERM") != "" {
initTerminal = initTerminalVT100
writeToTerminal = writeToTerminalVT100
}
err := initTerminal() err := initTerminal()
if err != nil { if err != nil {
fs.Errorf(nil, "Failed to start progress: %v", err) fs.Errorf(nil, "Failed to start progress: %v", err)

View File

@ -2,12 +2,8 @@
package cmd package cmd
import "os" func init() {
// Default terminal is VT100 for non Windows
func initTerminal() error { initTerminal = initTerminalVT100
return nil writeToTerminal = writeToTerminalVT100
}
func writeToTerminal(b []byte) {
_, _ = os.Stdout.Write(b)
} }

View File

@ -16,7 +16,13 @@ var (
ansiParser *ansiterm.AnsiParser ansiParser *ansiterm.AnsiParser
) )
func initTerminal() error { func init() {
// Default terminal is Windows console for Windows
initTerminal = initTerminalWindows
writeToTerminal = writeToTerminalWindows
}
func initTerminalWindows() error {
winEventHandler := winterm.CreateWinEventHandler(os.Stdout.Fd(), os.Stdout) winEventHandler := winterm.CreateWinEventHandler(os.Stdout.Fd(), os.Stdout)
if winEventHandler == nil { if winEventHandler == nil {
err := syscall.GetLastError() err := syscall.GetLastError()
@ -29,7 +35,7 @@ func initTerminal() error {
return nil return nil
} }
func writeToTerminal(b []byte) { func writeToTerminalWindows(b []byte) {
// Remove all non-ASCII characters until this is fixed // Remove all non-ASCII characters until this is fixed
// https://github.com/Azure/go-ansiterm/issues/26 // https://github.com/Azure/go-ansiterm/issues/26
r := []rune(string(b)) r := []rune(string(b))