-
-
Notifications
You must be signed in to change notification settings - Fork 23
/
print_tree.go
72 lines (62 loc) · 1.47 KB
/
print_tree.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package print
import (
"fmt"
"strings"
"github.com/jedib0t/go-pretty/v6/list"
"github.com/alajmo/mani/core"
"github.com/alajmo/mani/core/dao"
)
func PrintTree(config *dao.Config, theme dao.Theme, listFlags *core.ListFlags, tree []dao.TreeNode) {
var treeStyle list.Style
switch theme.Tree.Style {
case "ascii":
treeStyle = list.StyleDefault
case "bullet-flower":
treeStyle = list.StyleBulletFlower
case "bullet-square":
treeStyle = list.StyleBulletSquare
case "bullet-star":
treeStyle = list.StyleBulletStar
case "bullet-triangle":
treeStyle = list.StyleBulletTriangle
case "bold":
treeStyle = list.StyleConnectedBold
case "double":
treeStyle = list.StyleConnectedDouble
case "rounded":
treeStyle = list.StyleConnectedRounded
case "markdown":
treeStyle = list.StyleMarkdown
default: // light
treeStyle = list.StyleConnectedLight
}
l := list.NewWriter()
l.SetStyle(treeStyle)
printTreeNodes(l, tree, 0)
switch listFlags.Output {
case "markdown":
printTree(l.RenderMarkdown())
case "html":
printTree(l.RenderHTML())
default:
printTree(l.Render())
}
}
func printTreeNodes(l list.Writer, tree []dao.TreeNode, depth int) {
for _, n := range tree {
for i := 0; i < depth; i++ {
l.Indent()
}
l.AppendItem(n.Name)
printTreeNodes(l, n.Children, depth+1)
for i := 0; i < depth; i++ {
l.UnIndent()
}
}
}
func printTree(content string) {
for _, line := range strings.Split(content, "\n") {
fmt.Printf("%s\n", line)
}
fmt.Println()
}