1
0
mirror of https://github.com/go-acme/lego.git synced 2024-12-14 06:06:10 +02:00
lego/providers/dns/dynu/internal/client.go
2021-08-25 11:44:11 +02:00

182 lines
3.9 KiB
Go

package internal
import (
"bytes"
"encoding/json"
"errors"
"fmt"
"io"
"net/http"
"net/url"
"path"
"strconv"
"time"
"github.com/cenkalti/backoff/v4"
"github.com/go-acme/lego/v4/log"
)
const defaultBaseURL = "https://api.dynu.com/v2"
type Client struct {
HTTPClient *http.Client
BaseURL string
}
func NewClient() *Client {
return &Client{
HTTPClient: http.DefaultClient,
BaseURL: defaultBaseURL,
}
}
// GetRecords Get DNS records based on a hostname and resource record type.
func (c Client) GetRecords(hostname, recordType string) ([]DNSRecord, error) {
endpoint, err := c.createEndpoint("dns", "record", hostname)
if err != nil {
return nil, err
}
query := endpoint.Query()
query.Set("recordType", recordType)
endpoint.RawQuery = query.Encode()
apiResp := RecordsResponse{}
err = c.doRetry(http.MethodGet, endpoint.String(), nil, &apiResp)
if err != nil {
return nil, err
}
if apiResp.StatusCode/100 != 2 {
return nil, fmt.Errorf("API error: %w", apiResp.APIException)
}
return apiResp.DNSRecords, nil
}
// AddNewRecord Add a new DNS record for DNS service.
func (c Client) AddNewRecord(domainID int64, record DNSRecord) error {
endpoint, err := c.createEndpoint("dns", strconv.FormatInt(domainID, 10), "record")
if err != nil {
return err
}
reqBody, err := json.Marshal(record)
if err != nil {
return err
}
apiResp := RecordResponse{}
err = c.doRetry(http.MethodPost, endpoint.String(), reqBody, &apiResp)
if err != nil {
return err
}
if apiResp.StatusCode/100 != 2 {
return fmt.Errorf("API error: %w", apiResp.APIException)
}
return nil
}
// DeleteRecord Remove a DNS record from DNS service.
func (c Client) DeleteRecord(domainID, recordID int64) error {
endpoint, err := c.createEndpoint("dns", strconv.FormatInt(domainID, 10), "record", strconv.FormatInt(recordID, 10))
if err != nil {
return err
}
apiResp := APIException{}
err = c.doRetry(http.MethodDelete, endpoint.String(), nil, &apiResp)
if err != nil {
return err
}
if apiResp.StatusCode/100 != 2 {
return fmt.Errorf("API error: %w", apiResp)
}
return nil
}
// GetRootDomain Get the root domain name based on a hostname.
func (c Client) GetRootDomain(hostname string) (*DNSHostname, error) {
endpoint, err := c.createEndpoint("dns", "getroot", hostname)
if err != nil {
return nil, err
}
apiResp := DNSHostname{}
err = c.doRetry(http.MethodGet, endpoint.String(), nil, &apiResp)
if err != nil {
return nil, err
}
if apiResp.StatusCode/100 != 2 {
return nil, fmt.Errorf("API error: %w", apiResp.APIException)
}
return &apiResp, nil
}
// doRetry the API is really unstable so we need to retry on EOF.
func (c Client) doRetry(method, uri string, body []byte, data interface{}) error {
var resp *http.Response
operation := func() error {
var reqBody io.Reader
if len(body) > 0 {
reqBody = bytes.NewReader(body)
}
req, err := http.NewRequest(method, uri, reqBody)
if err != nil {
return err
}
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Accept", "application/json")
resp, err = c.HTTPClient.Do(req)
if errors.Is(err, io.EOF) {
return err
}
if err != nil {
return backoff.Permanent(fmt.Errorf("client error: %w", err))
}
return nil
}
notify := func(err error, duration time.Duration) {
log.Printf("client retries because of %v", err)
}
bo := backoff.NewExponentialBackOff()
bo.InitialInterval = 1 * time.Second
err := backoff.RetryNotify(operation, bo, notify)
if err != nil {
return err
}
defer func() { _ = resp.Body.Close() }()
all, err := io.ReadAll(resp.Body)
if err != nil {
return fmt.Errorf("failed to read response body: %w", err)
}
return json.Unmarshal(all, data)
}
func (c Client) createEndpoint(fragments ...string) (*url.URL, error) {
baseURL, err := url.Parse(c.BaseURL)
if err != nil {
return nil, err
}
return baseURL.Parse(path.Join(baseURL.Path, path.Join(fragments...)))
}