ficsit-cli-flake/tea/root.go

package tea

import (
	"github.com/Khan/genqlient/graphql"
	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"
	"github.com/pkg/errors"
	"github.com/satisfactorymodding/ficsit-cli/cli"
	"github.com/satisfactorymodding/ficsit-cli/tea/components"
	"github.com/satisfactorymodding/ficsit-cli/tea/scenes"
)

type rootModel struct {
	global             *cli.GlobalContext
	currentSize        tea.WindowSizeMsg
	headerComponent    tea.Model
	dependencyResolver cli.DependencyResolver
}

func newModel(global *cli.GlobalContext) *rootModel {
	m := &rootModel{
		global: global,
		currentSize: tea.WindowSizeMsg{
			Width:  20,
			Height: 14,
		},
		dependencyResolver: cli.NewDependencyResolver(global.APIClient),
	}

	m.headerComponent = components.NewHeaderComponent(m)

	return m
}

func (m *rootModel) GetCurrentProfile() *cli.Profile {
	return m.global.Profiles.GetProfile(m.global.Profiles.SelectedProfile)
}

func (m *rootModel) SetCurrentProfile(profile *cli.Profile) error {
	m.global.Profiles.SelectedProfile = profile.Name

	if err := m.GetCurrentInstallation().SetProfile(m.global, profile.Name); err != nil {
		return errors.Wrap(err, "failed setting profile on installation")
	}

	return m.global.Save()
}

func (m *rootModel) GetCurrentInstallation() *cli.Installation {
	return m.global.Installations.GetInstallation(m.global.Installations.SelectedInstallation)
}

func (m *rootModel) SetCurrentInstallation(installation *cli.Installation) error {
	m.global.Installations.SelectedInstallation = installation.Path
	m.global.Profiles.SelectedProfile = installation.Profile
	return m.global.Save()
}

func (m *rootModel) GetAPIClient() graphql.Client {
	return m.global.APIClient
}

func (m *rootModel) Size() tea.WindowSizeMsg {
	return m.currentSize
}

func (m *rootModel) SetSize(size tea.WindowSizeMsg) {
	m.currentSize = size
}

func (m *rootModel) View() string {
	return m.headerComponent.View()
}

func (m *rootModel) Height() int {
	return lipgloss.Height(m.View()) + 1
}

func (m *rootModel) GetGlobal() *cli.GlobalContext {
	return m.global
}

func (m *rootModel) ResolveModDependencies(constraints map[string]string) (map[string]cli.ModVersion, error) {
	return m.dependencyResolver.ResolveModDependencies(constraints)
}

func RunTea(global *cli.GlobalContext) error {
	if err := tea.NewProgram(scenes.NewMainMenu(newModel(global)), tea.WithAltScreen(), tea.WithMouseCellMotion()).Start(); err != nil {
		return errors.Wrap(err, "internal tea error")
	}
	return nil
}