prompt.go

package prompt

import (
	"github.com/charmbracelet/bubbles/textinput"
	tea "github.com/charmbracelet/bubbletea"
)

type State int

const (
	Active State = iota
	Submitted
	Canceled
)

func NewModel() Model {
	ti := textinput.NewModel()
	ti.Focus()
	return Model{
		Input: ti,
	}
}

type Model struct {
	Input textinput.Model
	State State
}

func (m Model) Init() tea.Cmd {
	return nil
}

func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	switch msg := msg.(type) {
	case tea.KeyMsg:
		switch msg.Type {

		case tea.KeyEnter:
			m.State = Submitted

		case tea.KeyEsc:
			m.State = Canceled

		default:
			var cmd tea.Cmd
			m.Input, cmd = m.Input.Update(msg)
			return m, cmd
		}
	}

	return m, nil
}

func (m Model) View() string {
	return "What's your favorite color? " + m.Input.View()
}