ana-ledger/cmd/clitest/main.go

63 lines
1.4 KiB
Go

package main
import (
"flag"
"fmt"
"sort"
"strings"
"gogs.inhome.blapointe.com/ana-ledger/ledger"
)
func main() {
foo := flag.String("foo", "bal", "bal or reg")
likePattern := flag.String("like", ".", "regexp to match")
flag.Parse()
if flag.NArg() < 1 {
panic(fmt.Errorf("positional arguments for files required"))
}
f, err := ledger.NewFiles(flag.Args()[0], flag.Args()[1:]...)
if err != nil {
panic(err)
}
deltas, err := f.Deltas()
if err != nil {
panic(err)
}
like := ledger.LikeName(*likePattern)
deltas = deltas.Like(like)
switch *foo {
case "reg":
sort.Slice(deltas, func(i, j int) bool {
return deltas[i].Debug() < deltas[j].Debug()
})
deltasSoFar := make(ledger.Deltas, 0)
for i := range deltas {
deltasSoFar = append(deltasSoFar, deltas[i])
bal, _ := deltasSoFar.Like(ledger.LikeName("^" + deltas[i].Name + "$")).Balances()
fmt.Printf("%s (%+v)\n", deltas[i].Debug(), bal[deltas[i].Name].Debug())
}
case "bal":
balances, err := deltas.Balances()
if err != nil {
panic(err)
}
for k, v := range balances {
results := []string{}
for subk, subv := range v {
if int(subv) != 0 {
results = append(results, fmt.Sprintf("%s %.2f", subk, subv))
}
}
if len(results) > 0 {
fmt.Printf("%s\t%s\n", k, strings.Join(results, " + "))
}
}
default:
panic(fmt.Errorf("not impl %q", flag.Args()[0]))
}
}