97 lines
1.7 KiB
Go
97 lines
1.7 KiB
Go
|
package ffmd
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"github.com/peterbourgon/ff/v3/ffcli"
|
||
|
"sort"
|
||
|
"strings"
|
||
|
)
|
||
|
|
||
|
// Tree returns a tree-representation of a ffcli.Command
|
||
|
func Tree(cmd *ffcli.Command) string {
|
||
|
t := tree{}
|
||
|
for _, path := range commandPaths(cmd, "") {
|
||
|
t.add(path)
|
||
|
}
|
||
|
s := t.String(true, "")
|
||
|
return strings.TrimSpace(s)
|
||
|
}
|
||
|
|
||
|
func commandPaths(cmd *ffcli.Command, path string) []string {
|
||
|
root := fmt.Sprintf("%s%s", path, cmd.Name)
|
||
|
s := []string{root}
|
||
|
for _, sub := range cmd.Subcommands {
|
||
|
s = append(s, commandPaths(sub, root+"/")...)
|
||
|
}
|
||
|
sort.SliceStable(s, func(i, j int) bool {
|
||
|
return s[i] < s[j]
|
||
|
})
|
||
|
return s
|
||
|
}
|
||
|
|
||
|
type tree map[string]tree
|
||
|
|
||
|
func (t tree) add(path string) {
|
||
|
t.addParts(strings.Split(path, "/"))
|
||
|
}
|
||
|
|
||
|
func (t tree) addParts(parts []string) {
|
||
|
if len(parts) == 0 {
|
||
|
return
|
||
|
}
|
||
|
next, ok := t[parts[0]]
|
||
|
if !ok {
|
||
|
next = tree{}
|
||
|
t[parts[0]] = next
|
||
|
}
|
||
|
next.addParts(parts[1:])
|
||
|
}
|
||
|
|
||
|
func (t tree) keys() []string {
|
||
|
k := make([]string, 0, len(t))
|
||
|
for key, _ := range t {
|
||
|
k = append(k, key)
|
||
|
}
|
||
|
sort.Strings(k)
|
||
|
return k
|
||
|
}
|
||
|
|
||
|
func (t tree) String(root bool, padding string) string {
|
||
|
var s strings.Builder
|
||
|
index := 0
|
||
|
for _, k := range t.keys() {
|
||
|
v := t[k]
|
||
|
s.WriteString(fmt.Sprintf("%s%s\n", padding+pipePad(root, pipe(index, len(t))), k))
|
||
|
s.WriteString(v.String(false, padding+pipePad(root, outerPipe(index, len(t)))))
|
||
|
index++
|
||
|
}
|
||
|
return s.String()
|
||
|
}
|
||
|
|
||
|
func pipe(index int, len int) string {
|
||
|
switch {
|
||
|
case index+1 == len:
|
||
|
return "└─"
|
||
|
case index+1 > len:
|
||
|
return " "
|
||
|
default:
|
||
|
return "├─"
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func outerPipe(index int, len int) string {
|
||
|
switch {
|
||
|
case index+1 == len:
|
||
|
return " "
|
||
|
default:
|
||
|
return "│ "
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func pipePad(root bool, box string) string {
|
||
|
if root {
|
||
|
return ""
|
||
|
}
|
||
|
return box + " "
|
||
|
}
|