117 lines
2.3 KiB
Go
117 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
|
|
"github.com/charmbracelet/bubbles/textinput"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"github.com/charmbracelet/lipgloss"
|
|
)
|
|
|
|
type Styles struct {
|
|
BorderColor lipgloss.Color
|
|
InputField lipgloss.Style
|
|
}
|
|
|
|
func DefaultStyles() *Styles {
|
|
s := new(Styles)
|
|
s.BorderColor = lipgloss.Color("36")
|
|
s.InputField = lipgloss.NewStyle().BorderForeground(s.BorderColor).BorderStyle(lipgloss.NormalBorder()).Padding(1).Width(80)
|
|
return s
|
|
}
|
|
|
|
type model struct {
|
|
questions []Question
|
|
width int
|
|
height int
|
|
answerField textinput.Model
|
|
index int
|
|
styles *Styles
|
|
}
|
|
|
|
type Question struct {
|
|
question string
|
|
answer string
|
|
}
|
|
|
|
func New(questions []Question) *model {
|
|
styles := DefaultStyles()
|
|
answerField := textinput.New()
|
|
answerField.Placeholder = "Your answer here"
|
|
answerField.Focus()
|
|
index := 0
|
|
return &model{questions: questions, answerField: answerField, index: index, styles: styles}
|
|
}
|
|
|
|
func (m model) Init() tea.Cmd {
|
|
return nil
|
|
}
|
|
|
|
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
var cmd tea.Cmd
|
|
var current = &m.questions[m.index]
|
|
switch msg := msg.(type) {
|
|
case tea.WindowSizeMsg:
|
|
m.width = msg.Width
|
|
m.height = msg.Height
|
|
case tea.KeyMsg:
|
|
switch msg.String() {
|
|
case "ctrl+c":
|
|
return m, tea.Quit
|
|
case "enter":
|
|
current.answer = m.answerField.Value()
|
|
log.Printf("question: %s, answer: %s", m.questions[m.index].question, m.questions[m.index].answer)
|
|
m.answerField.SetValue("")
|
|
m.Next()
|
|
return m, nil
|
|
}
|
|
}
|
|
m.answerField, cmd = m.answerField.Update(msg)
|
|
return m, cmd
|
|
}
|
|
|
|
func (m model) View() string {
|
|
if m.width == 0 {
|
|
return "loading..."
|
|
} else {
|
|
return lipgloss.Place(
|
|
m.width,
|
|
m.height,
|
|
lipgloss.Center,
|
|
lipgloss.Center,
|
|
lipgloss.JoinVertical(
|
|
lipgloss.Center,
|
|
m.questions[m.index].question,
|
|
m.styles.InputField.Render(m.answerField.View()),
|
|
),
|
|
)
|
|
}
|
|
}
|
|
|
|
func (m *model) Next() {
|
|
if m.index < len(m.questions)-1 {
|
|
m.index++
|
|
} else {
|
|
m.index = 0
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
questions := []Question{
|
|
{question: "what is your name?"},
|
|
{question: "what is your favorite editor?"},
|
|
{question: "best cite?"},
|
|
}
|
|
m := New(questions)
|
|
f, err := tea.LogToFile("debug.log", "debug")
|
|
if err != nil {
|
|
log.Fatalf("err: %w", err)
|
|
}
|
|
defer f.Close()
|
|
p := tea.NewProgram(m, tea.WithAltScreen())
|
|
if _, err := p.Run(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
}
|