forked from Mirrors/bubbletea
67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
//go:build windows
|
|
// +build windows
|
|
|
|
package tea
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/containerd/console"
|
|
"golang.org/x/sys/windows"
|
|
)
|
|
|
|
func (p *Program) initInput() error {
|
|
// If input's a file, use console to manage it
|
|
if f, ok := p.input.(*os.File); ok {
|
|
// Save a reference to the current stdin then replace stdin with our
|
|
// input. We do this so we can hand input off to containerd/console to
|
|
// set raw mode, and do it in this fashion because the method
|
|
// console.ConsoleFromFile isn't supported on Windows.
|
|
p.windowsStdin = os.Stdin
|
|
os.Stdin = f
|
|
|
|
// Note: this will panic if it fails.
|
|
c := console.Current()
|
|
p.console = c
|
|
}
|
|
|
|
enableAnsiColors(p.output)
|
|
|
|
return nil
|
|
}
|
|
|
|
// restoreInput restores stdout in the event that we placed it aside to handle
|
|
// input with CONIN$, above.
|
|
func (p *Program) restoreInput() error {
|
|
if p.windowsStdin != nil {
|
|
os.Stdin = p.windowsStdin
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Open the Windows equivalent of a TTY.
|
|
func openInputTTY() (*os.File, error) {
|
|
f, err := os.OpenFile("CONIN$", os.O_RDWR, 0644)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return f, nil
|
|
}
|
|
|
|
// enableAnsiColors enables support for ANSI color sequences in Windows
|
|
// default console. Note that this only works with Windows 10.
|
|
func enableAnsiColors(w io.Writer) {
|
|
f, ok := w.(*os.File)
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
stdout := windows.Handle(f.Fd())
|
|
var originalMode uint32
|
|
|
|
_ = windows.GetConsoleMode(stdout, &originalMode)
|
|
_ = windows.SetConsoleMode(stdout, originalMode|windows.ENABLE_VIRTUAL_TERMINAL_PROCESSING)
|
|
}
|