default_registry.go

package container

import (
	"fmt"
	"regexp"

	"github.com/docker/distribution/reference"
)

// The Host of a container registry where we can push images.
// ex)
// localhost:32000
// gcr.io/windmill-public-containers
type Registry string

var escapeRegex = regexp.MustCompile(`[/:@]`)

func escapeName(s string) string {
	return string(escapeRegex.ReplaceAll([]byte(s), []byte("_")))
}

// Produces a new image name that is in the specified registry.
// The name might be ugly, favoring uniqueness and simplicity and assuming that the prettiness of ephemeral dev image
// names is not that important.
func ReplaceRegistry(defaultRegistry Registry, rs RefSelector) (reference.Named, error) {
	if defaultRegistry == "" {
		return rs.AsNamedOnly(), nil
	}

	newNs := fmt.Sprintf("%s/%s", defaultRegistry, escapeName(rs.RefFamiliarName()))
	newN, err := reference.ParseNamed(newNs)
	if err != nil {
		return nil, err
	}

	return newN, nil
}