mirror of
https://github.com/go-acme/lego.git
synced 2024-12-23 09:15:11 +02:00
157 lines
3.4 KiB
Go
157 lines
3.4 KiB
Go
package internal
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
"unicode"
|
|
|
|
"github.com/go-acme/lego/v4/providers/dns/internal/errutils"
|
|
querystring "github.com/google/go-querystring/query"
|
|
)
|
|
|
|
const baseURL = "https://api.internet.bs"
|
|
|
|
// status SUCCESS, PENDING, FAILURE.
|
|
const statusSuccess = "SUCCESS"
|
|
|
|
// Client is the API client.
|
|
type Client struct {
|
|
apiKey string
|
|
password string
|
|
|
|
debug bool
|
|
|
|
baseURL *url.URL
|
|
HTTPClient *http.Client
|
|
}
|
|
|
|
// NewClient creates a new Client.
|
|
func NewClient(apiKey string, password string) *Client {
|
|
baseURL, _ := url.Parse(baseURL)
|
|
|
|
return &Client{
|
|
apiKey: apiKey,
|
|
password: password,
|
|
baseURL: baseURL,
|
|
HTTPClient: &http.Client{Timeout: 10 * time.Second},
|
|
}
|
|
}
|
|
|
|
// AddRecord The command is intended to add a new DNS record to a specific zone (domain).
|
|
func (c Client) AddRecord(ctx context.Context, query RecordQuery) error {
|
|
var r APIResponse
|
|
err := c.doRequest(ctx, "Add", query, &r)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if r.Status != statusSuccess {
|
|
return r
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// RemoveRecord The command is intended to remove a DNS record from a specific zone.
|
|
func (c Client) RemoveRecord(ctx context.Context, query RecordQuery) error {
|
|
var r APIResponse
|
|
err := c.doRequest(ctx, "Remove", query, &r)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if r.Status != statusSuccess {
|
|
return r
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// ListRecords The command is intended to retrieve the list of DNS records for a specific domain.
|
|
func (c Client) ListRecords(ctx context.Context, query ListRecordQuery) ([]Record, error) {
|
|
var l ListResponse
|
|
err := c.doRequest(ctx, "List", query, &l)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if l.Status != statusSuccess {
|
|
return nil, l.APIResponse
|
|
}
|
|
|
|
return l.Records, nil
|
|
}
|
|
|
|
func (c Client) doRequest(ctx context.Context, action string, params any, result any) error {
|
|
endpoint := c.baseURL.JoinPath("Domain", "DnsRecord", action)
|
|
|
|
values, err := querystring.Values(params)
|
|
if err != nil {
|
|
return fmt.Errorf("parse query parameters: %w", err)
|
|
}
|
|
|
|
values.Set("apiKey", c.apiKey)
|
|
values.Set("password", c.password)
|
|
values.Set("ResponseFormat", "JSON")
|
|
|
|
req, err := http.NewRequestWithContext(ctx, http.MethodPost, endpoint.String(), strings.NewReader(values.Encode()))
|
|
if err != nil {
|
|
return fmt.Errorf("unable to create request: %w", 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)
|
|
}
|
|
|
|
if c.debug {
|
|
return dump(endpoint, resp, result)
|
|
}
|
|
|
|
raw, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return errutils.NewReadResponseError(req, resp.StatusCode, err)
|
|
}
|
|
|
|
err = json.Unmarshal(raw, result)
|
|
if err != nil {
|
|
return errutils.NewUnmarshalError(req, resp.StatusCode, raw, err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func dump(endpoint *url.URL, resp *http.Response, response any) error {
|
|
raw, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fields := strings.FieldsFunc(endpoint.Path, func(r rune) bool {
|
|
return !unicode.IsLetter(r) && !unicode.IsNumber(r)
|
|
})
|
|
|
|
err = os.WriteFile(filepath.Join("fixtures", strings.Join(fields, "_")+".json"), raw, 0o666)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return json.Unmarshal(raw, response)
|
|
}
|