ficsit-cli-flake/tea/scenes/mods/mod.go
2023-06-15 10:50:10 +02:00

158 lines
4.4 KiB
Go

package mods
import (
"time"
"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/scenes/errors"
"github.com/satisfactorymodding/ficsit-cli/tea/scenes/keys"
"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(errors.ErrorFailedAddMod)
cmd := currentModel.list.NewStatusMessage(errors.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.StatusMessageLifetime = time.Second * 3
model.list.KeyMap.Quit.SetHelp("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 keys.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 keys.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())
}