Here have code

1 files changed, 10 insertions, 0 deletions

diff --git a/gomuks.go b/gomuks.go
index e87da59..13b6672 100644
--- a/gomuks.go
+++ b/gomuks.go
@@ -34,6 +34,7 @@ type Gomuks interface {
 
 	Start()
 	Stop()
+	Recover()
 }
 
 type gomuks struct {
@@ -79,6 +80,15 @@ func (gmx *gomuks) Stop() {
 	}
 }
 
+func (gmx *gomuks) Recover() {
+	if p := recover(); p != nil {
+		if gmx.App().GetScreen() != nil {
+			gmx.App().GetScreen().Fini()
+		}
+		panic(p)
+	}
+}
+
 func (gmx *gomuks) Start() {
 	if err := gmx.app.Run(); err != nil {
 		panic(err)