package util

import (
	"bytes"
	"fmt"
	"io/ioutil"
	"os"

	"git.sigsum.org/sigsum-go/pkg/hex"
)

func BytesFromStdin() ([]byte, error) {
	b, err := ioutil.ReadAll(os.Stdin)
	if err != nil {
		return nil, fmt.Errorf("util: read stdin: %v", err)
	}
	return b, nil
}

func HexFromStdin() ([]byte, error) {
	b, err := BytesFromStdin()
	if err != nil {
		return nil, err
	}
	b, err = hex.Deserialize(string(bytes.TrimSpace(b)))
	if err != nil {
		return nil, fmt.Errorf("util: parse stdin: %v", err)
	}
	return b, nil
}