38 lines
1.1 KiB
Go
38 lines
1.1 KiB
Go
package readline
|
|
|
|
import (
|
|
"golang.org/x/sys/windows"
|
|
)
|
|
|
|
type State struct {
|
|
mode uint32
|
|
}
|
|
|
|
// IsTerminal checks if the given file descriptor is associated with a terminal
|
|
func IsTerminal(fd int) bool {
|
|
var st uint32
|
|
err := windows.GetConsoleMode(windows.Handle(fd), &st)
|
|
return err == nil
|
|
}
|
|
|
|
func SetRawMode(fd int) (*State, error) {
|
|
var st uint32
|
|
if err := windows.GetConsoleMode(windows.Handle(fd), &st); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// this enables raw mode by turning off various flags in the console mode: https://pkg.go.dev/golang.org/x/sys/windows#pkg-constants
|
|
raw := st &^ (windows.ENABLE_ECHO_INPUT | windows.ENABLE_PROCESSED_INPUT | windows.ENABLE_LINE_INPUT | windows.ENABLE_PROCESSED_OUTPUT)
|
|
|
|
// turn on ENABLE_VIRTUAL_TERMINAL_INPUT to enable escape sequences
|
|
raw |= windows.ENABLE_VIRTUAL_TERMINAL_INPUT
|
|
if err := windows.SetConsoleMode(windows.Handle(fd), raw); err != nil {
|
|
return nil, err
|
|
}
|
|
return &State{st}, nil
|
|
}
|
|
|
|
func UnsetRawMode(fd int, state any) error {
|
|
s := state.(*State)
|
|
return windows.SetConsoleMode(windows.Handle(fd), s.mode)
|
|
}
|