From 4405a8d9741dbc08bf341683975270f514706780 Mon Sep 17 00:00:00 2001
From: deoxxa <deoxxa@fknsrs.biz>
Date: Sat, 9 Aug 2014 22:43:00 +1000
Subject: [PATCH] add vpc dns attribute support

---
 builtin/providers/aws/resource_aws_vpc.go     | 95 +++++++++++++++++--
 builtin/providers/aws/resources.go            |  1 +
 .../docs/providers/aws/r/vpc.html.markdown    |  4 +
 3 files changed, 93 insertions(+), 7 deletions(-)

diff --git a/builtin/providers/aws/resource_aws_vpc.go b/builtin/providers/aws/resource_aws_vpc.go
index 00f5b6054488..acb2b3002328 100644
--- a/builtin/providers/aws/resource_aws_vpc.go
+++ b/builtin/providers/aws/resource_aws_vpc.go
@@ -3,6 +3,7 @@ package aws
 import (
 	"fmt"
 	"log"
+	"strconv"
 	"time"
 
 	"github.com/hashicorp/terraform/helper/diff"
@@ -53,6 +54,36 @@ func resource_aws_vpc_create(
 			s.ID, err)
 	}
 
+	if attr, ok := d.Attributes["enable_dns_support"]; ok {
+		options := new(ec2.ModifyVpcAttribute)
+
+		options.EnableDnsSupport = attr.New != "" && attr.New != "false"
+		options.SetEnableDnsSupport = true
+
+		s.Attributes["enable_dns_support"] = strconv.FormatBool(options.EnableDnsSupport)
+
+		log.Printf("[INFO] Modifying vpc attributes for %s: %#v", s.ID, options)
+
+		if _, err := ec2conn.ModifyVpcAttribute(s.ID, options); err != nil {
+			return s, err
+		}
+	}
+
+	if attr, ok := d.Attributes["enable_dns_hostnames"]; ok {
+		options := new(ec2.ModifyVpcAttribute)
+
+		options.EnableDnsHostnames = attr.New != "" && attr.New != "false"
+		options.SetEnableDnsHostnames = true
+
+		s.Attributes["enable_dns_hostnames"] = strconv.FormatBool(options.EnableDnsHostnames)
+
+		log.Printf("[INFO] Modifying enable_dns_hostnames vpc attribute for %s: %#v", s.ID, options)
+
+		if _, err := ec2conn.ModifyVpcAttribute(s.ID, options); err != nil {
+			return s, err
+		}
+	}
+
 	// Update our attributes and return
 	return resource_aws_vpc_update_state(s, vpcRaw.(*ec2.VPC))
 }
@@ -61,11 +92,43 @@ func resource_aws_vpc_update(
 	s *terraform.ResourceState,
 	d *terraform.ResourceDiff,
 	meta interface{}) (*terraform.ResourceState, error) {
-	// This should never be called because we have no update-able
-	// attributes
-	panic("Update for VPC is not supported")
+	p := meta.(*ResourceProvider)
+	ec2conn := p.ec2conn
+	rs := s.MergeDiff(d)
+
+	log.Printf("[DEBUG] attributes: %#v", d.Attributes)
+
+	if attr, ok := d.Attributes["enable_dns_support"]; ok {
+		options := new(ec2.ModifyVpcAttribute)
+
+		options.EnableDnsSupport = attr.New != "" && attr.New != "false"
+		options.SetEnableDnsSupport = true
+
+		rs.Attributes["enable_dns_support"] = strconv.FormatBool(options.EnableDnsSupport)
 
-	return nil, nil
+		log.Printf("[INFO] Modifying enable_dns_support vpc attribute for %s: %#v", s.ID, options)
+
+		if _, err := ec2conn.ModifyVpcAttribute(s.ID, options); err != nil {
+			return s, err
+		}
+	}
+
+	if attr, ok := d.Attributes["enable_dns_hostnames"]; ok {
+		options := new(ec2.ModifyVpcAttribute)
+
+		options.EnableDnsHostnames = attr.New != "" && attr.New != "false"
+		options.SetEnableDnsHostnames = true
+
+		rs.Attributes["enable_dns_hostnames"] = strconv.FormatBool(options.EnableDnsHostnames)
+
+		log.Printf("[INFO] Modifying enable_dns_hostnames vpc attribute for %s: %#v", s.ID, options)
+
+		if _, err := ec2conn.ModifyVpcAttribute(s.ID, options); err != nil {
+			return s, err
+		}
+	}
+
+	return rs, nil
 }
 
 func resource_aws_vpc_destroy(
@@ -101,8 +164,19 @@ func resource_aws_vpc_refresh(
 		return nil, nil
 	}
 
-	vpc := vpcRaw.(*ec2.VPC)
-	return resource_aws_vpc_update_state(s, vpc)
+	if dnsSupportResp, err := ec2conn.VpcAttribute(s.ID, "enableDnsSupport"); err != nil {
+		return s, err
+	} else {
+		s.Attributes["enable_dns_support"] = strconv.FormatBool(dnsSupportResp.EnableDnsSupport)
+	}
+
+	if dnsHostnamesResp, err := ec2conn.VpcAttribute(s.ID, "enableDnsHostnames"); err != nil {
+		return s, err
+	} else {
+		s.Attributes["enable_dns_hostnames"] = strconv.FormatBool(dnsHostnamesResp.EnableDnsHostnames)
+	}
+
+	return resource_aws_vpc_update_state(s, vpcRaw.(*ec2.VPC))
 }
 
 func resource_aws_vpc_diff(
@@ -111,7 +185,14 @@ func resource_aws_vpc_diff(
 	meta interface{}) (*terraform.ResourceDiff, error) {
 	b := &diff.ResourceBuilder{
 		Attrs: map[string]diff.AttrType{
-			"cidr_block": diff.AttrTypeCreate,
+			"cidr_block":           diff.AttrTypeCreate,
+			"enable_dns_support":   diff.AttrTypeUpdate,
+			"enable_dns_hostnames": diff.AttrTypeUpdate,
+		},
+
+		ComputedAttrs: []string{
+			"enable_dns_support",
+			"enable_dns_hostnames",
 		},
 	}
 
diff --git a/builtin/providers/aws/resources.go b/builtin/providers/aws/resources.go
index 48465cb1249d..6d859a1d465d 100644
--- a/builtin/providers/aws/resources.go
+++ b/builtin/providers/aws/resources.go
@@ -157,6 +157,7 @@ func init() {
 				Destroy: resource_aws_vpc_destroy,
 				Diff:    resource_aws_vpc_diff,
 				Refresh: resource_aws_vpc_refresh,
+				Update:  resource_aws_vpc_update,
 			},
 		},
 	}
diff --git a/website/source/docs/providers/aws/r/vpc.html.markdown b/website/source/docs/providers/aws/r/vpc.html.markdown
index 9f16b82d8826..8104c2d6540a 100644
--- a/website/source/docs/providers/aws/r/vpc.html.markdown
+++ b/website/source/docs/providers/aws/r/vpc.html.markdown
@@ -21,6 +21,8 @@ resource "aws_vpc" "main" {
 The following arguments are supported:
 
 * `cidr_block` - (Required) The CIDR block for the VPC.
+* `enable_dns_support` - (Optional) A boolean flag to enable/disable DNS support in the VPC. Defaults true.
+* `enable_dns_hostnames` - (Optional) A boolean flag to enable/disable DNS hostnames in the VPC. Defaults false.
 
 ## Attributes Reference
 
@@ -28,4 +30,6 @@ The following attributes are exported:
 
 * `id` - The ID of the VPC
 * `cidr_block` - The CIDR block of the VPC
+* `enable_dns_support` - Whether or not the VPC has DNS support
+* `enable_dns_hostnames` - Whether or not the VPC has DNS hostname support