text_input.go

package utils

import (
	"fmt"

	tea "github.com/charmbracelet/bubbletea"
	"github.com/initia-labs/weave/styles"
)

type TextInput struct {
	Text        string
	Cursor      int // Cursor position within the text
	Placeholder string
	Entered     bool
}

func NewTextInput() TextInput {
	return TextInput{Text: "", Cursor: 0, Placeholder: "<todo: Jennie revisit placeholder>", Entered: false}
}

func (ti TextInput) Update(msg tea.Msg) (TextInput, tea.Cmd, bool) {
	switch msg := msg.(type) {
	case tea.KeyMsg:
		switch msg.Type {
		case tea.KeyEnter:
			ti.Entered = true
			return ti, nil, false
		case tea.KeyBackspace, tea.KeyCtrlH:
			if ti.Cursor > 0 && len(ti.Text) > 0 {
				ti.Text = ti.Text[:ti.Cursor-1] + ti.Text[ti.Cursor:]
				ti.Cursor--
			}
		case tea.KeyRunes:
			ti.Text = ti.Text[:ti.Cursor] + string(msg.Runes) + ti.Text[ti.Cursor:]
			ti.Cursor += len(msg.Runes)
		case tea.KeyLeft:
			if ti.Cursor > 0 {
				ti.Cursor--
			}
		case tea.KeyRight:
			if ti.Cursor < len(ti.Text) {
				ti.Cursor++
			}
		case tea.KeyCtrlC:
			return ti, tea.Quit, false
		}

	}
	return ti, nil, false
}

func (ti TextInput) View(questionText string) string {
	if !ti.Entered {
		questionText = styles.Text("? ", styles.Cyan) + questionText + "\n> "
	} else {
		questionText = styles.Text("✓ ", styles.Green) + questionText + "\n> "
	}

	var beforeCursor, cursorChar, afterCursor string
	if len(ti.Text) == 0 {
		return questionText + styles.Text(ti.Placeholder, styles.Gray) + styles.Cursor(" ") + "\n\nPress Enter to submit, or Ctrl+c to quit."
	} else if ti.Cursor < len(ti.Text) {
		// Cursor is within the text
		beforeCursor = styles.Text(ti.Text[:ti.Cursor], styles.White)
		cursorChar = styles.Cursor(ti.Text[ti.Cursor : ti.Cursor+1])
		afterCursor = styles.Text(ti.Text[ti.Cursor+1:], styles.White)
	} else {
		// Cursor is at the end of the text
		beforeCursor = styles.Text(ti.Text, styles.White)
		cursorChar = styles.Cursor(" ")
	}

	// Compose the full view string
	return fmt.Sprintf("%s%s%s%s\n\nPress Enter to submit, or Ctrl+c to quit.", questionText, beforeCursor, cursorChar, afterCursor)
}