459 lines
12 KiB
Go
459 lines
12 KiB
Go
package scenes
|
|
|
|
import (
|
|
"context"
|
|
"sort"
|
|
"time"
|
|
|
|
"github.com/satisfactorymodding/ficsit-cli/cli"
|
|
|
|
"github.com/charmbracelet/bubbles/key"
|
|
|
|
"github.com/charmbracelet/bubbles/list"
|
|
"github.com/charmbracelet/bubbles/spinner"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"github.com/charmbracelet/lipgloss"
|
|
"github.com/satisfactorymodding/ficsit-cli/ficsit"
|
|
"github.com/satisfactorymodding/ficsit-cli/tea/components"
|
|
"github.com/satisfactorymodding/ficsit-cli/tea/utils"
|
|
)
|
|
|
|
var _ tea.Model = (*modsList)(nil)
|
|
|
|
type sortOrder string
|
|
|
|
const (
|
|
sortOrderAsc sortOrder = "asc"
|
|
sortOrderDesc sortOrder = "desc"
|
|
)
|
|
|
|
const modsTitle = "Mods"
|
|
|
|
type modsList struct {
|
|
root components.RootModel
|
|
list list.Model
|
|
parent tea.Model
|
|
items chan []list.Item
|
|
|
|
sortingField string
|
|
sortingOrder sortOrder
|
|
|
|
showSortFieldList bool
|
|
sortFieldList list.Model
|
|
|
|
showSortOrderList bool
|
|
sortOrderList list.Model
|
|
|
|
err chan string
|
|
error *components.ErrorComponent
|
|
}
|
|
|
|
var _ list.DefaultItem = (*SimpleItemMod[tea.Model])(nil)
|
|
|
|
type SimpleItemMod[T tea.Model] struct {
|
|
utils.SimpleItem[T]
|
|
Mod ficsit.ModsGetModsModsMod
|
|
Context *cli.GlobalContext
|
|
}
|
|
|
|
func (n SimpleItemMod[any]) Title() string {
|
|
if n.Context != nil {
|
|
profile := n.Context.Profiles.Profiles[n.Context.Profiles.SelectedProfile]
|
|
if profile != nil {
|
|
if profile.HasMod(n.Mod.Mod_reference) {
|
|
return lipgloss.NewStyle().Foreground(lipgloss.Color("40")).Render("✓ " + n.ItemTitle)
|
|
}
|
|
}
|
|
}
|
|
|
|
return n.ItemTitle
|
|
}
|
|
|
|
func NewMods(root components.RootModel, parent tea.Model) tea.Model {
|
|
l := list.New([]list.Item{}, utils.NewItemDelegate(), root.Size().Width, root.Size().Height-root.Height())
|
|
l.SetShowStatusBar(true)
|
|
l.SetShowFilter(true)
|
|
l.SetFilteringEnabled(true)
|
|
l.SetSpinner(spinner.MiniDot)
|
|
l.Title = modsTitle
|
|
l.Styles = utils.ListStyles
|
|
l.SetSize(l.Width(), l.Height())
|
|
l.KeyMap.Quit.SetHelp("q", "back")
|
|
l.DisableQuitKeybindings()
|
|
|
|
l.AdditionalShortHelpKeys = func() []key.Binding {
|
|
return []key.Binding{
|
|
key.NewBinding(key.WithHelp("q", "back")),
|
|
key.NewBinding(key.WithHelp("s", "sort")),
|
|
key.NewBinding(key.WithHelp("o", "order")),
|
|
}
|
|
}
|
|
|
|
l.AdditionalFullHelpKeys = func() []key.Binding {
|
|
return []key.Binding{
|
|
key.NewBinding(key.WithHelp("q", "back")),
|
|
key.NewBinding(key.WithHelp("s", "sort")),
|
|
key.NewBinding(key.WithHelp("o", "order")),
|
|
}
|
|
}
|
|
|
|
sortFieldList := list.New([]list.Item{
|
|
utils.SimpleItem[modsList]{
|
|
ItemTitle: "Name",
|
|
Activate: func(msg tea.Msg, m modsList) (tea.Model, tea.Cmd) {
|
|
m.sortingField = "name"
|
|
cmd := m.list.SetItems(sortItems(m.list.Items(), m.sortingField, m.sortingOrder))
|
|
m.list.ResetSelected()
|
|
return m, cmd
|
|
},
|
|
},
|
|
utils.SimpleItem[modsList]{
|
|
ItemTitle: "Last Version Date",
|
|
Activate: func(msg tea.Msg, m modsList) (tea.Model, tea.Cmd) {
|
|
m.sortingField = "last_version_date"
|
|
cmd := m.list.SetItems(sortItems(m.list.Items(), m.sortingField, m.sortingOrder))
|
|
m.list.ResetSelected()
|
|
return m, cmd
|
|
},
|
|
},
|
|
utils.SimpleItem[modsList]{
|
|
ItemTitle: "Creation Date",
|
|
Activate: func(msg tea.Msg, m modsList) (tea.Model, tea.Cmd) {
|
|
m.sortingField = "created_at"
|
|
cmd := m.list.SetItems(sortItems(m.list.Items(), m.sortingField, m.sortingOrder))
|
|
m.list.ResetSelected()
|
|
return m, cmd
|
|
},
|
|
},
|
|
utils.SimpleItem[modsList]{
|
|
ItemTitle: "Downloads",
|
|
Activate: func(msg tea.Msg, m modsList) (tea.Model, tea.Cmd) {
|
|
m.sortingField = "downloads"
|
|
cmd := m.list.SetItems(sortItems(m.list.Items(), m.sortingField, m.sortingOrder))
|
|
m.list.ResetSelected()
|
|
return m, cmd
|
|
},
|
|
},
|
|
utils.SimpleItem[modsList]{
|
|
ItemTitle: "Views",
|
|
Activate: func(msg tea.Msg, m modsList) (tea.Model, tea.Cmd) {
|
|
m.sortingField = "views"
|
|
cmd := m.list.SetItems(sortItems(m.list.Items(), m.sortingField, m.sortingOrder))
|
|
m.list.ResetSelected()
|
|
return m, cmd
|
|
},
|
|
},
|
|
utils.SimpleItem[modsList]{
|
|
ItemTitle: "Popularity (recent downloads)",
|
|
Activate: func(msg tea.Msg, m modsList) (tea.Model, tea.Cmd) {
|
|
m.sortingField = "popularity"
|
|
cmd := m.list.SetItems(sortItems(m.list.Items(), m.sortingField, m.sortingOrder))
|
|
m.list.ResetSelected()
|
|
return m, cmd
|
|
},
|
|
},
|
|
utils.SimpleItem[modsList]{
|
|
ItemTitle: "Hotness (recent views)",
|
|
Activate: func(msg tea.Msg, m modsList) (tea.Model, tea.Cmd) {
|
|
m.sortingField = "hotness"
|
|
cmd := m.list.SetItems(sortItems(m.list.Items(), m.sortingField, m.sortingOrder))
|
|
m.list.ResetSelected()
|
|
return m, cmd
|
|
},
|
|
},
|
|
}, utils.NewItemDelegate(), root.Size().Width, root.Size().Height-root.Height())
|
|
sortFieldList.SetShowStatusBar(true)
|
|
sortFieldList.SetShowFilter(false)
|
|
sortFieldList.SetFilteringEnabled(false)
|
|
sortFieldList.Title = modsTitle
|
|
sortFieldList.Styles = utils.ListStyles
|
|
sortFieldList.SetSize(l.Width(), l.Height())
|
|
sortFieldList.KeyMap.Quit.SetHelp("q", "back")
|
|
sortFieldList.DisableQuitKeybindings()
|
|
|
|
sortOrderList := list.New([]list.Item{
|
|
utils.SimpleItem[modsList]{
|
|
ItemTitle: "Ascending",
|
|
Activate: func(msg tea.Msg, m modsList) (tea.Model, tea.Cmd) {
|
|
m.sortingOrder = sortOrderAsc
|
|
cmd := m.list.SetItems(sortItems(m.list.Items(), m.sortingField, m.sortingOrder))
|
|
m.list.ResetSelected()
|
|
return m, cmd
|
|
},
|
|
},
|
|
utils.SimpleItem[modsList]{
|
|
ItemTitle: "Descending",
|
|
Activate: func(msg tea.Msg, m modsList) (tea.Model, tea.Cmd) {
|
|
m.sortingOrder = sortOrderDesc
|
|
cmd := m.list.SetItems(sortItems(m.list.Items(), m.sortingField, m.sortingOrder))
|
|
m.list.ResetSelected()
|
|
return m, cmd
|
|
},
|
|
},
|
|
}, utils.NewItemDelegate(), root.Size().Width, root.Size().Height-root.Height())
|
|
sortOrderList.SetShowStatusBar(true)
|
|
sortOrderList.SetShowFilter(false)
|
|
sortOrderList.SetFilteringEnabled(false)
|
|
sortOrderList.Title = modsTitle
|
|
sortOrderList.Styles = utils.ListStyles
|
|
sortOrderList.SetSize(l.Width(), l.Height())
|
|
sortOrderList.KeyMap.Quit.SetHelp("q", "back")
|
|
sortOrderList.DisableQuitKeybindings()
|
|
|
|
m := &modsList{
|
|
root: root,
|
|
list: l,
|
|
parent: parent,
|
|
items: make(chan []list.Item),
|
|
sortingField: "last_version_date",
|
|
sortingOrder: sortOrderDesc,
|
|
sortFieldList: sortFieldList,
|
|
sortOrderList: sortOrderList,
|
|
err: make(chan string),
|
|
}
|
|
|
|
go func() {
|
|
items := make([]list.Item, 0)
|
|
allMods := make([]ficsit.ModsGetModsModsMod, 0)
|
|
offset := 0
|
|
for {
|
|
mods, err := ficsit.Mods(context.TODO(), root.GetAPIClient(), ficsit.ModFilter{
|
|
Limit: 100,
|
|
Offset: offset,
|
|
Order_by: ficsit.ModFieldsLastVersionDate,
|
|
Order: ficsit.OrderDesc,
|
|
})
|
|
|
|
if err != nil {
|
|
m.err <- err.Error()
|
|
return
|
|
}
|
|
|
|
if len(mods.GetMods.Mods) == 0 {
|
|
break
|
|
}
|
|
|
|
allMods = append(allMods, mods.GetMods.Mods...)
|
|
|
|
for i := 0; i < len(mods.GetMods.Mods); i++ {
|
|
currentOffset := offset
|
|
currentI := i
|
|
items = append(items, SimpleItemMod[modsList]{
|
|
SimpleItem: utils.SimpleItem[modsList]{
|
|
ItemTitle: mods.GetMods.Mods[i].Name,
|
|
Activate: func(msg tea.Msg, currentModel modsList) (tea.Model, tea.Cmd) {
|
|
mod := allMods[currentOffset+currentI]
|
|
return NewModMenu(root, currentModel, utils.Mod{
|
|
Name: mod.Name,
|
|
ID: mod.Id,
|
|
Reference: mod.Mod_reference,
|
|
}), nil
|
|
},
|
|
},
|
|
Mod: allMods[currentOffset+currentI],
|
|
Context: root.GetGlobal(),
|
|
})
|
|
}
|
|
|
|
offset += len(mods.GetMods.Mods)
|
|
}
|
|
|
|
m.items <- items
|
|
}()
|
|
|
|
return m
|
|
}
|
|
|
|
func (m modsList) Init() tea.Cmd {
|
|
return utils.Ticker()
|
|
}
|
|
|
|
func (m modsList) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
// List enables its own keybindings when they were previously disabled
|
|
m.list.DisableQuitKeybindings()
|
|
|
|
switch msg := msg.(type) {
|
|
case tea.KeyMsg:
|
|
if m.list.SettingFilter() {
|
|
var cmd tea.Cmd
|
|
m.list, cmd = m.list.Update(msg)
|
|
return m, cmd
|
|
}
|
|
|
|
switch keypress := msg.String(); keypress {
|
|
case "s":
|
|
m.showSortFieldList = !m.showSortFieldList
|
|
return m, nil
|
|
case "o":
|
|
m.showSortOrderList = !m.showSortOrderList
|
|
return m, nil
|
|
case KeyControlC:
|
|
return m, tea.Quit
|
|
case "q":
|
|
if m.showSortFieldList {
|
|
m.showSortFieldList = false
|
|
return m, nil
|
|
}
|
|
|
|
if m.showSortOrderList {
|
|
m.showSortOrderList = false
|
|
return m, nil
|
|
}
|
|
|
|
if m.parent != nil {
|
|
m.parent.Update(m.root.Size())
|
|
return m.parent, nil
|
|
}
|
|
return m, tea.Quit
|
|
case KeyEnter:
|
|
if m.showSortFieldList {
|
|
m.showSortFieldList = false
|
|
i, ok := m.sortFieldList.SelectedItem().(utils.SimpleItem[modsList])
|
|
if ok {
|
|
return m.processActivation(i, msg)
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
if m.showSortOrderList {
|
|
m.showSortOrderList = false
|
|
i, ok := m.sortOrderList.SelectedItem().(utils.SimpleItem[modsList])
|
|
if ok {
|
|
return m.processActivation(i, msg)
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
i, ok := m.list.SelectedItem().(SimpleItemMod[modsList])
|
|
if ok {
|
|
return m.processActivation(i.SimpleItem, msg)
|
|
}
|
|
return m, nil
|
|
}
|
|
case tea.WindowSizeMsg:
|
|
top, right, bottom, left := lipgloss.NewStyle().Margin(m.root.Height(), 2, 0).GetMargin()
|
|
m.list.SetSize(msg.Width-left-right, msg.Height-top-bottom)
|
|
m.root.SetSize(msg)
|
|
case utils.TickMsg:
|
|
select {
|
|
case items := <-m.items:
|
|
m.list.StopSpinner()
|
|
cmd := m.list.SetItems(items)
|
|
return m, cmd
|
|
case err := <-m.err:
|
|
errorComponent, cmd := components.NewErrorComponent(err, time.Second*5)
|
|
m.error = errorComponent
|
|
return m, cmd
|
|
default:
|
|
start := m.list.StartSpinner()
|
|
return m, tea.Batch(utils.Ticker(), start)
|
|
}
|
|
}
|
|
|
|
if m.showSortFieldList {
|
|
var cmd tea.Cmd
|
|
m.sortFieldList, cmd = m.sortFieldList.Update(msg)
|
|
return m, cmd
|
|
} else if m.showSortOrderList {
|
|
var cmd tea.Cmd
|
|
m.sortOrderList, cmd = m.sortOrderList.Update(msg)
|
|
return m, cmd
|
|
}
|
|
|
|
var cmd tea.Cmd
|
|
m.list, cmd = m.list.Update(msg)
|
|
return m, cmd
|
|
}
|
|
|
|
func (m modsList) View() string {
|
|
var bottomList list.Model
|
|
if m.showSortFieldList {
|
|
bottomList = m.sortFieldList
|
|
} else if m.showSortOrderList {
|
|
bottomList = m.sortOrderList
|
|
} else {
|
|
bottomList = m.list
|
|
}
|
|
|
|
if m.error != nil {
|
|
err := (*m.error).View()
|
|
bottomList.SetSize(bottomList.Width(), m.root.Size().Height-m.root.Height()-lipgloss.Height(err))
|
|
return lipgloss.JoinVertical(lipgloss.Left, m.root.View(), err, bottomList.View())
|
|
}
|
|
|
|
bottomList.SetSize(bottomList.Width(), m.root.Size().Height-m.root.Height())
|
|
return lipgloss.JoinVertical(lipgloss.Left, m.root.View(), bottomList.View())
|
|
}
|
|
|
|
func sortItems(items []list.Item, field string, direction sortOrder) []list.Item {
|
|
sortedItems := make([]list.Item, len(items))
|
|
copy(sortedItems, items)
|
|
|
|
switch field {
|
|
case "last_version_date":
|
|
sort.Slice(sortedItems, func(i, j int) bool {
|
|
a := sortedItems[i].(SimpleItemMod[modsList])
|
|
b := sortedItems[j].(SimpleItemMod[modsList])
|
|
return ascDesc(direction, a.Mod.Last_version_date.Before(b.Mod.Last_version_date))
|
|
})
|
|
case "created_at":
|
|
sort.Slice(sortedItems, func(i, j int) bool {
|
|
a := sortedItems[i].(SimpleItemMod[modsList])
|
|
b := sortedItems[j].(SimpleItemMod[modsList])
|
|
return ascDesc(direction, a.Mod.Created_at.Before(b.Mod.Created_at))
|
|
})
|
|
case "name":
|
|
sort.Slice(sortedItems, func(i, j int) bool {
|
|
a := sortedItems[i].(SimpleItemMod[modsList])
|
|
b := sortedItems[j].(SimpleItemMod[modsList])
|
|
return ascDesc(direction, a.Mod.Name < b.Mod.Name)
|
|
})
|
|
case "downloads":
|
|
sort.Slice(sortedItems, func(i, j int) bool {
|
|
a := sortedItems[i].(SimpleItemMod[modsList])
|
|
b := sortedItems[j].(SimpleItemMod[modsList])
|
|
return ascDesc(direction, a.Mod.Downloads < b.Mod.Downloads)
|
|
})
|
|
case "views":
|
|
sort.Slice(sortedItems, func(i, j int) bool {
|
|
a := sortedItems[i].(SimpleItemMod[modsList])
|
|
b := sortedItems[j].(SimpleItemMod[modsList])
|
|
return ascDesc(direction, a.Mod.Views < b.Mod.Views)
|
|
})
|
|
case "popularity":
|
|
sort.Slice(sortedItems, func(i, j int) bool {
|
|
a := sortedItems[i].(SimpleItemMod[modsList])
|
|
b := sortedItems[j].(SimpleItemMod[modsList])
|
|
return ascDesc(direction, a.Mod.Popularity < b.Mod.Popularity)
|
|
})
|
|
case "hotness":
|
|
sort.Slice(sortedItems, func(i, j int) bool {
|
|
a := sortedItems[i].(SimpleItemMod[modsList])
|
|
b := sortedItems[j].(SimpleItemMod[modsList])
|
|
return ascDesc(direction, a.Mod.Hotness < b.Mod.Hotness)
|
|
})
|
|
}
|
|
|
|
return sortedItems
|
|
}
|
|
|
|
func (m modsList) processActivation(item utils.SimpleItem[modsList], msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
if item.Activate != nil {
|
|
newModel, cmd := item.Activate(msg, m)
|
|
if newModel != nil || cmd != nil {
|
|
if newModel == nil {
|
|
newModel = m
|
|
}
|
|
return newModel, cmd
|
|
}
|
|
return m, nil
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
func ascDesc(order sortOrder, result bool) bool {
|
|
if order == sortOrderAsc {
|
|
return result
|
|
}
|
|
return !result
|
|
}
|