From 304e432e330d244283fff0da146e5d6ee1245504 Mon Sep 17 00:00:00 2001
From: skanehira <sho19921005@gmail.com>
Date: Wed, 30 Oct 2019 10:49:57 +0900
Subject: [PATCH] add expand keybindings

---
 gui/tree.go | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/gui/tree.go b/gui/tree.go
index bd600c7..8e64ee8 100644
--- a/gui/tree.go
+++ b/gui/tree.go
@@ -92,4 +92,21 @@ func (t *Tree) SetKeybindings() {
 			node.SetExpanded(!node.IsExpanded())
 		}
 	})
+
+	t.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
+		switch event.Rune() {
+		case 'h':
+			node := t.GetCurrentNode()
+			if node != nil {
+				node.SetExpanded(false)
+			}
+		case 'l':
+			node := t.GetCurrentNode()
+			if node != nil {
+				node.SetExpanded(true)
+			}
+		}
+
+		return event
+	})
 }