cloudflare-dns-cli/vendor/github.com/libdns/cloudflare/provider.go

175 lines
4.7 KiB
Go
Raw Normal View History

package cloudflare
import (
"context"
"fmt"
"net/http"
"sync"
"github.com/libdns/libdns"
)
// Provider implements the libdns interfaces for Cloudflare.
// TODO: Support pagination and retries, handle rate limits.
type Provider struct {
// API token is used for authentication. Make sure to use a
// scoped API **token**, NOT a global API **key**. It will
// need two permissions: Zone-Zone-Read and Zone-DNS-Edit,
// unless you are only using `GetRecords()`, in which case
// the second can be changed to Read.
APIToken string `json:"api_token,omitempty"`
zones map[string]cfZone
zonesMu sync.Mutex
}
// GetRecords lists all the records in the zone.
func (p *Provider) GetRecords(ctx context.Context, zone string) ([]libdns.Record, error) {
zoneInfo, err := p.getZoneInfo(ctx, zone)
if err != nil {
return nil, err
}
reqURL := fmt.Sprintf("%s/zones/%s/dns_records", baseURL, zoneInfo.ID)
req, err := http.NewRequestWithContext(ctx, "GET", reqURL, nil)
if err != nil {
return nil, err
}
var result []cfDNSRecord
_, err = p.doAPIRequest(req, &result)
if err != nil {
return nil, err
}
recs := make([]libdns.Record, 0, len(result))
for _, rec := range result {
recs = append(recs, rec.libdnsRecord(zone))
}
return recs, nil
}
// AppendRecords adds records to the zone. It returns the records that were added.
func (p *Provider) AppendRecords(ctx context.Context, zone string, records []libdns.Record) ([]libdns.Record, error) {
zoneInfo, err := p.getZoneInfo(ctx, zone)
if err != nil {
return nil, err
}
var created []libdns.Record
for _, rec := range records {
result, err := p.createRecord(ctx, zoneInfo, rec)
if err != nil {
return nil, err
}
created = append(created, result.libdnsRecord(zone))
}
return created, nil
}
// DeleteRecords deletes the records from the zone. If a record does not have an ID,
// it will be looked up. It returns the records that were deleted.
func (p *Provider) DeleteRecords(ctx context.Context, zone string, records []libdns.Record) ([]libdns.Record, error) {
zoneInfo, err := p.getZoneInfo(ctx, zone)
if err != nil {
return nil, err
}
var recs []libdns.Record
for _, rec := range records {
// we create a "delete queue" for each record
// requested for deletion; if the record ID
// is known, that is the only one to fill the
// queue, but if it's not known, we try to find
// a match theoretically there could be more
// than one
var deleteQueue []libdns.Record
if rec.ID == "" {
// record ID is required; try to find it with what was provided
exactMatches, err := p.getDNSRecords(ctx, zoneInfo, rec, true)
if err != nil {
return nil, err
}
for _, rec := range exactMatches {
deleteQueue = append(deleteQueue, rec.libdnsRecord(zone))
}
} else {
deleteQueue = []libdns.Record{rec}
}
for _, delRec := range deleteQueue {
reqURL := fmt.Sprintf("%s/zones/%s/dns_records/%s", baseURL, zoneInfo.ID, delRec.ID)
req, err := http.NewRequestWithContext(ctx, "DELETE", reqURL, nil)
if err != nil {
return nil, err
}
var result cfDNSRecord
_, err = p.doAPIRequest(req, &result)
if err != nil {
return nil, err
}
recs = append(recs, result.libdnsRecord(zone))
}
}
return recs, nil
}
// SetRecords sets the records in the zone, either by updating existing records
// or creating new ones. It returns the updated records.
func (p *Provider) SetRecords(ctx context.Context, zone string, records []libdns.Record) ([]libdns.Record, error) {
zoneInfo, err := p.getZoneInfo(ctx, zone)
if err != nil {
return nil, err
}
var results []libdns.Record
for _, rec := range records {
oldRec := cloudflareRecord(rec)
oldRec.ZoneID = zoneInfo.ID
if rec.ID == "" {
// the record might already exist, even if we don't know the ID yet
matches, err := p.getDNSRecords(ctx, zoneInfo, rec, false)
if err != nil {
return nil, err
}
if len(matches) == 0 {
// record doesn't exist; create it
result, err := p.createRecord(ctx, zoneInfo, rec)
if err != nil {
return nil, err
}
results = append(results, result.libdnsRecord(zone))
continue
}
if len(matches) > 1 {
return nil, fmt.Errorf("unexpectedly found more than 1 record for %v", rec)
}
// record does exist, fill in the ID so that we can update it
oldRec.ID = matches[0].ID
}
// record exists; update it
result, err := p.updateRecord(ctx, oldRec, cloudflareRecord(rec))
if err != nil {
return nil, err
}
results = append(results, result.libdnsRecord(zone))
}
return results, nil
}
// Interface guards
var (
_ libdns.RecordGetter = (*Provider)(nil)
_ libdns.RecordAppender = (*Provider)(nil)
_ libdns.RecordSetter = (*Provider)(nil)
_ libdns.RecordDeleter = (*Provider)(nil)
)