|
| 1 | +//go:build windows |
| 2 | + |
1 | 3 | package cli |
2 | 4 |
|
3 | 5 | import ( |
4 | | - "syscall" |
| 6 | + "errors" |
| 7 | + |
| 8 | + "golang.org/x/sys/windows" |
5 | 9 | ) |
6 | 10 |
|
7 | 11 | type terminalState struct { |
8 | | - mode uint32 |
| 12 | + handle windows.Handle |
| 13 | + mode uint32 |
9 | 14 | } |
10 | 15 |
|
11 | 16 | func makeRaw(fd int) (*terminalState, error) { |
12 | | - handle := syscall.Handle(fd) |
| 17 | + handle, err := windows.GetStdHandle(windows.STD_INPUT_HANDLE) |
| 18 | + if err != nil { |
| 19 | + return nil, err |
| 20 | + } |
13 | 21 | var original uint32 |
14 | | - if err := syscall.GetConsoleMode(handle, &original); err != nil { |
| 22 | + if err := windows.GetConsoleMode(handle, &original); err != nil { |
| 23 | + if errors.Is(err, windows.ERROR_INVALID_HANDLE) { |
| 24 | + return nil, nil |
| 25 | + } |
15 | 26 | return nil, err |
16 | 27 | } |
17 | 28 |
|
18 | 29 | raw := original |
19 | | - raw &^= syscall.ENABLE_ECHO_INPUT |
20 | | - raw &^= syscall.ENABLE_LINE_INPUT |
21 | | - raw &^= syscall.ENABLE_PROCESSED_INPUT |
22 | | - raw |= syscall.ENABLE_VIRTUAL_TERMINAL_INPUT |
| 30 | + raw &^= windows.ENABLE_ECHO_INPUT |
| 31 | + raw &^= windows.ENABLE_LINE_INPUT |
| 32 | + raw &^= windows.ENABLE_PROCESSED_INPUT |
| 33 | + raw |= windows.ENABLE_VIRTUAL_TERMINAL_INPUT |
23 | 34 |
|
24 | | - if err := syscall.SetConsoleMode(handle, raw); err != nil { |
| 35 | + if err := windows.SetConsoleMode(handle, raw); err != nil { |
25 | 36 | return nil, err |
26 | 37 | } |
27 | 38 |
|
28 | | - return &terminalState{mode: original}, nil |
| 39 | + return &terminalState{handle: handle, mode: original}, nil |
29 | 40 | } |
30 | 41 |
|
31 | 42 | func restore(fd int, state *terminalState) { |
32 | 43 | if state == nil { |
33 | 44 | return |
34 | 45 | } |
35 | | - handle := syscall.Handle(fd) |
36 | | - _ = syscall.SetConsoleMode(handle, state.mode) |
| 46 | + _ = windows.SetConsoleMode(state.handle, state.mode) |
37 | 47 | } |
0 commit comments