Gracefully recover from panics (but still print stack traces) by default

This commit is contained in:
Christian Rocha 2020-09-21 13:09:32 -04:00
parent 4609b8afd9
commit fb48e03639
No known key found for this signature in database
GPG Key ID: D6CC7A16E5878018
1 changed files with 20 additions and 1 deletions

19
tea.go
View File

@ -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.
@ -96,6 +103,7 @@ func NewProgram(init Init, update Update, view View) *Program {
view: view,
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()