145 lines
3.0 KiB
Go
145 lines
3.0 KiB
Go
package pages
|
|
|
|
import (
|
|
"log/slog"
|
|
"slices"
|
|
"tasksquire/common"
|
|
"tasksquire/components/picker"
|
|
"tasksquire/taskwarrior"
|
|
|
|
"github.com/charmbracelet/bubbles/key"
|
|
"github.com/charmbracelet/bubbles/list"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"github.com/charmbracelet/lipgloss"
|
|
)
|
|
|
|
type ContextPickerPage struct {
|
|
common *common.Common
|
|
contexts taskwarrior.Contexts
|
|
picker *picker.Picker
|
|
}
|
|
|
|
func NewContextPickerPage(common *common.Common) *ContextPickerPage {
|
|
p := &ContextPickerPage{
|
|
common: common,
|
|
contexts: common.TW.GetContexts(),
|
|
}
|
|
|
|
selected := common.TW.GetActiveContext().Name
|
|
|
|
itemProvider := func() []list.Item {
|
|
contexts := common.TW.GetContexts()
|
|
options := make([]string, 0)
|
|
for _, c := range contexts {
|
|
if c.Name != "none" {
|
|
options = append(options, c.Name)
|
|
}
|
|
}
|
|
slices.Sort(options)
|
|
options = append([]string{"(none)"}, options...)
|
|
|
|
items := []list.Item{}
|
|
for _, opt := range options {
|
|
items = append(items, picker.NewItem(opt))
|
|
}
|
|
return items
|
|
}
|
|
|
|
onSelect := func(item list.Item) tea.Cmd {
|
|
return func() tea.Msg { return contextSelectedMsg{item: item} }
|
|
}
|
|
|
|
p.picker = picker.New(common, "Contexts", itemProvider, onSelect)
|
|
|
|
// Set active context
|
|
if selected == "" {
|
|
selected = "(none)"
|
|
}
|
|
p.picker.SelectItemByFilterValue(selected)
|
|
|
|
p.SetSize(common.Width(), common.Height())
|
|
|
|
return p
|
|
}
|
|
|
|
func (p *ContextPickerPage) SetSize(width, height int) {
|
|
p.common.SetSize(width, height)
|
|
|
|
// Set list size with some padding/limits to look like a picker
|
|
listWidth := width - 4
|
|
if listWidth > 40 {
|
|
listWidth = 40
|
|
}
|
|
listHeight := height - 6
|
|
if listHeight > 20 {
|
|
listHeight = 20
|
|
}
|
|
p.picker.SetSize(listWidth, listHeight)
|
|
}
|
|
|
|
func (p *ContextPickerPage) Init() tea.Cmd {
|
|
return p.picker.Init()
|
|
}
|
|
|
|
type contextSelectedMsg struct {
|
|
item list.Item
|
|
}
|
|
|
|
func (p *ContextPickerPage) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
var cmd tea.Cmd
|
|
|
|
switch msg := msg.(type) {
|
|
case tea.WindowSizeMsg:
|
|
p.SetSize(msg.Width, msg.Height)
|
|
case contextSelectedMsg:
|
|
name := msg.item.FilterValue() // Use FilterValue (which is the name/text)
|
|
if name == "(none)" {
|
|
name = ""
|
|
}
|
|
|
|
ctx := p.common.TW.GetContext(name)
|
|
|
|
model, err := p.common.PopPage()
|
|
if err != nil {
|
|
slog.Error("page stack empty")
|
|
return nil, tea.Quit
|
|
}
|
|
return model, func() tea.Msg { return UpdateContextMsg(ctx) }
|
|
case tea.KeyMsg:
|
|
if !p.picker.IsFiltering() {
|
|
switch {
|
|
case key.Matches(msg, p.common.Keymap.Back):
|
|
model, err := p.common.PopPage()
|
|
if err != nil {
|
|
slog.Error("page stack empty")
|
|
return nil, tea.Quit
|
|
}
|
|
return model, BackCmd
|
|
}
|
|
}
|
|
}
|
|
|
|
_, cmd = p.picker.Update(msg)
|
|
return p, cmd
|
|
}
|
|
|
|
func (p *ContextPickerPage) View() string {
|
|
width := p.common.Width() - 4
|
|
if width > 40 {
|
|
width = 40
|
|
}
|
|
|
|
content := p.picker.View()
|
|
styledContent := lipgloss.NewStyle().Width(width).Render(content)
|
|
|
|
return lipgloss.Place(
|
|
p.common.Width(),
|
|
p.common.Height(),
|
|
lipgloss.Center,
|
|
lipgloss.Center,
|
|
p.common.Styles.Base.Render(styledContent),
|
|
)
|
|
}
|
|
|
|
type UpdateContextMsg *taskwarrior.Context
|