bubbletea/examples/input/main.go

package main

import (
	"errors"
	"fmt"
	"log"

	"github.com/charmbracelet/boba"
	input "github.com/charmbracelet/boba/textinput"
)

type Model struct {
	textInput input.Model
	err       error
}

type tickMsg struct{}
type errMsg error

func main() {
	p := boba.NewProgram(
		initialize,
		update,
		view,
		subscriptions,
	)

	if err := p.Start(); err != nil {
		log.Fatal(err)
	}
}

func initialize() (boba.Model, boba.Cmd) {
	inputModel := input.NewModel()
	inputModel.Placeholder = "Pikachu"
	inputModel.Focus()

	return Model{
		textInput: inputModel,
		err:       nil,
	}, nil
}

func update(msg boba.Msg, model boba.Model) (boba.Model, boba.Cmd) {
	var cmd boba.Cmd
	m, ok := model.(Model)
	if !ok {
		// When we encounter errors in Update we simply add the error to the
		// model so we can handle it in the view. We could also return a command
		// that does something else with the error, like logs it via IO.
		return Model{
			err: errors.New("could not perform assertion on model in update"),
		}, nil
	}

	switch msg := msg.(type) {
	case boba.KeyMsg:
		switch msg.Type {
		case boba.KeyCtrlC:
			fallthrough
		case boba.KeyEsc:
			fallthrough
		case boba.KeyEnter:
			return m, boba.Quit
		}

	// We handle errors just like any other message
	case errMsg:
		m.err = msg
		return m, nil
	}

	m.textInput, cmd = input.Update(msg, m.textInput)
	return m, cmd
}

func subscriptions(model boba.Model) boba.Subs {
	m, ok := model.(Model)
	if !ok {
		return nil
	}
	sub, err := input.MakeSub(m.textInput)
	if err != nil {
		return nil
	}
	return boba.Subs{
		"input": sub,
	}
}

func view(model boba.Model) string {
	m, ok := model.(Model)
	if !ok {
		return "Oh no: could not perform assertion on model."
	} else if m.err != nil {
		return fmt.Sprintf("Uh oh: %s", m.err)
	}
	return fmt.Sprintf(
		"What’s your favorite Pokémon?\n\n%s\n\n%s",
		input.View(m.textInput),
		"(esc to quit)",
	) + "\n"
}