diff --git a/kustomize/commands/edit/set/setimage.go b/kustomize/commands/edit/set/setimage.go index f03501dcfd..91381f7a6f 100644 --- a/kustomize/commands/edit/set/setimage.go +++ b/kustomize/commands/edit/set/setimage.go @@ -9,6 +9,7 @@ import ( "sort" "strings" + "sigs.k8s.io/kustomize/api/image" "sigs.k8s.io/kustomize/api/types" "github.com/spf13/cobra" @@ -31,8 +32,10 @@ var ( errImageInvalidArgs = errors.New(`invalid format of image, use one of the following options: - =: - =@ +- =:@ - = - : +- :@ - @`) ) @@ -76,7 +79,7 @@ images: to the kustomization file if it doesn't exist, and overwrite the previous ones if the image name exists. -The image tag can only contain alphanumeric, '.', '_' and '-'. Passing * (asterisk) either as the new name, +The image tag can only contain alphanumeric, '.', '_' and '-'. Passing * (asterisk) either as the new name, the new tag, or the digest will preserve the appropriate values from the kustomization file. `, RunE: func(cmd *cobra.Command, args []string) error { @@ -204,52 +207,37 @@ func replaceDigest(image types.Image, digest string) types.Image { } func parse(arg string) (types.Image, error) { - // matches if there is an image name to overwrite - // =<:|@> - if s := strings.Split(arg, separator); len(s) == 2 { - p, err := parseOverwrite(s[1], true) - return types.Image{ - Name: s[0], - NewName: p.name, - NewTag: p.tag, - Digest: p.digest, - }, err + key, value, err := imageArgParse(arg) + if err != nil { + return types.Image{}, err } - // matches only for overwrites - // <:|@> - p, err := parseOverwrite(arg, false) - return types.Image{ - Name: p.name, - NewTag: p.tag, - Digest: p.digest, - }, err -} + name, tag, digest := image.Split(value) + if name == arg { + return types.Image{}, errImageInvalidArgs + } -// parseOverwrite parses the overwrite parameters -// from the given arg into a struct -func parseOverwrite(arg string, overwriteImage bool) (overwrite, error) { - // match @ - if d := strings.Split(arg, "@"); len(d) > 1 { - return overwrite{ - name: d[0], - digest: d[1], - }, nil + newImage := types.Image{ + NewTag: tag, + Digest: digest, } - // match : - if t := pattern.FindStringSubmatch(arg); len(t) == 3 { - return overwrite{ - name: t[1], - tag: t[2], - }, nil + if key == "" { + newImage.Name = name + } else { + newImage.Name = key + newImage.NewName = name } - // match - if len(arg) > 0 && overwriteImage { - return overwrite{ - name: arg, - }, nil + return newImage, nil +} + +func imageArgParse(arg string) (key string, value string, err error) { + s := strings.SplitN(arg, separator, 2) + if len(s) == 2 { + return s[0], s[1], nil + } else if len(s) == 1 { + return "", s[0], nil } - return overwrite{}, errImageInvalidArgs + return "", "", errImageInvalidArgs } diff --git a/kustomize/commands/edit/set/setimage_test.go b/kustomize/commands/edit/set/setimage_test.go index ad2ff1f6db..076ff4024a 100644 --- a/kustomize/commands/edit/set/setimage_test.go +++ b/kustomize/commands/edit/set/setimage_test.go @@ -96,6 +96,31 @@ func TestSetImage(t *testing.T) { " name: image1", }}, }, + { + given: given{ + args: []string{"my-image1:my-tag@sha256:24a0c4b4a4c0eb97a1aabb8e29f18e917d05abfe1b7a7c07857230879ce7d3d3"}, + }, + expected: expected{ + fileOutput: []string{ + "images:", + "- digest: sha256:24a0c4b4a4c0eb97a1aabb8e29f18e917d05abfe1b7a7c07857230879ce7d3d3", + " name: my-image1", + " newTag: my-tag", + }}, + }, + { + given: given{ + args: []string{"image1=my-image1:my-tag@sha256:24a0c4b4a4c0eb97a1aabb8e29f18e917d05abfe1b7a7c07857230879ce7d3d3"}, + }, + expected: expected{ + fileOutput: []string{ + "images:", + "- digest: sha256:24a0c4b4a4c0eb97a1aabb8e29f18e917d05abfe1b7a7c07857230879ce7d3d3", + " name: image1", + " newName: my-image1", + " newTag: my-tag", + }}, + }, { description: "=", given: given{