cview/demos/presentation/treeview.go

166 lines
5.1 KiB
Go

package main
import (
"strings"
"github.com/gdamore/tcell/v2"
"gitlab.com/tslocum/cview"
)
const treeAllCode = `[green]package[white] main
[green]import[white] [red]"gitlab.com/tslocum/cview"[white]
[green]func[white] [yellow]main[white]() {
$$$
root := cview.[yellow]NewTreeNode[white]([red]"Root"[white]).
[yellow]AddChild[white](cview.[yellow]NewTreeNode[white]([red]"First child"[white]).
[yellow]AddChild[white](cview.[yellow]NewTreeNode[white]([red]"Grandchild A"[white])).
[yellow]AddChild[white](cview.[yellow]NewTreeNode[white]([red]"Grandchild B"[white]))).
[yellow]AddChild[white](cview.[yellow]NewTreeNode[white]([red]"Second child"[white]).
[yellow]AddChild[white](cview.[yellow]NewTreeNode[white]([red]"Grandchild C"[white])).
[yellow]AddChild[white](cview.[yellow]NewTreeNode[white]([red]"Grandchild D"[white]))).
[yellow]AddChild[white](cview.[yellow]NewTreeNode[white]([red]"Third child"[white]))
tree.[yellow]SetRoot[white](root).
[yellow]SetCurrentNode[white](root)
cview.[yellow]NewApplication[white]().
[yellow]SetRoot[white](tree, true).
[yellow]Run[white]()
}`
const treeBasicCode = `tree := cview.[yellow]NewTreeView[white]()`
const treeTopLevelCode = `tree := cview.[yellow]NewTreeView[white]().
[yellow]SetTopLevel[white]([red]1[white])`
const treeAlignCode = `tree := cview.[yellow]NewTreeView[white]().
[yellow]SetAlign[white](true)`
const treePrefixCode = `tree := cview.[yellow]NewTreeView[white]().
[yellow]SetGraphics[white](false).
[yellow]SetTopLevel[white]([red]1[white]).
[yellow]SetPrefixes[white]([][green]string[white]{
[red]"[red[]* "[white],
[red]"[darkcyan[]- "[white],
[red]"[darkmagenta[]- "[white],
})`
type node struct {
text string
expand bool
selected func()
children []*node
}
var (
tree = cview.NewTreeView()
treeNextSlide func()
treeCode = cview.NewTextView()
)
var rootNode = &node{
text: "Root",
children: []*node{
{text: "Expand all", selected: func() { tree.GetRoot().ExpandAll() }},
{text: "Collapse all", selected: func() {
for _, child := range tree.GetRoot().GetChildren() {
child.CollapseAll()
}
}},
{text: "Hide root node", expand: true, children: []*node{
{text: "Tree list starts one level down"},
{text: "Works better for lists where no top node is needed"},
{text: "Switch to this layout", selected: func() {
tree.SetAlign(false)
tree.SetTopLevel(1)
tree.SetGraphics(true)
tree.SetPrefixes(nil)
treeCode.SetText(strings.Replace(treeAllCode, "$$$", treeTopLevelCode, -1))
}},
}},
{text: "Align node text", expand: true, children: []*node{
{text: "For trees that are similar to lists"},
{text: "Hierarchy shown only in line drawings"},
{text: "Switch to this layout", selected: func() {
tree.SetAlign(true)
tree.SetTopLevel(0)
tree.SetGraphics(true)
tree.SetPrefixes(nil)
treeCode.SetText(strings.Replace(treeAllCode, "$$$", treeAlignCode, -1))
}},
}},
{text: "Prefixes", expand: true, children: []*node{
{text: "Best for hierarchical bullet point lists"},
{text: "You can define your own prefixes per level"},
{text: "Switch to this layout", selected: func() {
tree.SetAlign(false)
tree.SetTopLevel(1)
tree.SetGraphics(false)
tree.SetPrefixes([]string{"[red]* ", "[darkcyan]- ", "[darkmagenta]- "})
treeCode.SetText(strings.Replace(treeAllCode, "$$$", treePrefixCode, -1))
}},
}},
{text: "Basic tree with graphics", expand: true, children: []*node{
{text: "Lines illustrate hierarchy"},
{text: "Basic indentation"},
{text: "Switch to this layout", selected: func() {
tree.SetAlign(false)
tree.SetTopLevel(0)
tree.SetGraphics(true)
tree.SetPrefixes(nil)
treeCode.SetText(strings.Replace(treeAllCode, "$$$", treeBasicCode, -1))
}},
}},
{text: "Next slide", selected: func() { treeNextSlide() }},
}}
// TreeView demonstrates the tree view.
func TreeView(nextSlide func()) (title string, content cview.Primitive) {
treeNextSlide = nextSlide
tree.SetBorder(true)
tree.SetTitle("TreeView")
// Add nodes.
var add func(target *node) *cview.TreeNode
add = func(target *node) *cview.TreeNode {
node := cview.NewTreeNode(target.text)
node.SetSelectable(target.expand || target.selected != nil)
node.SetExpanded(target == rootNode)
node.SetReference(target)
if target.expand {
node.SetColor(tcell.ColorGreen.TrueColor())
} else if target.selected != nil {
node.SetColor(tcell.ColorRed.TrueColor())
}
for _, child := range target.children {
node.AddChild(add(child))
}
return node
}
root := add(rootNode)
tree.SetRoot(root)
tree.SetCurrentNode(root)
tree.SetSelectedFunc(func(n *cview.TreeNode) {
original := n.GetReference().(*node)
if original.expand {
n.SetExpanded(!n.IsExpanded())
} else if original.selected != nil {
original.selected()
}
})
treeCode.SetWrap(false)
treeCode.SetDynamicColors(true)
treeCode.SetText(strings.Replace(treeAllCode, "$$$", treeBasicCode, -1))
treeCode.SetPadding(1, 1, 2, 0)
flex := cview.NewFlex()
flex.AddItem(tree, 0, 1, true)
flex.AddItem(treeCode, codeWidth, 1, false)
return "TreeView", flex
}