bubbles/examples/inputs/main.go

package main

import (
	"fmt"
	"os"

	"github.com/charmbracelet/tea"
	"github.com/charmbracelet/teaparty/input"
)

func main() {
	if err := tea.NewProgram(
		initialize,
		update,
		view,
		subscriptions,
	).Start(); err != nil {
		fmt.Printf("could not start program: %s\n", err)
		os.Exit(1)
	}
}

type Model struct {
	index      int
	nameInput  input.Model
	emailInput input.Model
}

func initialize() (tea.Model, tea.Cmd) {
	n := input.DefaultModel()
	n.Placeholder = "Name"
	n.Focus = true

	e := input.DefaultModel()
	e.Placeholder = "Email"

	return Model{0, n, e}, nil
}

func update(msg tea.Msg, model tea.Model) (tea.Model, tea.Cmd) {
	m, ok := model.(Model)
	if !ok {
		panic("could not perform assertion on model")
	}

	switch msg := msg.(type) {

	case tea.KeyMsg:
		switch msg.String() {
		case "ctrl+c":
			return m, tea.Quit
		case "tab":
			fallthrough
		case "enter":
			fallthrough
		case "up":
			fallthrough
		case "down":
			inputs := []input.Model{
				m.nameInput,
				m.emailInput,
			}

			if msg.String() == "up" {
				m.index--
			} else {
				m.index++
			}

			if m.index > len(inputs)-1 {
				m.index = 0
			} else if m.index < 0 {
				m.index = len(inputs) - 1
			}

			for i := 0; i < len(inputs); i++ {
				if i == m.index {
					inputs[i].Focus = true
					continue
				}
				inputs[i].Focus = false
			}

			m.nameInput = inputs[0]
			m.emailInput = inputs[1]

			return m, nil
		default:
			m.nameInput, _ = input.Update(msg, m.nameInput)
			m.emailInput, _ = input.Update(msg, m.emailInput)
			return m, nil
		}

	default:
		m.nameInput, _ = input.Update(msg, m.nameInput)
		m.emailInput, _ = input.Update(msg, m.emailInput)
		return m, nil
	}
}

func subscriptions(model tea.Model) tea.Subs {
	return tea.Subs{
		"blink": func(model tea.Model) tea.Msg {
			m, _ := model.(Model)
			return input.Blink(m.nameInput)
		},
	}
}

func view(model tea.Model) string {
	m, ok := model.(Model)
	if !ok {
		return "[error] could not perform assertion on model"
	}

	s := "\n"

	inputs := []string{
		input.View(m.nameInput),
		input.View(m.emailInput),
	}

	for i := 0; i < len(inputs); i++ {
		s += inputs[i]
		if i < len(inputs)-1 {
			s += "\n"
		}
	}

	s += "\n"

	return s
}