lucky/ddns/dnspod.go

package ddns

import (
	"fmt"
	"net/http"
	"net/url"

	"github.com/gdy666/lucky/config"
	"github.com/gdy666/lucky/thirdlib/gdylib/httputils"
)

const (
	recordListAPI   string = "https://dnsapi.cn/Record.List"
	recordModifyURL string = "https://dnsapi.cn/Record.Modify"
	recordCreateAPI string = "https://dnsapi.cn/Record.Create"
)

// https://cloud.tencent.com/document/api/302/8516
// Dnspod 腾讯云dns实现
type Dnspod struct {
	DNSCommon
	TTL string
}

// DnspodRecordListResp recordListAPI结果
type DnspodRecordListResp struct {
	DnspodStatus
	Records []struct {
		ID      string
		Name    string
		Type    string
		Value   string
		Enabled string
	}
}

// DnspodStatus DnspodStatus
type DnspodStatus struct {
	Status struct {
		Code    string
		Message string
	}
}

// Init 初始化
func (dnspod *Dnspod) Init(task *config.DDNSTask) {
	dnspod.DNSCommon.Init(task)

	if task.TTL == "" {
		// 默认600s
		dnspod.TTL = "600"
	} else {
		dnspod.TTL = task.TTL
	}
	dnspod.SetCreateUpdateDomainFunc(dnspod.createUpdateDomain)
}

func (dnspod *Dnspod) createUpdateDomain(recordType, ipAddr string, domain *config.Domain) {
	result, err := dnspod.getRecordList(domain, recordType)
	if err != nil {
		return
	}

	if len(result.Records) > 0 {
		// 更新
		dnspod.modify(result, domain, recordType, ipAddr)
	} else {
		// 新增
		dnspod.create(result, domain, recordType, ipAddr)
	}
}

// 创建
func (dnspod *Dnspod) create(result DnspodRecordListResp, domain *config.Domain, recordType string, ipAddr string) {
	status, err := dnspod.commonRequest(
		recordCreateAPI,
		url.Values{
			"login_token": {dnspod.task.DNS.ID + "," + dnspod.task.DNS.Secret},
			"domain":      {domain.DomainName},
			"sub_domain":  {domain.GetSubDomain()},
			"record_type": {recordType},
			"record_line": {"默认"},
			"value":       {ipAddr},
			"ttl":         {dnspod.TTL},
			"format":      {"json"},
		},
		domain,
	)
	if err == nil && status.Status.Code == "1" {
		//log.Printf("新增域名解析 %s 成功！IP: %s", domain, ipAddr)
		domain.SetDomainUpdateStatus(config.UpdatedSuccess, "")
	} else {
		//log.Printf("新增域名解析 %s 失败！Code: %s, Message: %s", domain, status.Status.Code, status.Status.Message)
		domain.SetDomainUpdateStatus(config.UpdatedFailed, fmt.Sprintf("Code: %s, Message: %s", status.Status.Code, status.Status.Message))
	}
}

// 修改
func (dnspod *Dnspod) modify(result DnspodRecordListResp, domain *config.Domain, recordType string, ipAddr string) {
	for _, record := range result.Records {
		// 相同不修改
		if record.Value == ipAddr {
			//log.Printf("你的IP %s 没有变化, 域名 %s", ipAddr, domain)
			domain.SetDomainUpdateStatus(config.UpdatedNothing, "")
			continue
		}
		status, err := dnspod.commonRequest(
			recordModifyURL,
			url.Values{
				"login_token": {dnspod.task.DNS.ID + "," + dnspod.task.DNS.Secret},
				"domain":      {domain.DomainName},
				"sub_domain":  {domain.GetSubDomain()},
				"record_type": {recordType},
				"record_line": {"默认"},
				"record_id":   {record.ID},
				"value":       {ipAddr},
				"ttl":         {dnspod.TTL},
				"format":      {"json"},
			},
			domain,
		)
		if err == nil && status.Status.Code == "1" {
			//log.Printf("更新域名解析 %s 成功！IP: %s", domain, ipAddr)
			domain.SetDomainUpdateStatus(config.UpdatedSuccess, "")
		} else {
			//log.Printf("更新域名解析 %s 失败！Code: %s, Message: %s", domain, status.Status.Code, status.Status.Message)
			domain.SetDomainUpdateStatus(config.UpdatedFailed, fmt.Sprintf("Code: %s, Message: %s", status.Status.Code, status.Status.Message))
		}
	}
}

// 公共
func (dnspod *Dnspod) commonRequest(apiAddr string, values url.Values, domain *config.Domain) (status DnspodStatus, err error) {
	resp, err := http.PostForm(
		apiAddr,
		values,
	)

	err = httputils.GetAndParseJSONResponseFromHttpResponse(resp, &status)

	return
}

// 获得域名记录列表
func (dnspod *Dnspod) getRecordList(domain *config.Domain, typ string) (result DnspodRecordListResp, err error) {
	values := url.Values{
		"login_token": {dnspod.task.DNS.ID + "," + dnspod.task.DNS.Secret},
		"domain":      {domain.DomainName},
		"record_type": {typ},
		"sub_domain":  {domain.GetSubDomain()},
		"format":      {"json"},
	}

	client, e := dnspod.CreateHTTPClient()
	if e != nil {
		err = e
		return
	}

	resp, err := client.PostForm(
		recordListAPI,
		values,
	)

	err = httputils.GetAndParseJSONResponseFromHttpResponse(resp, result)
	return
}