From 9c97ce2d2f0268c228566398d4477a17103262d2 Mon Sep 17 00:00:00 2001
From: Mohammed Sohail <sohailsameja@gmail.com>
Date: Thu, 13 Mar 2025 12:20:10 +0300
Subject: [PATCH] fix: load multiple registries via env

---
 cmd/service/main.go   |  2 +-
 config.toml           |  2 +-
 internal/util/init.go | 17 +++++++++++++----
 3 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/cmd/service/main.go b/cmd/service/main.go
index f5e3dd5..2fb9834 100644
--- a/cmd/service/main.go
+++ b/cmd/service/main.go
@@ -74,7 +74,7 @@ func main() {
 
 	cacheOpts := cache.CacheOpts{
 		Chain:      chain,
-		Registries: []string{ko.MustString("bootstrap.ge_registry")},
+		Registries: ko.MustStrings("bootstrap.ge_registry"),
 		Watchlist:  ko.Strings("bootstrap.watchlist"),
 		Blacklist:  ko.Strings("bootstrap.blacklist"),
 		CacheType:  ko.MustString("core.cache_type"),
diff --git a/config.toml b/config.toml
index 82dc646..eb20b2d 100644
--- a/config.toml
+++ b/config.toml
@@ -27,7 +27,7 @@ start_block = 0
 
 [bootstrap]
 # This will bootstrap the cache on which addresses to track
-ge_registry = "0x0f8E97ef2d6A42CF62549D4924FCBdcE83A1C6A5"
+ge_registry = ["0x0f8E97ef2d6A42CF62549D4924FCBdcE83A1C6A5"]
 watchlist = [""]
 blacklist = [""]
 
diff --git a/internal/util/init.go b/internal/util/init.go
index 137f452..e8a684a 100644
--- a/internal/util/init.go
+++ b/internal/util/init.go
@@ -41,13 +41,22 @@ func InitConfig(lo *slog.Logger, confFilePath string) *koanf.Koanf {
 		os.Exit(1)
 	}
 
-	if err := ko.Load(env.Provider("TRACKER_", ".", func(s string) string {
-		return strings.ReplaceAll(strings.ToLower(
-			strings.TrimPrefix(s, "TRACKER_")), "__", ".")
-	}), nil); err != nil {
+	err := ko.Load(env.ProviderWithValue("TRACKER_", ".", func(s string, v string) (string, interface{}) {
+		key := strings.ReplaceAll(strings.ToLower(strings.TrimPrefix(s, "TRACKER_")), "__", ".")
+		if strings.Contains(v, " ") {
+			return key, strings.Split(v, " ")
+		}
+		return key, v
+	}), nil)
+
+	if err != nil {
 		lo.Error("could not override config from env vars", "error", err)
 		os.Exit(1)
 	}
 
+	if os.Getenv("DEBUG") != "" {
+		ko.Print()
+	}
+
 	return ko
 }