150 lines
3.8 KiB
Go
150 lines
3.8 KiB
Go
package internal
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
)
|
|
|
|
// defaultBaseURL represents the API endpoint to call.
|
|
const defaultBaseURL = "https://api.luadns.com"
|
|
|
|
// Client Lua DNS API client.
|
|
type Client struct {
|
|
HTTPClient *http.Client
|
|
BaseURL string
|
|
|
|
apiUsername string
|
|
apiToken string
|
|
}
|
|
|
|
// NewClient creates a new Client.
|
|
func NewClient(apiUsername, apiToken string) *Client {
|
|
return &Client{
|
|
HTTPClient: http.DefaultClient,
|
|
BaseURL: defaultBaseURL,
|
|
apiUsername: apiUsername,
|
|
apiToken: apiToken,
|
|
}
|
|
}
|
|
|
|
// ListZones gets all the hosted zones.
|
|
// https://luadns.com/api.html#list-zones
|
|
func (d *Client) ListZones() ([]DNSZone, error) {
|
|
resp, err := d.do(http.MethodGet, "/v1/zones", nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
defer func() { _ = resp.Body.Close() }()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
bodyBytes, _ := ioutil.ReadAll(resp.Body)
|
|
|
|
var errResp errorResponse
|
|
err = json.Unmarshal(bodyBytes, &errResp)
|
|
if err == nil {
|
|
return nil, fmt.Errorf("api call error: Status=%v: %w", resp.StatusCode, errResp)
|
|
}
|
|
|
|
return nil, fmt.Errorf("api call error: Status=%d: %s", resp.StatusCode, string(bodyBytes))
|
|
}
|
|
|
|
var zones []DNSZone
|
|
err = json.NewDecoder(resp.Body).Decode(&zones)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to unmarshal response body: %w", err)
|
|
}
|
|
|
|
return zones, nil
|
|
}
|
|
|
|
// CreateRecord creates a new record in a zone.
|
|
// https://luadns.com/api.html#create-a-record
|
|
func (d *Client) CreateRecord(zone DNSZone, newRecord DNSRecord) (*DNSRecord, error) {
|
|
body, err := json.Marshal(newRecord)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to marshal request body: %w", err)
|
|
}
|
|
|
|
resource := fmt.Sprintf("/v1/zones/%d/records", zone.ID)
|
|
|
|
resp, err := d.do(http.MethodPost, resource, bytes.NewReader(body))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
defer func() { _ = resp.Body.Close() }()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
bodyBytes, _ := ioutil.ReadAll(resp.Body)
|
|
|
|
var errResp errorResponse
|
|
err = json.Unmarshal(bodyBytes, &errResp)
|
|
if err == nil {
|
|
return nil, fmt.Errorf("could not create record %v: Status=%d: %w",
|
|
string(body), resp.StatusCode, errResp)
|
|
}
|
|
|
|
return nil, fmt.Errorf("could not create record %v: Status=%d: %s",
|
|
string(body), resp.StatusCode, string(bodyBytes))
|
|
}
|
|
|
|
var record *DNSRecord
|
|
err = json.NewDecoder(resp.Body).Decode(&record)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to unmarshal response body: %w", err)
|
|
}
|
|
|
|
return record, nil
|
|
}
|
|
|
|
// DeleteRecord deletes a record.
|
|
// https://luadns.com/api.html#delete-a-record
|
|
func (d *Client) DeleteRecord(record *DNSRecord) error {
|
|
body, err := json.Marshal(record)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to marshal request body: %w", err)
|
|
}
|
|
|
|
resource := fmt.Sprintf("/v1/zones/%d/records/%d", record.ZoneID, record.ID)
|
|
|
|
resp, err := d.do(http.MethodDelete, resource, bytes.NewReader(body))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer func() { _ = resp.Body.Close() }()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
bodyBytes, _ := ioutil.ReadAll(resp.Body)
|
|
|
|
var errResp errorResponse
|
|
err = json.Unmarshal(bodyBytes, &errResp)
|
|
if err == nil {
|
|
return fmt.Errorf("could not delete record %v: Status=%d: %w",
|
|
string(body), resp.StatusCode, errResp)
|
|
}
|
|
|
|
return fmt.Errorf("could not delete record %v: Status=%d: %s",
|
|
string(body), resp.StatusCode, string(bodyBytes))
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *Client) do(method, uri string, body io.Reader) (*http.Response, error) {
|
|
req, err := http.NewRequest(method, fmt.Sprintf("%s%s", d.BaseURL, uri), body)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to create request: %w", err)
|
|
}
|
|
|
|
req.Header.Set("Accept", "application/json")
|
|
req.Header.Set("Content-Type", "application/json")
|
|
req.SetBasicAuth(d.apiUsername, d.apiToken)
|
|
|
|
return d.HTTPClient.Do(req)
|
|
}
|