added functionality to interrupt the terminal when it is waiting for an input

Signed-off-by: Avi Vaid <avaid1996@gmail.com>
This commit is contained in:
Avi Vaid 2016-08-04 09:35:29 -07:00
parent e10f6a2e54
commit c4b07f4683
1 changed files with 9 additions and 3 deletions

View File

@ -6,6 +6,7 @@ package term
import (
"errors"
"fmt"
"io"
"os"
"os/signal"
@ -109,9 +110,14 @@ func SetRawTerminalOutput(fd uintptr) (*State, error) {
func handleInterrupt(fd uintptr, state *State) {
sigchan := make(chan os.Signal, 1)
signal.Notify(sigchan, os.Interrupt)
go func() {
_ = <-sigchan
RestoreTerminal(fd, state)
for range sigchan {
// quit cleanly and the new terminal item is on a new line
fmt.Println()
signal.Stop(sigchan)
close(sigchan)
RestoreTerminal(fd, state)
os.Exit(1)
}
}()
}