From 2803f845b8bf3bb245c40e2e52343714c4497e42 Mon Sep 17 00:00:00 2001
From: kylecarbs <kyle@carberry.com>
Date: Tue, 17 May 2022 01:07:01 +0000
Subject: [PATCH] fix: Parse resources from Terraform Modules

Fixes when Terraform modules are used to primariy provision
infrastructure!
---
 provisioner/terraform/provision.go | 28 +++++++++++++++++++++++-----
 1 file changed, 23 insertions(+), 5 deletions(-)

diff --git a/provisioner/terraform/provision.go b/provisioner/terraform/provision.go
index 4e9865e829344..9bcaaf2e513a7 100644
--- a/provisioner/terraform/provision.go
+++ b/provisioner/terraform/provision.go
@@ -290,8 +290,17 @@ func parseTerraformPlan(ctx context.Context, terraform *tfexec.Terraform, planfi
 	resources := make([]*proto.Resource, 0)
 	agents := map[string]*proto.Agent{}
 
+	tfResources := plan.Config.RootModule.Resources
+	var appendResources func(mod *tfjson.ConfigModule)
+	appendResources = func(mod *tfjson.ConfigModule) {
+		for _, module := range mod.ModuleCalls {
+			appendResources(module.Module)
+		}
+		tfResources = append(tfResources, mod.Resources...)
+	}
+
 	// Store all agents inside the maps!
-	for _, resource := range plan.Config.RootModule.Resources {
+	for _, resource := range tfResources {
 		if resource.Type != "coder_agent" {
 			continue
 		}
@@ -340,7 +349,7 @@ func parseTerraformPlan(ctx context.Context, terraform *tfexec.Terraform, planfi
 		agents[resource.Address] = agent
 	}
 
-	for _, resource := range plan.PlannedValues.RootModule.Resources {
+	for _, resource := range tfResources {
 		if resource.Mode == tfjson.DataResourceMode {
 			continue
 		}
@@ -407,8 +416,17 @@ func parseTerraformApply(ctx context.Context, terraform *tfexec.Terraform, state
 		}
 		agents := map[string]*proto.Agent{}
 
+		tfResources := state.Values.RootModule.Resources
+		var appendResources func(resource *tfjson.StateModule)
+		appendResources = func(mod *tfjson.StateModule) {
+			for _, module := range mod.ChildModules {
+				appendResources(module)
+			}
+			tfResources = append(tfResources, mod.Resources...)
+		}
+
 		// Store all agents inside the maps!
-		for _, resource := range state.Values.RootModule.Resources {
+		for _, resource := range tfResources {
 			if resource.Type != "coder_agent" {
 				continue
 			}
@@ -439,7 +457,7 @@ func parseTerraformApply(ctx context.Context, terraform *tfexec.Terraform, state
 		}
 
 		// Manually associate agents with instance IDs.
-		for _, resource := range state.Values.RootModule.Resources {
+		for _, resource := range tfResources {
 			if resource.Type != "coder_agent_instance" {
 				continue
 			}
@@ -471,7 +489,7 @@ func parseTerraformApply(ctx context.Context, terraform *tfexec.Terraform, state
 			}
 		}
 
-		for _, resource := range state.Values.RootModule.Resources {
+		for _, resource := range tfResources {
 			if resource.Mode == tfjson.DataResourceMode {
 				continue
 			}