package ledger

import (
	"slices"
)

type Deltas []Delta

func (deltas Deltas) Like(like ...Like) Deltas {
	result := make(Deltas, 0, len(deltas))
	for i := range deltas {
		if Likes(like).All(deltas[i]) {
			result = append(result, deltas[i])
		}
	}
	return result
}

func (deltas Deltas) Register() map[string]Balances {
	dateToBalances := map[string]Balances{}
	for _, delta := range deltas {
		if _, ok := dateToBalances[delta.Date]; !ok {
			dateToBalances[delta.Date] = make(Balances)
		}
		dateToBalances[delta.Date].Push(delta)
	}
	dates := make([]string, 0, len(dateToBalances)+1)
	for k := range dateToBalances {
		dates = append(dates, k)
	}
	slices.Sort(dates)
	for i := range dates {
		if i == 0 {
			continue
		}
		dateToBalances[dates[i]].PushAll(dateToBalances[dates[i-1]])
	}
	return dateToBalances
}

func (deltas Deltas) Balances() Balances {
	result := make(Balances)
	for _, delta := range deltas {
		if _, ok := result[delta.Name]; !ok {
			result[delta.Name] = make(Balance)
		}
		if _, ok := result[delta.Name][delta.Currency]; !ok {
			result[delta.Name][delta.Currency] = 0
		}
		result[delta.Name][delta.Currency] += delta.Value
		if result[delta.Name][delta.Currency] < 0.000000001 && result[delta.Name][delta.Currency] > -0.000000001 {
			delete(result[delta.Name], delta.Currency)
			if len(result[delta.Name]) == 0 {
				delete(result, delta.Name)
			}
		}
	}

	return result
}