introduce func to build Mapping from a set of KEY=VALUE strings

Signed-off-by: Nicolas De Loof <nicolas.deloof@gmail.com>

types/types.go

@@ -262,6 +262,20 @@ type StringOrNumberList []string
 // For the key without value (`key`), the mapped value is set to nil.
 type MappingWithEquals map[string]*string
 
+// NewMappingWithEquals build a new Mapping from a set of KEY=VALUE strings
+func NewMappingWithEquals(values []string) MappingWithEquals {
+	mapping := MappingWithEquals{}
+	for _, env := range values {
+		tokens := strings.SplitN(env, "=", 2)
+		if len(tokens) > 1 {
+			mapping[tokens[0]] = &tokens[1]
+		} else {
+			mapping[env] = nil
+		}
+	}
+	return mapping
+}
+
 // OverrideBy update MappingWithEquals with values from another MappingWithEquals
 func (e MappingWithEquals) OverrideBy(other MappingWithEquals) MappingWithEquals {
 	for k, v := range other {
@@ -298,6 +312,22 @@ func (e MappingWithEquals) RemoveEmpty() MappingWithEquals {
 // mapped value is set to an empty string `""`.
 type Mapping map[string]string
 
+// NewMapping build a new Mapping from a set of KEY=VALUE strings
+func NewMapping(values []string) Mapping {
+	mapping := Mapping{}
+	for _, value := range values {
+		parts := strings.SplitN(value, "=", 2)
+		key := parts[0]
+		switch {
+		case len(parts) == 1:
+			mapping[key] = ""
+		default:
+			mapping[key] = parts[1]
+		}
+	}
+	return mapping
+}
+
 // Labels is a mapping type for labels
 type Labels map[string]string
 

types/types_test.go

@@ -235,3 +235,22 @@ func TestExtension(t *testing.T) {
 	assert.NilError(t, err)
 	assert.Check(t, ok == false)
 }
+
+func TestNewMapping(t *testing.T) {
+	m := NewMapping([]string{
+		"FOO=BAR",
+		"ZOT=",
+		"QIX",
+	})
+	mw := NewMappingWithEquals([]string{
+		"FOO=BAR",
+		"ZOT=",
+		"QIX",
+	})
+	assert.Check(t, m["FOO"] == "BAR")
+	assert.Check(t, m["ZOT"] == "")
+	assert.Check(t, m["QIX"] == "")
+	assert.Check(t, *mw["FOO"] == "BAR")
+	assert.Check(t, *mw["ZOT"] == "")
+	assert.Check(t, mw["QIX"] == nil)
+}