feat: add stdin input handling and logging helpers

This commit is contained in:
Maas Lalani 2022-07-06 12:09:21 -04:00
parent 454040cf4d
commit 2320f10278
No known key found for this signature in database
GPG key ID: 5A6ED5CBF1A0A000
2 changed files with 44 additions and 0 deletions

8
internal/log/log.go Normal file
View file

@ -0,0 +1,8 @@
package log
import "fmt"
// Error prints an error message to the user.
func Error(message string) {
fmt.Println("Error:", message)
}

36
internal/stdin/stdin.go Normal file
View file

@ -0,0 +1,36 @@
package stdin
import (
"bufio"
"io"
"os"
"strings"
)
// Read reads input from an stdin pipe.
func Read() (string, error) {
stat, err := os.Stdin.Stat()
if err != nil {
return "", 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 "", err
}
}
return b.String(), nil
}