实现在域名解析中使用EdgeDNS

internal/dnsclients/types.go

@@ -1,40 +1,70 @@
 package dnsclients
 
-import "github.com/iwind/TeaGo/maps"
+import (
+	teaconst "github.com/TeaOSLab/EdgeAPI/internal/const"
+	"github.com/iwind/TeaGo/maps"
+)
 
 type ProviderType = string
 
 // 服务商代号
 const (
-	ProviderTypeDNSPod     ProviderType = "dnspod"
-	ProviderTypeAliDNS     ProviderType = "alidns"
-	ProviderTypeDNSCom     ProviderType = "dnscom"
-	ProviderTypeCloudFlare ProviderType = "cloudFlare"
-	ProviderTypeCustomHTTP ProviderType = "customHTTP"
+	ProviderTypeDNSPod       ProviderType = "dnspod"       // DNSPod
+	ProviderTypeAliDNS       ProviderType = "alidns"       // 阿里云DNS
+	ProviderTypeDNSCom       ProviderType = "dnscom"       // dns.com
+	ProviderTypeCloudFlare   ProviderType = "cloudFlare"   // CloudFlare DNS
+	ProviderTypeLocalEdgeDNS ProviderType = "localEdgeDNS" // 和当前系统集成的EdgeDNS
+	ProviderTypeUserEdgeDNS  ProviderType = "userEdgeDNS"  // 通过API连接的EdgeDNS
+	ProviderTypeCustomHTTP   ProviderType = "customHTTP"   // 自定义HTTP接口
 )
 
-// AllProviderTypes 所有的服务商类型
-var AllProviderTypes = []maps.Map{
-	{
-		"name": "阿里云DNS",
-		"code": ProviderTypeAliDNS,
-	},
-	{
-		"name": "DNSPod",
-		"code": ProviderTypeDNSPod,
-	},
-	/**{
-		"name": "帝恩思DNS.COM",
-		"code": ProviderTypeDNSCom,
-	},**/
-	{
-		"name": "CloudFlare DNS",
-		"code": ProviderTypeCloudFlare,
-	},
-	{
-		"name": "自定义HTTP DNS",
-		"code": ProviderTypeCustomHTTP,
-	},
+// FindAllProviderTypes 所有的服务商类型
+func FindAllProviderTypes() []maps.Map {
+	typeMaps := []maps.Map{
+		{
+			"name":        "阿里云DNS",
+			"code":        ProviderTypeAliDNS,
+			"description": "阿里云提供的DNS服务。",
+		},
+		{
+			"name":        "DNSPod",
+			"code":        ProviderTypeDNSPod,
+			"description": "DNSPod提供的DNS服务。",
+		},
+		/**{
+			"name": "帝恩思DNS.COM",
+			"code": ProviderTypeDNSCom,
+			"description": "DNS.com提供的DNS服务。",
+		},**/
+		{
+			"name":        "CloudFlare DNS",
+			"code":        ProviderTypeCloudFlare,
+			"description": "CloudFlare提供的DNS服务。",
+		},
+	}
+
+	if teaconst.IsPlus {
+		typeMaps = append(typeMaps, []maps.Map{
+			{
+				"name":        "集成EdgeDNS",
+				"code":        ProviderTypeLocalEdgeDNS,
+				"description": "当前企业版提供的DNS服务。",
+			},
+			// TODO 需要实现用户使用AccessId/AccessKey来连接DNS服务
+			/**{
+				"name":        "用户EdgeDNS",
+				"code":        ProviderTypeUserEdgeDNS,
+				"description": "通过API连接企业版提供的DNS服务。",
+			},**/
+		}...)
+	}
+
+	typeMaps = append(typeMaps, maps.Map{
+		"name":        "自定义HTTP DNS",
+		"code":        ProviderTypeCustomHTTP,
+		"description": "通过自定义的HTTP接口提供DNS服务。",
+	})
+	return typeMaps
 }
 
 // FindProvider 查找服务商实例
@@ -46,6 +76,10 @@ func FindProvider(providerType ProviderType) ProviderInterface {
 		return &AliDNSProvider{}
 	case ProviderTypeCloudFlare:
 		return &CloudFlareProvider{}
+	case ProviderTypeLocalEdgeDNS:
+		return &LocalEdgeDNSProvider{}
+	case ProviderTypeUserEdgeDNS:
+		return &UserEdgeDNSProvider{}
 	case ProviderTypeCustomHTTP:
 		return &CustomHTTPProvider{}
 	}
@@ -54,7 +88,7 @@ func FindProvider(providerType ProviderType) ProviderInterface {
 
 // FindProviderTypeName 查找服务商名称
 func FindProviderTypeName(providerType ProviderType) string {
-	for _, t := range AllProviderTypes {
+	for _, t := range FindAllProviderTypes() {
 		if t.GetString("code") == providerType {
 			return t.GetString("name")
 		}