support for test dc1-3

loading custom dc ips from config

internal/utils/read_config.go

@@ -2,11 +2,82 @@ package utils
 
 import (
 	"fmt"
+	"log"
 	"os"
 
+	"github.com/9seconds/mtg/v2/essentials"
 	"github.com/9seconds/mtg/v2/internal/config"
 )
 
+func readCustomDC(cfg *config.Config) {
+	if len(cfg.DCs.DC1) > 0 {
+		var ips []string
+		for _, addr := range cfg.DCs.DC1 {
+			ips = append(ips, addr.Value)
+		}
+		essentials.TelegramCoreAddresses[1] = ips
+	}
+	if len(cfg.DCs.DC2) > 0 {
+		var ips []string
+		for _, addr := range cfg.DCs.DC2 {
+			ips = append(ips, addr.Value)
+		}
+		essentials.TelegramCoreAddresses[2] = ips
+	}
+	if len(cfg.DCs.DC3) > 0 {
+		var ips []string
+		for _, addr := range cfg.DCs.DC3 {
+			ips = append(ips, addr.Value)
+		}
+		essentials.TelegramCoreAddresses[3] = ips
+	}
+	if len(cfg.DCs.DC4) > 0 {
+		var ips []string
+		for _, addr := range cfg.DCs.DC4 {
+			ips = append(ips, addr.Value)
+		}
+		essentials.TelegramCoreAddresses[4] = ips
+	}
+	if len(cfg.DCs.DC5) > 0 {
+		var ips []string
+		for _, addr := range cfg.DCs.DC5 {
+			ips = append(ips, addr.Value)
+		}
+		essentials.TelegramCoreAddresses[5] = ips
+	}
+	if len(cfg.DCs.DC203) > 0 {
+		var ips []string
+		for _, addr := range cfg.DCs.DC203 {
+			ips = append(ips, addr.Value)
+		}
+		essentials.TelegramCoreAddresses[203] = ips
+	}
+
+	if len(cfg.DCs.DC10001) > 0 {
+		var ips []string
+		for _, addr := range cfg.DCs.DC10001 {
+			ips = append(ips, addr.Value)
+		}
+		essentials.TelegramCoreAddresses[10001] = ips
+	}
+	if len(cfg.DCs.DC10002) > 0 {
+		var ips []string
+		for _, addr := range cfg.DCs.DC203 {
+			ips = append(ips, addr.Value)
+		}
+		essentials.TelegramCoreAddresses[10002] = ips
+	}
+	if len(cfg.DCs.DC10003) > 0 {
+		var ips []string
+		for _, addr := range cfg.DCs.DC203 {
+			ips = append(ips, addr.Value)
+		}
+		essentials.TelegramCoreAddresses[10003] = ips
+	}
+	log.Println(essentials.TelegramCoreAddresses)
+	log.Println(cfg.DCs)
+}
+
 func ReadConfig(path string) (*config.Config, error) {
 	content, err := os.ReadFile(path)
 	if err != nil {
@@ -21,6 +92,7 @@ func ReadConfig(path string) (*config.Config, error) {
 	if err := conf.Validate(); err != nil {
 		return nil, fmt.Errorf("invalid config: %w", err)
 	}
+	readCustomDC(conf)
 
 	return conf, nil
 }