Added ui:prompt-keypress to support single key presses

Have also fixed a bug in ui:prompt which was keeping the script running when the prompt was being cancelled

internal/common/ui/commandctrl/cmdpacks/modui.go

@@ -2,6 +2,7 @@ package cmdpacks
 
 import (
 	"context"
+
 	tea "github.com/charmbracelet/bubbletea"
 	"lmika.dev/cmd/dynamo-browse/internal/common/ui/commandctrl"
 	"lmika.dev/cmd/dynamo-browse/internal/common/ui/events"
@@ -38,16 +39,26 @@ func (m *uiModule) uiPrompt(ctx context.Context, args ucl.CallArgs) (any, error)
 	}
 
 	resChan := make(chan string)
+	cancelChan := make(chan struct{})
 	go func() {
-		commandctrl.PostMsg(ctx, events.PromptForInput(prompt, nil, func(value string) tea.Msg {
-			resChan <- value
-			return nil
-		}))
+		commandctrl.PostMsg(ctx, events.PromptForInputMsg{
+			Prompt: prompt,
+			OnDone: func(value string) tea.Msg {
+				resChan <- value
+				return nil
+			},
+			OnCancel: func() tea.Msg {
+				cancelChan <- struct{}{}
+				return nil
+			},
+		})
 	}()
 
 	select {
 	case value := <-resChan:
 		return value, nil
+	case <-cancelChan:
+		return nil, nil
 	case <-ctx.Done():
 		return nil, ctx.Err()
 	}
@@ -75,6 +86,38 @@ func (m *uiModule) uiConfirm(ctx context.Context, args ucl.CallArgs) (any, error
 	}
 }
 
+func (m *uiModule) uiInKey(ctx context.Context, args ucl.CallArgs) (any, error) {
+	var prompt string
+	if err := args.Bind(&prompt); err != nil {
+		return nil, err
+	}
+
+	resChan := make(chan string)
+	cancelChan := make(chan struct{})
+	go func() {
+		commandctrl.PostMsg(ctx, events.PromptForKeyMsg{
+			Prompt: prompt,
+			OnDone: func(value string) tea.Msg {
+				resChan <- value
+				return nil
+			},
+			OnCancel: func() tea.Msg {
+				cancelChan <- struct{}{}
+				return nil
+			},
+		})
+	}()
+
+	select {
+	case value := <-resChan:
+		return value, nil
+	case <-cancelChan:
+		return nil, nil
+	case <-ctx.Done():
+		return nil, ctx.Err()
+	}
+}
+
 func (m *uiModule) uiPromptTable(ctx context.Context, args ucl.CallArgs) (any, error) {
 	tables, err := m.tableService.ListTables(context.Background())
 	if err != nil {
@@ -166,13 +209,14 @@ func moduleUI(
 	return ucl.Module{
 		Name: "ui",
 		Builtins: map[string]ucl.BuiltinHandler{
-			"command":      m.uiCommand,
-			"prompt":       m.uiPrompt,
-			"prompt-table": m.uiPromptTable,
-			"confirm":      m.uiConfirm,
-			"query":        m.uiQuery,
-			"filter":       m.uiFilter,
-			"bind":         m.uiBind,
+			"command":         m.uiCommand,
+			"prompt":          m.uiPrompt,
+			"prompt-table":    m.uiPromptTable,
+			"prompt-keypress": m.uiInKey,
+			"confirm":         m.uiConfirm,
+			"query":           m.uiQuery,
+			"filter":          m.uiFilter,
+			"bind":            m.uiBind,
 		},
 	}, m.ckb
 }