108 lines
2.1 KiB
Go
108 lines
2.1 KiB
Go
package task
|
|
|
|
import (
|
|
"charm.land/bubbletea/v2"
|
|
"charm.land/huh/v2"
|
|
"charm.land/lipgloss/v2"
|
|
"github.com/LazyBachelor/LazyPM/internal/models"
|
|
"github.com/LazyBachelor/LazyPM/internal/style"
|
|
)
|
|
|
|
type Questions = models.Questions
|
|
|
|
type QuestionnaireModel struct {
|
|
Questions
|
|
form *huh.Form
|
|
keys []string
|
|
width, height int
|
|
userQuit bool
|
|
}
|
|
|
|
func NewQuestionnaireModel(questions Questions, keys []string) *QuestionnaireModel {
|
|
form := huh.NewForm(questions...).
|
|
WithTheme(style.HuhCenterTheme()).
|
|
WithLayout(huh.LayoutGrid(1, 1)).
|
|
WithWidth(80)
|
|
|
|
return &QuestionnaireModel{
|
|
Questions: questions,
|
|
form: form,
|
|
keys: keys,
|
|
}
|
|
}
|
|
|
|
func (q *QuestionnaireModel) Init() tea.Cmd {
|
|
return q.form.Init()
|
|
}
|
|
|
|
func (q *QuestionnaireModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
switch msg := msg.(type) {
|
|
case tea.WindowSizeMsg:
|
|
q.SetSize(msg.Width, msg.Height)
|
|
// Update form width to match window
|
|
q.form.WithWidth(msg.Width)
|
|
case tea.KeyPressMsg:
|
|
switch msg.String() {
|
|
case "ctrl+c":
|
|
q.userQuit = true
|
|
return q, tea.Quit
|
|
}
|
|
}
|
|
|
|
form, cmd := q.form.Update(msg)
|
|
if f, ok := form.(*huh.Form); ok {
|
|
q.form = f
|
|
}
|
|
|
|
if q.form.State == huh.StateCompleted {
|
|
return q, tea.Quit
|
|
}
|
|
|
|
return q, cmd
|
|
}
|
|
|
|
func (q *QuestionnaireModel) View() tea.View {
|
|
form := lipgloss.NewStyle().
|
|
Width(q.width).Align(lipgloss.Center).
|
|
Render(q.form.View())
|
|
|
|
content := lipgloss.Place(
|
|
q.width, q.height, lipgloss.Center, lipgloss.Center, form,
|
|
)
|
|
|
|
v := tea.NewView(content)
|
|
v.AltScreen = true
|
|
return v
|
|
}
|
|
|
|
func (q *QuestionnaireModel) SetSize(width, height int) {
|
|
q.width, q.height = width, height
|
|
}
|
|
|
|
func (q QuestionnaireModel) GetUserQuit() bool {
|
|
return q.userQuit
|
|
}
|
|
|
|
func (q QuestionnaireModel) GetCompleted() bool {
|
|
return q.form != nil && q.form.State == huh.StateCompleted
|
|
}
|
|
|
|
func (q QuestionnaireModel) GetAnswers() map[string]any {
|
|
if q.form == nil || len(q.keys) == 0 {
|
|
return nil
|
|
}
|
|
|
|
answers := make(map[string]any)
|
|
for _, key := range q.keys {
|
|
if key == "" {
|
|
continue
|
|
}
|
|
answers[key] = q.form.Get(key)
|
|
}
|
|
if len(answers) == 0 {
|
|
return nil
|
|
}
|
|
|
|
return answers
|
|
}
|