client.go

package leaseweb

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
	"os"
	"strings"
)

func handleLeasewebHttpError(res *http.Response, source string) error {
	if res.StatusCode >= 200 && res.StatusCode < 300 {
		// valid response, nothing to handle
		return nil
	}

	var message = fmt.Sprintf("Received StatusCode %d from Leaseweb API for %s operation", res.StatusCode, source)
	fmt.Fprintf(os.Stderr, "%s\n", message)
	data, err := ioutil.ReadAll(res.Body)
	if err != nil {
		return err
	}
	// add the full response from the Leaseweb API in text to the error message,
	// this makes it significantly easier to debug errors
	message += fmt.Sprintf(", response: %s", data)

	var errorResponseModel leasewebHttpError
	json.Unmarshal([]byte(data), &errorResponseModel)
	fmt.Fprintf(os.Stderr, "%+v", errorResponseModel)
	return fmt.Errorf(message)
}

func (p *Provider) listRecordSets(domainName string) (leasewebRecordSets, error) {
	httpClient := &http.Client{}

	req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("https://api.leaseweb.com/hosting/v2/domains/%s/resourceRecordSets", domainName), nil)
	if err != nil {
		return leasewebRecordSets{}, err
	}

	req.Header.Add(LeasewebApiKeyHeader, p.APIKey)

	res, err := httpClient.Do(req)
	defer res.Body.Close()
	if err != nil {
		return leasewebRecordSets{}, err
	}
	// if res.StatusCode == 401 {
	// 	return nil, fmt.Errorf("Received StatusCode %d from Leaseweb API, used APIKey: %s", res.StatusCode, p.APIKey)
	// }
	err = handleLeasewebHttpError(res, "list")
	if err != nil {
		return leasewebRecordSets{}, err
	}
	data, err := ioutil.ReadAll(res.Body)
	if err != nil {
		return leasewebRecordSets{}, err
	}

	var recordSets leasewebRecordSets
	json.Unmarshal([]byte(data), &recordSets)

	return recordSets, nil
}

func (p *Provider) createRecordSet(zone string, recordSet leasewebRecordSet) (leasewebRecordSet, error) {
	client := &http.Client{}

	bodyBuffer := new(bytes.Buffer)
	json.NewEncoder(bodyBuffer).Encode(recordSet)

	var domainName = strings.TrimSuffix(zone, ".")

	req, err := http.NewRequest(http.MethodPost, fmt.Sprintf("https://api.leaseweb.com/hosting/v2/domains/%s/resourceRecordSets", domainName), bodyBuffer)
	if err != nil {
		return leasewebRecordSet{}, err
	}

	req.Header.Add(LeasewebApiKeyHeader, p.APIKey)

	res, err := client.Do(req)
	defer res.Body.Close()
	if err != nil {
		return leasewebRecordSet{}, err
	}
	err = handleLeasewebHttpError(res, "create")
	if err != nil {
		return leasewebRecordSet{}, err
	}
	data, err := ioutil.ReadAll(res.Body)
	if err != nil {
		return leasewebRecordSet{}, err
	}

	json.Unmarshal([]byte(data), &recordSet)
	return recordSet, nil
}

func (p *Provider) updateRecordSet(domainName string, recordSet leasewebRecordSet) (leasewebRecordSets, error) {
	client := &http.Client{}

	bodyBuffer := new(bytes.Buffer)
	json.NewEncoder(bodyBuffer).Encode(&updateRecordSetRequest{
		Content: recordSet.Content,
		TTL:     recordSet.TTL,
	})

	// https://developer.leaseweb.com/api-docs/domains_v2.html#operation/put/domains/{domainName}/resourceRecordSets/{name}/{type}
	// https://api.leaseweb.com/hosting/v2/domains/{domainName}/resourceRecordSets/{name}/{type}
	req, err := http.NewRequest(http.MethodPut, fmt.Sprintf("https://api.leaseweb.com/hosting/v2/domains/%s/resourceRecordSets/%s/%s", domainName, recordSet.Name, recordSet.Type), bodyBuffer)
	if err != nil {
		return leasewebRecordSets{}, err
	}
	req.Header.Add(LeasewebApiKeyHeader, p.APIKey)

	res, err := client.Do(req)
	defer res.Body.Close()
	if err != nil {
		return leasewebRecordSets{}, err
	}
	err = handleLeasewebHttpError(res, "update")
	if err != nil {
		return leasewebRecordSets{}, err
	}

	return leasewebRecordSets{}, nil
}