package main

import (
	"encoding/json"
	"flag"
	"fmt"
	"os"
	"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-*")
	groupName := flag.String("group-name", ".*", "grouping to apply to names")
	groupDate := flag.String("group-date", ".*", "grouping to apply to dates")
	jsonOutput := flag.Bool("json", false, "json output")
	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)
	}

	deltas = deltas.Group(ledger.GroupName(*groupName), ledger.GroupDate(*groupDate))

	like := ledger.Likes{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))
	}

	jsonResult := []any{}

	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 {
			if like.All(deltas[i]) {
				if !*jsonOutput {
					fmt.Printf("%s (%+v)\n", deltas[i].Debug(), register[deltas[i].Date][deltas[i].Name].Debug())
				} else {
					jsonResult = append(jsonResult, map[string]any{
						"name":    deltas[i].Name,
						"delta":   deltas[i],
						"balance": register[deltas[i].Date][deltas[i].Name],
					})
				}
			}
		}
	case "bal":
		deltas = deltas.Like(like...)
		for k, v := range deltas.Balances() {
			results := []string{}
			for subk, subv := range v {
				results = append(results, fmt.Sprintf("%s %.2f", subk, subv))
			}
			if len(results) > 0 {
				if !*jsonOutput {
					fmt.Printf("%s\t%s\n", k, strings.Join(results, " + "))
				} else {
					jsonResult = append(jsonResult, map[string]any{
						"name":    k,
						"balance": v,
					})
				}
			}
		}
	default:
		panic(fmt.Errorf("not impl %q", flag.Args()[0]))
	}

	if *jsonOutput {
		json.NewEncoder(os.Stdout).Encode(jsonResult)
	}
}