mirror of
https://github.com/charmbracelet/gum
synced 2026-03-15 14:15:44 +01:00
2fe0291e02* stdin: add trim of trailing newline * choose: remove trimsuffix * filter: remove trimsuffix * style: remove trimsuffix
46 lines
734 B
Go
46 lines
734 B
Go
package stdin
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
// Read reads input from an stdin pipe.
|
|
func Read() (string, error) {
|
|
if IsEmpty() {
|
|
return "", fmt.Errorf("stdin is empty")
|
|
}
|
|
|
|
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 strings.TrimSuffix(b.String(), "\n"), nil
|
|
}
|
|
|
|
// IsEmpty returns whether stdin is empty.
|
|
func IsEmpty() bool {
|
|
stat, err := os.Stdin.Stat()
|
|
if err != nil {
|
|
return true
|
|
}
|
|
|
|
if stat.Mode()&os.ModeNamedPipe == 0 && stat.Size() == 0 {
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|