From 36e5a32e78ca1c40a825824c62e7ad68a8099c14 Mon Sep 17 00:00:00 2001
From: Adam Tauber <asciimoo@gmail.com>
Date: Sun, 5 Feb 2017 12:02:36 +0100
Subject: [PATCH] [fix] exit if terminal is too small - closes #5

---
 wuzz.go | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/wuzz.go b/wuzz.go
index c7a6335..eea3b97 100644
--- a/wuzz.go
+++ b/wuzz.go
@@ -544,7 +544,10 @@ func (a *App) restoreRequest(g *gocui.Gui, idx int) {
 func (a *App) ParseArgs(g *gocui.Gui) error {
 	a.Layout(g)
 	g.SetCurrentView(VIEWS[a.viewIndex])
-	vheader, _ := g.View("headers")
+	vheader, err := g.View("headers")
+	if err != nil {
+		return errors.New("Too small screen")
+	}
 	vheader.Clear()
 	vget, _ := g.View("get")
 	vget.Clear()
@@ -676,7 +679,6 @@ func main() {
 	if err != nil {
 		log.Panicln(err)
 	}
-	defer g.Close()
 
 	app := &App{history: make([]*Request, 0, 31)}
 
@@ -685,7 +687,15 @@ func main() {
 
 	initApp(app, g)
 
-	app.ParseArgs(g)
+	err = app.ParseArgs(g)
+
+	if err != nil {
+		g.Close()
+		fmt.Println("Error!", err)
+		os.Exit(1)
+	}
+
+	defer g.Close()
 
 	if err := g.MainLoop(); err != nil && err != gocui.ErrQuit {
 		log.Panicln(err)