增加DNS域名管理

internal/dnsproviders/provider_dnspod.go

@@ -0,0 +1,106 @@
+package dnsproviders
+
+import (
+	"encoding/json"
+	"errors"
+	"github.com/iwind/TeaGo/maps"
+	"github.com/iwind/TeaGo/types"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"strings"
+)
+
+type DNSPodProvider struct {
+	apiId    string
+	apiToken string
+}
+
+// 认证
+func (this *DNSPodProvider) Auth(params maps.Map) error {
+	this.apiId = params.GetString("id")
+	this.apiToken = params.GetString("token")
+
+	if len(this.apiId) == 0 {
+		return errors.New("'id' should be not empty")
+	}
+	if len(this.apiToken) == 0 {
+		return errors.New("'token' should not be empty")
+	}
+	return nil
+}
+
+// 读取线路数据
+func (this *DNSPodProvider) GetRoutes(domain string) ([][]string, error) {
+	infoResp, err := this.post("/Domain.info", map[string]string{
+		"domain": domain,
+	})
+	if err != nil {
+		return nil, err
+	}
+	domainInfo := infoResp.GetMap("domain")
+	grade := domainInfo.GetString("grade")
+
+	linesResp, err := this.post("/Record.Line", map[string]string{
+		"domain":       domain,
+		"domain_grade": grade,
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	lines := linesResp.GetSlice("lines")
+	if len(lines) == 0 {
+		return nil, nil
+	}
+	lineStrings := []string{}
+	for _, line := range lines {
+		lineStrings = append(lineStrings, types.String(line))
+	}
+
+	return [][]string{lineStrings}, nil
+}
+
+// 发送请求
+func (this *DNSPodProvider) post(path string, params map[string]string) (maps.Map, error) {
+	apiHost := "https://dnsapi.cn"
+	query := url.Values{
+		"login_token": []string{this.apiId + "," + this.apiToken},
+		"format":      []string{"json"},
+	}
+	for p, v := range params {
+		query[p] = []string{v}
+	}
+	req, err := http.NewRequest(http.MethodPost, apiHost+path, strings.NewReader(query.Encode()))
+	if err != nil {
+		return nil, err
+	}
+	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
+	req.Header.Set("User-Agent", "GoEdge Client/1.0.0 (iwind.liu@gmail.com)")
+
+	client := http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer func() {
+		_ = resp.Body.Close()
+		client.CloseIdleConnections()
+	}()
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return nil, err
+	}
+	m := maps.Map{}
+	err = json.Unmarshal(body, &m)
+	if err != nil {
+		return nil, err
+	}
+	status := m.GetMap("status")
+	code := status.GetString("code")
+	if code != "1" {
+		return nil, errors.New("code: " + code + ", message: " + status.GetString("message"))
+	}
+
+	return m, nil
+}

internal/dnsproviders/provider_dnspod_test.go

@@ -0,0 +1,22 @@
+package dnsproviders
+
+import (
+	"github.com/iwind/TeaGo/maps"
+	"testing"
+)
+
+func TestDNSPodProvider_GetRoutes(t *testing.T) {
+	provider := &DNSPodProvider{}
+	err := provider.Auth(maps.Map{
+		"id":    "191996",
+		"token": "366964e0f8ed4d8990a7f5d4b3cdec60",
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	routes, err := provider.GetRoutes("yun4s.cn")
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log(routes)
+}

internal/dnsproviders/provider_interface.go

@@ -1,7 +1,12 @@
 package dnsproviders
 
+import "github.com/iwind/TeaGo/maps"
+
 // DNS操作接口
 type ProviderInterface interface {
-	// 检查账号
+	// 认证
+	Auth(params maps.Map) error
 
+	// 读取线路数据
+	GetRoutes(domain string) ([][]string, error)
 }

internal/dnsproviders/types.go

@@ -4,15 +4,27 @@ import "github.com/iwind/TeaGo/maps"
 
 type ProviderType = string
 
+// 服务商代号
 const (
 	ProviderTypeDNSPod ProviderType = "dnspod"
+	ProviderTypeAliyun ProviderType = "aliyun"
+	ProviderTypeDNSCom ProviderType = "dnscom"
 )
 
+// 所有的服务商类型
 var AllProviderTypes = []maps.Map{
 	{
 		"name": "DNSPod",
 		"code": ProviderTypeDNSPod,
 	},
+	{
+		"name": "阿里云",
+		"code": ProviderTypeAliyun,
+	},
+	{
+		"name": "帝恩思",
+		"code": ProviderTypeDNSCom,
+	},
 }
 
 func FindProviderTypeName(providerType string) string {