mirror of
https://github.com/go-acme/lego.git
synced 2024-12-23 09:15:11 +02:00
129 lines
3.0 KiB
Go
129 lines
3.0 KiB
Go
|
package vinyldns
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
|
||
|
"github.com/go-acme/lego/v4/challenge/dns01"
|
||
|
"github.com/go-acme/lego/v4/platform/wait"
|
||
|
"github.com/vinyldns/go-vinyldns/vinyldns"
|
||
|
)
|
||
|
|
||
|
func (d *DNSProvider) getRecordSet(fqdn string) (*vinyldns.RecordSet, error) {
|
||
|
zoneName, hostName, err := splitDomain(fqdn)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
zone, err := d.client.ZoneByName(zoneName)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
allRecordSets, err := d.client.RecordSetsListAll(zone.ID, vinyldns.ListFilter{NameFilter: hostName})
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
var recordSets []vinyldns.RecordSet
|
||
|
for _, i := range allRecordSets {
|
||
|
if i.Type == "TXT" {
|
||
|
recordSets = append(recordSets, i)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
switch {
|
||
|
case len(recordSets) > 1:
|
||
|
return nil, fmt.Errorf("ambiguous recordset definition of %s", fqdn)
|
||
|
case len(recordSets) == 1:
|
||
|
return &recordSets[0], nil
|
||
|
default:
|
||
|
return nil, nil
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (d *DNSProvider) createRecordSet(fqdn string, records []vinyldns.Record) error {
|
||
|
zoneName, hostName, err := splitDomain(fqdn)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
zone, err := d.client.ZoneByName(zoneName)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
recordSet := vinyldns.RecordSet{
|
||
|
Name: hostName,
|
||
|
ZoneID: zone.ID,
|
||
|
Type: "TXT",
|
||
|
TTL: d.config.TTL,
|
||
|
Records: records,
|
||
|
}
|
||
|
|
||
|
resp, err := d.client.RecordSetCreate(&recordSet)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return d.waitForChanges("CreateRS", resp)
|
||
|
}
|
||
|
|
||
|
func (d *DNSProvider) updateRecordSet(recordSet *vinyldns.RecordSet, newRecords []vinyldns.Record) error {
|
||
|
operation := "delete"
|
||
|
if len(recordSet.Records) < len(newRecords) {
|
||
|
operation = "add"
|
||
|
}
|
||
|
|
||
|
recordSet.Records = newRecords
|
||
|
recordSet.TTL = d.config.TTL
|
||
|
|
||
|
resp, err := d.client.RecordSetUpdate(recordSet)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return d.waitForChanges("UpdateRS - "+operation, resp)
|
||
|
}
|
||
|
|
||
|
func (d *DNSProvider) deleteRecordSet(existingRecord *vinyldns.RecordSet) error {
|
||
|
resp, err := d.client.RecordSetDelete(existingRecord.ZoneID, existingRecord.ID)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return d.waitForChanges("DeleteRS", resp)
|
||
|
}
|
||
|
|
||
|
func (d *DNSProvider) waitForChanges(operation string, resp *vinyldns.RecordSetUpdateResponse) error {
|
||
|
return wait.For("vinyldns", d.config.PropagationTimeout, d.config.PollingInterval,
|
||
|
func() (bool, error) {
|
||
|
change, err := d.client.RecordSetChange(resp.Zone.ID, resp.RecordSet.ID, resp.ChangeID)
|
||
|
if err != nil {
|
||
|
return false, fmt.Errorf("failed to query change status: %w", err)
|
||
|
}
|
||
|
|
||
|
if change.Status == "Complete" {
|
||
|
return true, nil
|
||
|
}
|
||
|
|
||
|
return false, fmt.Errorf("waiting operation: %s, zoneID: %s, recordsetID: %s, changeID: %s",
|
||
|
operation, resp.Zone.ID, resp.RecordSet.ID, resp.ChangeID)
|
||
|
},
|
||
|
)
|
||
|
}
|
||
|
|
||
|
// splitDomain splits the hostname from the authoritative zone, and returns both parts.
|
||
|
func splitDomain(fqdn string) (string, string, error) {
|
||
|
zone, err := dns01.FindZoneByFqdn(fqdn)
|
||
|
if err != nil {
|
||
|
return "", "", fmt.Errorf("could not find zone for FDQN %q: %w", fqdn, err)
|
||
|
}
|
||
|
|
||
|
subDomain, err := dns01.ExtractSubDomain(fqdn, zone)
|
||
|
if err != nil {
|
||
|
return "", "", err
|
||
|
}
|
||
|
|
||
|
return zone, subDomain, nil
|
||
|
}
|