MujicaUI/economic_calendar.go

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
}