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