111 lines
2.8 KiB
Go
111 lines
2.8 KiB
Go
package ddns
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/naiba/nezha/pkg/utils"
|
|
)
|
|
|
|
type ProviderWebHook struct {
|
|
url string
|
|
requestMethod string
|
|
requestBody string
|
|
requestHeader string
|
|
domainConfig *DomainConfig
|
|
}
|
|
|
|
func NewProviderWebHook(s, rm, rb, rh string) *ProviderWebHook {
|
|
return &ProviderWebHook{
|
|
url: s,
|
|
requestMethod: rm,
|
|
requestBody: rb,
|
|
requestHeader: rh,
|
|
}
|
|
}
|
|
|
|
func (provider *ProviderWebHook) UpdateDomain(domainConfig *DomainConfig) error {
|
|
if domainConfig == nil {
|
|
return fmt.Errorf("获取 DDNS 配置失败")
|
|
}
|
|
provider.domainConfig = domainConfig
|
|
|
|
if provider.domainConfig.FullDomain == "" {
|
|
return fmt.Errorf("failed to update an empty domain")
|
|
}
|
|
|
|
if provider.domainConfig.EnableIPv4 && provider.domainConfig.Ipv4Addr != "" {
|
|
req, err := provider.prepareRequest(true)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to update a domain: %s. Cause by: %v", provider.domainConfig.FullDomain, err)
|
|
}
|
|
if _, err := utils.HttpClient.Do(req); err != nil {
|
|
return fmt.Errorf("failed to update a domain: %s. Cause by: %v", provider.domainConfig.FullDomain, err)
|
|
}
|
|
}
|
|
|
|
if provider.domainConfig.EnableIpv6 && provider.domainConfig.Ipv6Addr != "" {
|
|
req, err := provider.prepareRequest(false)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to update a domain: %s. Cause by: %v", provider.domainConfig.FullDomain, err)
|
|
}
|
|
if _, err := utils.HttpClient.Do(req); err != nil {
|
|
return fmt.Errorf("failed to update a domain: %s. Cause by: %v", provider.domainConfig.FullDomain, err)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (provider *ProviderWebHook) prepareRequest(isIPv4 bool) (*http.Request, error) {
|
|
u, err := url.Parse(provider.url)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed parsing url: %v", err)
|
|
}
|
|
|
|
// Only handle queries here
|
|
q := u.Query()
|
|
for p, vals := range q {
|
|
for n, v := range vals {
|
|
vals[n] = provider.formatWebhookString(v, isIPv4)
|
|
}
|
|
q[p] = vals
|
|
}
|
|
|
|
u.RawQuery = q.Encode()
|
|
body := provider.formatWebhookString(provider.requestBody, isIPv4)
|
|
header := provider.formatWebhookString(provider.requestHeader, isIPv4)
|
|
headers := strings.Split(header, "\n")
|
|
|
|
req, err := http.NewRequest(provider.requestMethod, u.String(), bytes.NewBufferString(body))
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed creating new request: %v", err)
|
|
}
|
|
|
|
utils.SetStringHeadersToRequest(req, headers)
|
|
return req, nil
|
|
}
|
|
|
|
func (provider *ProviderWebHook) formatWebhookString(s string, isIPv4 bool) string {
|
|
var ipAddr, ipType string
|
|
if isIPv4 {
|
|
ipAddr = provider.domainConfig.Ipv4Addr
|
|
ipType = "ipv4"
|
|
} else {
|
|
ipAddr = provider.domainConfig.Ipv6Addr
|
|
ipType = "ipv6"
|
|
}
|
|
|
|
r := strings.NewReplacer(
|
|
"{ip}", ipAddr,
|
|
"{domain}", provider.domainConfig.FullDomain,
|
|
"{type}", ipType,
|
|
"\r", "",
|
|
)
|
|
|
|
result := r.Replace(strings.TrimSpace(s))
|
|
return result
|
|
}
|