spilltea/internal/ui/plugins/model.go

package plugins

import (
	"strings"

	"charm.land/bubbles/v2/help"
	"charm.land/bubbles/v2/key"
	"charm.land/bubbles/v2/paginator"
	"charm.land/bubbles/v2/textarea"
	"charm.land/bubbles/v2/textinput"
	"charm.land/bubbles/v2/viewport"
	tea "charm.land/bubbletea/v2"
	"github.com/anotherhadi/spilltea/internal/keys"
	"github.com/anotherhadi/spilltea/internal/plugins"
	"github.com/anotherhadi/spilltea/internal/style"
)

type Model struct {
	manager  *plugins.Manager
	items    []plugins.Info
	cursor   int
	editing  bool
	filter   string
	filtered []plugins.Info

	listViewport   viewport.Model
	detailViewport viewport.Model
	textarea       textarea.Model
	filterInput    textinput.Model
	filtering      bool
	pager          paginator.Model
	help           help.Model

	width  int
	height int
}

func New(mgr *plugins.Manager) Model {
	ta := style.NewTextarea(true)
	ta.Placeholder = "plugin configuration..."
	ta.Blur()

	fi := textinput.New()
	fi.Prompt = ""

	return Model{
		manager:        mgr,
		listViewport:   style.NewViewport(),
		detailViewport: style.NewViewport(),
		textarea:       ta,
		filterInput:    fi,
		pager:          style.NewPaginator(),
		help:           style.NewHelp(),
	}
}

func (m Model) Init() tea.Cmd { return nil }

func (m Model) IsEditing() bool { return m.editing || m.filtering }

func (m *Model) SetSize(w, h int) {
	m.width = w
	m.height = h
	m.recalcSizes()
}

func (m *Model) recalcSizes() {
	if m.width == 0 {
		return
	}
	m.help.SetWidth(m.width - 2)

	listH, detailH := style.SplitH(m.height, m.renderStatusBar(), 0.4)

	inner := m.width - 2
	if inner < 0 {
		inner = 0
	}

	listVH := style.PanelContentH(listH) - 1 // -1 for the pager dots row
	if listVH < 0 {
		listVH = 0
	}
	m.listViewport.SetWidth(inner)
	m.listViewport.SetHeight(listVH)
	m.pager.PerPage = listVH
	if m.pager.PerPage < 1 {
		m.pager.PerPage = 1
	}

	m.filterInput.SetWidth(inner - 2)

	detailContentH := style.PanelContentH(detailH)
	const headerH = 2
	const configFixedH = 2 // blank line + label line
	textareaH := max(3, detailContentH/3)
	if textareaH > 12 {
		textareaH = 12
	}
	configTotalH := 0
	if m.hasConfig() {
		configTotalH = configFixedH + textareaH
	}
	descVH := max(1, detailContentH-headerH-configTotalH)

	m.detailViewport.SetWidth(inner)
	m.detailViewport.SetHeight(descVH)
	m.textarea.SetWidth(max(1, inner-2))
	m.textarea.SetHeight(max(3, textareaH))

	m.refreshListViewport()
	m.syncDetailViewport()
}

// Refresh reloads the plugin list from the manager.
func (m *Model) Refresh() {
	if m.manager == nil {
		return
	}
	pl := m.manager.GetPlugins()
	m.items = make([]plugins.Info, len(pl))
	for i, p := range pl {
		m.items[i] = p.Info()
	}
	m.applyFilter()
}

func (m *Model) applyFilter() {
	if m.filter == "" {
		m.filtered = m.items
	} else {
		f := strings.ToLower(m.filter)
		filtered := make([]plugins.Info, 0, len(m.items))
		for _, p := range m.items {
			if strings.Contains(strings.ToLower(p.Name), f) {
				filtered = append(filtered, p)
			}
		}
		m.filtered = filtered
	}
	m.pager.SetTotalPages(len(m.filtered))
	if m.cursor >= len(m.filtered) {
		m.cursor = max(0, len(m.filtered)-1)
	}
	m.refreshListViewport()
	m.syncTextarea()
	m.syncDetailViewport()
}

func (m *Model) selected() (plugins.Info, bool) {
	if len(m.filtered) == 0 {
		return plugins.Info{}, false
	}
	return m.filtered[m.cursor], true
}

func (m *Model) hasConfig() bool {
	info, ok := m.selected()
	if !ok {
		return false
	}
	_, has := info.Hooks["on_config"]
	return has
}

func (m *Model) syncTextarea() {
	if m.editing {
		return
	}
	info, ok := m.selected()
	if !ok {
		m.textarea.SetValue("")
		return
	}
	m.textarea.SetValue(info.ConfigText)
}

func (m *Model) syncDetailViewport() {
	info, ok := m.selected()
	if !ok || info.Description == "" {
		m.detailViewport.SetContent("")
		return
	}
	desc := renderPluginDescription(info.Description, m.width-6)
	m.detailViewport.SetContent(desc)
}

func (m *Model) refreshListViewport() {
	if m.pager.PerPage > 0 {
		if len(m.filtered) == 0 {
			m.pager.Page = 0
			m.pager.TotalPages = 0
		} else {
			m.pager.Page = m.cursor / m.pager.PerPage
			m.pager.SetTotalPages(len(m.filtered))
		}
	}
	m.listViewport.SetContent(m.renderList())
}

type pluginsKeyMap struct {
	editing   bool
	hasConfig bool
	width     int
}

func (k pluginsKeyMap) ShortHelp() []key.Binding {
	pk := keys.Keys.Plugins
	g := keys.Keys.Global
	if k.editing {
		esc := key.NewBinding(key.WithKeys(g.Escape.Keys()...), key.WithHelp(g.Escape.Help().Key, "save & exit"))
		return []key.Binding{esc}
	}
	scrollHint := key.NewBinding(
		key.WithKeys(g.ScrollUp.Keys()...),
		key.WithHelp(g.ScrollUp.Help().Key+"/"+g.ScrollDown.Help().Key, "scroll detail"),
	)
	if k.hasConfig {
		return []key.Binding{pk.Toggle, pk.EditConfig, scrollHint, g.Help}
	}
	return []key.Binding{pk.Toggle, scrollHint, g.Help}
}

func (k pluginsKeyMap) FullHelp() [][]key.Binding {
	g := keys.Keys.Global
	if k.editing {
		return [][]key.Binding{k.ShortHelp()}
	}
	pk := keys.Keys.Plugins
	pageGlobals := []key.Binding{g.Up, g.Down, g.ScrollUp, g.ScrollDown, g.Escape}
	all := []key.Binding{pk.Toggle, pk.EditConfig, pk.Filter}
	all = append(all, pageGlobals...)
	all = append(all, g.CommonBindings()...)
	return keys.ChunkByWidth(all, k.width)
}