cmd/listen: Exit when stdin is closed (#1155)

Co-authored-by: Ian Jabour <[email protected]>

pkg/cmd/listen.go

@@ -1,6 +1,7 @@
 package cmd
 
 import (
+	"bufio"
 	"context"
 	"fmt"
 	"net/url"
@@ -130,6 +131,8 @@ func (lc *listenCmd) runListenCmd(cmd *cobra.Command, args []string) error {
 		}).Debug("Ctrl+C received, cleaning up...")
 	})
 
+	ctx = withStdinCloseCancel(ctx)
+
 	client := &stripe.Client{
 		APIKey:  key,
 		BaseURL: apiBase,
@@ -198,6 +201,20 @@ func withSIGTERMCancel(ctx context.Context, onCancel func()) context.Context {
 	return ctx
 }
 
+func withStdinCloseCancel(ctx context.Context) context.Context {
+	ctx, cancel := context.WithCancel(ctx)
+
+	go func() {
+		scanner := bufio.NewScanner(os.Stdin)
+		for scanner.Scan() {
+		}
+		if scanner.Err() == nil {
+			cancel()
+		}
+	}()
+	return ctx
+}
+
 func createVisitor(logger *log.Logger, format string, printJSON bool) *websocket.Visitor {
 	var s *spinner.Spinner