From fb48e036396b9dc44699f4aa793e1462751f92ef Mon Sep 17 00:00:00 2001
From: Christian Rocha <christian@rocha.is>
Date: Mon, 21 Sep 2020 13:09:32 -0400
Subject: [PATCH] Gracefully recover from panics (but still print stack traces)
 by default

---
 tea.go | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/tea.go b/tea.go
index 61fffe8..42d9f19 100644
--- a/tea.go
+++ b/tea.go
@@ -12,6 +12,7 @@ package tea
 import (
 	"fmt"
 	"os"
+	"runtime/debug"
 	"sync"
 
 	te "github.com/muesli/termenv"
@@ -66,6 +67,12 @@ type Program struct {
 	output          *os.File // where to send output. this will usually be os.Stdout.
 	renderer        *renderer
 	altScreenActive bool
+
+	// CatchPanics is incredibly useful for restoring the terminal to a useable
+	// state after a panic occurs. When this is set, Bubble Tea will recover
+	// from panics, print the stack trace, and disable raw mode. This feature
+	// is on by default.
+	CatchPanics bool
 }
 
 // Quit is a special command that tells the Bubble Tea program to exit.
@@ -95,7 +102,8 @@ func NewProgram(init Init, update Update, view View) *Program {
 		update: update,
 		view:   view,
 
-		output: os.Stdout,
+		output:      os.Stdout,
+		CatchPanics: true,
 	}
 }
 
@@ -108,6 +116,17 @@ func (p *Program) Start() error {
 		done = make(chan struct{})
 	)
 
+	if p.CatchPanics {
+		defer func() {
+			if r := recover(); r != nil {
+				p.ExitAltScreen()
+				fmt.Print("Caught panic. Restoring terminal...\n\n")
+				debug.PrintStack()
+				return
+			}
+		}()
+	}
+
 	p.renderer = newRenderer(p.output, &p.mtx)
 
 	err := initTerminal()