lego/providers/dns/webnames/internal/client.go

97 lines
2.3 KiB
Go
Raw Permalink Normal View History

package internal
import (
"context"
"encoding/json"
"fmt"
"io"
"net/http"
"net/url"
"strings"
"time"
"github.com/go-acme/lego/v4/providers/dns/internal/errutils"
)
const defaultBaseURL = "https://www.webnames.ru/scripts/json_domain_zone_manager.pl"
// Client the Webnames API client.
type Client struct {
apiKey string
baseURL string
HTTPClient *http.Client
}
// NewClient Creates a new Client.
func NewClient(apiKey string) *Client {
return &Client{
apiKey: apiKey,
baseURL: defaultBaseURL,
HTTPClient: &http.Client{Timeout: 10 * time.Second},
}
}
// AddTXTRecord adds a TXT record.
// Inspired by https://github.com/regtime-ltd/certbot-dns-webnames/blob/master/authenticator.sh
func (c *Client) AddTXTRecord(ctx context.Context, domain, subDomain, value string) error {
data := url.Values{}
data.Set("domain", domain)
data.Set("type", "TXT")
data.Set("record", subDomain+":"+value)
data.Set("action", "add")
return c.doRequest(ctx, data)
}
// RemoveTXTRecord removes a TXT record.
// Inspired by https://github.com/regtime-ltd/certbot-dns-webnames/blob/master/cleanup.sh
func (c *Client) RemoveTXTRecord(ctx context.Context, domain, subDomain, value string) error {
data := url.Values{}
data.Set("domain", domain)
data.Set("type", "TXT")
data.Set("record", subDomain+":"+value)
data.Set("action", "delete")
return c.doRequest(ctx, data)
}
func (c *Client) doRequest(ctx context.Context, data url.Values) error {
data.Set("apikey", c.apiKey)
req, err := http.NewRequestWithContext(ctx, http.MethodPost, c.baseURL, strings.NewReader(data.Encode()))
if err != nil {
return err
}
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
resp, err := c.HTTPClient.Do(req)
if err != nil {
return errutils.NewHTTPDoError(req, err)
}
defer func() { _ = resp.Body.Close() }()
if resp.StatusCode/100 != 2 {
return errutils.NewUnexpectedResponseStatusCodeError(req, resp)
}
raw, err := io.ReadAll(resp.Body)
if err != nil {
return errutils.NewReadResponseError(req, resp.StatusCode, err)
}
var r APIResponse
err = json.Unmarshal(raw, &r)
if err != nil {
return errutils.NewUnmarshalError(req, resp.StatusCode, raw, err)
}
if r.Result == "OK" {
return nil
}
return fmt.Errorf("%s: %s", r.Result, r.Details)
}