package internal import ( "bytes" "encoding/json" "fmt" "io/ioutil" "net/http" "net/url" "path" ) const defaultBaseURL = "https://api.netlify.com/api/v1" // Client Netlify API client. type Client struct { HTTPClient *http.Client BaseURL string token string } // NewClient creates a new Client. func NewClient(token string) *Client { return &Client{ HTTPClient: http.DefaultClient, BaseURL: defaultBaseURL, token: token, } } // GetRecords gets a DNS records. func (c *Client) GetRecords(zoneID string) ([]DNSRecord, error) { endpoint, err := c.createEndpoint("dns_zones", zoneID, "dns_records") if err != nil { return nil, fmt.Errorf("failed to parse endpoint: %w", err) } req, err := http.NewRequest(http.MethodGet, endpoint, nil) if err != nil { return nil, fmt.Errorf("failed to create request: %w", err) } req.Header.Set("Accept", "application/json") req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", c.token)) resp, err := c.HTTPClient.Do(req) if err != nil { return nil, fmt.Errorf("API call failed: %w", err) } defer func() { _ = resp.Body.Close() }() body, err := ioutil.ReadAll(resp.Body) if err != nil { return nil, fmt.Errorf("failed to read response body: %w", err) } if resp.StatusCode != http.StatusOK { return nil, fmt.Errorf("invalid status code: %s: %s", resp.Status, string(body)) } var records []DNSRecord err = json.Unmarshal(body, &records) if err != nil { return nil, fmt.Errorf("failed to marshal response body: %w", err) } return records, nil } // CreateRecord creates a DNS records. func (c *Client) CreateRecord(zoneID string, record DNSRecord) (*DNSRecord, error) { endpoint, err := c.createEndpoint("dns_zones", zoneID, "dns_records") if err != nil { return nil, fmt.Errorf("failed to parse endpoint: %w", err) } marshaledRecord, err := json.Marshal(record) if err != nil { return nil, fmt.Errorf("failed to marshal request body: %w", err) } req, err := http.NewRequest(http.MethodPost, endpoint, bytes.NewReader(marshaledRecord)) 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; charset=utf-8") req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", c.token)) resp, err := c.HTTPClient.Do(req) if err != nil { return nil, fmt.Errorf("API call failed: %w", err) } defer func() { _ = resp.Body.Close() }() body, err := ioutil.ReadAll(resp.Body) if err != nil { return nil, fmt.Errorf("failed to read response body: %w", err) } if resp.StatusCode != http.StatusCreated { return nil, fmt.Errorf("invalid status code: %s: %s", resp.Status, string(body)) } var recordResp DNSRecord err = json.Unmarshal(body, &recordResp) if err != nil { return nil, fmt.Errorf("failed to marshal response body: %w", err) } return &recordResp, nil } // RemoveRecord removes a DNS records. func (c *Client) RemoveRecord(zoneID, recordID string) error { endpoint, err := c.createEndpoint("dns_zones", zoneID, "dns_records", recordID) if err != nil { return fmt.Errorf("failed to parse endpoint: %w", err) } req, err := http.NewRequest(http.MethodDelete, endpoint, nil) if err != nil { return fmt.Errorf("failed to create request: %w", err) } req.Header.Set("Accept", "application/json") req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", c.token)) resp, err := c.HTTPClient.Do(req) if err != nil { return fmt.Errorf("API call failed: %w", err) } defer func() { _ = resp.Body.Close() }() body, err := ioutil.ReadAll(resp.Body) if err != nil { return fmt.Errorf("failed to read response body: %w", err) } if resp.StatusCode != http.StatusNoContent { return fmt.Errorf("invalid status code: %s: %s", resp.Status, string(body)) } return nil } func (c *Client) createEndpoint(parts ...string) (string, error) { base, err := url.Parse(c.BaseURL) if err != nil { return "", fmt.Errorf("failed to parse base URL: %w", err) } endpoint, err := base.Parse(path.Join(base.Path, path.Join(parts...))) if err != nil { return "", fmt.Errorf("failed to parse endpoint path: %w", err) } return endpoint.String(), nil }