feat: add UI element for periodic arguments (#25)

accounting/filter.go

@@ -25,8 +25,6 @@ func NewFilter(filterType FilterType, value interface{}) Filter {
 		return NewAccountFilter(value.(string))
 	case FilterTypeDate:
 		return NewDateFilter()
-	case FilterTypePeriod:
-		return NewPeriodFilter()
 	case FilterTypeAccountDepth:
 		return NewAccountDepthFilter()
 	case FilterTypeStartDate:

ui/keys.go

@@ -22,15 +22,20 @@ type keyMap struct {
 	ResetFilters      key.Binding
 	AcctDepthDecr     key.Binding
 	AcctDepthIncr     key.Binding
-	Yearly            key.Binding
 	Monthly           key.Binding
+	Quarterly         key.Binding
+	Yearly            key.Binding
 }
 
 var allKeys = keyMap{
 	Monthly: key.NewBinding(
 		key.WithKeys("m", "m"),
 		key.WithHelp("m", "monthly period"),
 	),
+	Quarterly: key.NewBinding(
+		key.WithKeys("u", "u"),
+		key.WithHelp("u", "quarterly period"),
+	),
 	Yearly: key.NewBinding(
 		key.WithKeys("y", "y"),
 		key.WithHelp("y", "yearly period"),

ui/model.go

@@ -27,9 +27,9 @@ type model struct {
 	isFormDisplay        bool
 	filterGroup          *filterGroup
 	spinner              spinner.Model
+	period               *Period
 
 	searchFilter             accounting.Filter
-	periodFilter             accounting.Filter
 	acctDepth                accounting.AccountDepthFilter
 	isTxnsSortedByMostRecent bool
 
@@ -53,8 +53,8 @@ func newModel(hlcmd accounting.HledgerCmd) *model {
 		quitting:                 false,
 		isFormDisplay:            false,
 		filterGroup:              newFilterGroup(),
+		period:                   newPeriod(),
 		searchFilter:             accounting.NoFilter{},
-		periodFilter:             accounting.NewPeriodFilter().Yearly(),
 		acctDepth:                accounting.NewAccountDepthFilter(),
 		isTxnsSortedByMostRecent: true,
 		width:                    0,
@@ -121,12 +121,14 @@ func (m *model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		case key.Matches(msg, m.help.keys.Search):
 			form := newFilterForm(m, searchFilter)
 			return form.Update(nil)
-		case key.Matches(msg, m.help.keys.Yearly):
-			m.periodFilter = accounting.NewPeriodFilter().Yearly()
-			return m, m.refresh()
-		case key.Matches(msg, m.help.keys.Monthly):
-			m.periodFilter = accounting.NewPeriodFilter().Monthly()
+
+		case key.Matches(msg, m.help.keys.Yearly),
+			key.Matches(msg, m.help.keys.Monthly),
+			key.Matches(msg, m.help.keys.Quarterly):
+			p, _ := m.period.Update(msg)
+			m.period = p.(*Period)
 			return m, m.refresh()
+
 		case key.Matches(msg, m.help.keys.ResetFilters):
 			m.resetFilters()
 			return m, m.refresh()
@@ -229,6 +231,7 @@ func (m *model) View() string {
 				lipgloss.Right,
 				m.tabs.View(),
 				m.filterGroup.View(),
+				m.period.View(),
 			),
 			activeItemStyle.Render(v),
 		),
@@ -259,8 +262,6 @@ func (m *model) updateAllModels(msg tea.Msg) {
 func (m *model) refresh() tea.Cmd {
 	m.msgError = nil // reset the msgError
 
-	pf := m.periodFilter.(accounting.PeriodFilter)
-
 	registerOpts := hledger.NewOptions().
 		WithAccount(m.filterGroup.account.Value()).
 		WithStartDate(m.filterGroup.startDate.Value()).
@@ -271,7 +272,7 @@ func (m *model) refresh() tea.Cmd {
 		WithStartDate(m.filterGroup.startDate.Value()).
 		WithEndDate(m.filterGroup.endDate.Value()).
 		WithAccountDepth(m.acctDepth.RawValue()).
-		WithPeriod(hledger.PeriodType(pf.RawValue()))
+		WithPeriod(hledger.PeriodType(m.period.periodType))
 
 	optsPretty := opts.WithPretty().WithLayout(hledger.LayoutBare).WithAccountDrop(1)
 

ui/period.go

@@ -0,0 +1,82 @@
+package ui
+
+import (
+	tea "github.com/charmbracelet/bubbletea"
+	"github.com/charmbracelet/lipgloss"
+	"github.com/siddhantac/hledger"
+)
+
+type periodType string
+
+const (
+	monthly periodType = "monthly"
+	quarter periodType = "quarterly"
+	yearly  periodType = "yearly"
+)
+
+type Period struct {
+	periodType hledger.PeriodType
+}
+
+func newPeriod() *Period {
+	return &Period{periodType: hledger.PeriodYearly}
+}
+
+func (p *Period) Init() tea.Cmd { return nil }
+
+func (p *Period) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
+	switch msg := msg.(type) {
+	case tea.KeyMsg:
+		switch msg.String() {
+		case "m":
+			p.periodType = hledger.PeriodMonthly
+		case "u":
+			p.periodType = hledger.PeriodQuarterly
+		case "y":
+			p.periodType = hledger.PeriodYearly
+		case "q", "esc":
+			return p, tea.Quit
+		}
+	}
+	return p, nil
+}
+
+func (p *Period) View() string {
+	sectionTitleStyle := lipgloss.NewStyle().
+		MarginTop(1).
+		MarginRight(1).
+		PaddingRight(1).
+		PaddingLeft(1).
+		Foreground(theme.Accent)
+	sectionTitle := sectionTitleStyle.Render("PERIOD")
+
+	inactiveTextStyle := lipgloss.NewStyle().
+		Foreground(theme.PrimaryForeground).
+		MarginRight(2)
+	textStyle := lipgloss.NewStyle().
+		MarginRight(2)
+
+	var monthView, quarterView, yearView string
+	switch p.periodType {
+	case hledger.PeriodMonthly:
+		monthView = textStyle.Render("monthly")
+		quarterView = inactiveTextStyle.Render("quarterly")
+		yearView = inactiveTextStyle.Render("yearly")
+	case hledger.PeriodQuarterly:
+		monthView = inactiveTextStyle.Render("monthly")
+		quarterView = textStyle.Render("quarterly")
+		yearView = inactiveTextStyle.Render("yearly")
+	case hledger.PeriodYearly:
+		monthView = inactiveTextStyle.Render("monthly")
+		quarterView = inactiveTextStyle.Render("quarterly")
+		yearView = textStyle.Render("yearly")
+	}
+
+	return lipgloss.JoinVertical(
+		lipgloss.Right,
+		sectionTitle,
+		monthView,
+		quarterView,
+		yearView,
+	)
+}