116 lines
2.5 KiB
Go
116 lines
2.5 KiB
Go
|
package scenes
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/charmbracelet/bubbles/help"
|
||
|
|
"github.com/charmbracelet/bubbles/list"
|
||
|
|
tea "github.com/charmbracelet/bubbletea"
|
||
|
|
"github.com/charmbracelet/lipgloss"
|
||
|
|
"io"
|
||
|
|
)
|
||
|
|
|
||
|
|
type mainMenu struct {
|
||
|
|
root RootModel
|
||
|
|
help help.Model
|
||
|
|
inputStyle lipgloss.Style
|
||
|
|
lastKey string
|
||
|
|
quitting bool
|
||
|
|
list list.Model
|
||
|
|
}
|
||
|
|
|
||
|
|
type menuItem struct {
|
||
|
|
Title string
|
||
|
|
ModelFn func(model RootModel) tea.Model
|
||
|
|
}
|
||
|
|
|
||
|
|
func (i menuItem) FilterValue() string { return i.Title }
|
||
|
|
|
||
|
|
type itemDelegate struct{}
|
||
|
|
|
||
|
|
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.(menuItem)
|
||
|
|
if !ok {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
style := lipgloss.NewStyle().PaddingLeft(2)
|
||
|
|
|
||
|
|
str := style.Render("o " + i.Title)
|
||
|
|
if index == m.Index() {
|
||
|
|
str = style.Foreground(lipgloss.Color("202")).Render("• " + i.Title)
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Fprintf(w, str)
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewMainMenu(root RootModel) tea.Model {
|
||
|
|
items := []list.Item{
|
||
|
|
menuItem{
|
||
|
|
Title: "Installations",
|
||
|
|
ModelFn: NewInstallations,
|
||
|
|
},
|
||
|
|
menuItem{
|
||
|
|
Title: "Profiles",
|
||
|
|
ModelFn: NewProfiles,
|
||
|
|
},
|
||
|
|
menuItem{
|
||
|
|
Title: "Mods",
|
||
|
|
ModelFn: NewMods,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
l := list.NewModel(items, itemDelegate{}, 20, 14)
|
||
|
|
l.SetShowStatusBar(false)
|
||
|
|
l.SetFilteringEnabled(false)
|
||
|
|
l.SetShowTitle(false)
|
||
|
|
l.Styles.PaginationStyle = list.DefaultStyles().PaginationStyle.PaddingLeft(2)
|
||
|
|
l.Styles.HelpStyle = list.DefaultStyles().HelpStyle.PaddingLeft(2).PaddingBottom(1)
|
||
|
|
|
||
|
|
return mainMenu{
|
||
|
|
root: root,
|
||
|
|
list: l,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m mainMenu) Init() tea.Cmd {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m mainMenu) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||
|
|
switch msg := msg.(type) {
|
||
|
|
case tea.KeyMsg:
|
||
|
|
switch keypress := msg.String(); keypress {
|
||
|
|
case "ctrl+c":
|
||
|
|
fallthrough
|
||
|
|
case "q":
|
||
|
|
m.quitting = true
|
||
|
|
return m, tea.Quit
|
||
|
|
case "enter":
|
||
|
|
i, ok := m.list.SelectedItem().(menuItem)
|
||
|
|
if ok {
|
||
|
|
if i.ModelFn != nil {
|
||
|
|
m.root.ChangeScene(i.ModelFn(m.root))
|
||
|
|
return m, nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return m, tea.Quit
|
||
|
|
default:
|
||
|
|
var cmd tea.Cmd
|
||
|
|
m.list, cmd = m.list.Update(msg)
|
||
|
|
return m, cmd
|
||
|
|
}
|
||
|
|
case tea.WindowSizeMsg:
|
||
|
|
top, right, bottom, left := lipgloss.NewStyle().Margin(1, 2).GetMargin()
|
||
|
|
m.list.SetSize(msg.Width-left-right, msg.Height-top-bottom)
|
||
|
|
}
|
||
|
|
|
||
|
|
return m, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m mainMenu) View() string {
|
||
|
|
return m.list.View()
|
||
|
|
}
|