EdgeAPI/internal/dnsclients/provider_custom_http.go

package dnsclients

import (
	"bytes"
	"crypto/sha1"
	"crypto/tls"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"strconv"
	"time"

	teaconst "github.com/TeaOSLab/EdgeAPI/internal/const"
	"github.com/TeaOSLab/EdgeAPI/internal/dnsclients/dnstypes"
	"github.com/TeaOSLab/EdgeAPI/internal/errors"
	"github.com/iwind/TeaGo/maps"
)

var customHTTPClient = &http.Client{
	Timeout: 10 * time.Second,
	Transport: &http.Transport{
		TLSClientConfig: &tls.Config{
			InsecureSkipVerify: true,
		},
	},
}

// CustomHTTPProvider HTTP自定义DNS
type CustomHTTPProvider struct {
	url    string
	secret string

	ProviderId int64

	BaseProvider
}

// Auth 认证
// 参数：
//   - url
//   - secret
func (this *CustomHTTPProvider) Auth(params maps.Map) error {
	this.url = params.GetString("url")
	if len(this.url) == 0 {
		return errors.New("'url' should not be empty")
	}

	this.secret = params.GetString("secret")
	if len(this.secret) == 0 {
		return errors.New("'secret' should not be empty")
	}

	return nil
}

// MaskParams 对参数进行掩码
func (this *CustomHTTPProvider) MaskParams(params maps.Map) {
	// 这里暂时不要掩码，避免用户忘记
}

// GetDomains 获取所有域名列表
func (this *CustomHTTPProvider) GetDomains() (domains []string, err error) {
	resp, err := this.post(maps.Map{
		"action": "GetDomains",
	})
	if err != nil {
		return nil, err
	}
	err = json.Unmarshal(resp, &domains)
	return
}

// GetRecords 获取域名解析记录列表
func (this *CustomHTTPProvider) GetRecords(domain string) (records []*dnstypes.Record, err error) {
	resp, err := this.post(maps.Map{
		"action": "GetRecords",
		"domain": domain,
	})
	if err != nil {
		return nil, err
	}
	err = json.Unmarshal(resp, &records)
	return
}

// GetRoutes 读取域名支持的线路数据
func (this *CustomHTTPProvider) GetRoutes(domain string) (routes []*dnstypes.Route, err error) {
	resp, err := this.post(maps.Map{
		"action": "GetRoutes",
		"domain": domain,
	})
	if err != nil {
		return nil, err
	}
	err = json.Unmarshal(resp, &routes)
	return
}

// QueryRecord 查询单个记录
func (this *CustomHTTPProvider) QueryRecord(domain string, name string, recordType dnstypes.RecordType) (*dnstypes.Record, error) {
	resp, err := this.post(maps.Map{
		"action":     "QueryRecord",
		"domain":     domain,
		"name":       name,
		"recordType": recordType,
	})
	if err != nil {
		return nil, err
	}
	if len(resp) == 0 || string(resp) == "null" {
		return nil, nil
	}
	var record = &dnstypes.Record{}
	err = json.Unmarshal(resp, record)
	if err != nil {
		return nil, err
	}
	if len(record.Value) == 0 {
		return nil, nil
	}
	return record, nil
}

// QueryRecords 查询多个记录
func (this *CustomHTTPProvider) QueryRecords(domain string, name string, recordType dnstypes.RecordType) (result []*dnstypes.Record, err error) {
	resp, err := this.post(maps.Map{
		"action":     "QueryRecords",
		"domain":     domain,
		"name":       name,
		"recordType": recordType,
	})
	if err != nil {
		return nil, err
	}
	if len(resp) == 0 || string(resp) == "null" {
		return nil, nil
	}
	result = []*dnstypes.Record{}
	err = json.Unmarshal(resp, &result)
	if err != nil {
		return nil, err
	}
	return result, nil
}

// AddRecord 设置记录
func (this *CustomHTTPProvider) AddRecord(domain string, newRecord *dnstypes.Record) error {
	_, err := this.post(maps.Map{
		"action":    "AddRecord",
		"domain":    domain,
		"newRecord": newRecord,
	})
	return this.WrapError(err, domain, newRecord)
}

// UpdateRecord 修改记录
func (this *CustomHTTPProvider) UpdateRecord(domain string, record *dnstypes.Record, newRecord *dnstypes.Record) error {
	_, err := this.post(maps.Map{
		"action":    "UpdateRecord",
		"domain":    domain,
		"record":    record,
		"newRecord": newRecord,
	})
	return this.WrapError(err, domain, newRecord)
}

// DeleteRecord 删除记录
func (this *CustomHTTPProvider) DeleteRecord(domain string, record *dnstypes.Record) error {
	_, err := this.post(maps.Map{
		"action": "DeleteRecord",
		"domain": domain,
		"record": record,
	})
	return this.WrapError(err, domain, record)
}

// DefaultRoute 默认线路
func (this *CustomHTTPProvider) DefaultRoute() string {
	resp, err := this.post(maps.Map{
		"action": "DefaultRoute",
	})
	if err != nil {
		return ""
	}
	return string(resp)
}

// 执行操作
func (this *CustomHTTPProvider) post(params maps.Map) (respData []byte, err error) {
	data, err := json.Marshal(params)
	if err != nil {
		return nil, err
	}
	req, err := http.NewRequest(http.MethodPost, this.url, bytes.NewReader(data))
	if err != nil {
		return nil, err
	}
	var timestamp = strconv.FormatInt(time.Now().Unix(), 10)
	req.Header.Set("Timestamp", timestamp)
	req.Header.Set("Token", fmt.Sprintf("%x", sha1.Sum([]byte(this.secret+"@"+timestamp))))
	req.Header.Set("User-Agent", teaconst.ProductName+"/"+teaconst.Version)
	req.Header.Set("Content-Type", "application/json")

	resp, err := customHTTPClient.Do(req)
	if err != nil {
		return nil, err
	}
	defer func() {
		_ = resp.Body.Close()
	}()
	if resp.StatusCode != http.StatusOK {
		return nil, errors.New("status should be 200, but got '" + strconv.Itoa(resp.StatusCode) + "'")
	}
	return io.ReadAll(resp.Body)
}