From 01db1ff8bdeddc66917539f342c906109ef41ac0 Mon Sep 17 00:00:00 2001
From: Armon Dadgar <armon.dadgar@gmail.com>
Date: Tue, 18 Nov 2014 15:38:40 -0800
Subject: [PATCH] terraform: diff handler in expansion avoids duplicate destroy

---
 terraform/graph.go | 42 ++++++++++++++++++++++++++++++------------
 1 file changed, 30 insertions(+), 12 deletions(-)

diff --git a/terraform/graph.go b/terraform/graph.go
index 6762ed8fa..ff1e32ffb 100644
--- a/terraform/graph.go
+++ b/terraform/graph.go
@@ -1751,24 +1751,42 @@ func (n *GraphNodeResource) expand(g *depgraph.Graph, count int, diff *ModuleDif
 			}
 		}
 
+		// Initialize a default state if not available
 		if state == nil {
 			state = &ResourceState{
 				Type: r.Type,
 			}
 		}
 
-		if inDiff != nil && n.ExpandMode != ResourceExpandDestroy {
-			// Disable Destroy if we aren't doing a destroy expansion.
-			// There is a seperate expansion for the destruction action.
-			d := new(InstanceDiff)
-			*d = *inDiff
-			inDiff = d
-			inDiff.Destroy = false
-		} else if inDiff != nil && inDiff.Destroy && n.ExpandMode == ResourceExpandDestroy {
-			// If we are doing a destroy, make sure it is exclusively
-			// a destroy, since there is a seperate expansion for the apply
-			inDiff = new(InstanceDiff)
-			inDiff.Destroy = true
+		// Prepare the diff if it exists
+		if inDiff != nil {
+			switch n.ExpandMode {
+			case ResourceExpandApply:
+				// Disable Destroy if we aren't doing a destroy expansion.
+				// There is a seperate expansion for the destruction action.
+				d := new(InstanceDiff)
+				*d = *inDiff
+				inDiff = d
+				inDiff.Destroy = false
+
+				// If we require a new resource, there is a seperate delete
+				// phase, so the create phase must not have access to the ID.
+				if inDiff.RequiresNew() {
+					s := new(ResourceState)
+					*s = *state
+					state = s
+					state.Primary = nil
+				}
+
+			case ResourceExpandDestroy:
+				// If we are doing a destroy, make sure it is exclusively
+				// a destroy, since there is a seperate expansion for the apply
+				inDiff = new(InstanceDiff)
+				inDiff.Destroy = true
+
+			default:
+				panic(fmt.Sprintf("Unhandled expansion mode %d", n.ExpandMode))
+			}
 		}
 
 		// Inherit the existing flags!