Cancel process on interrupt

Author: AdamSLevy

cmd/godotenv/cmd.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"flag"
 	"fmt"
 	"log"
@@ -54,12 +55,16 @@ example
 	cmd := args[0]
 	cmdArgs := args[1:]
 
-	command := exec.Command(cmd, cmdArgs...)
+	ctx, cancel := context.WithCancel(context.Background())
+	notify := make(chan os.Signal, 1)
+	signal.Notify(notify, os.Interrupt)
+	go func() { <-notify; cancel() }()
+
+	command := exec.CommandContext(ctx, cmd, cmdArgs...)
 	command.Stdin = os.Stdin
 	command.Stdout = os.Stdout
 	command.Stderr = os.Stderr
 
-	signal.Ignore(os.Interrupt)
 	if err := command.Run(); err != nil {
 		log.Fatal(err)
 	}