chore: store handlers and simplify teardown

tea.go

@@ -16,6 +16,7 @@ import (
 	"os"
 	"os/signal"
 	"runtime/debug"
+	"sync"
 	"syscall"
 	"time"
 
@@ -178,6 +179,7 @@ func (p *Program) handleSignals() chan struct{} {
 			select {
 			case <-p.ctx.Done():
 				return
+
 			case <-sig:
 				if !p.ignoreSignals {
 					p.msgs <- quitMsg{}
@@ -332,6 +334,7 @@ func (p *Program) eventLoop(model Model, cmds chan Cmd) (Model, error) {
 
 // StartReturningModel initializes the program. Returns the final model.
 func (p *Program) StartReturningModel() (Model, error) {
+	var handlers []chan struct{}
 	cmds := make(chan Cmd)
 	p.errs = make(chan error)
 
@@ -360,7 +363,6 @@ func (p *Program) StartReturningModel() (Model, error) {
 		if !isFile {
 			break
 		}
-
 		if isatty.IsTerminal(f.Fd()) {
 			break
 		}
@@ -376,11 +378,8 @@ func (p *Program) StartReturningModel() (Model, error) {
 	}
 
 	// Handle signals.
-	sigintLoopDone := make(chan struct{})
 	if !p.startupOptions.has(withoutSignalHandler) {
-		sigintLoopDone = p.handleSignals()
-	} else {
-		close(sigintLoopDone)
+		handlers = append(handlers, p.handleSignals())
 	}
 
 	// Recover from panics.
@@ -417,18 +416,19 @@ func (p *Program) StartReturningModel() (Model, error) {
 	}
 
 	// Initialize the program.
-	initSignalDone := make(chan struct{})
 	model := p.initialModel
 	if initCmd := model.Init(); initCmd != nil {
+		ch := make(chan struct{})
+		handlers = append(handlers, ch)
+
 		go func() {
-			defer close(initSignalDone)
+			defer close(ch)
+
 			select {
 			case cmds <- initCmd:
 			case <-p.ctx.Done():
 			}
 		}()
-	} else {
-		close(initSignalDone)
 	}
 
 	// Start the renderer.
@@ -442,16 +442,16 @@ func (p *Program) StartReturningModel() (Model, error) {
 		if err := p.initCancelReader(); err != nil {
 			return model, err
 		}
-	} else {
-		defer close(p.readLoopDone)
+		defer p.cancelReader.Close() //nolint:errcheck
+
+		handlers = append(handlers, p.readLoopDone)
 	}
-	defer p.cancelReader.Close() //nolint:errcheck
 
 	// Handle resize events.
-	resizeLoopDone := p.handleResize()
+	handlers = append(handlers, p.handleResize())
 
 	// Process commands.
-	cmdLoopDone := p.handleCommands(cmds)
+	handlers = append(handlers, p.handleCommands(cmds))
 
 	// Run event loop, handle updates and draw.
 	model, err := p.eventLoop(model, cmds)
@@ -463,10 +463,19 @@ func (p *Program) StartReturningModel() (Model, error) {
 	if p.cancelReader.Cancel() {
 		p.waitForReadLoop()
 	}
-	<-cmdLoopDone
-	<-resizeLoopDone
-	<-sigintLoopDone
-	<-initSignalDone
+
+	// Wait for all handlers to finish.
+	var wg sync.WaitGroup
+	for _, ch := range handlers {
+		wg.Add(1)
+		go func(ch chan struct{}) {
+			<-ch
+			wg.Done()
+		}(ch)
+	}
+	wg.Wait()
+
+	// Restore terminal state.
 	p.shutdown(false)
 
 	return model, err