1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61
|
package term
import (
"os"
"golang.org/x/sys/windows"
"src.elv.sh/pkg/errutil"
)
const (
inMode = windows.ENABLE_WINDOW_INPUT |
windows.ENABLE_MOUSE_INPUT | windows.ENABLE_PROCESSED_INPUT
outMode = windows.ENABLE_PROCESSED_OUTPUT |
windows.ENABLE_WRAP_AT_EOL_OUTPUT |
windows.ENABLE_VIRTUAL_TERMINAL_PROCESSING
)
func setupForTUIOnce(_, _ *os.File) func() {
return func() {}
}
func setupForTUI(in, out *os.File) (func() error, error) {
hIn := windows.Handle(in.Fd())
hOut := windows.Handle(out.Fd())
var oldInMode, oldOutMode uint32
err := windows.GetConsoleMode(hIn, &oldInMode)
if err != nil {
return nil, err
}
err = windows.GetConsoleMode(hOut, &oldOutMode)
if err != nil {
return nil, err
}
errSetIn := windows.SetConsoleMode(hIn, inMode)
errSetOut := windows.SetConsoleMode(hOut, outMode)
errVT := setupVT(out)
return func() error {
return errutil.Multi(
restoreVT(out),
windows.SetConsoleMode(hOut, oldOutMode),
windows.SetConsoleMode(hIn, oldInMode))
}, errutil.Multi(errSetIn, errSetOut, errVT)
}
// We need ENABLE_VIRTUAL_TERMINAL_PROCESSING for styled text to function. This
// includes texts created by the user (with the "styled" builtin) or by Elvish
// itself (like exception stack traces).
const outFlagForEval = windows.ENABLE_VIRTUAL_TERMINAL_PROCESSING
func setupForEval(_, out *os.File) func() {
h := windows.Handle(out.Fd())
var oldOutMode uint32
err := windows.GetConsoleMode(h, &oldOutMode)
if err == nil {
windows.SetConsoleMode(h, oldOutMode|outFlagForEval)
}
return func() {}
}
|