feat: 尝试支持设置阿里云 dns

dns/aliyun/aliyun.go

@@ -1,14 +1,161 @@
 package aliyun
 
 import (
-	"github.com/Mmx233/BitSrunLoginGo/dns/models"
-	"github.com/Mmx233/BitSrunLoginGo/dns/util"
+	"crypto/hmac"
+	"crypto/sha1"
+	"encoding/base64"
+	"encoding/json"
+	"errors"
+	"fmt"
+	dnsUtil "github.com/Mmx233/BitSrunLoginGo/dns/util"
+	"github.com/Mmx233/tool"
+	"math/rand"
+	"net/http"
+	"net/url"
+	"sort"
+	"time"
 )
 
 type DnsProvider struct {
+	TTL             uint       `mapstructure:"-"`
+	Http            *tool.Http `mapstructure:"-"`
+	AccessKeyId     string     `mapstructure:"access_key_id"`
+	AccessKeySecret string     `mapstructure:"access_key_secret"`
 }
 
-func New(conf map[string]interface{}) (models.DnsProvider, error) {
-	var p DnsProvider
-	return &p, util.DecodeConfig(conf, &p)
+func New(ttl uint, conf map[string]interface{}, Http *http.Client) (*DnsProvider, error) {
+	var p = DnsProvider{
+		Http: tool.NewHttpTool(Http),
+	}
+	return &p, dnsUtil.DecodeConfig(conf, &p)
+}
+
+func (a DnsProvider) SendRequest(Type, Action string, data map[string]interface{}) (*http.Response, error) {
+	var reqOpt = tool.DoHttpReq{
+		Url: "https://alidns.aliyuncs.com",
+	}
+	rand.Seed(time.Now().UnixNano())
+	data["Format"] = "json"
+	data["Version"] = "2015-01-09"
+	data["SignatureMethod"] = "HMAC-SHA1"
+	data["SignatureNonce"] = fmt.Sprint(tool.Rand.Num(10000000, 90000000))
+	data["Timestamp"] = time.Now().UTC().Format("2006-01-02T15:04:05Z")
+	data["Action"] = Action
+
+	signStr := Type + "&" + url.QueryEscape("/")
+	var keys = make([]string, len(data))
+	var i int
+	for k := range data {
+		keys[i] = k
+		i++
+	}
+	sort.Strings(keys)
+	for _, k := range keys {
+		signStr += "&" + k + "=" + url.QueryEscape(fmt.Sprint(data[k]))
+	}
+
+	mac := hmac.New(sha1.New, []byte(a.AccessKeySecret+"&"))
+	_, e := mac.Write([]byte(signStr))
+	if e != nil {
+		return nil, e
+	}
+	data["Signature"] = base64.StdEncoding.EncodeToString([]byte(fmt.Sprintf("%x", mac.Sum(nil))))
+
+	if Type == "GET" || Type == "DELETE" {
+		reqOpt.Query = data
+	} else {
+		reqOpt.Body = data
+	}
+
+	resp, e := a.Http.Request(Type, &reqOpt)
+	if e != nil {
+		return nil, e
+	}
+
+	if resp.StatusCode > 299 {
+		defer resp.Body.Close()
+		var res Response
+		if e = json.NewDecoder(resp.Body).Decode(&res); e != nil {
+			return nil, e
+		}
+
+		return nil, errors.New(res.Message)
+	}
+
+	return resp, nil
+}
+
+func (a DnsProvider) DomainRecordStatus(subDomain, rootDomain string) (*DomainStatus, bool, error) {
+	resp, e := a.SendRequest("GET", "DescribeDomainRecords", map[string]interface{}{
+		"DomainName": rootDomain,
+		"SearchMode": "EXACT",
+		"KeyWord":    subDomain,
+		"PageSize":   1,
+		"Type":       "A",
+	})
+	if e != nil {
+		return nil, false, e
+	}
+	defer resp.Body.Close()
+
+	var res DomainStatusRes
+	if e = json.NewDecoder(resp.Body).Decode(&res); e != nil {
+		return nil, false, e
+	}
+
+	if res.TotalCount == 0 || len(res.DomainRecords.Record) == 0 {
+		return nil, false, nil
+	}
+
+	return &res.DomainRecords.Record[0], true, nil
+}
+
+func (a DnsProvider) UpdateRecord(RecordId, subDomain, ip string) error {
+	resp, e := a.SendRequest("POST", "UpdateDomainRecord", map[string]interface{}{
+		"RecordId": RecordId,
+		"RR":       subDomain,
+		"Type":     "A",
+		"Value":    ip,
+		"TTL":      a.TTL,
+	})
+	if e != nil {
+		return e
+	}
+	defer resp.Body.Close()
+
+	return nil
+}
+
+func (a DnsProvider) NewRecord(subDomain, rootDomain, ip string) error {
+	resp, e := a.SendRequest("POST", "AddDomainRecord", map[string]interface{}{
+		"DomainName": rootDomain,
+		"RR":         subDomain,
+		"Type":       "A",
+		"Value":      ip,
+		"TTL":        a.TTL,
+	})
+	if e != nil {
+		return e
+	}
+	defer resp.Body.Close()
+
+	return nil
+}
+
+func (a DnsProvider) SetDomainRecord(domain, ip string) error {
+	subDomain, rootDomain, e := dnsUtil.DecodeDomain(domain)
+	if e != nil {
+		return e
+	}
+
+	record, exist, e := a.DomainRecordStatus(subDomain, rootDomain)
+	if e != nil {
+		return e
+	}
+
+	if exist {
+		return a.UpdateRecord(record.RecordId, subDomain, ip)
+	} else {
+		return a.NewRecord(subDomain, rootDomain, ip)
+	}
 }

dns/aliyun/models.go

@@ -0,0 +1,19 @@
+package aliyun
+
+type Response struct {
+	Code    string `json:"code"`
+	Message string `json:"message"`
+}
+
+type DomainStatus struct {
+	DomainName string `json:"DomainName"`
+	RecordId   string `json:"RecordId"`
+	Value      string `json:"Value"`
+}
+
+type DomainStatusRes struct {
+	TotalCount    uint `json:"TotalCount"`
+	DomainRecords struct {
+		Record []DomainStatus `json:"Record"`
+	} `json:"DomainRecords"`
+}

dns/initer.go

@@ -14,12 +14,16 @@ func Run(c *Config) error {
 		return e
 	}
 
+	if meta.TTL == 0 {
+		meta.TTL = 600
+	}
+
 	// 配置解析
 
 	var dns Provider
 	switch c.Provider {
 	case "aliyun":
-		dns, e = aliyun.New(meta.Other)
+		dns, e = aliyun.New(meta.TTL, meta.Other, c.Http)
 	default:
 		log.Warnf("DDNS 模块 dns 运营商 %s 不支持", c.Provider)
 		return nil

dns/models.go

@@ -15,5 +15,6 @@ type Config struct {
 
 type BasicConfig struct {
 	Domain string                 `mapstructure:"domain"`
+	TTL    uint                   `mapstructure:"ttl"`
 	Other  map[string]interface{} `mapstructure:",remain"`
 }

dns/util/domain.go

@@ -0,0 +1,18 @@
+package dnsUtil
+
+import (
+	"errors"
+	"strings"
+)
+
+func DecodeDomain(domain string) (subStr string, rootDomain string, e error) {
+	t := strings.Split(domain, ".")
+	if len(t) == 1 {
+		return "", "", errors.New("域名不合法")
+	} else if len(t) == 2 {
+		return "@", domain, nil
+	}
+
+	l := len(t)
+	return strings.Join(t[:l-2], "."), strings.Join(t[l-2:l], "."), nil
+}