forked from Archives/tson
make json tree view
parent
203e88ccdf
commit
1d87821121
@ -0,0 +1,38 @@
|
||||
package gui
|
||||
|
||||
import (
|
||||
"log"
|
||||
|
||||
"github.com/rivo/tview"
|
||||
)
|
||||
|
||||
type Gui struct {
|
||||
Tree *Tree
|
||||
App *tview.Application
|
||||
Pages *tview.Pages
|
||||
}
|
||||
|
||||
func New() *Gui {
|
||||
g := &Gui{
|
||||
Tree: NewTree(),
|
||||
App: tview.NewApplication(),
|
||||
Pages: tview.NewPages(),
|
||||
}
|
||||
return g
|
||||
}
|
||||
|
||||
func (g *Gui) Run(i interface{}) error {
|
||||
g.Tree.UpdateView(g, i)
|
||||
|
||||
grid := tview.NewGrid().
|
||||
AddItem(g.Tree, 0, 0, 1, 1, 0, 0, true)
|
||||
|
||||
g.Pages.AddAndSwitchToPage("main", grid, true)
|
||||
|
||||
if err := g.App.SetRoot(g.Pages, true).Run(); err != nil {
|
||||
log.Println(err)
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
@ -0,0 +1,95 @@
|
||||
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)
|
||||
for _, node := range t.AddNode(i) {
|
||||
root.AddChild(node)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
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)
|
||||
}
|
||||
for _, n := range t.AddNode(v) {
|
||||
newNode.AddChild(n)
|
||||
}
|
||||
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))
|
||||
for _, n := range t.AddNode(v) {
|
||||
numberNode.AddChild(n)
|
||||
}
|
||||
nodes = append(nodes, numberNode)
|
||||
} else if m, isMap := v.(map[string]interface{}); isMap && len(m) > 0 {
|
||||
numberNode := tview.NewTreeNode(fmt.Sprintf("[%d]", i+1))
|
||||
for _, n := range t.AddNode(v) {
|
||||
numberNode.AddChild(n)
|
||||
}
|
||||
nodes = append(nodes, numberNode)
|
||||
} else {
|
||||
nodes = append(nodes, t.AddNode(v)...)
|
||||
}
|
||||
}
|
||||
default:
|
||||
nodes = append(nodes, t.NewNodeWithLiteral(node))
|
||||
}
|
||||
return nodes
|
||||
}
|
||||
|
||||
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)
|
||||
}
|
Loading…
Reference in New Issue