wip markets

economic_calendar.go

@@ -0,0 +1,139 @@
+package mujica
+
+import (
+	"cmp"
+	"fmt"
+	"log/slog"
+	"slices"
+	"strings"
+	"time"
+
+	"github.com/egoist/mygo/ui"
+
+	"mujicaui/internal"
+	"mujicaui/theme"
+)
+
+// EconomicEvent is a scheduled release; Actual stays empty until published.
+type EconomicEvent struct {
+	ID, Country, Title string
+	Time               time.Time
+	// Importance is 1 (low) to 3 (high).
+	Importance                 int
+	Forecast, Previous, Actual string
+}
+
+// EconomicCalendarState is the country filter (empty shows all) and the table's state.
+type EconomicCalendarState struct {
+	Countries []string
+	// MinImportance hides events below it; zero shows all.
+	MinImportance int
+	Table         DataTableState[string]
+	cache         economicCache
+}
+
+type economicCache struct {
+	key  string
+	rows []EconomicEvent
+}
+
+// EconomicCalendarOptions configure an EconomicCalendar.
+type EconomicCalendarOptions struct {
+	Version  int
+	Status   DataStatus
+	Error    string
+	Disabled bool
+}
+
+// EconomicCalendar lists economic events with importance, forecast, previous and actual, filtered by country.
+func EconomicCalendar(c *ui.Context, s *EconomicCalendarState, events []EconomicEvent, opts EconomicCalendarOptions) DataTableView {
+	if s == nil {
+		panic("mujica: EconomicCalendar needs a state")
+	}
+	if s.MinImportance < 0 || s.MinImportance > 3 {
+		panic("mujica: EconomicCalendar MinImportance must be 0..3")
+	}
+	u := Density(c).Unit()
+	var v DataTableView
+	col := ui.Column(c).Gap(u * 2).MinHeight(0).Grow(1).Disabled(opts.Disabled)
+	col.Children(func() {
+		var countries []SelectOption[string]
+		for _, e := range events {
+			if e.ID == "" || e.Importance < 1 || e.Importance > 3 {
+				panic("mujica: EconomicEvent needs an ID and Importance 1..3")
+			}
+			if !slices.ContainsFunc(countries, func(o SelectOption[string]) bool { return o.Value == e.Country }) {
+				countries = append(countries, SelectOption[string]{Value: e.Country, Label: e.Country})
+			}
+		}
+		slices.SortFunc(countries, func(a, b SelectOption[string]) int { return cmp.Compare(a.Value, b.Value) })
+		ui.Row(c).Gap(u * 3).AlignItems(ui.Center).Wrap().Children(func() {
+			MultiSelect(c, &s.Countries, countries, MultiSelectOptions{Placeholder: msg(c, "economic_calendar.all_countries"),
+				Label: msg(c, "economic_calendar.countries"), Disabled: opts.Disabled}).Element.MinWidth(220)
+			imp := []RadioItem[int]{{Value: 0, Label: msg(c, "economic_calendar.all")}, {Value: 2, Label: msg(c, "economic_calendar.medium_up")},
+				{Value: 3, Label: msg(c, "economic_calendar.high")}}
+			RadioGroup(c, &s.MinImportance, imp, RadioGroupOptions{Horizontal: true, Label: msg(c, "economic_calendar.importance"), Disabled: opts.Disabled})
+		})
+		rows := economicFilter(s, events, opts.Version)
+		cols := []DataColumn[EconomicEvent]{
+			{ID: "time", Title: msg(c, "economic_calendar.time"), Width: 130, Sortable: true,
+				Compare: func(a, b EconomicEvent) int { return a.Time.Compare(b.Time) },
+				Text:    func(e EconomicEvent) string { return e.Time.Format("01-02 15:04") }},
+			{ID: "country", Title: msg(c, "economic_calendar.country"), Width: 80, Sortable: true,
+				Compare: func(a, b EconomicEvent) int { return cmp.Compare(a.Country, b.Country) },
+				Text:    func(e EconomicEvent) string { return e.Country }},
+			{ID: "importance", Title: msg(c, "economic_calendar.importance"), Width: 110, Sortable: true,
+				Compare: func(a, b EconomicEvent) int { return cmp.Compare(a.Importance, b.Importance) },
+				Text:    func(e EconomicEvent) string { return msg(c, fmt.Sprintf("economic_calendar.level_%d", e.Importance)) },
+				Cell:    func(e EconomicEvent) { economicImportance(c, e.Importance) }},
+			{ID: "event", Title: msg(c, "economic_calendar.event"), Text: func(e EconomicEvent) string { return e.Title }},
+			{ID: "actual", Title: msg(c, "economic_calendar.actual"), Width: 90, Align: ui.End,
+				Text: func(e EconomicEvent) string { return cmp.Or(e.Actual, "—") },
+				Cell: func(e EconomicEvent) {
+					ui.Text(c, cmp.Or(e.Actual, "—")).FontWeight(700).FontFeatures("tnum").JustifySelf(ui.End).FillWidth().TextAlign(ui.End)
+				}},
+			{ID: "forecast", Title: msg(c, "economic_calendar.forecast"), Width: 90, Align: ui.End, Text: func(e EconomicEvent) string { return cmp.Or(e.Forecast, "—") }},
+			{ID: "previous", Title: msg(c, "economic_calendar.previous"), Width: 90, Align: ui.End, Text: func(e EconomicEvent) string { return cmp.Or(e.Previous, "—") }},
+		}
+		v = DataTable(c, &s.Table, rows, DataTableOptions[EconomicEvent, string]{Columns: cols, Key: func(e EconomicEvent) string { return e.ID },
+			Status: opts.Status, Error: opts.Error, Empty: msg(c, "economic_calendar.empty")})
+		v.Element.Grow(1)
+	})
+	v.Element = col
+	return v
+}
+
+// economicImportance draws one to three diamonds and names the level.
+func economicImportance(c *ui.Context, level int) {
+	k := Tokens(c)
+	word := msg(c, fmt.Sprintf("economic_calendar.level_%d", level))
+	ui.Row(c).Gap(4).AlignItems(ui.Center).Label(word).Children(func() {
+		ui.Box(c).Size(34, 10).Draw(func(p *ui.Painter, r ui.Rect) {
+			for i := range 3 {
+				d := internal.Diamond(r.X+5+float32(i)*12, r.Y+5, 9)
+				if i < level {
+					p.FillPath(d, k.AccentText)
+				} else {
+					p.StrokePath(d, 1, k.ControlBorder)
+				}
+			}
+		})
+		ui.Text(c, word).FontSize(FontSize(c, theme.CaptionSize)).TextColor(k.TextMuted)
+	})
+}
+
+// economicFilter keeps events of the chosen countries and importance, only refiltering on change.
+func economicFilter(s *EconomicCalendarState, events []EconomicEvent, version int) []EconomicEvent {
+	key := fmt.Sprint(chartID(events), version, s.MinImportance, strings.Join(s.Countries, "\x00"))
+	if key == s.cache.key {
+		return s.cache.rows
+	}
+	s.cache.key, s.cache.rows = key, nil
+	for _, e := range events {
+		if e.Importance >= s.MinImportance && (len(s.Countries) == 0 || slices.Contains(s.Countries, e.Country)) {
+			s.cache.rows = append(s.cache.rows, e)
+		}
+	}
+	slog.Debug("mujica: economic_calendar filter", "shown", len(s.cache.rows))
+	return s.cache.rows
+}