git.iko.soy/crush
on 94de6bc help sign in
crush/internal/tui/components/dialogs/commands/arguments.go
254 lines 6.3 kB history raw
1package commands
3import (
4 "cmp"
6 "charm.land/bubbles/v2/help"
7 "charm.land/bubbles/v2/key"
8 "charm.land/bubbles/v2/textinput"
9 tea "charm.land/bubbletea/v2"
10 "charm.land/lipgloss/v2"
11 "github.com/charmbracelet/crush/internal/tui/components/dialogs"
12 "github.com/charmbracelet/crush/internal/tui/styles"
13 "github.com/charmbracelet/crush/internal/tui/util"
16const (
17 argumentsDialogID dialogs.DialogID = "arguments"
20// ShowArgumentsDialogMsg is a message that is sent to show the arguments dialog.
21type ShowArgumentsDialogMsg struct {
22 CommandID string
23 Description string
24 ArgNames []string
25 OnSubmit func(args map[string]string) tea.Cmd
28// CloseArgumentsDialogMsg is a message that is sent when the arguments dialog is closed.
29type CloseArgumentsDialogMsg struct {
30 Submit bool
31 CommandID string
32 Content string
33 Args map[string]string
36// CommandArgumentsDialog represents the commands dialog.
37type CommandArgumentsDialog interface {
38 dialogs.DialogModel
41type commandArgumentsDialogCmp struct {
42 wWidth, wHeight int
43 width, height int
45 inputs []textinput.Model
46 focused int
47 keys ArgumentsDialogKeyMap
48 arguments []Argument
49 help help.Model
51 id string
52 title string
53 name string
54 description string
56 onSubmit func(args map[string]string) tea.Cmd
59type Argument struct {
60 Name, Title, Description string
61 Required bool
64func NewCommandArgumentsDialog(
65 id, title, name, description string,
66 arguments []Argument,
67 onSubmit func(args map[string]string) tea.Cmd,
68) CommandArgumentsDialog {
69 t := styles.CurrentTheme()
70 inputs := make([]textinput.Model, len(arguments))
72 for i, arg := range arguments {
73 ti := textinput.New()
74 ti.Placeholder = cmp.Or(arg.Description, "Enter value for "+arg.Title)
75 ti.SetWidth(40)
76 ti.SetVirtualCursor(false)
77 ti.Prompt = ""
79 ti.SetStyles(t.S().TextInput)
80 // Only focus the first input initially
81 if i == 0 {
82 ti.Focus()
83 } else {
84 ti.Blur()
85 }
87 inputs[i] = ti
88 }
90 return &commandArgumentsDialogCmp{
91 inputs: inputs,
92 keys: DefaultArgumentsDialogKeyMap(),
93 id: id,
94 name: name,
95 title: title,
96 description: description,
97 arguments: arguments,
98 width: 60,
99 help: help.New(),
100 onSubmit: onSubmit,
101 }
104// Init implements CommandArgumentsDialog.
105func (c *commandArgumentsDialogCmp) Init() tea.Cmd {
106 return nil
109// Update implements CommandArgumentsDialog.
110func (c *commandArgumentsDialogCmp) Update(msg tea.Msg) (util.Model, tea.Cmd) {
111 switch msg := msg.(type) {
112 case tea.WindowSizeMsg:
113 c.wWidth = msg.Width
114 c.wHeight = msg.Height
115 c.width = min(90, c.wWidth)
116 c.height = min(15, c.wHeight)
117 for i := range c.inputs {
118 c.inputs[i].SetWidth(c.width - (paddingHorizontal * 2))
119 }
120 case tea.KeyPressMsg:
121 switch {
122 case key.Matches(msg, c.keys.Close):
123 return c, util.CmdHandler(dialogs.CloseDialogMsg{})
124 case key.Matches(msg, c.keys.Confirm):
125 if c.focused == len(c.inputs)-1 {
126 args := make(map[string]string)
127 for i, arg := range c.arguments {
128 value := c.inputs[i].Value()
129 args[arg.Name] = value
130 }
131 return c, tea.Sequence(
132 util.CmdHandler(dialogs.CloseDialogMsg{}),
133 c.onSubmit(args),
134 )
135 }
136 // Otherwise, move to the next input
137 c.inputs[c.focused].Blur()
138 c.focused++
139 c.inputs[c.focused].Focus()
140 case key.Matches(msg, c.keys.Next):
141 // Move to the next input
142 c.inputs[c.focused].Blur()
143 c.focused = (c.focused + 1) % len(c.inputs)
144 c.inputs[c.focused].Focus()
145 case key.Matches(msg, c.keys.Previous):
146 // Move to the previous input
147 c.inputs[c.focused].Blur()
148 c.focused = (c.focused - 1 + len(c.inputs)) % len(c.inputs)
149 c.inputs[c.focused].Focus()
150 case key.Matches(msg, c.keys.Close):
151 return c, util.CmdHandler(dialogs.CloseDialogMsg{})
152 default:
153 var cmd tea.Cmd
154 c.inputs[c.focused], cmd = c.inputs[c.focused].Update(msg)
155 return c, cmd
156 }
157 case tea.PasteMsg:
158 var cmd tea.Cmd
159 c.inputs[c.focused], cmd = c.inputs[c.focused].Update(msg)
160 return c, cmd
161 }
162 return c, nil
165// View implements CommandArgumentsDialog.
166func (c *commandArgumentsDialogCmp) View() string {
167 t := styles.CurrentTheme()
168 baseStyle := t.S().Base
170 title := lipgloss.NewStyle().
171 Foreground(t.Primary).
172 Bold(true).
173 Padding(0, 1).
174 Render(cmp.Or(c.title, c.name))
176 promptName := t.S().Text.
177 Padding(0, 1).
178 Render(c.description)
180 inputFields := make([]string, len(c.inputs))
181 for i, input := range c.inputs {
182 labelStyle := baseStyle.Padding(1, 1, 0, 1)
184 if i == c.focused {
185 labelStyle = labelStyle.Foreground(t.FgBase).Bold(true)
186 } else {
187 labelStyle = labelStyle.Foreground(t.FgMuted)
188 }
190 arg := c.arguments[i]
191 argName := cmp.Or(arg.Title, arg.Name)
192 if arg.Required {
193 argName += "*"
194 }
195 label := labelStyle.Render(argName + ":")
197 field := t.S().Text.
198 Padding(0, 1).
199 Render(input.View())
201 inputFields[i] = lipgloss.JoinVertical(lipgloss.Left, label, field)
202 }
204 elements := []string{title, promptName}
205 elements = append(elements, inputFields...)
207 c.help.ShowAll = false
208 helpText := baseStyle.Padding(0, 1).Render(c.help.View(c.keys))
209 elements = append(elements, "", helpText)
211 content := lipgloss.JoinVertical(lipgloss.Left, elements...)
213 return baseStyle.Padding(1, 1, 0, 1).
214 Border(lipgloss.RoundedBorder()).
215 BorderForeground(t.BorderFocus).
216 Width(c.width).
217 Render(content)
220func (c *commandArgumentsDialogCmp) Cursor() *tea.Cursor {
221 if len(c.inputs) == 0 {
222 return nil
223 }
224 cursor := c.inputs[c.focused].Cursor()
225 if cursor != nil {
226 cursor = c.moveCursor(cursor)
227 }
228 return cursor
231const (
232 headerHeight = 3
233 itemHeight = 3
234 paddingHorizontal = 3
237func (c *commandArgumentsDialogCmp) moveCursor(cursor *tea.Cursor) *tea.Cursor {
238 row, col := c.Position()
239 offset := row + headerHeight + (1+c.focused)*itemHeight
240 cursor.Y += offset
241 cursor.X = cursor.X + col + paddingHorizontal
242 return cursor
245func (c *commandArgumentsDialogCmp) Position() (int, int) {
246 row := (c.wHeight / 2) - (c.height / 2)
247 col := (c.wWidth / 2) - (c.width / 2)
248 return row, col
251// ID implements CommandArgumentsDialog.
252func (c *commandArgumentsDialogCmp) ID() dialogs.DialogID {
253 return argumentsDialogID