Better logic in input checks

tea.go

@@ -272,29 +272,35 @@ func (p *Program) Start() error {
 		p.outputIsTTY = isatty.IsTerminal(f.Fd())
 	}
 
+	switch {
+	case p.startupOptions.has(withInputTTY):
 		// Open a new TTY, by request
-	if p.startupOptions.has(withInputTTY) {
 		f, err := openInputTTY()
 		if err != nil {
 			return err
 		}
 		p.input = f
+
+	case !p.startupOptions.has(withCustomInput):
+		// If the user hasn't set a custom input, and input's not a terminal,
+		// open a TTY so we can capture input as normal. This will allow things
+		// to "just work" in cases where data was piped or redirected into this
+		// application.
+		f, isFile := p.input.(*os.File)
+		if !isFile {
+			break
 		}
 
-	// If input is not a terminal, and the user hasn't set a custom input, open
-	// a TTY so we can capture input as normal. This will allow things to "just
-	// work" in cases where data was piped or redirected into this application.
-	if f, ok := p.input.(*os.File); ok {
-		inputIsTTY := isatty.IsTerminal(f.Fd())
+		if isatty.IsTerminal(f.Fd()) {
+			break
+		}
 
-		if !inputIsTTY && !p.startupOptions.has(withCustomInput) {
 		f, err := openInputTTY()
 		if err != nil {
 			return err
 		}
 		p.input = f
 	}
-	}
 
 	// Listen for SIGINT. Note that in most cases ^C will not send an
 	// interrupt because the terminal will be in raw mode and thus capture