code clane

cmd/ddns/main.go

@@ -43,12 +43,7 @@ func Init() error {
 		notify.AddNotifier(notifier)
 	}
 
-	switch config.GetConfig().DNSProvider {
-	case "aliyun":
-		dnsProvider = dns.NewAliyun()
-	default:
-		dnsProvider = dns.NewCloudFlare()
-	}
+	dnsProvider = dns.New(config.GetConfig())
 
 	return nil
 }

dns/dns.go

@@ -2,6 +2,7 @@ package dns
 
 import (
 	"github.com/libdns/alidns"
+	"github.com/libdns/cloudflare"
 	"github.com/libdns/libdns"
 	"github.com/orvice/ddns/internal/config"
 )
@@ -12,6 +13,22 @@ type LibDNS interface {
 	libdns.RecordSetter
 }
 
+func New(conf *config.Config) LibDNS {
+	switch conf.DNSProvider {
+	case "cloudflare":
+		return NewCloudFlare()
+	case "aliyun":
+		return NewAliyun()
+	}
+	return nil
+}
+
+// cloudflare
+func NewCloudFlare() LibDNS {
+	provider := cloudflare.Provider{APIToken: config.GetConfig().CFToken}
+	return &provider
+}
+
 // aliyun
 func NewAliyun() LibDNS {
 	provider := alidns.Provider{

internal/config/config.go

@@ -17,7 +17,7 @@ type Config struct {
 }
 
 var (
-	config Config
+	config = &Config{}
 )
 
 func Init() (err error) {
@@ -28,7 +28,7 @@ func Init() (err error) {
 	return
 }
 
-func GetConfig() Config {
+func GetConfig() *Config {
 	return config
 }
 
@@ -39,6 +39,6 @@ func LoadConfig(path string) (err error) {
 	viper.SetConfigType("env")
 	viper.AutomaticEnv()
 	// viper.ReadInConfig()
-	err = viper.Unmarshal(&config)
+	err = viper.Unmarshal(config)
 	return
 }