Add Base64 Encode and Decode methods

script.go

@@ -2,8 +2,10 @@ package script
 
 import (
  "bufio"
+ "bytes"
  "container/ring"
  "crypto/sha256"
+ "encoding/base64"
  "encoding/hex"
  "encoding/json"
  "fmt"
@@ -759,6 +761,43 @@ func (p *Pipe) SetError(err error) {
  p.err = err
 }
 
+// Base64Encode returns the returns the base64 encoding of the
+// pipe, or an error.
+func (p *Pipe) Base64Encode(enc *base64.Encoding) (string, error) {
+ if p.Error() != nil {
+ return "", p.Error()
+ }
+ var encodedData bytes.Buffer
+ encoder := base64.NewEncoder(enc, &encodedData)
+
+ _, err := io.Copy(encoder, p)
+ if err != nil {
+ p.SetError(err)
+ return "", err
+ }
+ encoder.Close()
+
+ return encodedData.String(), nil
+}
+
+// Base64Decode returns the string represented by the base64
+// pipe, or an error.
+func (p *Pipe) Base64Decode(enc *base64.Encoding) (string, error) {
+ if p.Error() != nil {
+ return "", p.Error()
+ }
+ decoder := base64.NewDecoder(enc, p)
+
+ var decodedData bytes.Buffer
+ _, err := io.Copy(&decodedData, decoder)
+ if err != nil {
+ p.SetError(err)
+ return "", err
+ }
+
+ return decodedData.String(), nil
+}
+
 // SHA256Sum returns the hex-encoded SHA-256 hash of the entire contents of the
 // pipe, or an error.
 func (p *Pipe) SHA256Sum() (string, error) {

script_test.go

@@ -3,6 +3,7 @@ package script_test
 import (
  "bufio"
  "bytes"
+ "encoding/base64"
  "errors"
  "fmt"
  "io"
@@ -1491,6 +1492,60 @@ func TestCountLines_ReturnsErrorGivenReadErrorOnPipe(t *testing.T) {
  }
 }
 
+func TestBase64Encode(t *testing.T) {
+ t.Parallel()
+ tcs := []struct {
+ name, input, want string
+ enc *base64.Encoding
+ }{
+ {
+ name: "for short string",
+ input: "Hello, world!",
+ want: "SGVsbG8sIHdvcmxkIQ==",
+ enc: base64.StdEncoding,
+ },
+ }
+
+ for _, tc := range tcs {
+ t.Run(tc.name, func(t *testing.T) {
+ got, err := script.Echo(tc.input).Base64Encode(tc.enc)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if got != tc.want {
+ t.Errorf("want %q, got %q", tc.want, got)
+ }
+ })
+ }
+}
+
+func TestBase64Decode(t *testing.T) {
+ t.Parallel()
+ tcs := []struct {
+ name, input, want string
+ enc *base64.Encoding
+ }{
+ {
+ name: "for short string",
+ input: "SGVsbG8sIHdvcmxkIQ==",
+ want: "Hello, world!",
+ enc: base64.StdEncoding,
+ },
+ }
+
+ for _, tc := range tcs {
+ t.Run(tc.name, func(t *testing.T) {
+ got, err := script.Echo(tc.input).Base64Decode(tc.enc)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if got != tc.want {
+ t.Errorf("want %q, got %q", tc.want, got)
+ }
+ })
+ }
+}
+
 func TestSHA256Sum_OutputsCorrectHash(t *testing.T) {
  t.Parallel()
  tcs := []struct {