93 lines
2.1 KiB
Go
93 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/libdns/libdns"
|
|
"github.com/libdns/vultr/v2"
|
|
)
|
|
|
|
func main() {
|
|
token := os.Getenv("VULTR_API_TOKEN")
|
|
if token == "" {
|
|
fmt.Printf("VULTR_API_TOKEN not set\n")
|
|
return
|
|
}
|
|
zone := os.Getenv("ZONE")
|
|
if zone == "" {
|
|
fmt.Printf("ZONE not set\n")
|
|
return
|
|
}
|
|
shouldCleanup := os.Getenv("DELETE_RECORDS") == "true"
|
|
|
|
provider := vultr.Provider{APIToken: token}
|
|
|
|
zones, err := provider.ListZones(context.TODO())
|
|
if err != nil {
|
|
fmt.Printf("ERROR: %s\n", err.Error())
|
|
}
|
|
|
|
for _, zone := range zones {
|
|
fmt.Printf("%s\n", zone.Name)
|
|
}
|
|
|
|
records, err := provider.GetRecords(context.TODO(), zone)
|
|
if err != nil {
|
|
fmt.Printf("ERROR: %s\n", err.Error())
|
|
}
|
|
|
|
testName := "libdns-test"
|
|
testId := ""
|
|
for _, record := range records {
|
|
fmt.Printf("%s (.%s): %s, %s\n", record.RR().Name, zone, record.RR().Data, record.RR().Type)
|
|
|
|
if record.RR().Name == testName {
|
|
testId = record.(vultr.VultrRecord).ID
|
|
}
|
|
}
|
|
|
|
if testId != "" {
|
|
if shouldCleanup {
|
|
fmt.Printf("Delete entry for %s (id:%s)\n", testName, testId)
|
|
_, err = provider.DeleteRecords(context.TODO(), zone, []libdns.Record{vultr.VultrRecord{
|
|
ID: testId,
|
|
}})
|
|
if err != nil {
|
|
fmt.Printf("ERROR: %s\n", err.Error())
|
|
}
|
|
} else {
|
|
// Set only works if we have a record.ID
|
|
fmt.Printf("Replacing entry for %s\n", testName)
|
|
_, err = provider.SetRecords(context.TODO(), zone, []libdns.Record{vultr.VultrRecord{
|
|
Record: libdns.RR{
|
|
Name: testName,
|
|
Type: "TXT",
|
|
Data: fmt.Sprintf("Replacement test entry created by libdns %s", time.Now()),
|
|
TTL: time.Duration(90) * time.Second,
|
|
},
|
|
ID: testId,
|
|
}})
|
|
if err != nil {
|
|
fmt.Printf("ERROR: %s\n", err.Error())
|
|
}
|
|
}
|
|
} else {
|
|
fmt.Printf("Creating new entry for %s\n", testName)
|
|
_, err = provider.AppendRecords(context.TODO(), zone, []libdns.Record{vultr.VultrRecord{
|
|
Record: libdns.RR{
|
|
Type: "TXT",
|
|
Name: testName,
|
|
Data: fmt.Sprintf("This is a test entry created by libdns %s", time.Now()),
|
|
TTL: time.Duration(60) * time.Second,
|
|
},
|
|
ID: testId,
|
|
}})
|
|
if err != nil {
|
|
fmt.Printf("ERROR: %s\n", err.Error())
|
|
}
|
|
}
|
|
}
|