cmdlist.go

package boa

import (
	"fmt"
	"io"

	"github.com/charmbracelet/bubbles/list"
	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"
	"github.com/spf13/cobra"
)

// screen length of the list
const listHeight = 10

// item is the object that will appear in our list
type item struct {
	cmd *cobra.Command
}

func (i item) FilterValue() string { return i.cmd.Name() }

// itemDelegate encapsulates the general functionality for all list items
type itemDelegate struct {
	styles *Styles
}

func (d itemDelegate) Height() int                               { return 1 }
func (d itemDelegate) Spacing() int                              { return 0 }
func (d itemDelegate) Update(msg tea.Msg, m *list.Model) tea.Cmd { return nil }
func (d itemDelegate) Render(w io.Writer, m list.Model, index int, listItem list.Item) {
	i, ok := listItem.(item)
	if !ok {
		return
	}
	str := i.cmd.Name() + lipgloss.NewStyle().Bold(true).
		PaddingLeft(i.cmd.NamePadding()-len(i.cmd.Name())+1).Render(i.cmd.Short)

	fn := d.styles.Item.Render
	if index == m.Index() {
		fn = func(strs ...string) string {
			return d.styles.SelectedItem.Render("> " + fmt.Sprint(strs))
		}
	}
	fmt.Fprint(w, fn(str))
}

// newSubCmdsList returns a new list.Model filled with the values in []list.Items
func newSubCmdsList(styles *Styles, items []list.Item) list.Model {
	l := list.New(items, itemDelegate{styles: styles}, 0, listHeight)
	l.Styles.TitleBar.Padding(0, 0)
	l.Styles.Title = styles.Section
	l.Title = "Available Sub Commands:"
	l.SetShowHelp(false)
	l.SetShowStatusBar(false)
	l.SetShowPagination(false)
	return l
}

// getSubCommands returns a []list.Item filled with any available sub command from the supplied *cobra.Command.
// This does not follow the command chain past a depth of 1.
func getSubCommands(c *cobra.Command) []list.Item {
	subs := make([]list.Item, 0)
	if c.HasAvailableSubCommands() {
		for _, subcmd := range c.Commands() {
			if subcmd.Name() == "help" || subcmd.IsAvailableCommand() {
				subs = append(subs, item{cmd: subcmd})
			}
		}
	}
	return subs
}