mirror of https://github.com/skanehira/tson
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
94 lines
2.1 KiB
Go
94 lines
2.1 KiB
Go
package gui
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/gdamore/tcell"
|
|
"github.com/rivo/tview"
|
|
)
|
|
|
|
type Tree struct {
|
|
*tview.TreeView
|
|
}
|
|
|
|
func NewTree() *Tree {
|
|
t := &Tree{
|
|
TreeView: tview.NewTreeView(),
|
|
}
|
|
|
|
t.SetBorder(true).SetTitle("json tree").SetTitleAlign(tview.AlignLeft)
|
|
return t
|
|
}
|
|
|
|
func (t *Tree) UpdateView(g *Gui, i interface{}) {
|
|
g.App.QueueUpdateDraw(func() {
|
|
root := tview.NewTreeNode(".")
|
|
t.SetRoot(root).SetCurrentNode(root)
|
|
t.AddNodes(root, t.AddNode(i))
|
|
})
|
|
}
|
|
|
|
func (t *Tree) AddNode(node interface{}) []*tview.TreeNode {
|
|
// e.g child is {"name": "gorilla", "lang": {"ja":"japan", "en": "english"}}
|
|
var nodes []*tview.TreeNode
|
|
|
|
switch node := node.(type) {
|
|
case map[string]interface{}:
|
|
for k, v := range node {
|
|
newNode := t.NewNodeWithLiteral(k).
|
|
SetColor(tcell.ColorMediumSlateBlue).SetReference(k)
|
|
|
|
list, isList := v.([]interface{})
|
|
if isList && len(list) > 0 {
|
|
newNode.SetSelectable(true)
|
|
}
|
|
t.AddNodes(newNode, t.AddNode(v))
|
|
nodes = append(nodes, newNode)
|
|
}
|
|
case []interface{}:
|
|
for i, v := range node {
|
|
if list, isList := v.([]interface{}); isList && len(list) > 0 {
|
|
numberNode := tview.NewTreeNode(fmt.Sprintf("[%d]", i+1))
|
|
t.AddNodes(numberNode, t.AddNode(v))
|
|
nodes = append(nodes, numberNode)
|
|
} else if m, isMap := v.(map[string]interface{}); isMap && len(m) > 0 {
|
|
numberNode := tview.NewTreeNode(fmt.Sprintf("[%d]", i+1))
|
|
t.AddNodes(numberNode, t.AddNode(v))
|
|
nodes = append(nodes, numberNode)
|
|
} else {
|
|
nodes = append(nodes, t.AddNode(v)...)
|
|
}
|
|
}
|
|
default:
|
|
nodes = append(nodes, t.NewNodeWithLiteral(node))
|
|
}
|
|
return nodes
|
|
}
|
|
|
|
func (t *Tree) AddNodes(target *tview.TreeNode, nodes []*tview.TreeNode) {
|
|
for _, node := range nodes {
|
|
target.AddChild(node)
|
|
}
|
|
}
|
|
|
|
func (t *Tree) NewNodeWithLiteral(i interface{}) *tview.TreeNode {
|
|
var text string
|
|
node := tview.NewTreeNode("")
|
|
switch v := i.(type) {
|
|
case int32:
|
|
text = fmt.Sprintf("%d", v)
|
|
case int64:
|
|
text = fmt.Sprintf("%d", v)
|
|
case float32:
|
|
text = fmt.Sprintf("%f", v)
|
|
case float64:
|
|
text = fmt.Sprintf("%f", v)
|
|
case bool:
|
|
text = fmt.Sprintf("%t", v)
|
|
case string:
|
|
text = v
|
|
}
|
|
|
|
return node.SetText(text)
|
|
}
|