add expand keybindings

5 years ago · d3307f7342
parent f91a6e2074
commit d3307f7342
1 changed files with 6 additions and 8 deletions
--- a/gui/tree.go
+++ b/gui/tree.go
@ -96,15 +96,13 @@ func (t *Tree) SetKeybindings(g *Gui) {
 	t.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
 		switch event.Rune() {
 		case 'h':
-			node := t.GetCurrentNode()
-			if node != nil {
-				node.SetExpanded(false)
-			}
+			t.GetCurrentNode().SetExpanded(false)
+		case 'H':
+			t.GetRoot().CollapseAll()
+		case 'L':
+			t.GetRoot().ExpandAll()
 		case 'l':
-			node := t.GetCurrentNode()
-			if node != nil {
-				node.SetExpanded(true)
-			}
+			t.GetCurrentNode().SetExpanded(true)
 		}

 		return event