Open a TTY if input is not a TTY, unless the user has spec'd otherwise

tty_windows.go

@@ -6,9 +6,55 @@ import (
 	"io"
 	"os"
 
+	"github.com/containerd/console"
 	"golang.org/x/sys/windows"
 )
 
+func (p *Program) initInput() error {
+	if !p.inputIsTTY {
+		return nil
+	}
+
+	// If input's a TTY this should always succeed.
+	f, ok := p.input.(*os.File)
+	if !ok {
+		return errInputIsNotAFile
+	}
+
+	if p.inputStatus == managedInput {
+		// 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
+
+	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
+}
+
+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) {
@@ -20,6 +66,6 @@ func enableAnsiColors(w io.Writer) {
 	stdout := windows.Handle(f.Fd())
 	var originalMode uint32
 
-	windows.GetConsoleMode(stdout, &originalMode)
-	windows.SetConsoleMode(stdout, originalMode|windows.ENABLE_VIRTUAL_TERMINAL_PROCESSING)
+	_ = windows.GetConsoleMode(stdout, &originalMode)
+	_ = windows.SetConsoleMode(stdout, originalMode|windows.ENABLE_VIRTUAL_TERMINAL_PROCESSING)
 }