mirror of
https://github.com/containers/podman.git
synced 2025-08-06 19:44:14 +08:00
7030b559fb
In a different PR review, it was noted that defined error types for artifacts was lacking. We have these for most other commands and they help with error differentiation. The changes here are to define the errors, implement them in the library, and adopt test verifications to match. Signed-off-by: Brent Baude <bbaude@redhat.com>
80 lines
2.1 KiB
Go
80 lines
2.1 KiB
Go
package libartifact
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/containers/image/v5/manifest"
|
|
"github.com/containers/podman/v5/pkg/libartifact/types"
|
|
"github.com/opencontainers/go-digest"
|
|
)
|
|
|
|
type Artifact struct {
|
|
// Manifest is the OCI manifest for the artifact with the name.
|
|
// In a valid artifact the Manifest is guaranteed to not be nil.
|
|
Manifest *manifest.OCI1
|
|
Name string
|
|
}
|
|
|
|
// TotalSizeBytes returns the total bytes of the all the artifact layers
|
|
func (a *Artifact) TotalSizeBytes() int64 {
|
|
var s int64
|
|
for _, layer := range a.Manifest.Layers {
|
|
s += layer.Size
|
|
}
|
|
return s
|
|
}
|
|
|
|
// GetName returns the "name" or "image reference" of the artifact
|
|
func (a *Artifact) GetName() (string, error) {
|
|
if a.Name != "" {
|
|
return a.Name, nil
|
|
}
|
|
// We don't have a concept of None for artifacts yet, but if we do,
|
|
// then we should probably not error but return `None`
|
|
return "", types.ErrArtifactUnamed
|
|
}
|
|
|
|
// SetName is a accessor for setting the artifact name
|
|
// Note: long term this may not be needed, and we would
|
|
// be comfortable with simply using the exported field
|
|
// called Name
|
|
func (a *Artifact) SetName(name string) {
|
|
a.Name = name
|
|
}
|
|
|
|
func (a *Artifact) GetDigest() (*digest.Digest, error) {
|
|
b, err := json.Marshal(a.Manifest)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
artifactDigest := digest.FromBytes(b)
|
|
return &artifactDigest, nil
|
|
}
|
|
|
|
type ArtifactList []*Artifact
|
|
|
|
// GetByNameOrDigest returns an artifact, if present, by a given name
|
|
// Returns an error if not found
|
|
func (al ArtifactList) GetByNameOrDigest(nameOrDigest string) (*Artifact, bool, error) {
|
|
// This is the hot route through
|
|
for _, artifact := range al {
|
|
if artifact.Name == nameOrDigest {
|
|
return artifact, false, nil
|
|
}
|
|
}
|
|
// Before giving up, check by digest
|
|
for _, artifact := range al {
|
|
artifactDigest, err := artifact.GetDigest()
|
|
if err != nil {
|
|
return nil, false, err
|
|
}
|
|
// If the artifact's digest matches or is a prefix of ...
|
|
if artifactDigest.Encoded() == nameOrDigest || strings.HasPrefix(artifactDigest.Encoded(), nameOrDigest) {
|
|
return artifact, true, nil
|
|
}
|
|
}
|
|
return nil, false, fmt.Errorf("%s: %w", nameOrDigest, types.ErrArtifactNotExist)
|
|
}
|