package main

import (
	"flag"
	"fmt"
	"time"

	"git.lattuga.net/blallo/ruspa/tree"
)

type UnitValue struct {
	unit string
}

func (u *UnitValue) String() string {
	return u.unit
}

func (u *UnitValue) Set(value string) error {
	switch value {
	case "B":
		u.unit = "B"
		return nil
	case "KB":
		u.unit = "KB"
		return nil
	case "MB":
		u.unit = "MB"
		return nil
	case "GB":
		u.unit = "GB"
		return nil
	case "TB":
		u.unit = "TB"
		return nil
	case "PB":
		u.unit = "PB"
		return nil
	default:
		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()

	if depth == 0 {
		t = tree.NewSingle(path)
	} else {
		t = tree.NewTop(path)
	}
	t.SetUnit(unit.String())
	go t.Spawn(depth)
	go t.Collect()
	for !t.Complete() {
		select {
		case <-time.After(500 * time.Millisecond):
			fmt.Printf("\033[H\033[2J")
			fmt.Println(t)
		}
	}
}