impl ana.Normalizer

src/ana/normalize.go

@@ -0,0 +1,63 @@
+package ana
+
+import (
+	"regexp"
+	"slices"
+	"strings"
+
+	"gogs.inhome.blapointe.com/ana-ledger/src/ledger"
+)
+
+type Normalizer struct {
+	m map[string][]normalize
+}
+
+type normalize struct {
+	startDate string
+	factor    float64
+}
+
+func NewNormalizer() Normalizer {
+	return Normalizer{
+		m: make(map[string][]normalize),
+	}
+}
+
+func (n Normalizer) With(pattern, startDate string, factor float64) Normalizer {
+	n.m[pattern] = append(n.m[pattern], normalize{startDate: startDate, factor: factor})
+	slices.SortFunc(n.m[pattern], func(a, b normalize) int {
+		return -1 * strings.Compare(a.startDate, b.startDate)
+	})
+	return n
+}
+
+func (n Normalizer) Normalize(deltas ledger.Deltas) ledger.Deltas {
+	deltas = slices.Clone(deltas)
+
+	patterns := []string{}
+	for pattern := range n.m {
+		patterns = append(patterns, pattern)
+	}
+	like := ledger.LikeName(strings.Join(patterns, "|"))
+
+	for i, delta := range deltas {
+		if !like(delta) {
+			continue
+		}
+		func() {
+			for pattern := range n.m {
+				if regexp.MustCompile(pattern).MatchString(delta.Name) {
+					for _, normalize := range n.m[pattern] {
+						if normalize.startDate < delta.Date {
+							delta.Value /= normalize.factor
+							return
+						}
+					}
+				}
+			}
+		}()
+		deltas[i] = delta
+	}
+
+	return deltas
+}