From bf49ccb878069e853485b8fcb9f5406ba1c6e28e Mon Sep 17 00:00:00 2001
From: skanehira <sho19921005@gmail.com>
Date: Wed, 6 Nov 2019 12:24:51 +0900
Subject: [PATCH] use `H` to collapse value nodes

---
 gui/tree.go | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/gui/tree.go b/gui/tree.go
index ffe5323..68d2de6 100644
--- a/gui/tree.go
+++ b/gui/tree.go
@@ -139,7 +139,7 @@ func (t *Tree) SetKeybindings(g *Gui) {
 		case 'h':
 			t.GetCurrentNode().SetExpanded(false)
 		case 'H':
-			t.GetRoot().CollapseAll()
+			t.CollapseValues(t.GetRoot())
 		case 'd':
 			t.GetCurrentNode().ClearChildren()
 			newRoot := *g.Tree.GetRoot()
@@ -176,6 +176,20 @@ func (t *Tree) SetKeybindings(g *Gui) {
 	})
 }
 
+func (t *Tree) CollapseValues(node *tview.TreeNode) {
+	node.Walk(func(node, parent *tview.TreeNode) bool {
+		ref := node.GetReference().(Reference)
+		if ref.JSONType == Value {
+			pRef := parent.GetReference().(Reference)
+			t := pRef.JSONType
+			if t == Key || t == Array {
+				parent.SetExpanded(false)
+			}
+		}
+		return true
+	})
+}
+
 func (t *Tree) moveParent(movement int) {
 	current := t.GetCurrentNode()
 	t.GetRoot().Walk(func(node, parent *tview.TreeNode) bool {