Switch to go mod

vendor/github.com/heroku/docker-registry-client/registry/blob.go

@@ -0,0 +1,108 @@
+package registry
+
+import (
+	"io"
+	"net/http"
+	"net/url"
+
+	"github.com/docker/distribution"
+	digest "github.com/opencontainers/go-digest"
+)
+
+func (registry *Registry) DownloadBlob(repository string, digest digest.Digest) (io.ReadCloser, error) {
+	url := registry.url("/v2/%s/blobs/%s", repository, digest)
+	registry.Logf("registry.blob.download url=%s repository=%s digest=%s", url, repository, digest)
+
+	resp, err := registry.Client.Get(url)
+	if err != nil {
+		return nil, err
+	}
+
+	return resp.Body, nil
+}
+
+func (registry *Registry) UploadBlob(repository string, digest digest.Digest, content io.Reader) error {
+	uploadUrl, err := registry.initiateUpload(repository)
+	if err != nil {
+		return err
+	}
+	q := uploadUrl.Query()
+	q.Set("digest", digest.String())
+	uploadUrl.RawQuery = q.Encode()
+
+	registry.Logf("registry.blob.upload url=%s repository=%s digest=%s", uploadUrl, repository, digest)
+
+	upload, err := http.NewRequest("PUT", uploadUrl.String(), content)
+	if err != nil {
+		return err
+	}
+	upload.Header.Set("Content-Type", "application/octet-stream")
+
+	_, err = registry.Client.Do(upload)
+	return err
+}
+
+func (registry *Registry) HasBlob(repository string, digest digest.Digest) (bool, error) {
+	checkUrl := registry.url("/v2/%s/blobs/%s", repository, digest)
+	registry.Logf("registry.blob.check url=%s repository=%s digest=%s", checkUrl, repository, digest)
+
+	resp, err := registry.Client.Head(checkUrl)
+	if resp != nil {
+		defer resp.Body.Close()
+	}
+	if err == nil {
+		return resp.StatusCode == http.StatusOK, nil
+	}
+
+	urlErr, ok := err.(*url.Error)
+	if !ok {
+		return false, err
+	}
+	httpErr, ok := urlErr.Err.(*HttpStatusError)
+	if !ok {
+		return false, err
+	}
+	if httpErr.Response.StatusCode == http.StatusNotFound {
+		return false, nil
+	}
+
+	return false, err
+}
+
+func (registry *Registry) BlobMetadata(repository string, digest digest.Digest) (distribution.Descriptor, error) {
+	checkUrl := registry.url("/v2/%s/blobs/%s", repository, digest)
+	registry.Logf("registry.blob.check url=%s repository=%s digest=%s", checkUrl, repository, digest)
+
+	resp, err := registry.Client.Head(checkUrl)
+	if resp != nil {
+		defer resp.Body.Close()
+	}
+	if err != nil {
+		return distribution.Descriptor{}, err
+	}
+
+	return distribution.Descriptor{
+		Digest: digest,
+		Size:   resp.ContentLength,
+	}, nil
+}
+
+func (registry *Registry) initiateUpload(repository string) (*url.URL, error) {
+	initiateUrl := registry.url("/v2/%s/blobs/uploads/", repository)
+	registry.Logf("registry.blob.initiate-upload url=%s repository=%s", initiateUrl, repository)
+
+	resp, err := registry.Client.Post(initiateUrl, "application/octet-stream", nil)
+	if resp != nil {
+		defer resp.Body.Close()
+	}
+	if err != nil {
+		return nil, err
+	}
+
+	location := resp.Header.Get("Location")
+	locationUrl, err := url.Parse(location)
+	if err != nil {
+		return nil, err
+	}
+	return locationUrl, nil
+}