forked from Mirrors/bubbletea
88 lines
1.6 KiB
Go
88 lines
1.6 KiB
Go
package main
|
|
|
|
// A simple example that shows how to retrieve a value from a Bubble Tea
|
|
// program after the Bubble Tea has exited.
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
)
|
|
|
|
var choices = []string{"Taro", "Coffee", "Lychee"}
|
|
|
|
type model struct {
|
|
cursor int
|
|
choice string
|
|
}
|
|
|
|
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.String() {
|
|
case "ctrl+c", "q", "esc":
|
|
return m, tea.Quit
|
|
|
|
case "enter":
|
|
// Send the choice on the channel and exit.
|
|
m.choice = choices[m.cursor]
|
|
return m, tea.Quit
|
|
|
|
case "down", "j":
|
|
m.cursor++
|
|
if m.cursor >= len(choices) {
|
|
m.cursor = 0
|
|
}
|
|
|
|
case "up", "k":
|
|
m.cursor--
|
|
if m.cursor < 0 {
|
|
m.cursor = len(choices) - 1
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
return m, nil
|
|
}
|
|
|
|
func (m model) View() string {
|
|
s := strings.Builder{}
|
|
s.WriteString("What kind of Bubble Tea would you like to order?\n\n")
|
|
|
|
for i := 0; i < len(choices); i++ {
|
|
if m.cursor == i {
|
|
s.WriteString("(•) ")
|
|
} else {
|
|
s.WriteString("( ) ")
|
|
}
|
|
s.WriteString(choices[i])
|
|
s.WriteString("\n")
|
|
}
|
|
s.WriteString("\n(press q to quit)\n")
|
|
|
|
return s.String()
|
|
}
|
|
|
|
func main() {
|
|
p := tea.NewProgram(model{})
|
|
|
|
// StartReturningModel returns the model as a tea.Model.
|
|
m, err := p.StartReturningModel()
|
|
if err != nil {
|
|
fmt.Println("Oh no:", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Assert the final tea.Model to our local model and print the choice.
|
|
if m, ok := m.(model); ok && m.choice != "" {
|
|
fmt.Printf("\n---\nYou chose %s!\n", m.choice)
|
|
}
|
|
}
|