Restore validate DDNS provider function

nezhahq · Apr 25, 2024 · d5f5d57 · d5f5d57
1 parent f607362
commit d5f5d57
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/service/singleton/ddns.go b/service/singleton/ddns.go
@@ -74,3 +74,15 @@ func GetDDNSProviderFromProfile(profileName string) (ddns2.Provider, error) {
 	}
 	return ddns2.ProviderDummy{}, errors.New(fmt.Sprintf("无法找到配置的DDNS提供者%s", profile.Provider))
 }
+
+func ValidateDDNSProvidersFromProfiles() error {
+    validProviders := map[string]bool{"webhook": true, "dummy": true, "cloudflare": true, "tencentcloud": true}
+    providers := make(map[string]string)
+    for profileName, profile := range Conf.DDNS.Profiles {
+        if _, ok := validProviders[profile.Provider]; !ok {
+            return errors.New(fmt.Sprintf("无法找到配置的DDNS提供者%s", profile.Provider))
+        }
+        providers[profileName] = profile.Provider
+    }
+    return nil
+}
diff --git a/service/singleton/singleton.go b/service/singleton/singleton.go
@@ -52,6 +52,15 @@ func InitConfigFromPath(path string) {
 
 // ValidateConfig 验证配置文件有效性
 func ValidateConfig() {
+	var err error
+	if Conf.DDNS.Provider == "" {
+		err = ValidateDDNSProvidersFromProfiles()
+	} else {
+		_, err = GetDDNSProviderFromString(Conf.DDNS.Provider)
+	}
+	if err != nil {
+		panic(err)
+	}
 	if Conf.DDNS.Enable {
 		if Conf.DDNS.MaxRetries < 1 || Conf.DDNS.MaxRetries > 10 {
 			panic(fmt.Errorf("DDNS.MaxRetries值域为[1, 10]的整数, 当前为 %d", Conf.DDNS.MaxRetries))