137 lines
3.5 KiB
Go
137 lines
3.5 KiB
Go
package registry
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"path/filepath"
|
|
"strings"
|
|
"text/template"
|
|
|
|
"github.com/containers/image/v5/docker/reference"
|
|
digest "github.com/opencontainers/go-digest"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// Image holds information about an image.
|
|
type Image struct {
|
|
Domain string
|
|
Path string
|
|
Tag string
|
|
Digest digest.Digest
|
|
HubLink string
|
|
|
|
named reference.Named
|
|
opts ParseImageOptions
|
|
}
|
|
|
|
// ParseImageOptions holds image options for parsing.
|
|
type ParseImageOptions struct {
|
|
Name string
|
|
HubTpl string
|
|
}
|
|
|
|
// ParseImage returns an Image struct with all the values filled in for a given image.
|
|
func ParseImage(parseOpts ParseImageOptions) (Image, error) {
|
|
// Parse the image name and tag.
|
|
named, err := reference.ParseNormalizedNamed(parseOpts.Name)
|
|
if err != nil {
|
|
return Image{}, errors.Wrapf(err, "parsing image %s failed", parseOpts.Name)
|
|
}
|
|
// Add the latest lag if they did not provide one.
|
|
named = reference.TagNameOnly(named)
|
|
|
|
i := Image{
|
|
opts: parseOpts,
|
|
named: named,
|
|
Domain: reference.Domain(named),
|
|
Path: reference.Path(named),
|
|
}
|
|
|
|
// Hub link
|
|
i.HubLink, err = i.hubLink()
|
|
if err != nil {
|
|
return Image{}, errors.Wrap(err, fmt.Sprintf("resolving hub link for image %s failed", parseOpts.Name))
|
|
}
|
|
|
|
// Add the tag if there was one.
|
|
if tagged, ok := named.(reference.Tagged); ok {
|
|
i.Tag = tagged.Tag()
|
|
}
|
|
|
|
// Add the digest if there was one.
|
|
if canonical, ok := named.(reference.Canonical); ok {
|
|
i.Digest = canonical.Digest()
|
|
}
|
|
|
|
return i, nil
|
|
}
|
|
|
|
// Name returns the full name representation of an image.
|
|
func (i Image) Name() string {
|
|
return i.named.Name()
|
|
}
|
|
|
|
// String returns the string representation of an image.
|
|
func (i Image) String() string {
|
|
return i.named.String()
|
|
}
|
|
|
|
// Reference returns either the digest if it is non-empty or the tag for the image.
|
|
func (i Image) Reference() string {
|
|
if len(i.Digest.String()) > 1 {
|
|
return i.Digest.String()
|
|
}
|
|
|
|
return i.Tag
|
|
}
|
|
|
|
// WithDigest sets the digest for an image.
|
|
func (i Image) WithDigest(digest digest.Digest) (err error) {
|
|
i.Digest = digest
|
|
i.named, err = reference.WithDigest(i.named, digest)
|
|
return err
|
|
}
|
|
|
|
func (i Image) hubLink() (string, error) {
|
|
if i.opts.HubTpl != "" {
|
|
var out bytes.Buffer
|
|
tmpl, err := template.New("tmpl").
|
|
Option("missingkey=error").
|
|
Parse(i.opts.HubTpl)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
err = tmpl.Execute(&out, i)
|
|
return out.String(), err
|
|
}
|
|
|
|
switch i.Domain {
|
|
case "docker.io":
|
|
prefix := "r"
|
|
path := i.Path
|
|
if strings.HasPrefix(i.Path, "library/") {
|
|
prefix = "_"
|
|
path = strings.Replace(i.Path, "library/", "", 1)
|
|
}
|
|
return fmt.Sprintf("https://hub.docker.com/%s/%s", prefix, path), nil
|
|
case "docker.bintray.io", "jfrog-docker-reg2.bintray.io":
|
|
return fmt.Sprintf("https://bintray.com/jfrog/reg2/%s", strings.ReplaceAll(i.Path, "/", "%3A")), nil
|
|
case "docker.pkg.github.com":
|
|
return fmt.Sprintf("https://github.com/%s/packages", filepath.ToSlash(filepath.Dir(i.Path))), nil
|
|
case "gcr.io":
|
|
return fmt.Sprintf("https://%s/%s", i.Domain, i.Path), nil
|
|
case "ghcr.io":
|
|
ref := strings.Split(i.Path, "/")
|
|
ghUser, ghPackage := ref[0], ref[1]
|
|
return fmt.Sprintf("https://github.com/users/%s/packages/container/package/%s", ghUser, ghPackage), nil
|
|
case "quay.io":
|
|
return fmt.Sprintf("https://quay.io/repository/%s", i.Path), nil
|
|
case "registry.access.redhat.com":
|
|
return fmt.Sprintf("https://access.redhat.com/containers/#/registry.access.redhat.com/%s", i.Path), nil
|
|
case "registry.gitlab.com":
|
|
return fmt.Sprintf("https://gitlab.com/%s/container_registry", i.Path), nil
|
|
default:
|
|
return "", nil
|
|
}
|
|
}
|