forked from TrueCloudLab/lego
140 lines
3 KiB
Go
140 lines
3 KiB
Go
package internal
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"path"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
"unicode"
|
|
|
|
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 {
|
|
HTTPClient *http.Client
|
|
baseURL *url.URL
|
|
debug bool
|
|
|
|
apiKey string
|
|
password string
|
|
}
|
|
|
|
// NewClient creates a new Client.
|
|
func NewClient(apiKey string, password string) *Client {
|
|
baseURL, _ := url.Parse(baseURL)
|
|
|
|
return &Client{
|
|
HTTPClient: &http.Client{Timeout: 10 * time.Second},
|
|
baseURL: baseURL,
|
|
apiKey: apiKey,
|
|
password: password,
|
|
}
|
|
}
|
|
|
|
// AddRecord The command is intended to add a new DNS record to a specific zone (domain).
|
|
func (c Client) AddRecord(query RecordQuery) error {
|
|
var r APIResponse
|
|
err := c.do("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(query RecordQuery) error {
|
|
var r APIResponse
|
|
err := c.do("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(query ListRecordQuery) ([]Record, error) {
|
|
var l ListResponse
|
|
err := c.do("List", query, &l)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if l.Status != statusSuccess {
|
|
return nil, l.APIResponse
|
|
}
|
|
|
|
return l.Records, nil
|
|
}
|
|
|
|
func (c Client) do(action string, params interface{}, response interface{}) error {
|
|
endpoint, err := c.baseURL.Parse(path.Join(c.baseURL.Path, "Domain", "DnsRecord", action))
|
|
if err != nil {
|
|
return fmt.Errorf("create endpoint: %w", err)
|
|
}
|
|
|
|
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")
|
|
|
|
resp, err := c.HTTPClient.PostForm(endpoint.String(), values)
|
|
if err != nil {
|
|
return fmt.Errorf("post request: %w", err)
|
|
}
|
|
|
|
defer func() { _ = resp.Body.Close() }()
|
|
|
|
if resp.StatusCode/100 != 2 {
|
|
data, _ := ioutil.ReadAll(resp.Body)
|
|
return fmt.Errorf("status code: %d, %s", resp.StatusCode, string(data))
|
|
}
|
|
|
|
if c.debug {
|
|
return dump(endpoint, resp, response)
|
|
}
|
|
|
|
return json.NewDecoder(resp.Body).Decode(response)
|
|
}
|
|
|
|
func dump(endpoint *url.URL, resp *http.Response, response interface{}) error {
|
|
data, err := ioutil.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 = ioutil.WriteFile(filepath.Join("fixtures", strings.Join(fields, "_")+".json"), data, 0o666)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return json.Unmarshal(data, response)
|
|
}
|