Fixed CTRL+C behavior

claudiodangelis · Apr 19, 2021 · 581e917 · 581e917
2 parents 017defe + b510e15
commit 581e917
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/cmd/receive.go b/cmd/receive.go
@@ -53,8 +53,8 @@ func receiveCmdFunc(command *cobra.Command, args []string) error {
 	}()
 	go func() {
 		for {
-			char, _, _ := keyboard.GetKey()
-			if string(char) == "q" {
+			char, key, _ := keyboard.GetKey()
+			if string(char) == "q" || key == keyboard.KeyCtrlC {
 				srv.Shutdown()
 			}
 		}

diff --git a/cmd/send.go b/cmd/send.go
@@ -53,8 +53,8 @@ func sendCmdFunc(command *cobra.Command, args []string) error {
 	}()
 	go func() {
 		for {
-			char, _, _ := keyboard.GetKey()
-			if string(char) == "q" {
+			char, key, _ := keyboard.GetKey()
+			if string(char) == "q" || key == keyboard.KeyCtrlC {
 				srv.Shutdown()
 			}
 		}