package main

import (
	"flag"
	"fmt"
	"sort"
	"strings"

	"gogs.inhome.blapointe.com/ana-ledger/ledger"
)

func main() {
	foo := flag.String("foo", "bal", "bal or reg")
	likeName := flag.String("like", ".", "regexp to match")
	likeBefore := flag.String("like-before", "9", "date str to compare")
	likeAfter := flag.String("like-after", "0", "date str to compare")
	likeLedger := flag.Bool("like-ledger", false, "limit data to these -like-* rather than zoom to these -like-*")
	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.Like{ledger.LikeName(*likeName)}
	if *likeLedger {
		like = append(like, ledger.LikeBefore(*likeBefore))
		like = append(like, ledger.LikeAfter(*likeAfter))
		deltas = deltas.Like(like...)
	} else {
		deltas = deltas.Like(like...)
		like = append(like, ledger.LikeBefore(*likeBefore))
		like = append(like, ledger.LikeAfter(*likeAfter))
	}

	switch *foo {
	case "reg":
		sort.Slice(deltas, func(i, j int) bool {
			return deltas[i].Debug() < deltas[j].Debug()
		})
		register := deltas.Register()
		for i := range deltas {
			all := true
			for _, like := range like {
				all = all && like(deltas[i])
			}
			if all {
				fmt.Printf("%s (%+v)\n", deltas[i].Debug(), register[deltas[i].Date][deltas[i].Name].Debug())
			}
		}
	case "bal":
		for k, v := range deltas.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]))
	}
}