add tree browser concept
This commit is contained in:
Alex Goodman
parent
0d2b6551b3
commit
8f3e4b42d7
200
main.go
200
main.go
@ -9,8 +9,16 @@ import (
|
||||
|
||||
"github.com/docker/docker/client"
|
||||
"golang.org/x/net/context"
|
||||
"github.com/jroimartin/gocui"
|
||||
"log"
|
||||
)
|
||||
|
||||
var data struct {
|
||||
tree *FileTree
|
||||
absPosition uint
|
||||
}
|
||||
|
||||
|
||||
func check(e error) {
|
||||
if e != nil {
|
||||
panic(e)
|
||||
@ -101,3 +109,195 @@ func demo() {
|
||||
}
|
||||
fmt.Println("See './image' for the cached image tar")
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
func getAbsPositionNode() (node *Node) {
|
||||
var visiter func(*Node) error
|
||||
var evaluator func(*Node) bool
|
||||
var dfsCounter uint
|
||||
|
||||
visiter = func(curNode *Node) error {
|
||||
if dfsCounter == data.absPosition {
|
||||
node = curNode
|
||||
}
|
||||
dfsCounter++
|
||||
return nil
|
||||
}
|
||||
|
||||
evaluator = func(curNode *Node) bool {
|
||||
return !curNode.collapsed
|
||||
}
|
||||
|
||||
err := data.tree.VisitDepthParentFirst(visiter, evaluator)
|
||||
if err != nil {
|
||||
// todo: you guessed it, check errors
|
||||
}
|
||||
|
||||
return node
|
||||
}
|
||||
|
||||
func showCurNodeInSideBar(g *gocui.Gui, v *gocui.View) error {
|
||||
g.Update(func(g *gocui.Gui) error {
|
||||
v, _ := g.View("side")
|
||||
// todo: handle above error.
|
||||
v.Clear()
|
||||
_, err := fmt.Fprintf(v, "Node:\n%+v\n", getAbsPositionNode())
|
||||
return err
|
||||
})
|
||||
// todo: blerg
|
||||
return nil
|
||||
}
|
||||
|
||||
func cursorDown(g *gocui.Gui, v *gocui.View) error {
|
||||
if v != nil {
|
||||
cx, cy := v.Cursor()
|
||||
|
||||
// if there isn't a next line
|
||||
line, err := v.Line(cy+1)
|
||||
if err != nil {
|
||||
// todo: handle error
|
||||
}
|
||||
if len(line) == 0 {
|
||||
return nil
|
||||
}
|
||||
if err := v.SetCursor(cx, cy+1); err != nil {
|
||||
ox, oy := v.Origin()
|
||||
if err := v.SetOrigin(ox, oy+1); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
data.absPosition++
|
||||
showCurNodeInSideBar(g, v)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func cursorUp(g *gocui.Gui, v *gocui.View) error {
|
||||
if v != nil {
|
||||
ox, oy := v.Origin()
|
||||
cx, cy := v.Cursor()
|
||||
if err := v.SetCursor(cx, cy-1); err != nil && oy > 0 {
|
||||
if err := v.SetOrigin(ox, oy-1); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
data.absPosition--
|
||||
showCurNodeInSideBar(g, v)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
func toggleCollapse(g *gocui.Gui, v *gocui.View) error {
|
||||
node := getAbsPositionNode()
|
||||
node.collapsed = !node.collapsed
|
||||
return drawTree(g, v)
|
||||
}
|
||||
|
||||
func drawTree(g *gocui.Gui, v *gocui.View) error {
|
||||
g.Update(func(g *gocui.Gui) error {
|
||||
v, _ := g.View("main")
|
||||
// todo: handle above error.
|
||||
v.Clear()
|
||||
_, err := fmt.Fprintln(v, data.tree.String())
|
||||
return err
|
||||
})
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
func quit(g *gocui.Gui, v *gocui.View) error {
|
||||
return gocui.ErrQuit
|
||||
}
|
||||
|
||||
func keybindings(g *gocui.Gui) error {
|
||||
if err := g.SetKeybinding("", gocui.KeyCtrlC, gocui.ModNone, quit); err != nil {
|
||||
return err
|
||||
}
|
||||
//if err := g.SetKeybinding("main", gocui.MouseLeft, gocui.ModNone, toggleCollapse); err != nil {
|
||||
// return err
|
||||
//}
|
||||
if err := g.SetKeybinding("main", gocui.KeyArrowDown, gocui.ModNone, cursorDown); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := g.SetKeybinding("main", gocui.KeyArrowUp, gocui.ModNone, cursorUp); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := g.SetKeybinding("main", gocui.KeySpace, gocui.ModNone, toggleCollapse); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
func layout(g *gocui.Gui) error {
|
||||
maxX, maxY := g.Size()
|
||||
splitCol := 50
|
||||
if v, err := g.SetView("side", -1, -1, splitCol, maxY); err != nil {
|
||||
if err != gocui.ErrUnknownView {
|
||||
return err
|
||||
}
|
||||
v.Wrap = true
|
||||
}
|
||||
if v, err := g.SetView("main", splitCol, -1, maxX, maxY); err != nil {
|
||||
if err != gocui.ErrUnknownView {
|
||||
return err
|
||||
}
|
||||
v.Editable = false
|
||||
v.Wrap = false
|
||||
v.Highlight = true
|
||||
v.SelBgColor = gocui.ColorGreen
|
||||
v.SelFgColor = gocui.ColorBlack
|
||||
if _, err := g.SetCurrentView("main"); err != nil {
|
||||
return err
|
||||
}
|
||||
drawTree(g, v)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
func main() {
|
||||
data.tree = NewTree()
|
||||
data.tree.AddPath("/etc/nginx/nginx.conf", nil)
|
||||
data.tree.AddPath("/etc/nginx/public", nil)
|
||||
data.tree.AddPath("/var/run/systemd", nil)
|
||||
data.tree.AddPath("/var/run/bashful", nil)
|
||||
data.tree.AddPath("/tmp", nil)
|
||||
data.tree.AddPath("/tmp/nonsense", nil)
|
||||
data.tree.AddPath("/tmp/wifi/coffeeyo", nil)
|
||||
|
||||
g, err := gocui.NewGui(gocui.OutputNormal)
|
||||
if err != nil {
|
||||
log.Panicln(err)
|
||||
}
|
||||
defer g.Close()
|
||||
|
||||
g.Cursor = false
|
||||
//g.Mouse = true
|
||||
g.SetManagerFunc(layout)
|
||||
|
||||
if err := keybindings(g); err != nil {
|
||||
log.Panicln(err)
|
||||
}
|
||||
|
||||
if err := g.MainLoop(); err != nil && err != gocui.ErrQuit {
|
||||
log.Panicln(err)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
@ -11,7 +11,7 @@ import (
|
||||
"strings"
|
||||
)
|
||||
|
||||
func main() {
|
||||
func tarReadDemo() {
|
||||
f, err := os.Open("image/cache.tar")
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
|
||||
42
tree.go
42
tree.go
@ -30,7 +30,7 @@ type Node struct {
|
||||
parent *Node
|
||||
name string
|
||||
collapsed bool
|
||||
data interface{}
|
||||
data *FileChangeInfo
|
||||
children map[string]*Node
|
||||
}
|
||||
|
||||
@ -124,7 +124,8 @@ func (tree *FileTree) String() string {
|
||||
for idx, name := range keys {
|
||||
child := node.children[name]
|
||||
last := idx == (len(node.children) - 1)
|
||||
result += renderLine(child.String(), spaces, last, child.collapsed)
|
||||
showCollapsed := child.collapsed && len(child.children) > 0
|
||||
result += renderLine(child.String(), spaces, last, showCollapsed)
|
||||
if len(child.children) > 0 && !child.collapsed {
|
||||
spacesChild := append(spaces, last)
|
||||
result += walkTree(child, spacesChild, depth+1)
|
||||
@ -156,13 +157,24 @@ func (tree *FileTree) Copy() *FileTree {
|
||||
}
|
||||
|
||||
type Visiter func(*Node) error
|
||||
type VisitEvaluator func(*Node) bool
|
||||
|
||||
func (tree *FileTree) Visit(visiter Visiter) error {
|
||||
return tree.root.Visit(visiter)
|
||||
}
|
||||
|
||||
func (tree *FileTree) VisitDepthParentFirst(visiter Visiter, evaluator VisitEvaluator) error {
|
||||
return tree.root.VisitDepthParentFirst(visiter, evaluator)
|
||||
}
|
||||
|
||||
func (node *Node) Visit(visiter Visiter) error {
|
||||
for _, child := range node.children {
|
||||
var keys []string
|
||||
for key := range node.children {
|
||||
keys = append(keys, key)
|
||||
}
|
||||
sort.Strings(keys)
|
||||
for _, name := range keys {
|
||||
child := node.children[name]
|
||||
err := child.Visit(visiter)
|
||||
if err != nil {
|
||||
return err
|
||||
@ -171,6 +183,30 @@ func (node *Node) Visit(visiter Visiter) error {
|
||||
return visiter(node)
|
||||
}
|
||||
|
||||
func (node *Node) VisitDepthParentFirst(visiter Visiter, evaluator VisitEvaluator) error {
|
||||
err := visiter(node)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var keys []string
|
||||
for key := range node.children {
|
||||
keys = append(keys, key)
|
||||
}
|
||||
sort.Strings(keys)
|
||||
for _, name := range keys {
|
||||
child := node.children[name]
|
||||
if evaluator == nil || !evaluator(node) {
|
||||
continue
|
||||
}
|
||||
err = child.VisitDepthParentFirst(visiter, evaluator)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
func (node *Node) IsWhiteout() bool {
|
||||
return strings.HasPrefix(node.name, whiteoutPrefix)
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user