Package the logic. Create cli entrypoint.

cmd/ru/main.go

@@ -4,6 +4,8 @@ import (
 	"flag"
 	"fmt"
 	"time"
+
+	"git.lattuga.net/blallo/ruspa/tree"
 )
 
 type UnitValue struct {
@@ -35,24 +37,29 @@ func (u *UnitValue) Set(value string) error {
 		u.unit = "PB"
 		return nil
 	default:
-		return ErrUnknownUnit
+		return tree.ErrUnknownUnit
 	}
 }
 
 func main() {
 	var path string
 	var depth int
+	var t tree.Node
 	var unit = &UnitValue{unit: "KB"}
 	flag.StringVar(&path, "path", ".", "Path from where to start the walk from")
 	flag.IntVar(&depth, "depth", 0, "Depth to display")
 	flag.Var(unit, "unit", "Unit in which to report size")
 	flag.Parse()
 
-	t := NewTop(path)
+	if depth == 0 {
+		t = tree.NewSingle(path)
+	} else {
+		t = tree.NewTop(path)
+	}
 	t.SetUnit(unit.String())
 	go t.Spawn(depth)
 	go t.Collect()
-	for {
+	for !t.Complete() {
 		select {
 		case <-time.After(500 * time.Millisecond):
 			fmt.Printf("\033[H\033[2J")

tree/dir_utils.go

@@ -1,4 +1,4 @@
-package main
+package tree
 
 import (
 	"os"

tree/dir_utils_test.go

@@ -1,4 +1,4 @@
-package main
+package tree
 
 import (
 	"io/ioutil"

tree/dynamic_tree.go

@@ -1,4 +1,4 @@
-package main
+package tree
 
 import (
 	"errors"

tree/dynamic_tree_test.go

@@ -1,4 +1,4 @@
-package main
+package tree
 
 import (
 	"fmt"

tree/real_walker.go

@@ -1,4 +1,4 @@
-package main
+package tree
 
 import (
 	"os"