ficsit-cli-flake/tea/scenes/mod.go

package scenes

import (
	"time"

	"github.com/charmbracelet/bubbles/key"
	"github.com/charmbracelet/bubbles/list"
	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"
	"github.com/rs/zerolog/log"

	"github.com/satisfactorymodding/ficsit-cli/tea/components"
	"github.com/satisfactorymodding/ficsit-cli/tea/utils"
)

var _ tea.Model = (*modMenu)(nil)

type modMenu struct {
	root   components.RootModel
	list   list.Model
	parent tea.Model
}

func NewModMenu(root components.RootModel, parent tea.Model, mod utils.Mod) tea.Model {
	model := modMenu{
		root:   root,
		parent: parent,
	}

	var items []list.Item
	if root.GetCurrentProfile().HasMod(mod.Reference) {
		items = []list.Item{
			utils.SimpleItem[modMenu]{
				ItemTitle: "Remove Mod",
				Activate: func(msg tea.Msg, currentModel modMenu) (tea.Model, tea.Cmd) {
					root.GetCurrentProfile().RemoveMod(mod.Reference)
					return currentModel.parent, currentModel.parent.Init()
				},
			},
			utils.SimpleItem[modMenu]{
				ItemTitle: "Change Version",
				Activate: func(msg tea.Msg, currentModel modMenu) (tea.Model, tea.Cmd) {
					newModel := NewModVersion(root, currentModel.parent, mod)
					return newModel, newModel.Init()
				},
			},
		}

		if root.GetCurrentProfile().IsModEnabled(mod.Reference) {
			items = append(items, utils.SimpleItem[modMenu]{
				ItemTitle: "Disable Mod",
				Activate: func(msg tea.Msg, currentModel modMenu) (tea.Model, tea.Cmd) {
					root.GetCurrentProfile().SetModEnabled(mod.Reference, false)
					return currentModel.parent, currentModel.parent.Init()
				},
			})
		} else {
			items = append(items, utils.SimpleItem[modMenu]{
				ItemTitle: "Enable Mod",
				Activate: func(msg tea.Msg, currentModel modMenu) (tea.Model, tea.Cmd) {
					root.GetCurrentProfile().SetModEnabled(mod.Reference, true)
					return currentModel.parent, currentModel.parent.Init()
				},
			})
		}
	} else {
		items = []list.Item{
			utils.SimpleItem[modMenu]{
				ItemTitle: "Install Mod",
				Activate: func(msg tea.Msg, currentModel modMenu) (tea.Model, tea.Cmd) {
					err := root.GetCurrentProfile().AddMod(mod.Reference, ">=0.0.0")
					if err != nil {
						log.Error().Err(err).Msg(ErrorFailedAddMod)
						cmd := currentModel.list.NewStatusMessage(ErrorFailedAddMod)
						return currentModel, cmd
					}
					return currentModel.parent, nil
				},
			},
			utils.SimpleItem[modMenu]{
				ItemTitle: "Install Mod with specific version",
				Activate: func(msg tea.Msg, currentModel modMenu) (tea.Model, tea.Cmd) {
					newModel := NewModVersion(root, currentModel.parent, mod)
					return newModel, newModel.Init()
				},
			},
		}
	}

	items = append(items, utils.SimpleItem[modMenu]{
		ItemTitle: "View Mod info",
		Activate: func(msg tea.Msg, currentModel modMenu) (tea.Model, tea.Cmd) {
			newModel := NewModInfo(root, currentModel, mod)
			return newModel, newModel.Init()
		},
	})

	model.list = list.New(items, utils.NewItemDelegate(), root.Size().Width, root.Size().Height-root.Height())
	model.list.SetShowStatusBar(false)
	model.list.SetFilteringEnabled(false)
	model.list.Title = mod.Name
	model.list.Styles = utils.ListStyles
	model.list.SetSize(model.list.Width(), model.list.Height())
	model.list.KeyMap.Quit.SetHelp("q", "back")
	model.list.StatusMessageLifetime = time.Second * 3
	model.list.DisableQuitKeybindings()

	model.list.AdditionalShortHelpKeys = func() []key.Binding {
		return []key.Binding{
			key.NewBinding(key.WithHelp("q", "back")),
		}
	}

	model.list.AdditionalFullHelpKeys = func() []key.Binding {
		return []key.Binding{
			key.NewBinding(key.WithHelp("q", "back")),
		}
	}

	return model
}

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

func (m modMenu) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	switch msg := msg.(type) {
	case tea.KeyMsg:
		switch keypress := msg.String(); keypress {
		case KeyControlC:
			return m, tea.Quit
		case "q":
			if m.parent != nil {
				m.parent.Update(m.root.Size())
				return m.parent, nil
			}
			return m, tea.Quit
		case KeyEnter:
			i, ok := m.list.SelectedItem().(utils.SimpleItem[modMenu])
			if ok {
				if i.Activate != nil {
					newModel, cmd := i.Activate(msg, m)
					if newModel != nil || cmd != nil {
						if newModel == nil {
							newModel = m
						}
						return newModel, cmd
					}
					return m, nil
				}
			}
			return m, nil
		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(2, 2).GetMargin()
		m.list.SetSize(msg.Width-left-right, msg.Height-top-bottom)
		m.root.SetSize(msg)
	}

	return m, nil
}

func (m modMenu) View() string {
	return lipgloss.JoinVertical(lipgloss.Left, m.root.View(), m.list.View())
}