forked from toolshed/abra
		
	We were running behind and there were quite some deprecations to update. This was mostly in the upstream copy/pasta package but seems quite minimal.
		
			
				
	
	
		
			88 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			88 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| //go:build windows
 | |
| // +build windows
 | |
| 
 | |
| package term
 | |
| 
 | |
| import (
 | |
| 	"os"
 | |
| 
 | |
| 	"golang.org/x/sys/windows"
 | |
| )
 | |
| 
 | |
| type state struct {
 | |
| 	Mode uint32
 | |
| }
 | |
| 
 | |
| func isTerminal(fd uintptr) bool {
 | |
| 	var st uint32
 | |
| 	err := windows.GetConsoleMode(windows.Handle(fd), &st)
 | |
| 	return err == nil
 | |
| }
 | |
| 
 | |
| func makeRaw(fd uintptr) (*State, error) {
 | |
| 	var st uint32
 | |
| 	if err := windows.GetConsoleMode(windows.Handle(fd), &st); err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 	raw := st &^ (windows.ENABLE_ECHO_INPUT | windows.ENABLE_PROCESSED_INPUT | windows.ENABLE_LINE_INPUT | windows.ENABLE_PROCESSED_OUTPUT)
 | |
| 	raw |= windows.ENABLE_VIRTUAL_TERMINAL_INPUT
 | |
| 	if err := windows.SetConsoleMode(windows.Handle(fd), raw); err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 	return &State{state{st}}, nil
 | |
| }
 | |
| 
 | |
| func setState(fd uintptr, state *State) error {
 | |
| 	var mode uint32
 | |
| 	if state != nil {
 | |
| 		mode = state.Mode
 | |
| 	}
 | |
| 	return windows.SetConsoleMode(windows.Handle(fd), mode)
 | |
| }
 | |
| 
 | |
| func getState(fd uintptr) (*State, error) {
 | |
| 	var st uint32
 | |
| 	if err := windows.GetConsoleMode(windows.Handle(fd), &st); err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 	return &State{state{st}}, nil
 | |
| }
 | |
| 
 | |
| func restore(fd uintptr, state *State) error {
 | |
| 	return windows.SetConsoleMode(windows.Handle(fd), state.Mode)
 | |
| }
 | |
| 
 | |
| func getSize(fd uintptr) (width, height int, err error) {
 | |
| 	var info windows.ConsoleScreenBufferInfo
 | |
| 	if err := windows.GetConsoleScreenBufferInfo(windows.Handle(fd), &info); err != nil {
 | |
| 		return 0, 0, err
 | |
| 	}
 | |
| 	return int(info.Window.Right - info.Window.Left + 1), int(info.Window.Bottom - info.Window.Top + 1), nil
 | |
| }
 | |
| 
 | |
| func readPassword(fd uintptr) ([]byte, error) {
 | |
| 	var st uint32
 | |
| 	if err := windows.GetConsoleMode(windows.Handle(fd), &st); err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 	old := st
 | |
| 
 | |
| 	st &^= (windows.ENABLE_ECHO_INPUT | windows.ENABLE_LINE_INPUT)
 | |
| 	st |= (windows.ENABLE_PROCESSED_OUTPUT | windows.ENABLE_PROCESSED_INPUT)
 | |
| 	if err := windows.SetConsoleMode(windows.Handle(fd), st); err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 
 | |
| 	defer windows.SetConsoleMode(windows.Handle(fd), old)
 | |
| 
 | |
| 	var h windows.Handle
 | |
| 	p, _ := windows.GetCurrentProcess()
 | |
| 	if err := windows.DuplicateHandle(p, windows.Handle(fd), p, &h, 0, false, windows.DUPLICATE_SAME_ACCESS); err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 
 | |
| 	f := os.NewFile(uintptr(h), "stdin")
 | |
| 	defer f.Close()
 | |
| 	return readPasswordLine(f)
 | |
| }
 |