Scaffold Azure RM Route Table resource

Author: stack72

builtin/providers/azurerm/config.go

@@ -37,6 +37,7 @@ type ArmClient struct {
 	vnetGatewayConnectionsClient network.VirtualNetworkGatewayConnectionsClient
 	vnetGatewayClient            network.VirtualNetworkGatewaysClient
 	vnetClient                   network.VirtualNetworksClient
+	routeTablesClient            network.RouteTablesClient
 
 	providers           resources.ProvidersClient
 	resourceGroupClient resources.GroupsClient
@@ -186,6 +187,12 @@ func (c *Config) getArmClient() (*ArmClient, error) {
 	vnc.Sender = autorest.CreateSender(withRequestLogging())
 	client.vnetClient = vnc
 
+	rtc := network.NewRouteTablesClient(c.SubscriptionID)
+	setUserAgent(&rtc.Client)
+	rtc.Authorizer = spt
+	rtc.Sender = autorest.CreateSender(withRequestLogging())
+	client.routeTablesClient = rtc
+
 	rgc := resources.NewGroupsClient(c.SubscriptionID)
 	setUserAgent(&rgc.Client)
 	rgc.Authorizer = spt

builtin/providers/azurerm/provider.go

@@ -49,6 +49,7 @@ func Provider() terraform.ResourceProvider {
 			"azurerm_public_ip":              resourceArmPublicIp(),
 			"azurerm_subnet":                 resourceArmSubnet(),
 			"azurerm_network_interface":      resourceArmNetworkInterface(),
+			"azurerm_route_table":            resourceArmRouteTable(),
 		},
 		ConfigureFunc: providerConfigure,
 	}

builtin/providers/azurerm/resource_arm_route_table.go

@@ -0,0 +1,252 @@
+package azurerm
+
+import (
+	"bytes"
+	"fmt"
+	"log"
+	"net/http"
+	"strings"
+	"time"
+
+	"github.com/Azure/azure-sdk-for-go/arm/network"
+	"github.com/hashicorp/terraform/helper/hashcode"
+	"github.com/hashicorp/terraform/helper/resource"
+	"github.com/hashicorp/terraform/helper/schema"
+)
+
+func resourceArmRouteTable() *schema.Resource {
+	return &schema.Resource{
+		Create: resourceArmRouteTableCreate,
+		Read:   resourceArmRouteTableRead,
+		Update: resourceArmRouteTableCreate,
+		Delete: resourceArmRouteTableDelete,
+
+		Schema: map[string]*schema.Schema{
+			"name": &schema.Schema{
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: true,
+			},
+
+			"location": &schema.Schema{
+				Type:      schema.TypeString,
+				Required:  true,
+				ForceNew:  true,
+				StateFunc: azureRMNormalizeLocation,
+			},
+
+			"resource_group_name": &schema.Schema{
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: true,
+			},
+
+			"route": &schema.Schema{
+				Type:     schema.TypeSet,
+				Optional: true,
+				Computed: true,
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"name": &schema.Schema{
+							Type:     schema.TypeString,
+							Required: true,
+						},
+
+						"address_prefix": &schema.Schema{
+							Type:     schema.TypeString,
+							Required: true,
+						},
+
+						"next_hop_type": &schema.Schema{
+							Type:         schema.TypeString,
+							Required:     true,
+							ValidateFunc: validateRouteTableNextHopType,
+						},
+
+						"next_hop_in_ip_address": &schema.Schema{
+							Type:     schema.TypeString,
+							Optional: true,
+							Computed: true,
+						},
+					},
+				},
+				Set: resourceArmRouteTableRouteHash,
+			},
+
+			"subnets": &schema.Schema{
+				Type:     schema.TypeSet,
+				Optional: true,
+				Computed: true,
+				Elem:     &schema.Schema{Type: schema.TypeString},
+				Set:      schema.HashString,
+			},
+		},
+	}
+}
+
+func resourceArmRouteTableCreate(d *schema.ResourceData, meta interface{}) error {
+	client := meta.(*ArmClient)
+	routeTablesClient := client.routeTablesClient
+
+	log.Printf("[INFO] preparing arguments for Azure ARM Route Table creation.")
+
+	name := d.Get("name").(string)
+	location := d.Get("location").(string)
+	resGroup := d.Get("resource_group_name").(string)
+
+	routeSet := network.RouteTable{
+		Name:     &name,
+		Location: &location,
+	}
+
+	if _, ok := d.GetOk("route"); ok {
+		properties := network.RouteTablePropertiesFormat{}
+		routes, routeErr := expandAzureRmRouteTableRoutes(d)
+		if routeErr != nil {
+			return fmt.Errorf("Error Building list of Route Table Routes: %s", routeErr)
+		}
+		if len(routes) > 0 {
+			routeSet.Properties = &properties
+		}
+
+	}
+
+	resp, err := routeTablesClient.CreateOrUpdate(resGroup, name, routeSet)
+	if err != nil {
+		return err
+	}
+
+	d.SetId(*resp.ID)
+
+	log.Printf("[DEBUG] Waiting for Route Table (%s) to become available", name)
+	stateConf := &resource.StateChangeConf{
+		Pending: []string{"Accepted", "Updating"},
+		Target:  "Succeeded",
+		Refresh: routeTableStateRefreshFunc(client, resGroup, name),
+		Timeout: 10 * time.Minute,
+	}
+	if _, err := stateConf.WaitForState(); err != nil {
+		return fmt.Errorf("Error waiting forRoute Table (%s) to become available: %s", name, err)
+	}
+
+	return resourceArmRouteTableRead(d, meta)
+}
+
+func resourceArmRouteTableRead(d *schema.ResourceData, meta interface{}) error {
+	routeTablesClient := meta.(*ArmClient).routeTablesClient
+
+	id, err := parseAzureResourceID(d.Id())
+	if err != nil {
+		return err
+	}
+	resGroup := id.ResourceGroup
+	name := id.Path["routeTables"]
+
+	resp, err := routeTablesClient.Get(resGroup, name, "")
+	if resp.StatusCode == http.StatusNotFound {
+		d.SetId("")
+		return nil
+	}
+	if err != nil {
+		return fmt.Errorf("Error making Read request on Azure Route Table %s: %s", name, err)
+	}
+
+	if resp.Properties.Subnets != nil {
+		if len(*resp.Properties.Subnets) > 0 {
+			subnets := make([]string, 0, len(*resp.Properties.Subnets))
+			for _, subnet := range *resp.Properties.Subnets {
+				id := subnet.ID
+				subnets = append(subnets, *id)
+			}
+
+			if err := d.Set("subnets", subnets); err != nil {
+				return err
+			}
+		}
+	}
+
+	return nil
+}
+
+func resourceArmRouteTableDelete(d *schema.ResourceData, meta interface{}) error {
+	routeTablesClient := meta.(*ArmClient).routeTablesClient
+
+	id, err := parseAzureResourceID(d.Id())
+	if err != nil {
+		return err
+	}
+	resGroup := id.ResourceGroup
+	name := id.Path["routeTables"]
+
+	_, err = routeTablesClient.Delete(resGroup, name)
+
+	return err
+}
+
+func expandAzureRmRouteTableRoutes(d *schema.ResourceData) ([]network.Route, error) {
+	configs := d.Get("route").(*schema.Set).List()
+	routes := make([]network.Route, 0, len(configs))
+
+	for _, configRaw := range configs {
+		data := configRaw.(map[string]interface{})
+
+		address_prefix := data["address_prefix"].(string)
+		next_hop_type := data["next_hop_type"].(string)
+
+		properties := network.RoutePropertiesFormat{
+			AddressPrefix: &address_prefix,
+			NextHopType:   network.RouteNextHopType(next_hop_type),
+		}
+
+		if v := data["next_hop_in_ip_address"].(string); v != "" {
+			properties.NextHopIPAddress = &v
+		}
+
+		name := data["name"].(string)
+		route := network.Route{
+			Name:       &name,
+			Properties: &properties,
+		}
+
+		routes = append(routes, route)
+	}
+
+	return routes, nil
+}
+
+func routeTableStateRefreshFunc(client *ArmClient, resourceGroupName string, routeTableName string) resource.StateRefreshFunc {
+	return func() (interface{}, string, error) {
+		res, err := client.routeTablesClient.Get(resourceGroupName, routeTableName, "")
+		if err != nil {
+			return nil, "", fmt.Errorf("Error issuing read request in routeTableStateRefreshFunc to Azure ARM for route table '%s' (RG: '%s'): %s", routeTableName, resourceGroupName, err)
+		}
+
+		return res, *res.Properties.ProvisioningState, nil
+	}
+}
+
+func resourceArmRouteTableRouteHash(v interface{}) int {
+	var buf bytes.Buffer
+	m := v.(map[string]interface{})
+	buf.WriteString(fmt.Sprintf("%s-", m["name"].(string)))
+	buf.WriteString(fmt.Sprintf("%s-", m["address_prefix"].(string)))
+	buf.WriteString(fmt.Sprintf("%s-", m["next_hop_type"].(string)))
+
+	return hashcode.String(buf.String())
+}
+
+func validateRouteTableNextHopType(v interface{}, k string) (ws []string, errors []error) {
+	value := strings.ToLower(v.(string))
+	hopTypes := map[string]bool{
+		"virtualnetworkgateway": true,
+		"vnetlocal":             true,
+		"internet":              true,
+		"virtualappliance":      true,
+		"null":                  true,
+	}
+
+	if !hopTypes[value] {
+		errors = append(errors, fmt.Errorf("Route Table NextHopType Protocol can only be VirtualNetworkGateway, VnetLocal, Internet or VirtualAppliance"))
+	}
+	return
+}