ficsit-cli-flake/tea/root.go

package tea

import (
	"fmt"

	"github.com/Khan/genqlient/graphql"
	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"
	resolver "github.com/satisfactorymodding/ficsit-resolver"

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

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

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

	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 m.GetCurrentInstallation() != nil {
		if err := m.GetCurrentInstallation().SetProfile(m.global, profile.Name); err != nil {
			return fmt.Errorf("failed setting profile on installation: %w", err)
		}
	}

	return nil
}

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 nil
}

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

func (m *rootModel) GetProvider() provider.Provider {
	return m.global.Provider
}

func (m *rootModel) GetResolver() resolver.DependencyResolver {
	return m.dependencyResolver
}

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 RunTea(global *cli.GlobalContext) error {
	if _, err := tea.NewProgram(scenes.NewMainMenu(newModel(global)), tea.WithAltScreen(), tea.WithMouseCellMotion()).Run(); err != nil {
		return fmt.Errorf("internal tea error: %w", err)
	}
	return nil
}