mirror of
https://github.com/genuinetools/reg.git
synced 2024-05-09 16:28:32 -04:00
32589e90be
* passing context in layer calls * more contexting * clair folder and context in handlers * fixed token transport to reuse request context * tests * taking out context pass in server handlers
127 lines
3.5 KiB
Go
127 lines
3.5 KiB
Go
package registry
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
|
|
"github.com/docker/distribution"
|
|
"github.com/docker/distribution/manifest/manifestlist"
|
|
"github.com/docker/distribution/manifest/schema1"
|
|
"github.com/docker/distribution/manifest/schema2"
|
|
)
|
|
|
|
var (
|
|
// ErrUnexpectedSchemaVersion a specific schema version was requested, but was not returned
|
|
ErrUnexpectedSchemaVersion = errors.New("recieved a different schema version than expected")
|
|
)
|
|
|
|
// Manifest returns the manifest for a specific repository:tag.
|
|
func (r *Registry) Manifest(ctx context.Context, repository, ref string) (distribution.Manifest, error) {
|
|
uri := r.url("/v2/%s/manifests/%s", repository, ref)
|
|
r.Logf("registry.manifests uri=%s repository=%s ref=%s", uri, repository, ref)
|
|
|
|
req, err := http.NewRequest("GET", uri, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
req.Header.Add("Accept", fmt.Sprintf("%s;q=0.9", schema2.MediaTypeManifest))
|
|
|
|
resp, err := r.Client.Do(req.WithContext(ctx))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
r.Logf("registry.manifests resp.Status=%s, body=%s", resp.Status, body)
|
|
|
|
m, _, err := distribution.UnmarshalManifest(resp.Header.Get("Content-Type"), body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return m, nil
|
|
}
|
|
|
|
// ManifestList gets the registry v2 manifest list.
|
|
func (r *Registry) ManifestList(ctx context.Context, repository, ref string) (manifestlist.ManifestList, error) {
|
|
uri := r.url("/v2/%s/manifests/%s", repository, ref)
|
|
r.Logf("registry.manifests uri=%s repository=%s ref=%s", uri, repository, ref)
|
|
|
|
var m manifestlist.ManifestList
|
|
if _, err := r.getJSON(ctx, uri, &m); err != nil {
|
|
r.Logf("registry.manifests response=%v", m)
|
|
return m, err
|
|
}
|
|
|
|
return m, nil
|
|
}
|
|
|
|
// ManifestV2 gets the registry v2 manifest.
|
|
func (r *Registry) ManifestV2(ctx context.Context, repository, ref string) (schema2.Manifest, error) {
|
|
uri := r.url("/v2/%s/manifests/%s", repository, ref)
|
|
r.Logf("registry.manifests uri=%s repository=%s ref=%s", uri, repository, ref)
|
|
|
|
var m schema2.Manifest
|
|
if _, err := r.getJSON(ctx, uri, &m); err != nil {
|
|
r.Logf("registry.manifests response=%v", m)
|
|
return m, err
|
|
}
|
|
|
|
if m.Versioned.SchemaVersion != 2 {
|
|
return m, ErrUnexpectedSchemaVersion
|
|
}
|
|
|
|
return m, nil
|
|
}
|
|
|
|
// ManifestV1 gets the registry v1 manifest.
|
|
func (r *Registry) ManifestV1(ctx context.Context, repository, ref string) (schema1.SignedManifest, error) {
|
|
uri := r.url("/v2/%s/manifests/%s", repository, ref)
|
|
r.Logf("registry.manifests uri=%s repository=%s ref=%s", uri, repository, ref)
|
|
|
|
var m schema1.SignedManifest
|
|
if _, err := r.getJSON(ctx, uri, &m); err != nil {
|
|
r.Logf("registry.manifests response=%v", m)
|
|
return m, err
|
|
}
|
|
|
|
if m.Versioned.SchemaVersion != 1 {
|
|
return m, ErrUnexpectedSchemaVersion
|
|
}
|
|
|
|
return m, nil
|
|
}
|
|
|
|
// PutManifest calls a PUT for the specific manifest for an image.
|
|
func (r *Registry) PutManifest(ctx context.Context, repository, ref string, manifest distribution.Manifest) error {
|
|
url := r.url("/v2/%s/manifests/%s", repository, ref)
|
|
r.Logf("registry.manifest.put url=%s repository=%s reference=%s", url, repository, ref)
|
|
|
|
b, err := json.Marshal(manifest)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
req, err := http.NewRequest("PUT", url, bytes.NewBuffer(b))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
req.Header.Set("Content-Type", schema2.MediaTypeManifest)
|
|
resp, err := r.Client.Do(req.WithContext(ctx))
|
|
if resp != nil {
|
|
defer resp.Body.Close()
|
|
}
|
|
return err
|
|
}
|