adds collapse/expend all

jsonui.go

@@ -115,6 +115,12 @@ func main() {
 	if err := g.SetKeybinding(treeView, 'e', gocui.ModNone, toggleExpand); err != nil {
 		log.Panicln(err)
 	}
+	if err := g.SetKeybinding(treeView, 'E', gocui.ModNone, expandAll); err != nil {
+		log.Panicln(err)
+	}
+	if err := g.SetKeybinding(treeView, 'C', gocui.ModNone, collapseAll); err != nil {
+		log.Panicln(err)
+	}
 	if err := g.SetKeybinding("", 'h', gocui.ModNone, toggleHelp); err != nil {
 		log.Panicln(err)
 	}
@@ -137,6 +143,8 @@ k/ArrowUp 		═ 	Move a line up
 J/PageDown		═ 	Move 15 line down
 K/PageUp 		═ 	Move 15 line up
 e				═ 	Toggle expend/collapse node
+E				═ 	Expand all nodes
+C				═ 	Collapse all nodes
 h/?				═ 	Toggle help message
 `
 
@@ -288,6 +296,18 @@ func findTreePosition(v *gocui.View) treePosition {
 	return path[1:]
 }
 
+func expandAll(g *gocui.Gui, v *gocui.View) error {
+	tree.expandAll()
+	v.Clear()
+	tree.draw(v, 2, 0)
+	return nil
+}
+func collapseAll(g *gocui.Gui, v *gocui.View) error {
+	tree.collapseAll()
+	v.Clear()
+	tree.draw(v, 2, 0)
+	return nil
+}
 func toggleExpand(g *gocui.Gui, v *gocui.View) error {
 	tv, err := g.View(treeView)
 	if err != nil {

tree.go

@@ -45,6 +45,8 @@ type treeNode interface {
 	search(query string) (treeNode, error)
 	isCollapsable() bool
 	toggleExpanded()
+	collapseAll()
+	expandAll()
 	isExpanded() bool
 }
 
@@ -72,6 +74,18 @@ type complexNode struct {
 	data map[string]treeNode
 }
 
+func (n *complexNode) collapseAll() {
+	n.expanded = false
+	for _, v := range n.data {
+		v.collapseAll()
+	}
+}
+func (n *complexNode) expandAll() {
+	n.expanded = true
+	for _, v := range n.data {
+		v.expandAll()
+	}
+}
 func (n complexNode) isCollapsable() bool {
 	return true
 }
@@ -169,6 +183,18 @@ type listNode struct {
 	data []treeNode
 }
 
+func (n *listNode) collapseAll() {
+	n.expanded = false
+	for _, v := range n.data {
+		v.collapseAll()
+	}
+}
+func (n *listNode) expandAll() {
+	n.expanded = true
+	for _, v := range n.data {
+		v.expandAll()
+	}
+}
 func (n listNode) isCollapsable() bool {
 	return true
 }
@@ -248,6 +274,10 @@ type floatNode struct {
 	data float64
 }
 
+func (n *floatNode) collapseAll() {
+}
+func (n *floatNode) expandAll() {
+}
 func (n floatNode) isCollapsable() bool {
 	return false
 }
@@ -277,6 +307,11 @@ type stringNode struct {
 	data string
 }
 
+func (n *stringNode) collapseAll() {
+}
+func (n *stringNode) expandAll() {
+}
+
 func (n stringNode) isCollapsable() bool {
 	return false
 }