Make it easier to use other assets.
All checks were successful
Build & Test / Main (push) Successful in 1m31s
All checks were successful
Build & Test / Main (push) Successful in 1m31s
This commit is contained in:
@@ -1,31 +1,33 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
_ "embed"
|
||||
"slices"
|
||||
"time"
|
||||
|
||||
"code.humancabbage.net/sam/moonmath/coindesk"
|
||||
"code.humancabbage.net/sam/moonmath/moon"
|
||||
"github.com/knadh/koanf/parsers/toml"
|
||||
"github.com/knadh/koanf/parsers/yaml"
|
||||
"github.com/knadh/koanf/providers/file"
|
||||
"github.com/knadh/koanf/providers/rawbytes"
|
||||
"github.com/knadh/koanf/providers/structs"
|
||||
"github.com/knadh/koanf/v2"
|
||||
)
|
||||
|
||||
var k = koanf.New(".")
|
||||
|
||||
func Load(path string) (data Data, err error) {
|
||||
func Load(path string) (all All, err error) {
|
||||
err = k.Load(structs.Provider(Default, "koanf"), nil)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
if path != "" {
|
||||
err = k.Load(file.Provider(path), toml.Parser())
|
||||
err = k.Load(file.Provider(path), yaml.Parser())
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
}
|
||||
err = k.Unmarshal("", &data)
|
||||
err = k.Unmarshal("", &all)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
@@ -33,11 +35,11 @@ func Load(path string) (data Data, err error) {
|
||||
return
|
||||
}
|
||||
|
||||
var Default = Data{
|
||||
Asset: coindesk.BTC,
|
||||
Goals: moon.DefaultGoals,
|
||||
ConstantBases: moon.DefaultConstantBases,
|
||||
RelativeBases: moon.DefaultRelativeBases,
|
||||
var Default All
|
||||
|
||||
type All struct {
|
||||
Defaults Data `koanf:"defaults"`
|
||||
Assets map[coindesk.Asset]Data `koanf:"assets"`
|
||||
}
|
||||
|
||||
type Data struct {
|
||||
@@ -47,9 +49,24 @@ type Data struct {
|
||||
RelativeBases []moon.RelativeBase `koanf:"relativeBases"`
|
||||
}
|
||||
|
||||
type Goal struct {
|
||||
Name string `koanf:"name"`
|
||||
Value float64 `koanf:"value"`
|
||||
func (all All) GetData(asset coindesk.Asset) (data Data, err error) {
|
||||
data, ok := all.Assets[asset]
|
||||
if !ok {
|
||||
data = all.Defaults
|
||||
}
|
||||
if data.Asset == "" {
|
||||
data.Asset = asset
|
||||
}
|
||||
if data.Goals == nil || len(data.Goals) == 0 {
|
||||
data.Goals = all.Defaults.Goals
|
||||
}
|
||||
if data.ConstantBases == nil || len(data.ConstantBases) == 0 {
|
||||
data.ConstantBases = all.Defaults.ConstantBases
|
||||
}
|
||||
if data.RelativeBases == nil || len(data.RelativeBases) == 0 {
|
||||
data.RelativeBases = all.Defaults.RelativeBases
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// GetBases returns the concatenation of the constant and relative bases, sorted
|
||||
@@ -77,3 +94,18 @@ func GetBases(d *Data) (bases []moon.Base) {
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
//go:embed default.yaml
|
||||
var defaultYamlBytes []byte
|
||||
|
||||
func init() {
|
||||
var k = koanf.New(".")
|
||||
err := k.Load(rawbytes.Provider(defaultYamlBytes), yaml.Parser())
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = k.Unmarshal("", &Default)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
|
10
config/config_test.go
Normal file
10
config/config_test.go
Normal file
@@ -0,0 +1,10 @@
|
||||
package config_test
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestDefault(t *testing.T) {
|
||||
// config.Default is parsed in config.init(), which panics on error.
|
||||
// so just getting this far means that at least it parsed successfully.
|
||||
}
|
46
config/default.yaml
Normal file
46
config/default.yaml
Normal file
@@ -0,0 +1,46 @@
|
||||
defaults:
|
||||
goals:
|
||||
- name: $100k
|
||||
value: 100000
|
||||
- name: $150k
|
||||
value: 150000
|
||||
- name: $200k
|
||||
value: 200000
|
||||
- name: $250k
|
||||
value: 250000
|
||||
- name: $300k
|
||||
value: 300000
|
||||
- name: $500k
|
||||
value: 500000
|
||||
- name: $1m
|
||||
value: 1000000
|
||||
relativeBases:
|
||||
- name: Month
|
||||
offset: -720h0m0s
|
||||
- name: Quarter
|
||||
offset: -2160h0m0s
|
||||
- name: Half-Year
|
||||
offset: -4368h0m0s
|
||||
- name: Year
|
||||
offset: -8760h0m0s
|
||||
constantBases:
|
||||
- name: 2020-
|
||||
time: 2019-12-31T16:00:00-08:00
|
||||
- name: 2017-
|
||||
time: 2016-12-31T16:00:00-08:00
|
||||
|
||||
assets:
|
||||
ETH:
|
||||
goals:
|
||||
- name: $5k
|
||||
value: 5000
|
||||
- name: $7.5k
|
||||
value: 7500
|
||||
- name: $10k
|
||||
value: 10000
|
||||
- name: $15k
|
||||
value: 15000
|
||||
- name: $20k
|
||||
value: 20000
|
||||
- name: $25k
|
||||
value: 25000
|
@@ -1,25 +0,0 @@
|
||||
asset = "ETH"
|
||||
|
||||
[[goals]]
|
||||
name = "$5k"
|
||||
value = 5000
|
||||
|
||||
[[goals]]
|
||||
name = "$7.5k"
|
||||
value = 7500
|
||||
|
||||
[[goals]]
|
||||
name = "$10k"
|
||||
value = 10000
|
||||
|
||||
[[goals]]
|
||||
name = "$15k"
|
||||
value = 15000
|
||||
|
||||
[[goals]]
|
||||
name = "$20k"
|
||||
value = 15000
|
||||
|
||||
[[goals]]
|
||||
name = "$25k"
|
||||
value = 25000
|
Reference in New Issue
Block a user