add parsing of Secret from ENV var

davidmasek · Jan 14, 2025 · 0c3c780 · 0c3c780
1 parent bd0b6f6
commit 0c3c780
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/conf/config.go b/conf/config.go
@@ -8,6 +8,7 @@ import (
 	"log"
 	"os"
 	"path/filepath"
+	"reflect"
 	"time"
 
 	"github.com/caarlos0/env/v11"
@@ -215,6 +216,11 @@ func ConfigFromBytes(data []byte) (*Config, error) {
 	}
 	err = env.ParseWithOptions(config, env.Options{
 		Prefix: ENV_VAR_PREFIX,
+		FuncMap: map[reflect.Type]env.ParserFunc{
+			reflect.TypeOf(Secret{}): func(v string) (interface{}, error) {
+				return Secret{v}, nil
+			},
+		},
 	})
 	if err != nil {
 		return nil, err

diff --git a/conf/config_test.go b/conf/config_test.go
@@ -151,3 +151,12 @@ prefix: "[test]"
 	})
 	require.Equal(t, "h4xor", emailConfig.SmtpPassword.Get())
 }
+
+func TestSecretFromEnv(t *testing.T) {
+	err := os.Setenv("BEACON_EMAIL_SMTP_PASSWORD", "secr4t")
+	require.NoError(t, err)
+	conf, err := ConfigFromBytes([]byte(""))
+	require.NoError(t, err)
+	require.Equal(t, "secr4t", conf.EmailConf.SmtpPassword.Get())
+
+}