mirror of
https://github.com/mudler/luet.git
synced 2025-06-30 17:22:38 +00:00
109 lines
2.8 KiB
Go
109 lines
2.8 KiB
Go
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
|
|
}
|