feat(yaml): support array notation within yaml

Author: gsdevme

internal/pkg/gitops/version.go

@@ -6,6 +6,8 @@ import (
 	"fmt"
 	"gopkg.in/yaml.v3"
 	"io/ioutil"
+	"regexp"
+	"strconv"
 	"strings"
 )
 
@@ -41,8 +43,31 @@ func WriteVersion(f []byte, version string, newVersion string, filename string)
 
 func unwrapYaml(yaml map[string]interface{}, notion string) (string, error) {
 	d := yaml
+	r := regexp.MustCompile(`\d+]$`)
 
 	for _, k := range strings.Split(notion, ".") {
+		if r.MatchString(k) {
+			idx := r.FindString(k)[:1]
+
+			if len(idx) > 0 {
+				i, err := strconv.ParseInt(idx, 10, 8)
+
+				if err != nil {
+					return "", err
+				}
+
+				k = k[:len(k)-int(i)]
+
+				if a, ok := d[k].([]interface{}); ok {
+					if b, ok := a[i].(map[string]interface{}); ok {
+						d = b
+
+						continue
+					}
+				}
+			}
+		}
+
 		if _, ok := d[k]; !ok {
 			return "", fmt.Errorf("unable to find %s in yaml", notion)
 		}

internal/pkg/gitops/version_test.go

@@ -30,7 +30,7 @@ func TestReadCurrentVersion(t *testing.T) {
 		{
 			name: "invalid yaml",
 			args: args{
-				f:        []byte("wibble&&3..\nfff"),
+				f:  []byte("wibble&&3..\nfff"),
 				notation: "image.tag",
 			},
 			want:    "",
@@ -39,7 +39,7 @@ func TestReadCurrentVersion(t *testing.T) {
 		{
 			name: "simple yaml",
 			args: args{
-				f:        createSimpleYaml("v1.2.99"),
+				f:  createSimpleYaml("v1.2.99"),
 				notation: "image.tag",
 			},
 			want:    "v1.2.99",
@@ -48,15 +48,24 @@ func TestReadCurrentVersion(t *testing.T) {
 		{
 			name: "invalid notation to yaml",
 			args: args{
-				f:        createSimpleYaml("v1.2.99"),
+				f:  createSimpleYaml("v1.2.99"),
 				notation: "image.nope",
 			},
 			want:    "",
 			wantErr: true,
 		},
+		{
+			name: "array notation",
+			args: args{
+				f:  createSimpleArrayYaml("v1.9"),
+				notation: "images[3].tag",
+			},
+			want:    "v1.9",
+			wantErr: false,
+		},
 	}
 
-	fmt.Println(string(createSimpleYaml("v1.2.99")))
+	fmt.Println(string(createSimpleArrayYaml("v1.2.99")))
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
@@ -88,3 +97,20 @@ func createSimpleYaml(t string) []byte {
 
 	return marshal
 }
+
+func createSimpleArrayYaml(t string) []byte {
+	type tag struct {
+		Tag string `yaml:"tag"`
+	}
+
+	marshal, err := yaml.Marshal(&struct {
+		Images []tag `yaml:"images"`
+	}{
+		Images: []tag{{Tag: "v1"},{Tag: "v2"},{Tag: "v3"},{Tag: t}},
+	})
+	if err != nil {
+		return nil
+	}
+
+	return marshal
+}