mirror of
https://github.com/charmbracelet/gum
synced 2024-05-03 23:13:14 +02:00
38 lines
610 B
Go
38 lines
610 B
Go
package stdin
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
// Read reads input from an stdin pipe.
|
|
func Read() (string, error) {
|
|
stat, err := os.Stdin.Stat()
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to stat stdin: %w", err)
|
|
}
|
|
|
|
if stat.Mode()&os.ModeNamedPipe == 0 && stat.Size() == 0 {
|
|
return "", nil
|
|
}
|
|
|
|
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 {
|
|
return "", fmt.Errorf("failed to write rune: %w", err)
|
|
}
|
|
}
|
|
|
|
return b.String(), nil
|
|
}
|