2022-07-06 18:09:21 +02:00
|
|
|
package stdin
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bufio"
|
2022-08-05 00:45:19 +02:00
|
|
|
"fmt"
|
2022-07-06 18:09:21 +02:00
|
|
|
"io"
|
|
|
|
"os"
|
|
|
|
"strings"
|
|
|
|
)
|
|
|
|
|
|
|
|
// Read reads input from an stdin pipe.
|
|
|
|
func Read() (string, error) {
|
|
|
|
stat, err := os.Stdin.Stat()
|
|
|
|
if err != nil {
|
2022-08-05 00:45:19 +02:00
|
|
|
return "", fmt.Errorf("failed to stat stdin: %w", err)
|
2022-07-06 18:09:21 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
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 {
|
2022-08-05 00:45:19 +02:00
|
|
|
return "", fmt.Errorf("failed to write rune: %w", err)
|
2022-07-06 18:09:21 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return b.String(), nil
|
|
|
|
}
|