Allow App.Close() and App.Run() to be run concurrently

1 files changed, 7 insertions, 1 deletions

diff --git a/ui/ui.go b/ui/ui.go
index cda5a06..ec1fd09 100644
--- a/ui/ui.go
+++ b/ui/ui.go
@@ -60,8 +60,14 @@ func New(config Config) (ui *UI, err error) {
 	ui.Events = make(chan tcell.Event, 128)
 	go func() {
 		for !ui.ShouldExit() {
-			ui.Events <- ui.screen.PollEvent()
+			ev := ui.screen.PollEvent()
+			if ev == nil {
+				ui.Exit()
+				break
+			}
+			ui.Events <- ev
 		}
+		close(ui.Events)
 	}()
 
 	ui.bs = NewBufferList()