Merge pull request #304 from aqche/support_vult_dns_records

Add support for Vultr DNS record

README.md

@@ -753,6 +753,7 @@ List of supported Vultr resources:
     * `vultr_block_storage`
 *   `dns_domain`
     * `vultr_dns_domain`
+    * `vultr_dns_record`
 *   `firewall_group`
     * `vultr_firewall_group`
 *   `network`

providers/vultr/dns_domain.go

@@ -16,6 +16,7 @@ package vultr
 
 import (
 	"context"
+	"strconv"
 
 	"github.com/GoogleCloudPlatform/terraformer/terraform_utils"
 	"github.com/vultr/govultr"
@@ -25,25 +26,51 @@ type DNSDomainGenerator struct {
 	VultrService
 }
 
-func (g DNSDomainGenerator) createResources(domainList []govultr.DNSDomain) []terraform_utils.Resource {
-	var resources []terraform_utils.Resource
+func (g *DNSDomainGenerator) loadDNSDomains(client *govultr.Client) ([]govultr.DNSDomain, error) {
+	domainList, err := client.DNSDomain.List(context.Background())
+	if err != nil {
+		return nil, err
+	}
 	for _, domain := range domainList {
-		resources = append(resources, terraform_utils.NewSimpleResource(
+		g.Resources = append(g.Resources, terraform_utils.NewSimpleResource(
 			domain.Domain,
 			domain.Domain,
 			"vultr_dns_domain",
 			"vultr",
 			[]string{}))
 	}
-	return resources
+	return domainList, nil
+}
+
+func (g *DNSDomainGenerator) loadDNSRecords(client *govultr.Client, domain string) error {
+	recordList, err := client.DNSRecord.List(context.Background(), domain)
+	if err != nil {
+		return err
+	}
+	for _, record := range recordList {
+		g.Resources = append(g.Resources, terraform_utils.NewResource(
+			strconv.Itoa(record.RecordID),
+			strconv.Itoa(record.RecordID),
+			"vultr_dns_record",
+			"vultr",
+			map[string]string{"domain": domain},
+			[]string{},
+			map[string]interface{}{}))
+	}
+	return nil
 }
 
 func (g *DNSDomainGenerator) InitResources() error {
 	client := g.generateClient()
-	output, err := client.DNSDomain.List(context.Background())
+	domainList, err := g.loadDNSDomains(client)
 	if err != nil {
 		return err
 	}
-	g.Resources = g.createResources(output)
+	for _, domain := range domainList {
+		err := g.loadDNSRecords(client, domain.Domain)
+		if err != nil {
+			return err
+		}
+	}
 	return nil
 }