forked from Mirrors/bubbletea
94 lines
1.8 KiB
Go
94 lines
1.8 KiB
Go
package main
|
|
|
|
// An example illustrating how to pipe in data to a Bubble Tea application.
|
|
// Moreso, this serves as proof that Bubble Tea will automatically listen for
|
|
// keystrokes when input is not a TTY, such as when data is piped or redirected
|
|
// in.
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/charmbracelet/bubbles/textinput"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"github.com/charmbracelet/lipgloss"
|
|
)
|
|
|
|
func main() {
|
|
stat, err := os.Stdin.Stat()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if stat.Mode()&os.ModeNamedPipe == 0 && stat.Size() == 0 {
|
|
fmt.Println("Try piping in some text.")
|
|
os.Exit(1)
|
|
}
|
|
|
|
reader := bufio.NewReader(os.Stdin)
|
|
var b strings.Builder
|
|
|
|
for {
|
|
r, _, err := reader.ReadRune()
|
|
if err != nil && err == io.EOF {
|
|
break
|
|
}
|
|
_, err = b.WriteRune(r)
|
|
if err != nil {
|
|
fmt.Println("Error getting input:", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
model := newModel(strings.TrimSpace(b.String()))
|
|
|
|
if _, err := tea.NewProgram(model).Run(); err != nil {
|
|
fmt.Println("Couldn't start program:", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
type model struct {
|
|
userInput textinput.Model
|
|
}
|
|
|
|
func newModel(initialValue string) (m model) {
|
|
i := textinput.New()
|
|
i.Prompt = ""
|
|
i.CursorStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("63"))
|
|
i.Width = 48
|
|
i.SetValue(initialValue)
|
|
i.CursorEnd()
|
|
i.Focus()
|
|
|
|
m.userInput = i
|
|
return
|
|
}
|
|
|
|
func (m model) Init() tea.Cmd {
|
|
return textinput.Blink
|
|
}
|
|
|
|
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
if key, ok := msg.(tea.KeyMsg); ok {
|
|
switch key.Type {
|
|
case tea.KeyCtrlC, tea.KeyEscape, tea.KeyEnter:
|
|
return m, tea.Quit
|
|
}
|
|
}
|
|
|
|
var cmd tea.Cmd
|
|
m.userInput, cmd = m.userInput.Update(msg)
|
|
return m, cmd
|
|
}
|
|
|
|
func (m model) View() string {
|
|
return fmt.Sprintf(
|
|
"\nYou piped in: %s\n\nPress ^C to exit",
|
|
m.userInput.View(),
|
|
)
|
|
}
|