Merge pull request #139 from hoshsadiq/fix-stringmap

Fix StringMap and StringsMap
knadh · Mar 22, 2022 · f7f848b · f7f848b
2 parents a9bede6 + d468182
commit f7f848b
Show file tree

Hide file tree

Showing 2 changed files with 84 additions and 24 deletions.
diff --git a/getters.go b/getters.go
@@ -443,18 +443,22 @@ func (ko *Koanf) StringMap(path string) map[string]string {
 		return out
 	}
 
-	mp, ok := o.(map[string]interface{})
-	if !ok {
-		return out
-	}
-	out = make(map[string]string, len(mp))
-	for k, v := range mp {
-		switch s := v.(type) {
-		case string:
-			out[k] = s
-		default:
-			// There's a non string type. Return.
-			return map[string]string{}
+	switch mp := o.(type) {
+	case map[string]string:
+		out = make(map[string]string, len(mp))
+		for k, v := range mp {
+			out[k] = v
+		}
+	case map[string]interface{}:
+		out = make(map[string]string, len(mp))
+		for k, v := range mp {
+			switch s := v.(type) {
+			case string:
+				out[k] = s
+			default:
+				// There's a non string type. Return.
+				return map[string]string{}
+			}
 		}
 	}
 
@@ -483,25 +487,48 @@ func (ko *Koanf) StringsMap(path string) map[string][]string {
 		return out
 	}
 
-	mp, ok := o.(map[string]interface{})
-	if !ok {
-		return out
-	}
-	out = make(map[string][]string, len(mp))
-	for k, v := range mp {
-		switch s := v.(type) {
-		case []interface{}:
-			for _, v := range s {
+	switch mp := o.(type) {
+	case map[string][]string:
+		out = make(map[string][]string, len(mp))
+		for k, v := range mp {
+			out[k] = make([]string, 0, len(v))
+			for _, s := range v {
+				out[k] = append(out[k], s)
+			}
+		}
+	case map[string][]interface{}:
+		out = make(map[string][]string, len(mp))
+		for k, v := range mp {
+			for _, v := range v {
 				switch sv := v.(type) {
 				case string:
 					out[k] = append(out[k], sv)
 				default:
 					return map[string][]string{}
 				}
 			}
-		default:
-			// There's a non []interface type. Return.
-			return map[string][]string{}
+		}
+	case map[string]interface{}:
+		out = make(map[string][]string, len(mp))
+		for k, v := range mp {
+			switch s := v.(type) {
+			case []string:
+				for _, v := range s {
+					out[k] = append(out[k], v)
+				}
+			case []interface{}:
+				for _, v := range s {
+					switch sv := v.(type) {
+					case string:
+						out[k] = append(out[k], sv)
+					default:
+						return map[string][]string{}
+					}
+				}
+			default:
+				// There's a non []interface type. Return.
+				return map[string][]string{}
+			}
 		}
 	}
 

diff --git a/koanf_test.go b/koanf_test.go
@@ -1188,3 +1188,36 @@ func TestDelete(t *testing.T) {
 		assert.Equal(false, c.koanf.Exists("empty"))
 	}
 }
+
+func TestGetStringsMap(t *testing.T) {
+	assert := assert.New(t)
+
+	k := koanf.New(delim)
+
+	k.Load(confmap.Provider(map[string]interface{}{
+		"str": map[string]string{
+			"k1": "value",
+		},
+		"strs": map[string][]string{
+			"k1": {"value"},
+		},
+		"iface": map[string]interface{}{
+			"k2": "value",
+		},
+		"ifaces": map[string][]interface{}{
+			"k2": {"value"},
+		},
+		"ifaces2": map[string]interface{}{
+			"k2": []interface{}{"value"},
+		},
+		"ifaces3": map[string]interface{}{
+			"k2": []string{"value"},
+		},
+	}, "."), nil)
+	assert.Equal(map[string]string{"k1": "value"}, k.StringMap("str"), "types don't match")
+	assert.Equal(map[string]string{"k2": "value"}, k.StringMap("iface"), "types don't match")
+	assert.Equal(map[string][]string{"k1": {"value"}}, k.StringsMap("strs"), "types don't match")
+	assert.Equal(map[string][]string{"k2": {"value"}}, k.StringsMap("ifaces"), "types don't match")
+	assert.Equal(map[string][]string{"k2": {"value"}}, k.StringsMap("ifaces2"), "types don't match")
+	assert.Equal(map[string][]string{"k2": {"value"}}, k.StringsMap("ifaces3"), "types don't match")
+}