WithProgress option for crane push/pull

pkg/crane/options.go

@@ -109,6 +109,17 @@ func WithPlatform(platform *v1.Platform) Option {
 	}
 }
 
+// WithProgress is an option which takes a channel that will receive
+// progress updates as bytes are written.
+//
+// Sending updates to an unbuffered channel will block writes, so callers
+// should provide a buffered channel to avoid potential deadlocks.
+func WithProgress(updates chan<- v1.Update) Option {
+	return func(o *Options) {
+		o.Remote = append(o.Remote, remote.WithProgress(updates))
+	}
+}
+
 // WithAuthFromKeychain is a functional option for overriding the default
 // authenticator for remote operations, using an authn.Keychain to find
 // credentials.

pkg/crane/options_test.go

@@ -19,6 +19,7 @@ import (
 	"net/http"
 	"testing"
 
+	v1 "github.com/google/go-containerregistry/pkg/v1"
 	"github.com/google/go-containerregistry/pkg/v1/remote"
 )
 
@@ -56,3 +57,13 @@ func TestInsecureTransport(t *testing.T) {
 		t.Errorf("got: %t\nwant: %t", got, want)
 	}
 }
+
+func TestWithProgress(t *testing.T) {
+	c := make(chan v1.Update, 100)
+
+	opts := GetOptions(WithProgress(c))
+
+	if len(opts.Remote) != 2 {
+		t.Errorf("expected 1 remote option, got %d", len(opts.Remote))
+	}
+}