WTF-566 Support GitLab projects as list

cfg/parsers.go

@@ -0,0 +1,28 @@
+package cfg
+
+import (
+	"fmt"
+
+	"github.com/olebedev/config"
+)
+
+// ParseAsMapOrList takes a configuration key and attempts to parse it first as a map
+// and then as a list. Map entries are concatenated as "key/value"
+func ParseAsMapOrList(ymlConfig *config.Config, configKey string) []string {
+	result := []string{}
+
+	mapItems, err := ymlConfig.Map(configKey)
+	if err == nil {
+		for key, value := range mapItems {
+			result = append(result, fmt.Sprintf("%s/%s", value, key))
+		}
+		return result
+	}
+
+	listItems := ymlConfig.UList(configKey)
+	for _, listItem := range listItems {
+		result = append(result, listItem.(string))
+	}
+
+	return result
+}

cfg/parsers_test.go

@@ -0,0 +1,50 @@
+package cfg
+
+import (
+	"testing"
+
+	"github.com/olebedev/config"
+)
+
+func Test_ParseAsMapOrList(t *testing.T) {
+	tests := []struct {
+		name          string
+		configKey     string
+		yaml          string
+		expectedCount int
+	}{
+		{
+			name:          "as empty set",
+			configKey:     "data",
+			yaml:          "",
+			expectedCount: 0,
+		},
+		{
+			name:          "as map",
+			configKey:     "data",
+			yaml:          "data:\n  a: cat\n  b: dog",
+			expectedCount: 2,
+		},
+		{
+			name:          "as list",
+			configKey:     "data",
+			yaml:          "data:\n  - cat\n  - dog\n  - rat\n",
+			expectedCount: 3,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			ymlConfig, err := config.ParseYaml(tt.yaml)
+			if err != nil {
+				t.Errorf("\nexpected: no error\n     got: %v", err)
+			}
+
+			actual := ParseAsMapOrList(ymlConfig, tt.configKey)
+
+			if tt.expectedCount != len(actual) {
+				t.Errorf("\nexpected: %d\n     got: %d", tt.expectedCount, len(actual))
+			}
+		})
+	}
+}