forked from TrueCloudLab/distribution
671184e910
Defining an interface on the implementer side is generally not best practice in Go code. There is no code in the distribution module which consumes a ManifestBuilder value so there is no need to define the interface in the distribution module. Export the concrete ManifestBuilder types and modify the constructors to return concrete values. Co-authored-by: Sebastiaan van Stijn <github@gone.nl> Signed-off-by: Cory Snider <csnider@mirantis.com>
86 lines
2.8 KiB
Go
86 lines
2.8 KiB
Go
package testutil
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/distribution/distribution/v3"
|
|
"github.com/distribution/distribution/v3/internal/dcontext"
|
|
"github.com/distribution/distribution/v3/manifest/manifestlist"
|
|
"github.com/distribution/distribution/v3/manifest/ocischema"
|
|
"github.com/distribution/distribution/v3/manifest/schema2"
|
|
"github.com/opencontainers/go-digest"
|
|
)
|
|
|
|
// MakeManifestList constructs a manifest list out of a list of manifest digests
|
|
func MakeManifestList(blobstatter distribution.BlobStatter, manifestDigests []digest.Digest) (*manifestlist.DeserializedManifestList, error) {
|
|
ctx := dcontext.Background()
|
|
|
|
manifestDescriptors := make([]manifestlist.ManifestDescriptor, 0, len(manifestDigests))
|
|
for _, manifestDigest := range manifestDigests {
|
|
descriptor, err := blobstatter.Stat(ctx, manifestDigest)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
platformSpec := manifestlist.PlatformSpec{
|
|
Architecture: "atari2600",
|
|
OS: "CP/M",
|
|
Variant: "ternary",
|
|
Features: []string{"VLIW", "superscalaroutoforderdevnull"},
|
|
}
|
|
manifestDescriptor := manifestlist.ManifestDescriptor{
|
|
Descriptor: descriptor,
|
|
Platform: platformSpec,
|
|
}
|
|
manifestDescriptors = append(manifestDescriptors, manifestDescriptor)
|
|
}
|
|
|
|
return manifestlist.FromDescriptors(manifestDescriptors)
|
|
}
|
|
|
|
// MakeSchema2Manifest constructs a schema 2 manifest from a given list of digests and returns
|
|
// the digest of the manifest
|
|
func MakeSchema2Manifest(repository distribution.Repository, digests []digest.Digest) (distribution.Manifest, error) {
|
|
ctx := dcontext.Background()
|
|
blobStore := repository.Blobs(ctx)
|
|
|
|
var configJSON []byte
|
|
|
|
d, err := blobStore.Put(ctx, schema2.MediaTypeImageConfig, configJSON)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("unexpected error storing content in blobstore: %v", err)
|
|
}
|
|
builder := schema2.NewManifestBuilder(d, configJSON)
|
|
for _, dgst := range digests {
|
|
if err := builder.AppendReference(distribution.Descriptor{Digest: dgst}); err != nil {
|
|
return nil, fmt.Errorf("unexpected error building schema2 manifest: %v", err)
|
|
}
|
|
}
|
|
|
|
mfst, err := builder.Build(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("unexpected error generating schema2 manifest: %v", err)
|
|
}
|
|
|
|
return mfst, nil
|
|
}
|
|
|
|
func MakeOCIManifest(repository distribution.Repository, digests []digest.Digest) (distribution.Manifest, error) {
|
|
ctx := dcontext.Background()
|
|
blobStore := repository.Blobs(ctx)
|
|
|
|
var configJSON []byte
|
|
|
|
builder := ocischema.NewManifestBuilder(blobStore, configJSON, make(map[string]string))
|
|
for _, dgst := range digests {
|
|
if err := builder.AppendReference(distribution.Descriptor{Digest: dgst}); err != nil {
|
|
return nil, fmt.Errorf("unexpected error building OCI manifest: %v", err)
|
|
}
|
|
}
|
|
|
|
mfst, err := builder.Build(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("unexpected error generating OCI manifest: %v", err)
|
|
}
|
|
|
|
return mfst, nil
|
|
}
|