49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
package ana
|
|
|
|
import (
|
|
"maps"
|
|
"math"
|
|
"regexp"
|
|
"time"
|
|
|
|
"gogs.inhome.blapointe.com/ana-ledger/ledger"
|
|
)
|
|
|
|
type Predictor func(ledger.Balances, time.Duration) ledger.Balances
|
|
|
|
func NewInterestPredictor(namePattern string, currencyPattern string, apy float64) Predictor {
|
|
nameMatcher := regexp.MustCompile(namePattern)
|
|
currencyMatcher := regexp.MustCompile(currencyPattern)
|
|
return func(given ledger.Balances, delta time.Duration) ledger.Balances {
|
|
result := maps.Clone(given)
|
|
for k, v := range result {
|
|
result[k] = maps.Clone(v)
|
|
}
|
|
|
|
monthsPassed := float64(delta) / float64(time.Hour*24.0*365.0/12.0)
|
|
scalar := math.Pow(1.0+apy/12.0, monthsPassed)
|
|
for name := range result {
|
|
if !nameMatcher.MatchString(name) {
|
|
continue
|
|
}
|
|
for currency := range result[name] {
|
|
if !currencyMatcher.MatchString(string(currency)) {
|
|
continue
|
|
}
|
|
result[name][currency] *= scalar
|
|
}
|
|
}
|
|
return result
|
|
}
|
|
}
|
|
|
|
func NewContributionPredictor(reg ledger.Register) Predictor {
|
|
monthlyRate := getMonthlyContributionRate(reg)
|
|
return func(given ledger.Balances, delta time.Duration) ledger.Balances {
|
|
panic(nil)
|
|
}
|
|
}
|
|
|
|
func getMonthlyContributionRate(reg ledger.Register) map[string]ledger.Balance {
|
|
}
|