From 089ac467f881a0f21b0161b5aba54f5d1ee148a0 Mon Sep 17 00:00:00 2001
From: Chris Miller <chris@cmiller.me>
Date: Fri, 17 Apr 2020 02:16:20 +0000
Subject: [PATCH] Allow changing focus in handler

---
 treeview.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/treeview.go b/treeview.go
index 6e43477..4e0e3f5 100644
--- a/treeview.go
+++ b/treeview.go
@@ -742,6 +742,7 @@ func (t *TreeView) MouseHandler() func(action MouseAction, event *tcell.EventMou
 
 		switch action {
 		case MouseLeftClick:
+			setFocus(t)
 			_, rectY, _, _ := t.GetInnerRect()
 			y -= rectY
 			if y >= 0 && y < len(t.nodes) {
@@ -758,7 +759,6 @@ func (t *TreeView) MouseHandler() func(action MouseAction, event *tcell.EventMou
 				}
 			}
 			consumed = true
-			setFocus(t)
 		case MouseScrollUp:
 			t.movement = treeUp
 			consumed = true