Support multiple assets simultaneously.

Add config defaults for more assets.
* LTC * SOL * XRP * DOGE * ADA
2024-03-22 17:43:15 -07:00 · 2024-03-21 22:07:16 -07:00 · 2024-03-21 17:45:27 -07:00 · 2024-03-21 17:39:10 -07:00 · 2024-03-21 12:31:40 -07:00 · 2024-03-21 12:23:17 -07:00
14 changed files with 537 additions and 198 deletions
--- a/README.md
+++ b/README.md
@@ -1,6 +1,6 @@
 # moonmath

-## Bullshit BTC Price Projections, Now in Your CLI!
+## Bullshit Crypto Price Projections, Now in Your CLI!

 ![screenshot](./screenshot.png)

@@ -12,13 +12,27 @@ program. It's written in Go using the [Bubble Tea][tea] library, and uses
 [tea]: https://github.com/charmbracelet/bubbletea
 [coin]: https://www.coindesk.com

+### Installation
+
+Go to the [Releases page](https://code.humancabbage.net/sam/moonmath/releases)
+and download the archive for your operating system and architecture. (For the
+uninitiated, "Darwin" means macOS.)
+
 ### Configuration

-By default, the program will use Bitcoin along with various goals and bases
-of comparison. With the `--config-file` flag, however, one can specify a TOML
-file that overrides these defaults.
+By default, the program will use Bitcoin along with various goals and bases of
+comparison. With the `--asset` flag, another asset supported by Coindesk can be
+chosen. These can even be chained, e.g. `--asset BTC --asset ETH`, to show
+projections for multiple assets simultaneously.

-See [this one for Ethereum](./config/eth.toml) for an example.
+The [builtin default config](./config/default.yaml) only has special
+goals a handful of the most popular assets. With the `--config-file` flag,
+however, one can specify a YAML file that overrides these defaults and adds
+goals for other assets.
+
+Check out [coindesk/assets.go](./coindesk/assets.go) for a full list of
+supported assets. Keep in mind these have not been exhaustively tested, and
+it's likely that many will fail with the default configuration settings.

 ### "Theory"

--- a/bitcoinity/captured.json
+++ b/bitcoinity/captured.json
--- a/bitcoinity/model_test.go
+++ b/bitcoinity/model_test.go
@@ -21,3 +21,24 @@ func TestUnmarshalGetTickerResponse(t *testing.T) {

 //go:embed get_ticker.json
 var getTickerJson []byte
+
+func TestUnmarshalWebhookMessages(t *testing.T) {
+	var msgs []bitcoinity.Message
+	err := json.Unmarshal(capturedJson, &msgs)
+	if err != nil {
+		t.Errorf("failed to unmarshal webhook messages JSON: %v", err)
+	}
+	for _, msg := range msgs {
+		if msg.Event != "new_msg" {
+			continue
+		}
+		var payload bitcoinity.MarketPayload
+		err := json.Unmarshal(msg.Payload, &payload)
+		if err != nil {
+			t.Errorf("failed to unmarshal market payload JSON: %v", err)
+		}
+	}
+}
+
+//go:embed captured.json
+var capturedJson []byte
--- a/config/config.go
+++ b/config/config.go
@@ -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) {
+	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)
+	}
+}
--- a/config/config_test.go
+++ b/config/config_test.go
@@ -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.
+}
--- a/config/default.yaml
+++ b/config/default.yaml
@@ -0,0 +1,131 @@
+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
+  LTC:
+    goals:
+    - name: $100
+      value: 100
+    - name: $150
+      value: 150
+    - name: $200
+      value: 200
+    - name: $300
+      value: 300
+    - name: $500
+      value: 500
+    - name: $1k
+      value: 1000
+    constantBases:
+    - name: 2021-
+      time: 2020-12-31T16:00:00-08:00
+  SOL:
+    goals:
+    - name: $250
+      value: 250
+    - name: $375
+      value: 375
+    - name: $500
+      value: 500
+    - name: $750
+      value: 750
+    - name: $1k
+      value: 1000
+    - name: $2k
+      value: 2000
+    constantBases:
+    - name: 2022-
+      time: 2021-12-31T16:00:00-08:00
+  XRP:
+    goals:
+    - name: $1
+      value: 1
+    - name: $1.5
+      value: 1.5
+    - name: $2
+      value: 2
+    - name: $3
+      value: 3
+    - name: $5
+      value: 5
+    - name: $10
+      value: 10
+    constantBases:
+    - name: 2022-
+      time: 2021-12-31T16:00:00-08:00
+  DOGE:
+    goals:
+    - name: $1
+      value: 1
+    - name: $1.5
+      value: 1.5
+    - name: $2
+      value: 2
+    - name: $3
+      value: 3
+    - name: $5
+      value: 5
+    - name: $10
+      value: 10
+    constantBases:
+    - name: 2022-
+      time: 2021-12-31T16:00:00-08:00
+  ADA:
+    goals:
+    - name: $1
+      value: 1
+    - name: $1.5
+      value: 1.5
+    - name: $2
+      value: 2
+    - name: $3
+      value: 3
+    - name: $5
+      value: 5
+    - name: $10
+      value: 10
+    constantBases:
+    - name: 2022-
+      time: 2021-12-31T16:00:00-08:00
--- a/config/eth.toml
+++ b/config/eth.toml
@@ -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
--- a/go.mod
+++ b/go.mod
@@ -3,26 +3,28 @@ module code.humancabbage.net/sam/moonmath
 go 1.22.1

 require (
+	github.com/alecthomas/kong v0.9.0
 	github.com/carlmjohnson/requests v0.23.5
 	github.com/charmbracelet/bubbles v0.18.0
 	github.com/charmbracelet/bubbletea v0.25.0
 	github.com/charmbracelet/lipgloss v0.10.0
+	github.com/knadh/koanf/parsers/yaml v0.1.0
+	github.com/knadh/koanf/providers/file v0.1.0
+	github.com/knadh/koanf/providers/rawbytes v0.1.0
+	github.com/knadh/koanf/providers/structs v0.1.0
+	github.com/knadh/koanf/v2 v2.1.0
 	github.com/sourcegraph/conc v0.3.0
 	golang.org/x/net v0.22.0
 )

 require (
-	github.com/alecthomas/kong v0.9.0 // indirect
 	github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
 	github.com/containerd/console v1.0.4 // indirect
 	github.com/fatih/structs v1.1.0 // indirect
 	github.com/fsnotify/fsnotify v1.7.0 // indirect
 	github.com/go-viper/mapstructure/v2 v2.0.0-alpha.1 // indirect
 	github.com/knadh/koanf/maps v0.1.1 // indirect
-	github.com/knadh/koanf/parsers/toml v0.1.0 // indirect
-	github.com/knadh/koanf/providers/file v0.1.0 // indirect
-	github.com/knadh/koanf/providers/structs v0.1.0 // indirect
-	github.com/knadh/koanf/v2 v2.1.0 // indirect
+	github.com/kr/text v0.2.0 // indirect
 	github.com/lucasb-eyer/go-colorful v1.2.0 // indirect
 	github.com/mattn/go-isatty v0.0.20 // indirect
 	github.com/mattn/go-localereader v0.0.1 // indirect
@@ -33,7 +35,6 @@ require (
 	github.com/muesli/cancelreader v0.2.2 // indirect
 	github.com/muesli/reflow v0.3.0 // indirect
 	github.com/muesli/termenv v0.15.2 // indirect
-	github.com/pelletier/go-toml v1.9.5 // indirect
 	github.com/rivo/uniseg v0.4.7 // indirect
 	go.uber.org/atomic v1.7.0 // indirect
 	go.uber.org/multierr v1.9.0 // indirect
@@ -41,4 +42,5 @@ require (
 	golang.org/x/sys v0.18.0 // indirect
 	golang.org/x/term v0.18.0 // indirect
 	golang.org/x/text v0.14.0 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
 )
--- a/go.sum
+++ b/go.sum
@@ -1,5 +1,9 @@
+github.com/alecthomas/assert/v2 v2.6.0 h1:o3WJwILtexrEUk3cUVal3oiQY2tfgr/FHWiz/v2n4FU=
+github.com/alecthomas/assert/v2 v2.6.0/go.mod h1:Bze95FyfUr7x34QZrjL+XP+0qgp/zg8yS+TtBj1WA3k=
 github.com/alecthomas/kong v0.9.0 h1:G5diXxc85KvoV2f0ZRVuMsi45IrBgx9zDNGNj165aPA=
 github.com/alecthomas/kong v0.9.0/go.mod h1:Y47y5gKfHp1hDc7CH7OeXgLIpp+Q2m1Ni0L5s3bI8Os=
+github.com/alecthomas/repr v0.4.0 h1:GhI2A8MACjfegCPVq9f1FLvIBS+DrQ2KQBFZP1iFzXc=
+github.com/alecthomas/repr v0.4.0/go.mod h1:Fr0507jx4eOXV7AlPV6AVZLYrLIuIeSOWtW57eE/O/4=
 github.com/aymanbagabas/go-osc52/v2 v2.0.1 h1:HwpRHbFMcZLEVr42D4p7XBqjyuxQH5SMiErDT4WkJ2k=
 github.com/aymanbagabas/go-osc52/v2 v2.0.1/go.mod h1:uYgXzlJ7ZpABp8OJ+exZzJJhRNQ2ASbcXHWsFqH8hp8=
 github.com/carlmjohnson/requests v0.23.5 h1:NPANcAofwwSuC6SIMwlgmHry2V3pLrSqRiSBKYbNHHA=
@@ -12,6 +16,7 @@ github.com/charmbracelet/lipgloss v0.10.0 h1:KWeXFSexGcfahHX+54URiZGkBFazf70JNMt
 github.com/charmbracelet/lipgloss v0.10.0/go.mod h1:Wig9DSfvANsxqkRsqj6x87irdy123SR4dOXlKa91ciE=
 github.com/containerd/console v1.0.4 h1:F2g4+oChYvBTsASRTz8NP6iIAi97J3TtSAsLbIFn4ro=
 github.com/containerd/console v1.0.4/go.mod h1:YynlIjWYF8myEu6sdkwKIvGQq+cOckRm6So2avqoYAk=
+github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
@@ -21,16 +26,24 @@ github.com/fsnotify/fsnotify v1.7.0 h1:8JEhPFa5W2WU7YfeZzPNqzMP6Lwt7L2715Ggo0nos
 github.com/fsnotify/fsnotify v1.7.0/go.mod h1:40Bi/Hjc2AVfZrqy+aj+yEI+/bRxZnMJyTJwOpGvigM=
 github.com/go-viper/mapstructure/v2 v2.0.0-alpha.1 h1:TQcrn6Wq+sKGkpyPvppOz99zsMBaUOKXq6HSv655U1c=
 github.com/go-viper/mapstructure/v2 v2.0.0-alpha.1/go.mod h1:oJDH3BJKyqBA2TXFhDsKDGDTlndYOZ6rGS0BRZIxGhM=
+github.com/hexops/gotextdiff v1.0.3 h1:gitA9+qJrrTCsiCl7+kh75nPqQt1cx4ZkudSTLoUqJM=
+github.com/hexops/gotextdiff v1.0.3/go.mod h1:pSWU5MAI3yDq+fZBTazCSJysOMbxWL1BSow5/V2vxeg=
 github.com/knadh/koanf/maps v0.1.1 h1:G5TjmUh2D7G2YWf5SQQqSiHRJEjaicvU0KpypqB3NIs=
 github.com/knadh/koanf/maps v0.1.1/go.mod h1:npD/QZY3V6ghQDdcQzl1W4ICNVTkohC8E73eI2xW4yI=
-github.com/knadh/koanf/parsers/toml v0.1.0 h1:S2hLqS4TgWZYj4/7mI5m1CQQcWurxUz6ODgOub/6LCI=
-github.com/knadh/koanf/parsers/toml v0.1.0/go.mod h1:yUprhq6eo3GbyVXFFMdbfZSo928ksS+uo0FFqNMnO18=
+github.com/knadh/koanf/parsers/yaml v0.1.0 h1:ZZ8/iGfRLvKSaMEECEBPM1HQslrZADk8fP1XFUxVI5w=
+github.com/knadh/koanf/parsers/yaml v0.1.0/go.mod h1:cvbUDC7AL23pImuQP0oRw/hPuccrNBS2bps8asS0CwY=
 github.com/knadh/koanf/providers/file v0.1.0 h1:fs6U7nrV58d3CFAFh8VTde8TM262ObYf3ODrc//Lp+c=
 github.com/knadh/koanf/providers/file v0.1.0/go.mod h1:rjJ/nHQl64iYCtAW2QQnF0eSmDEX/YZ/eNFj5yR6BvA=
+github.com/knadh/koanf/providers/rawbytes v0.1.0 h1:dpzgu2KO6uf6oCb4aP05KDmKmAmI51k5pe8RYKQ0qME=
+github.com/knadh/koanf/providers/rawbytes v0.1.0/go.mod h1:mMTB1/IcJ/yE++A2iEZbY1MLygX7vttU+C+S/YmPu9c=
 github.com/knadh/koanf/providers/structs v0.1.0 h1:wJRteCNn1qvLtE5h8KQBvLJovidSdntfdyIbbCzEyE0=
 github.com/knadh/koanf/providers/structs v0.1.0/go.mod h1:sw2YZ3txUcqA3Z27gPlmmBzWn1h8Nt9O6EP/91MkcWE=
 github.com/knadh/koanf/v2 v2.1.0 h1:eh4QmHHBuU8BybfIJ8mB8K8gsGCD/AUQTdwGq/GzId8=
 github.com/knadh/koanf/v2 v2.1.0/go.mod h1:4mnTRbZCK+ALuBXHZMjDfG9y714L7TykVnZkXbMU3Es=
+github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=
+github.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk=
+github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
+github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
 github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69Aj6K7nkY=
 github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0=
 github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
@@ -52,14 +65,14 @@ github.com/muesli/reflow v0.3.0 h1:IFsN6K9NfGtjeggFP+68I4chLZV2yIKsXJFNZ+eWh6s=
 github.com/muesli/reflow v0.3.0/go.mod h1:pbwTDkVPibjO2kyvBQRBxTWEEGDGq0FlB1BIKtnHY/8=
 github.com/muesli/termenv v0.15.2 h1:GohcuySI0QmI3wN8Ok9PtKGkgkFIk7y6Vpb5PvrY+Wo=
 github.com/muesli/termenv v0.15.2/go.mod h1:Epx+iuz8sNs7mNKhxzH4fWXGNpZwUaJKRS1noLXviQ8=
-github.com/pelletier/go-toml v1.9.5 h1:4yBQzkHv+7BHq2PQUZF3Mx0IYxG7LsP222s7Agd3ve8=
-github.com/pelletier/go-toml v1.9.5/go.mod h1:u1nR/EPcESfeI/szUZKdtJ0xRNbUoANCkoOuaOx1Y+c=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/rivo/uniseg v0.1.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
 github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
 github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ=
 github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
+github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8=
+github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs=
 github.com/sourcegraph/conc v0.3.0 h1:OQTbbt6P72L20UqAkXXuLOj79LfEanQ+YQFNpLA9ySo=
 github.com/sourcegraph/conc v0.3.0/go.mod h1:Sdozi7LEKbFPqYX2/J+iBAM6HpqSLTASQIKqDmF7Mt0=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
@@ -82,5 +95,8 @@ golang.org/x/term v0.18.0 h1:FcHjZXDMxI8mM3nwhX9HlKop4C0YQvCVCdwYl2wOtE8=
 golang.org/x/term v0.18.0/go.mod h1:ILwASektA3OnRv7amZ1xhE/KTR+u50pbXfZ03+6Nx58=
 golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ=
 golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 h1:YR8cESwS4TdDjEe65xsg0ogRM/Nc3DYOhEAlW+xobZo=
+gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
--- a/moon/moon.go
+++ b/moon/moon.go
@@ -86,18 +86,18 @@ func (m *Math) Refresh(ctx context.Context) (err error) {
 			if err != nil {
 				return err
 			}
+			if len(resp.Data.Entries) == 0 {
+				c.Projections.Dates = nil
+				return nil
+			}
 			c.StartingPrice = resp.Data.Entries[0].Price
 			c.Gain = float64(m.CurrentPrice) / float64(c.StartingPrice)
 			days := now.Sub(c.StartingDate).Hours() / 24
 			c.CDPR = CDPR(days, c.Gain)
-			if c.CDPR > 1 {
-				c.Projections = ProjectDates(
-					now, float64(m.CurrentPrice),
-					c.CDPR, m.Goals,
-				)
-			} else {
-				c.Projections.Dates = nil
-			}
+			c.Projections = ProjectDates(
+				now, float64(m.CurrentPrice),
+				c.CDPR, m.Goals,
+			)
 			return nil
 		})
 	}
@@ -146,8 +146,6 @@ var DefaultGoals = []Goal{

 var DefaultConstantBases = []ConstantBase{
 	{"2020-", time.Unix(1577836800, 0)},
-	{"2019-", time.Unix(1546300800, 0)},
-	{"2018-", time.Unix(1514764800, 0)},
 	{"2017-", time.Unix(1483228800, 0)},
 }

--- a/moonmath.go
+++ b/moonmath.go
@@ -3,15 +3,17 @@ package main
 import (
 	"fmt"
 	"os"
+	"strings"

+	"code.humancabbage.net/sam/moonmath/coindesk"
 	"code.humancabbage.net/sam/moonmath/config"
 	"code.humancabbage.net/sam/moonmath/tui"
 	"github.com/alecthomas/kong"
-	tea "github.com/charmbracelet/bubbletea"
 )

 var CLI struct {
-	ConfigFile string `help:"Path to TOML configuration file."`
+	Asset      []string `short:"a" default:"BTC" help:"Asset(s) to project."`
+	ConfigFile string   `short:"c" help:"Path to YAML configuration file."`
 }

 func main() {
@@ -19,11 +21,16 @@ func main() {
 	if ctx.Error != nil {
 		fail(ctx.Error)
 	}
-	cfg, err := config.Load(CLI.ConfigFile)
+	allCfg, err := config.Load(CLI.ConfigFile)
 	if err != nil {
 		fail(err)
 	}
-	p := tea.NewProgram(tui.New(cfg))
+	var assets []coindesk.Asset
+	for i := range CLI.Asset {
+		asset := coindesk.Asset(strings.ToUpper(CLI.Asset[i]))
+		assets = append(assets, asset)
+	}
+	p := tui.New(assets, allCfg)
 	if _, err := p.Run(); err != nil {
 		fail(err)
 	}
--- a/screenshot.png
+++ b/screenshot.png
--- a/tui/asset/asset.go
+++ b/tui/asset/asset.go
@@ -0,0 +1,201 @@
+package asset
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"code.humancabbage.net/sam/moonmath/coindesk"
+	"code.humancabbage.net/sam/moonmath/config"
+	"code.humancabbage.net/sam/moonmath/moon"
+	"github.com/charmbracelet/bubbles/spinner"
+	"github.com/charmbracelet/bubbles/table"
+	tea "github.com/charmbracelet/bubbletea"
+	"github.com/charmbracelet/lipgloss"
+)
+
+type Model struct {
+	math moon.Math
+
+	refreshing bool
+	indicator  spinner.Model
+
+	properties  table.Model
+	projections table.Model
+}
+
+type Msg struct {
+	Asset coindesk.Asset
+	inner tea.Msg
+}
+
+func New(cfg config.Data) (m Model) {
+	m.math = moon.NewMath(
+		cfg.Asset,
+		cfg.Goals,
+		config.GetBases(&cfg),
+	)
+
+	tableStyle := table.DefaultStyles()
+	tableStyle.Selected = tableStyle.Cell.Copy().
+		Padding(0)
+	tableStyle.Header = tableStyle.Header.
+		Bold(true).
+		Foreground(lipgloss.Color("214")).
+		Border(baseStyle.GetBorderStyle(), false, false, true, false)
+
+	// properties table
+
+	m.properties = table.New(
+		table.WithColumns([]table.Column{
+			{Title: "Property", Width: 9},
+			{Title: "Value", Width: 9},
+		}),
+		table.WithHeight(2),
+		table.WithStyles(tableStyle),
+	)
+
+	// projections table
+
+	labelsWidth := 0
+	for _, l := range m.math.Labels {
+		if len(l) > labelsWidth {
+			labelsWidth = len(l)
+		}
+	}
+	projectionCols := []table.Column{
+		{Title: "Labels", Width: labelsWidth},
+	}
+	for _, c := range m.math.Columns {
+		projectionCols = append(projectionCols,
+			table.Column{
+				Title: c.Base.Label(),
+				Width: 10,
+			})
+	}
+	m.projections = table.New(
+		table.WithColumns(projectionCols),
+		table.WithHeight(len(m.math.Labels)),
+		table.WithStyles(tableStyle),
+	)
+
+	// indicator spinner
+
+	m.indicator = spinner.New()
+	m.indicator.Spinner = spinner.Points
+	m.indicator.Style = lipgloss.NewStyle().
+		Foreground(lipgloss.Color("69"))
+
+	return
+}
+
+func (m Model) Handles(a coindesk.Asset) bool {
+	return m.math.Asset == a
+}
+
+func (m Model) Init() tea.Cmd {
+	return tea.Batch(
+		m.indicator.Tick,
+		func() tea.Msg {
+			return Msg{m.math.Asset, refresh{}}
+		},
+	)
+}
+
+func (m Model) Update(msg tea.Msg) (Model, tea.Cmd) {
+	switch msg := msg.(type) {
+	case Msg:
+		switch msg := msg.inner.(type) {
+		case refresh:
+			m.refreshing = true
+			return m, func() tea.Msg {
+				// TODO: log errors
+				_ = m.math.Refresh(context.TODO())
+				return Msg{m.math.Asset, m.math}
+			}
+		case moon.Math:
+			m.math = msg
+			refillProperties(&m)
+			refillProjections(&m)
+			return m, tea.Batch(
+				// schedule the next refresh
+				tea.Tick(time.Second*30,
+					func(t time.Time) tea.Msg {
+						return Msg{m.math.Asset, refresh{}}
+					}),
+				// wait a bit to stop the indicator, so that it's more obvious
+				// even when the refresh completes quickly.
+				tea.Tick(time.Millisecond*500,
+					func(t time.Time) tea.Msg {
+						return Msg{m.math.Asset, stopIndicator{}}
+					}),
+			)
+		case stopIndicator:
+			m.refreshing = false
+			return m, nil
+		}
+	case spinner.TickMsg:
+		var cmd tea.Cmd
+		m.indicator, cmd = m.indicator.Update(msg)
+		return m, cmd
+	}
+	return m, nil
+}
+
+type refresh struct{}
+type stopIndicator struct{}
+
+func refillProperties(m *Model) {
+	rows := []table.Row{
+		{"Asset", string(m.math.Asset)},
+		{"Price", fmt.Sprintf("$%0.2f", m.math.CurrentPrice)},
+	}
+	m.properties.SetRows(rows)
+}
+
+func refillProjections(m *Model) {
+	rows := []table.Row{m.math.Labels}
+	for i := range m.math.Columns {
+		rows = append(rows, m.math.Columns[i].Column())
+	}
+	rows = transpose(rows)
+	m.projections.SetRows(rows)
+}
+
+func transpose(slice []table.Row) []table.Row {
+	xl := len(slice[0])
+	yl := len(slice)
+	result := make([]table.Row, xl)
+	for i := range result {
+		result[i] = make(table.Row, yl)
+	}
+	for i := 0; i < xl; i++ {
+		for j := 0; j < yl; j++ {
+			result[i][j] = slice[j][i]
+		}
+	}
+	return result
+}
+
+func (m Model) View() string {
+	var s string
+	indicator := ""
+	if m.refreshing {
+		indicator = m.indicator.View()
+	}
+	right := lipgloss.JoinVertical(
+		lipgloss.Center,
+		baseStyle.Render(m.properties.View()),
+		indicator,
+	)
+	s += lipgloss.JoinHorizontal(
+		lipgloss.Center,
+		right,
+		baseStyle.Render(m.projections.View()),
+	)
+	return s + "\n"
+}
+
+var baseStyle = lipgloss.NewStyle().
+	BorderStyle(lipgloss.RoundedBorder()).
+	BorderForeground(lipgloss.Color("240"))
--- a/tui/tui.go
+++ b/tui/tui.go
@@ -1,168 +1,100 @@
 package tui

 import (
-	"context"
 	"fmt"
-	"time"

+	"code.humancabbage.net/sam/moonmath/coindesk"
 	"code.humancabbage.net/sam/moonmath/config"
-	"code.humancabbage.net/sam/moonmath/moon"
-	"github.com/charmbracelet/bubbles/spinner"
-	"github.com/charmbracelet/bubbles/table"
+	"code.humancabbage.net/sam/moonmath/tui/asset"
 	tea "github.com/charmbracelet/bubbletea"
 	"github.com/charmbracelet/lipgloss"
 )

 type Model struct {
-	math moon.Math
-
-	reloading bool
-	indicator spinner.Model
-
-	prices      table.Model
-	projections table.Model
+	assets []asset.Model
 }

-func New(cfg config.Data) Model {
-	math := moon.NewMath(
-		cfg.Asset,
-		cfg.Goals,
-		config.GetBases(&cfg))
-
-	tableStyle := table.DefaultStyles()
-	tableStyle.Selected = lipgloss.NewStyle()
-	prices := table.New(
-		table.WithColumns([]table.Column{
-			{Title: "Asset", Width: 6},
-			{Title: "Price", Width: 9},
-		}),
-		table.WithHeight(1),
-		table.WithStyles(tableStyle),
+func New(assets []coindesk.Asset, cfg config.All) (p *tea.Program) {
+	model := newModel(assets, cfg)
+	p = tea.NewProgram(
+		model,
+		tea.WithAltScreen(),
+		tea.WithFPS(30),
 	)
+	return
+}

-	projectionCols := []table.Column{
-		{Title: "Labels", Width: 8},
-	}
-	for i := range math.Columns {
-		projectionCols = append(projectionCols,
-			table.Column{
-				Title: math.Columns[i].Base.Label(),
-				Width: 10,
-			})
-	}
-	projections := table.New(
-		table.WithColumns(projectionCols),
-		table.WithHeight(len(math.Labels)),
-		table.WithStyles(tableStyle),
-	)
-
-	indicator := spinner.New()
-	indicator.Spinner = spinner.Points
-	indicator.Style = lipgloss.NewStyle().
-		Foreground(lipgloss.Color("69"))
-
-	return Model{
-		math:        math,
-		indicator:   indicator,
-		prices:      prices,
-		projections: projections,
+func newModel(assets []coindesk.Asset, cfg config.All) (m Model) {
+	// construct models for each asset, but don't filter out dupes
+	seen := map[coindesk.Asset]struct{}{}
+	for _, a := range assets {
+		_, ok := seen[a]
+		if ok {
+			continue
+		}
+		assetCfg := cfg.GetData(a)
+		assetModel := asset.New(assetCfg)
+		m.assets = append(m.assets, assetModel)
+		seen[a] = struct{}{}
 	}
+	return
 }

 func (m Model) Init() tea.Cmd {
-	return tea.Batch(
-		m.indicator.Tick,
-		func() tea.Msg {
-			return refresh{}
-		},
-	)
+	// initialize child models, collecting their commands,
+	// then return them all in a batch
+	var inits []tea.Cmd
+	for i := range m.assets {
+		cmd := m.assets[i].Init()
+		inits = append(inits, cmd)
+	}
+	return tea.Batch(inits...)
 }

 func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	switch msg := msg.(type) {
-	case refresh:
-		m.reloading = true
-		return m, func() tea.Msg {
-			_ = m.math.Refresh(context.TODO())
-			return m.math
-		}
-	case moon.Math:
-		m.math = msg
-		m.reloading = false
-		refillPrice(&m)
-		refillProjections(&m)
-		return m, tea.Tick(time.Second*30,
-			func(t time.Time) tea.Msg {
-				return refresh{}
-			})
-	case spinner.TickMsg:
-		var cmd tea.Cmd
-		m.indicator, cmd = m.indicator.Update(msg)
-		return m, cmd
+	// handle keys for quitting
 	case tea.KeyMsg:
 		switch msg.String() {
 		case "ctrl+c", "q", "esc":
 			return m, tea.Quit
 		}
+	// forward asset messages to the appropriate model
+	case asset.Msg:
+		cmd := m.forward(msg.Asset, msg)
+		return m, cmd
+	// forward any other message to each child model.
+	// typically, this is for animation.
+	default:
+		var commands []tea.Cmd
+		for i := range m.assets {
+			var cmd tea.Cmd
+			m.assets[i], cmd = m.assets[i].Update(msg)
+			commands = append(commands, cmd)
+		}
+		return m, tea.Batch(commands...)
 	}
 	return m, nil
 }

-type refresh struct{}
-
-func refillPrice(m *Model) {
-	rows := []table.Row{
-		[]string{
-			string(m.math.Asset),
-			fmt.Sprintf("$%0.2f", m.math.CurrentPrice),
-		},
-	}
-	m.prices.SetRows(rows)
-}
-
-func refillProjections(m *Model) {
-	rows := []table.Row{m.math.Labels}
-	for i := range m.math.Columns {
-		rows = append(rows, m.math.Columns[i].Column())
-	}
-	rows = transpose(rows)
-	m.projections.SetRows(rows)
-}
-
-func transpose(slice []table.Row) []table.Row {
-	xl := len(slice[0])
-	yl := len(slice)
-	result := make([]table.Row, xl)
-	for i := range result {
-		result[i] = make(table.Row, yl)
-	}
-	for i := 0; i < xl; i++ {
-		for j := 0; j < yl; j++ {
-			result[i][j] = slice[j][i]
-		}
-	}
-	return result
-}
-
 func (m Model) View() string {
-	var s string
-	indicator := ""
-	if m.reloading {
-		indicator = m.indicator.View()
+	var ss []string
+	for i := range m.assets {
+		s := m.assets[i].View()
+		ss = append(ss, s)
 	}
-	right := lipgloss.JoinVertical(
-		lipgloss.Center,
-		baseStyle.Render(m.prices.View()),
-		indicator,
-	)
-	s += lipgloss.JoinHorizontal(
-		lipgloss.Top,
-		right,
-		baseStyle.Render(m.projections.View()),
-	)
-	return s + "\n"
+	r := lipgloss.JoinVertical(lipgloss.Center, ss...)
+	return r
 }

-var baseStyle = lipgloss.NewStyle().
-	BorderStyle(lipgloss.NormalBorder()).
-	BorderForeground(lipgloss.Color("240"))
+func (m Model) forward(a coindesk.Asset, msg tea.Msg) (cmd tea.Cmd) {
+	// O(n) is fine when n is small
+	for i := range m.assets {
+		if !m.assets[i].Handles(a) {
+			continue
+		}
+		m.assets[i], cmd = m.assets[i].Update(msg)
+		return
+	}
+	panic(fmt.Errorf("rogue message: %v", msg))
+}
Author	SHA1	Message	Date
Sam Fredrickson	c4dde38d23	Support multiple assets simultaneously. All checks were successful Build & Test / Main (push) Successful in 1m0s Details Release / Release (push) Successful in 1m1s Details	2024-03-22 17:43:15 -07:00
Sam Fredrickson	e14f0488c5	Add config defaults for more assets. All checks were successful Build & Test / Main (push) Successful in 59s Details * LTC * SOL * XRP * DOGE * ADA	2024-03-21 22:07:16 -07:00
Sam Fredrickson	7222c4e26b	Update screenshot. All checks were successful Build & Test / Main (push) Successful in 57s Details Release / Release (push) Successful in 1m1s Details	2024-03-21 17:45:27 -07:00
Sam Fredrickson	ba4e803eea	Make it easier to use other assets. All checks were successful Build & Test / Main (push) Successful in 1m31s Details	2024-03-21 17:39:10 -07:00
Sam Fredrickson	ac1a4bc16c	Move Bitcoinity websocket capture, and test it. All checks were successful Build & Test / Main (push) Successful in 57s Details	2024-03-21 12:31:40 -07:00
Sam Fredrickson	52508b7d4e	'go mod tidy' All checks were successful Build & Test / Main (push) Successful in 1m25s Details	2024-03-21 12:23:17 -07:00
Sam Fredrickson	4a40899653	Spice up the UI. All checks were successful Build & Test / Main (push) Successful in 58s Details * Full screen ("alt framebuffer"). * Rounded borders. * Spinner that starts during a refresh. * Colored table headers. * Table header have bottom border.	2024-03-21 02:43:38 -07:00