279 lines
7.9 KiB
Go
279 lines
7.9 KiB
Go
|
package aws
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"github.com/aws/aws-sdk-go/aws"
|
||
|
|
"github.com/aws/aws-sdk-go/aws/awsutil"
|
||
|
|
"github.com/aws/aws-sdk-go/service/ec2"
|
||
|
|
"github.com/hashicorp/terraform/helper/hashcode"
|
||
|
|
"github.com/hashicorp/terraform/helper/schema"
|
||
|
|
)
|
||
|
|
|
||
|
|
// AWS Route resource Schema delcaration
|
||
|
|
func resourceAwsRoute() *schema.Resource {
|
||
|
|
return &schema.Resource{
|
||
|
|
Create: resourceAwsRouteCreate,
|
||
|
|
Read: resourceAwsRouteRead,
|
||
|
|
Update: resourceAwsRouteUpdate,
|
||
|
|
Delete: resourceAwsRouteDelete,
|
||
|
|
|
||
|
|
Schema: map[string]*schema.Schema{
|
||
|
|
"destination_cidr_block": &schema.Schema{
|
||
|
|
Type: schema.TypeString,
|
||
|
|
Required: true,
|
||
|
|
},
|
||
|
|
|
||
|
|
"destination_prefix_list_id": &schema.Schema{
|
||
|
|
Type: schema.TypeString,
|
||
|
|
Computed: true,
|
||
|
|
},
|
||
|
|
|
||
|
|
"gateway_id": &schema.Schema{
|
||
|
|
Type: schema.TypeString,
|
||
|
|
Optional: true,
|
||
|
|
},
|
||
|
|
|
||
|
|
"instance_id": &schema.Schema{
|
||
|
|
Type: schema.TypeString,
|
||
|
|
Optional: true,
|
||
|
|
},
|
||
|
|
|
||
|
|
"instance_owner_id": &schema.Schema{
|
||
|
|
Type: schema.TypeString,
|
||
|
|
Computed: true,
|
||
|
|
},
|
||
|
|
|
||
|
|
"network_interface_id": &schema.Schema{
|
||
|
|
Type: schema.TypeString,
|
||
|
|
Optional: true,
|
||
|
|
},
|
||
|
|
|
||
|
|
"origin": &schema.Schema{
|
||
|
|
Type: schema.TypeString,
|
||
|
|
Computed: true,
|
||
|
|
},
|
||
|
|
|
||
|
|
"state": &schema.Schema{
|
||
|
|
Type: schema.TypeString,
|
||
|
|
Computed: true,
|
||
|
|
},
|
||
|
|
|
||
|
|
"route_table_id": &schema.Schema{
|
||
|
|
Type: schema.TypeString,
|
||
|
|
Required: true,
|
||
|
|
},
|
||
|
|
|
||
|
|
"vpc_peering_connection_id": &schema.Schema{
|
||
|
|
Type: schema.TypeString,
|
||
|
|
Optional: true,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func resourceAwsRouteCreate(d *schema.ResourceData, meta interface{}) error {
|
||
|
|
conn := meta.(*AWSClient).ec2conn
|
||
|
|
var numTargets int
|
||
|
|
var setTarget string
|
||
|
|
allowedTargets := []string{
|
||
|
|
"gateway_id",
|
||
|
|
"instance_id",
|
||
|
|
"network_interface_id",
|
||
|
|
"vpc_peering_connection_id",
|
||
|
|
}
|
||
|
|
createOpts := &ec2.CreateRouteInput{}
|
||
|
|
|
||
|
|
// Check if more than 1 target is specified
|
||
|
|
for _, target := range allowedTargets {
|
||
|
|
if len(d.Get(target).(string)) > 0 {
|
||
|
|
numTargets++
|
||
|
|
setTarget = target
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if numTargets > 1 {
|
||
|
|
fmt.Errorf("Error: more than 1 target specified. Only 1 of gateway_id" +
|
||
|
|
"instance_id, network_interface_id, route_table_id or" +
|
||
|
|
"vpc_peering_connection_id is allowed.")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Formulate CreateRouteInput based on the target type
|
||
|
|
switch setTarget {
|
||
|
|
case "gateway_id":
|
||
|
|
createOpts = &ec2.CreateRouteInput{
|
||
|
|
RouteTableID: aws.String(d.Get("route_table_id").(string)),
|
||
|
|
DestinationCIDRBlock: aws.String(d.Get("destination_cidr_block").(string)),
|
||
|
|
GatewayID: aws.String(d.Get("gateway_id").(string)),
|
||
|
|
}
|
||
|
|
case "instance_id":
|
||
|
|
createOpts = &ec2.CreateRouteInput{
|
||
|
|
RouteTableID: aws.String(d.Get("route_table_id").(string)),
|
||
|
|
DestinationCIDRBlock: aws.String(d.Get("destination_cidr_block").(string)),
|
||
|
|
InstanceID: aws.String(d.Get("instance_id").(string)),
|
||
|
|
}
|
||
|
|
case "network_interface_id":
|
||
|
|
createOpts = &ec2.CreateRouteInput{
|
||
|
|
RouteTableID: aws.String(d.Get("route_table_id").(string)),
|
||
|
|
DestinationCIDRBlock: aws.String(d.Get("destination_cidr_block").(string)),
|
||
|
|
NetworkInterfaceID: aws.String(d.Get("network_interface_id").(string)),
|
||
|
|
}
|
||
|
|
case "vpc_peering_connection_id":
|
||
|
|
createOpts = &ec2.CreateRouteInput{
|
||
|
|
RouteTableID: aws.String(d.Get("route_table_id").(string)),
|
||
|
|
DestinationCIDRBlock: aws.String(d.Get("destination_cidr_block").(string)),
|
||
|
|
VPCPeeringConnectionID: aws.String(d.Get("vpc_peering_connection_id").(string)),
|
||
|
|
}
|
||
|
|
default:
|
||
|
|
fmt.Errorf("Error: invalid target type specified.")
|
||
|
|
}
|
||
|
|
log.Printf("[DEBUG] Route create config: %s", awsutil.Prettify(createOpts))
|
||
|
|
|
||
|
|
// Create the route
|
||
|
|
_, err := conn.CreateRoute(createOpts)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("Error creating route: %s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
route, err := findResourceRoute(conn, d.Get("route_table_id").(string), d.Get("destination_cidr_block").(string))
|
||
|
|
if err != nil {
|
||
|
|
fmt.Errorf("Error: %s", awsutil.Prettify(err))
|
||
|
|
}
|
||
|
|
|
||
|
|
d.SetId(routeIDHash(d, route))
|
||
|
|
|
||
|
|
return resourceAwsRouteRead(d, meta)
|
||
|
|
}
|
||
|
|
|
||
|
|
func resourceAwsRouteRead(d *schema.ResourceData, meta interface{}) error {
|
||
|
|
conn := meta.(*AWSClient).ec2conn
|
||
|
|
route, err := findResourceRoute(conn, d.Get("route_table_id").(string), d.Get("destination_cidr_block").(string))
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
d.Set("destination_prefix_list_id", route.DestinationPrefixListID)
|
||
|
|
d.Set("gateway_id", route.DestinationPrefixListID)
|
||
|
|
d.Set("instance_id", route.InstanceID)
|
||
|
|
d.Set("instance_owner_id", route.InstanceOwnerID)
|
||
|
|
d.Set("network_interface_id", route.NetworkInterfaceID)
|
||
|
|
d.Set("origin", route.Origin)
|
||
|
|
d.Set("state", route.State)
|
||
|
|
d.Set("vpc_peering_connection_id", route.VPCPeeringConnectionID)
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func resourceAwsRouteUpdate(d *schema.ResourceData, meta interface{}) error {
|
||
|
|
conn := meta.(*AWSClient).ec2conn
|
||
|
|
var numTargets int
|
||
|
|
var setTarget string
|
||
|
|
allowedTargets := []string{
|
||
|
|
"gateway_id",
|
||
|
|
"instance_id",
|
||
|
|
"network_interface_id",
|
||
|
|
"vpc_peering_connection_id",
|
||
|
|
}
|
||
|
|
replaceOpts := &ec2.ReplaceRouteInput{}
|
||
|
|
|
||
|
|
// Check if more than 1 target is specified
|
||
|
|
for _, target := range allowedTargets {
|
||
|
|
if len(d.Get(target).(string)) > 0 {
|
||
|
|
numTargets++
|
||
|
|
setTarget = target
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if numTargets > 1 {
|
||
|
|
fmt.Errorf("Error: more than 1 target specified. Only 1 of gateway_id" +
|
||
|
|
"instance_id, network_interface_id, route_table_id or" +
|
||
|
|
"vpc_peering_connection_id is allowed.")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Formulate ReplaceRouteInput based on the target type
|
||
|
|
switch setTarget {
|
||
|
|
case "gateway_id":
|
||
|
|
replaceOpts = &ec2.ReplaceRouteInput{
|
||
|
|
RouteTableID: aws.String(d.Get("route_table_id").(string)),
|
||
|
|
DestinationCIDRBlock: aws.String(d.Get("destination_cidr_block").(string)),
|
||
|
|
GatewayID: aws.String(d.Get("gateway_id").(string)),
|
||
|
|
}
|
||
|
|
case "instance_id":
|
||
|
|
replaceOpts = &ec2.ReplaceRouteInput{
|
||
|
|
RouteTableID: aws.String(d.Get("route_table_id").(string)),
|
||
|
|
DestinationCIDRBlock: aws.String(d.Get("destination_cidr_block").(string)),
|
||
|
|
InstanceID: aws.String(d.Get("instance_id").(string)),
|
||
|
|
}
|
||
|
|
case "network_interface_id":
|
||
|
|
replaceOpts = &ec2.ReplaceRouteInput{
|
||
|
|
RouteTableID: aws.String(d.Get("route_table_id").(string)),
|
||
|
|
DestinationCIDRBlock: aws.String(d.Get("destination_cidr_block").(string)),
|
||
|
|
NetworkInterfaceID: aws.String(d.Get("network_interface_id").(string)),
|
||
|
|
}
|
||
|
|
case "vpc_peering_connection_id":
|
||
|
|
replaceOpts = &ec2.ReplaceRouteInput{
|
||
|
|
RouteTableID: aws.String(d.Get("route_table_id").(string)),
|
||
|
|
DestinationCIDRBlock: aws.String(d.Get("destination_cidr_block").(string)),
|
||
|
|
VPCPeeringConnectionID: aws.String(d.Get("vpc_peering_connection_id").(string)),
|
||
|
|
}
|
||
|
|
default:
|
||
|
|
fmt.Errorf("Error: invalid target type specified.")
|
||
|
|
}
|
||
|
|
log.Printf("[DEBUG] Route replace config: %s", awsutil.Prettify(replaceOpts))
|
||
|
|
|
||
|
|
// Replace the route
|
||
|
|
_, err := conn.ReplaceRoute(replaceOpts)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func resourceAwsRouteDelete(d *schema.ResourceData, meta interface{}) error {
|
||
|
|
conn := meta.(*AWSClient).ec2conn
|
||
|
|
deleteOpts := &ec2.DeleteRouteInput{
|
||
|
|
DestinationCIDRBlock: aws.String(d.Get("destination_cidr_block").(string)),
|
||
|
|
RouteTableID: aws.String(d.Get("route_table_id").(string)),
|
||
|
|
}
|
||
|
|
log.Printf("[DEBUG] Route delete opts: %s", awsutil.Prettify(deleteOpts))
|
||
|
|
|
||
|
|
resp, err := conn.DeleteRoute(deleteOpts)
|
||
|
|
log.Printf("[DEBUG] Route delete result: %s", awsutil.Prettify(resp))
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
d.SetId("")
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Create an ID for a route
|
||
|
|
func routeIDHash(d *schema.ResourceData, r *ec2.Route) string {
|
||
|
|
return fmt.Sprintf("r-%s%d", d.Get("route_table_id").(string), hashcode.String(*r.DestinationCIDRBlock))
|
||
|
|
}
|
||
|
|
|
||
|
|
// Helper: retrieve a route
|
||
|
|
func findResourceRoute(conn *ec2.EC2, rtbid string, cidr string) (*ec2.Route, error) {
|
||
|
|
routeTableID := rtbid
|
||
|
|
|
||
|
|
findOpts := &ec2.DescribeRouteTablesInput{
|
||
|
|
RouteTableIDs: []*string{&routeTableID},
|
||
|
|
}
|
||
|
|
|
||
|
|
resp, err := conn.DescribeRouteTables(findOpts)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, route := range (*resp.RouteTables[0]).Routes {
|
||
|
|
if *route.DestinationCIDRBlock == cidr {
|
||
|
|
return route, nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil, nil
|
||
|
|
}
|