package widgets import ( "github.com/gdamore/tcell" "git.sr.ht/~sircmpwn/aerc/config" "git.sr.ht/~sircmpwn/aerc/lib/ui" ) type Selecter struct { ui.Invalidatable chooser bool focused bool focus int options []string uiConfig config.UIConfig onChoose func(option string) onSelect func(option string) } func NewSelecter(options []string, focus int, uiConfig config.UIConfig) *Selecter { return &Selecter{ focus: focus, options: options, uiConfig: uiConfig, } } func (sel *Selecter) Chooser(chooser bool) *Selecter { sel.chooser = chooser return sel } func (sel *Selecter) Invalidate() { sel.DoInvalidate(sel) } func (sel *Selecter) Draw(ctx *ui.Context) { ctx.Fill(0, 0, ctx.Width(), ctx.Height(), ' ', sel.uiConfig.GetStyle(config.STYLE_SELECTER_DEFAULT)) x := 2 for i, option := range sel.options { style := sel.uiConfig.GetStyle(config.STYLE_SELECTER_DEFAULT) if sel.focus == i { if sel.focused { style = sel.uiConfig.GetStyle(config.STYLE_SELECTER_FOCUSED) } else if sel.chooser { style = sel.uiConfig.GetStyle(config.STYLE_SELECTER_CHOOSER) } } x += ctx.Printf(x, 1, style, "[%s]", option) x += 5 } } func (sel *Selecter) OnChoose(fn func(option string)) *Selecter { sel.onChoose = fn return sel } func (sel *Selecter) OnSelect(fn func(option string)) *Selecter { sel.onSelect = fn return sel } func (sel *Selecter) Selected() string { return sel.options[sel.focus] } func (sel *Selecter) Focus(focus bool) { sel.focused = focus sel.Invalidate() } func (sel *Selecter) Event(event tcell.Event) bool { switch event := event.(type) { case *tcell.EventKey: switch event.Key() { case tcell.KeyCtrlH: fallthrough case tcell.KeyLeft: if sel.focus > 0 { sel.focus-- sel.Invalidate() } if sel.onSelect != nil { sel.onSelect(sel.Selected()) } case tcell.KeyCtrlL: fallthrough case tcell.KeyRight: if sel.focus < len(sel.options)-1 { sel.focus++ sel.Invalidate() } if sel.onSelect != nil { sel.onSelect(sel.Selected()) } case tcell.KeyEnter: if sel.onChoose != nil { sel.onChoose(sel.Selected()) } } } return false }