ficsit-cli-flake/tea/scenes/profile/new_profile.go

package profile

import (
	"time"

	"github.com/charmbracelet/bubbles/help"
	"github.com/charmbracelet/bubbles/key"
	"github.com/charmbracelet/bubbles/textinput"
	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"

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

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

type keyMap struct {
	Back  key.Binding
	Quit  key.Binding
	Enter key.Binding
}

func (k keyMap) ShortHelp() []key.Binding {
	return []key.Binding{k.Enter, k.Back}
}

func (k keyMap) FullHelp() [][]key.Binding {
	return [][]key.Binding{
		{k.Enter, k.Back},
	}
}

type newProfile struct {
	root   components.RootModel
	parent tea.Model
	error  *components.ErrorComponent
	help   help.Model
	title  string
	keys   keyMap
	input  textinput.Model
}

func NewNewProfile(root components.RootModel, parent tea.Model) tea.Model {
	model := newProfile{
		root:   root,
		parent: parent,
		input:  textinput.New(),
		title:  utils.NonListTitleStyle.Render("New Profile"),
		help:   help.New(),
		keys: keyMap{
			Back: key.NewBinding(
				key.WithKeys(keys.KeyEscape, keys.KeyControlC),
				key.WithHelp(keys.KeyEscape, "back"),
			),
			Enter: key.NewBinding(
				key.WithKeys(keys.KeyEnter),
				key.WithHelp(keys.KeyEnter, "create"),
			),
			Quit: key.NewBinding(
				key.WithKeys(keys.KeyControlC),
			),
		},
	}

	model.input.Focus()
	model.input.Width = root.Size().Width

	return model
}

func (m newProfile) Init() tea.Cmd {
	return textinput.Blink
}

func (m newProfile) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	switch msg := msg.(type) {
	case tea.KeyMsg:
		switch {
		case key.Matches(msg, m.keys.Quit):
			return m, tea.Quit
		case key.Matches(msg, m.keys.Back):
			return m.parent, nil
		case key.Matches(msg, m.keys.Enter):
			if _, err := m.root.GetGlobal().Profiles.AddProfile(m.input.Value()); err != nil {
				errorComponent, cmd := components.NewErrorComponent(err.Error(), time.Second*5)
				m.error = errorComponent
				return m, cmd
			}

			return m.parent, updateProfileListCmd
		default:
			var cmd tea.Cmd
			m.input, cmd = m.input.Update(msg)
			return m, cmd
		}
	case tea.WindowSizeMsg:
		m.root.SetSize(msg)
	case components.ErrorComponentTimeoutMsg:
		m.error = nil
	default:
		var cmd tea.Cmd
		m.input, cmd = m.input.Update(msg)
		return m, cmd
	}

	return m, nil
}

func (m newProfile) View() string {
	style := lipgloss.NewStyle().Padding(1, 2)
	inputView := style.Render(m.input.View())

	if m.error != nil {
		return lipgloss.JoinVertical(lipgloss.Left, m.root.View(), m.title, m.error.View(), inputView)
	}

	infoBox := lipgloss.NewStyle().
		BorderStyle(lipgloss.ThickBorder()).
		BorderForeground(lipgloss.Color("39")).
		Padding(0, 1).
		Margin(0, 0, 0, 2).
		Render("Enter the name of the profile")

	return lipgloss.JoinVertical(lipgloss.Left, m.root.View(), m.title, inputView, infoBox, style.Render(m.help.View(m.keys)))
}