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]))
	}
}