mirror of
https://github.com/jeessy2/ddns-go.git
synced 2025-10-20 15:33:46 +08:00
7466211d5e
Compared to `ReplaceAll()`, `NewReplacer()` is probably a better choice, because you only need to maintain a list of old and new string pairs.
131 lines
3.2 KiB
Go
131 lines
3.2 KiB
Go
package dns
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"net/http"
|
||
"net/url"
|
||
"strings"
|
||
|
||
"github.com/jeessy2/ddns-go/v6/config"
|
||
"github.com/jeessy2/ddns-go/v6/util"
|
||
)
|
||
|
||
type Callback struct {
|
||
DNS config.DNS
|
||
Domains config.Domains
|
||
TTL string
|
||
lastIpv4 string
|
||
lastIpv6 string
|
||
}
|
||
|
||
// Init 初始化
|
||
func (cb *Callback) Init(dnsConf *config.DnsConfig, ipv4cache *util.IpCache, ipv6cache *util.IpCache) {
|
||
cb.Domains.Ipv4Cache = ipv4cache
|
||
cb.Domains.Ipv6Cache = ipv6cache
|
||
cb.lastIpv4 = ipv4cache.Addr
|
||
cb.lastIpv6 = ipv6cache.Addr
|
||
|
||
cb.DNS = dnsConf.DNS
|
||
cb.Domains.GetNewIp(dnsConf)
|
||
if dnsConf.TTL == "" {
|
||
// 默认600
|
||
cb.TTL = "600"
|
||
} else {
|
||
cb.TTL = dnsConf.TTL
|
||
}
|
||
}
|
||
|
||
// AddUpdateDomainRecords 添加或更新IPv4/IPv6记录
|
||
func (cb *Callback) AddUpdateDomainRecords() config.Domains {
|
||
cb.addUpdateDomainRecords("A")
|
||
cb.addUpdateDomainRecords("AAAA")
|
||
return cb.Domains
|
||
}
|
||
|
||
func (cb *Callback) addUpdateDomainRecords(recordType string) {
|
||
ipAddr, domains := cb.Domains.GetNewIpResult(recordType)
|
||
|
||
if ipAddr == "" {
|
||
return
|
||
}
|
||
|
||
// 防止多次发送Webhook通知
|
||
if recordType == "A" {
|
||
if cb.lastIpv4 == ipAddr {
|
||
util.Log("你的IPv4未变化, 未触发 %s 请求", "Callback")
|
||
return
|
||
}
|
||
} else {
|
||
if cb.lastIpv6 == ipAddr {
|
||
util.Log("你的IPv6未变化, 未触发 %s 请求", "Callback")
|
||
return
|
||
}
|
||
}
|
||
|
||
for _, domain := range domains {
|
||
method := "GET"
|
||
postPara := ""
|
||
contentType := "application/x-www-form-urlencoded"
|
||
if cb.DNS.Secret != "" {
|
||
method = "POST"
|
||
postPara = replacePara(cb.DNS.Secret, ipAddr, domain, recordType, cb.TTL)
|
||
if json.Valid([]byte(postPara)) {
|
||
contentType = "application/json"
|
||
}
|
||
}
|
||
requestURL := replacePara(cb.DNS.ID, ipAddr, domain, recordType, cb.TTL)
|
||
u, err := url.Parse(requestURL)
|
||
if err != nil {
|
||
util.Log("Callback的URL不正确")
|
||
return
|
||
}
|
||
req, err := http.NewRequest(method, u.String(), strings.NewReader(postPara))
|
||
if err != nil {
|
||
util.Log("异常信息: %s", err)
|
||
domain.UpdateStatus = config.UpdatedFailed
|
||
return
|
||
}
|
||
req.Header.Add("content-type", contentType)
|
||
|
||
clt := util.CreateHTTPClient()
|
||
resp, err := clt.Do(req)
|
||
body, err := util.GetHTTPResponseOrg(resp, err)
|
||
if err == nil {
|
||
util.Log("Callback调用成功, 域名: %s, IP: %s, 返回数据: %s", domain, ipAddr, string(body))
|
||
domain.UpdateStatus = config.UpdatedSuccess
|
||
} else {
|
||
util.Log("Callback调用失败, 异常信息: %s", err)
|
||
domain.UpdateStatus = config.UpdatedFailed
|
||
}
|
||
}
|
||
}
|
||
|
||
// replacePara 替换参数
|
||
func replacePara(orgPara, ipAddr string, domain *config.Domain, recordType string, ttl string) string {
|
||
// params 使用 map 以便添加更多参数
|
||
params := map[string]string{
|
||
"ip": ipAddr,
|
||
"domain": domain.String(),
|
||
"recordType": recordType,
|
||
"ttl": ttl,
|
||
}
|
||
|
||
// 也替换域名的自定义参数
|
||
for k, v := range domain.GetCustomParams() {
|
||
if len(v) == 1 {
|
||
params[k] = v[0]
|
||
}
|
||
}
|
||
|
||
// 将 map 转换为 [NewReplacer] 所需的参数
|
||
// map 中的每个元素占用 2 个位置（kv），因此需要预留 2 倍的空间
|
||
oldnew := make([]string, 0, len(params)*2)
|
||
for k, v := range params {
|
||
k = fmt.Sprintf("#{%s}", k)
|
||
oldnew = append(oldnew, k, v)
|
||
}
|
||
|
||
return strings.NewReplacer(oldnew...).Replace(orgPara)
|
||
}
|