refactor survery cmd

move package files to pkg
This commit is contained in:
Robin Olsen
2026-02-13 10:13:41 +01:00
parent fc5156a0ad
commit e994c890f4
11 changed files with 23 additions and 38 deletions

27
pkg/task/ui/help.go Normal file
View File

@@ -0,0 +1,27 @@
package taskui
import "github.com/charmbracelet/bubbles/key"
type TaskHelpKeys struct {
Quit key.Binding
Continue key.Binding
}
var DefaultTaskKeys = TaskHelpKeys{
Quit: key.NewBinding(
key.WithKeys("q", "ctrl+c"),
key.WithHelp("q", "Quit"),
),
Continue: key.NewBinding(
key.WithKeys(" "),
key.WithHelp("space", "Continue"),
),
}
func (h TaskHelpKeys) ShortHelp() []key.Binding {
return []key.Binding{h.Continue, h.Quit}
}
func (h TaskHelpKeys) FullHelp() [][]key.Binding {
return [][]key.Binding{{h.Continue, h.Quit}}
}

View File

@@ -0,0 +1,54 @@
package taskui
import (
"charm.land/lipgloss/v2"
"github.com/LazyBachelor/LazyPM/internal/style"
tea "github.com/charmbracelet/bubbletea"
"github.com/charmbracelet/huh"
)
func NewQuestionnaireModel(questions Questions) *QuestionnaireModel {
form := huh.NewForm(questions...).
WithTheme(style.HuhCenterTheme()).WithLayout(huh.LayoutGrid(1, 1))
return &QuestionnaireModel{
Questions: questions,
form: form,
}
}
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)
case tea.KeyMsg:
switch msg.String() {
case "q", "ctrl+c":
return q, tea.Quit
}
}
form, cmd := q.form.Update(msg)
if f, ok := form.(*huh.Form); ok {
q.form = f
}
return q, cmd
}
func (q *QuestionnaireModel) View() string {
form := lipgloss.NewStyle().
Width(q.width).Align(lipgloss.Center).
Render(q.form.View())
return lipgloss.Place(
q.width, q.height, lipgloss.Center, lipgloss.Center, form,
)
}
func (q *QuestionnaireModel) SetSize(width, height int) {
q.width, q.height = width, height
}

72
pkg/task/ui/task.go Normal file
View File

@@ -0,0 +1,72 @@
package taskui
import (
"fmt"
"charm.land/lipgloss/v2"
"github.com/charmbracelet/bubbles/help"
"github.com/charmbracelet/bubbles/key"
tea "github.com/charmbracelet/bubbletea"
)
func NewTaskModel(details TaskDetails) TaskModel {
return TaskModel{
TaskDetails: details,
keys: DefaultTaskKeys,
help: help.New(),
}
}
func (m TaskModel) Init() tea.Cmd {
return nil
}
func (m TaskModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
switch msg := msg.(type) {
case tea.WindowSizeMsg:
m.SetSize(msg.Width, msg.Height)
case tea.KeyMsg:
switch {
case key.Matches(msg, m.keys.Quit):
return m, tea.Quit
case key.Matches(msg, m.keys.Continue):
return m, tea.Quit
}
}
return m, nil
}
func (m TaskModel) View() string {
padding := 3
header := lipgloss.NewStyle().
PaddingTop(padding).Width(m.width).Align(lipgloss.Center).
Bold(true).Render(m.Title)
headerHeight := lipgloss.Height(header)
helpView := lipgloss.NewStyle().
PaddingBottom(padding).
Width(m.width).Align(lipgloss.Center).
Render(m.help.View(m.keys))
helpHeigh := lipgloss.Height(helpView)
detailsText := fmt.Sprintf("Time to complete: %s | Difficulty: %s", m.TimeToComplete, m.Difficulty)
details := lipgloss.NewStyle().Align(lipgloss.Center).
Width(m.width).PaddingBottom(1).Render(detailsText)
detailsHeight := lipgloss.Height(details)
content := lipgloss.NewStyle().
Width(m.width).Height(m.height-headerHeight-helpHeigh-detailsHeight).
Align(lipgloss.Center, lipgloss.Center).
Render(m.Description)
return lipgloss.JoinVertical(lipgloss.Top, header, content, details, helpView)
}
func (m *TaskModel) SetSize(width, height int) {
m.width, m.height = width, height
}

28
pkg/task/ui/types.go Normal file
View File

@@ -0,0 +1,28 @@
package taskui
import (
"github.com/charmbracelet/bubbles/help"
"github.com/charmbracelet/huh"
)
type TaskDetails struct {
Title string
Description string
TimeToComplete string
Difficulty string
}
type TaskModel struct {
TaskDetails
keys TaskHelpKeys
help help.Model
width, height int
}
type Questions []*huh.Group
type QuestionnaireModel struct {
Questions
form *huh.Form
width, height int
}