fix: Load array env

cmd/relayproxy/config/config.go

@@ -5,6 +5,7 @@ import (
 	"net/http"
 	"os"
 	"path/filepath"
+	"strconv"
 	"strings"
 	"time"
 
@@ -91,10 +92,14 @@ func New(flagSet *pflag.FlagSet, log *zap.Logger, version string) (*Config, erro
 			log.Error("error loading file", zap.Error(errBindFile))
 		}
 	}
-
+	configMap := k.Raw()
 	// Map environment variables
-	_ = k.Load(env.Provider("", ".", func(s string) string {
-		return strings.ReplaceAll(strings.ToLower(s), "_", ".")
+	_ = k.Load(env.ProviderWithValue("", ".", func(s string, v string) (string, interface{}) {
+		if strings.HasPrefix(s, "RETRIEVERS") || strings.HasPrefix(s, "NOTIFIERS") {
+			_ = loadArrayEnv(s, v, configMap)
+			return s, v
+		}
+		return strings.ReplaceAll(strings.ToLower(s), "_", "."), v
 	}), nil)
 
 	_ = k.Set("version", version)
@@ -322,3 +327,45 @@ func locateConfigFile(inputFilePath string) (string, error) {
 	return "", fmt.Errorf(
 		"impossible to find config file in the default locations [%s]", strings.Join(defaultLocations, ","))
 }
+
+// Load the ENV Like:RETRIEVERS_0_HEADERS_AUTHORIZATION
+func loadArrayEnv(s string, v string, configMap map[string]interface{}) error {
+	paths := strings.Split(s, "_")
+	for i, str := range paths {
+		paths[i] = strings.ToLower(str)
+	}
+	prefixKey := paths[0]
+	if configArray, ok := configMap[prefixKey].([]interface{}); ok {
+		index, err := strconv.Atoi(paths[1])
+		if err != nil {
+			return err
+		}
+		configItem := configArray[index].(map[string]interface{})
+		keys := paths[2:]
+		currentMap := configItem
+		for i, key := range keys {
+			hasKey := false
+			lowerKey := key
+			for y := range currentMap {
+				if y != lowerKey {
+					continue
+				}
+				if nextMap, ok := currentMap[y].(map[string]interface{}); ok {
+					currentMap = nextMap
+					hasKey = true
+					break
+				}
+			}
+			if !hasKey && i != len(keys)-1 {
+				newMap := make(map[string]interface{})
+				currentMap[lowerKey] = newMap
+				currentMap = newMap
+			}
+		}
+		lastKey := keys[len(keys)-1]
+		currentMap[lastKey] = v
+		configArray[index] = configItem
+		_ = k.Set(prefixKey, configArray)
+	}
+	return nil
+}

cmd/relayproxy/config/config_test.go

@@ -692,3 +692,75 @@ func TestConfig_APIAdminKeyExists(t *testing.T) {
 		})
 	}
 }
+
+func TestMergeConfig_FromOSEnv(t *testing.T) {
+	tests := []struct {
+		name                       string
+		want                       *config.Config
+		fileLocation               string
+		wantErr                    assert.ErrorAssertionFunc
+		disableDefaultFileCreation bool
+	}{
+		{
+			name:         "Valid file",
+			fileLocation: "../testdata/config/validate-array-env-file.yaml",
+			want: &config.Config{
+				ListenPort:      1031,
+				PollingInterval: 1000,
+				FileFormat:      "yaml",
+				Host:            "localhost",
+				Retrievers: &[]config.RetrieverConf{
+					config.RetrieverConf{
+						Kind: "http",
+						URL:  "https://github.com/thomaspoignant/go-feature-flag/main/examples/retriever_file/flags.goff.yaml",
+						HTTPHeaders: map[string][]string{
+							"authorization": []string{
+								"test",
+							},
+						},
+					},
+				},
+				Exporter: &config.ExporterConf{
+					Kind: "log",
+				},
+				StartWithRetrieverError: false,
+				RestAPITimeout:          5000,
+				Version:                 "1.X.X",
+				EnableSwagger:           true,
+				AuthorizedKeys: config.APIKeys{
+					Admin: []string{
+						"apikey3",
+					},
+					Evaluation: []string{
+						"apikey1",
+						"apikey2",
+					},
+				},
+			},
+			wantErr: assert.NoError,
+		},
+	}
+	for _, tt := range tests {
+		os.Setenv("RETRIEVERS_0_HEADERS_AUTHORIZATION", "test")
+		t.Run(tt.name, func(t *testing.T) {
+			_ = os.Remove("./goff-proxy.yaml")
+			if !tt.disableDefaultFileCreation {
+				source, _ := os.Open(tt.fileLocation)
+				destination, _ := os.Create("./goff-proxy.yaml")
+				defer destination.Close()
+				defer source.Close()
+				defer os.Remove("./goff-proxy.yaml")
+				_, _ = io.Copy(destination, source)
+			}
+
+			f := pflag.NewFlagSet("config", pflag.ContinueOnError)
+			f.String("config", "", "Location of your config file")
+			_ = f.Parse([]string{fmt.Sprintf("--config=%s", tt.fileLocation)})
+			got, err := config.New(f, zap.L(), "1.X.X")
+			if !tt.wantErr(t, err) {
+				return
+			}
+			assert.Equal(t, tt.want, got, "Config not matching")
+		})
+	}
+}

cmd/relayproxy/testdata/config/validate-array-env-file.yaml

@@ -0,0 +1,15 @@
+listen: 1031
+pollingInterval: 1000
+startWithRetrieverError: false
+retrievers:
+  - kind: http
+    url: https://github.com/thomaspoignant/go-feature-flag/main/examples/retriever_file/flags.goff.yaml
+exporter:
+  kind: log
+enableSwagger: true
+authorizedKeys:
+  evaluation:
+    - apikey1 # owner: userID1
+    - apikey2 # owner: userID2
+  admin:
+    - apikey3